ServerUtils/lxd-btrfs-manager at master · janxb/ServerUtils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

#!/bin/bash
set -e

ARG_SELF=$0

while [ $# -gt 0 ]; do
  case "$1" in
    --pool=*)
      ARG_POOL="${1#*=}"
      ;;
    --path=*)
      ARG_PATH="${1#*=}"
      ;;
    mount|unmount|list|fstrim|defrag|size)
      COMMAND="$1"
      ;;
    *)
      echo "
        LXD BTRFS Manager
		Usage:
		  lxd-btrfs-manager mount [--pool=<name>]
		  lxd-btrfs-manager unmount [--pool=<name>]
		  lxd-btrfs-manager list [--pool=<name>]
		  lxd-btrfs-manager size [--pool=<name>]
		  lxd-btrfs-manager fstrim [--pool=<name>]
		  lxd-btrfs-manager defrag [--pool=<name>] [--path=<path>]

		Options:
		  --pool=<name>  Select a custom BTRFS pool name. Defaults to first found pool if empty.
		  --path=<path>  Only defrag the specified sub-path. Defaults to the complete pool if empty.
      "
      exit 1
  esac
  shift
done

ROOT_FS_TYPE=$(df -Th / | awk 'FNR == 2 {print $2}')

if [ -z "$ARG_POOL" ]; then
	if [ $ROOT_FS_TYPE == 'btrfs' ]; then
		ARG_POOL='/'
	else
		ARG_POOL=$(lxc storage list | awk 'FNR == 4 {print $6}');
	fi
fi

if [ -z "$ARG_PATH" ]; then ARG_PATH="/"; fi

btrfs filesystem show $ARG_POOL >/dev/null

LOOPFILE_PATH=$(btrfs filesystem show $ARG_POOL | awk 'FNR == 3 {print $8}')
IMAGEFILE_PATH=$(losetup | grep $LOOPFILE_PATH | awk 'FNR == 1 {print $6}')
BTRFS_UUID=$(btrfs filesystem show $ARG_POOL | awk 'FNR == 1 {print $4}')
TEMP_MOUNT_PATH=/tmp/btrfs_$BTRFS_UUID

if [ $ROOT_FS_TYPE == 'btrfs' ]; then TEMP_MOUNT_PATH='/'; fi

case "$COMMAND" in
	mount)
		if [ $ROOT_FS_TYPE != 'btrfs' ]; then
			mkdir $TEMP_MOUNT_PATH
			mount $LOOPFILE_PATH $TEMP_MOUNT_PATH
		fi
		echo "Mounted LXD storage $ARG_POOL on $TEMP_MOUNT_PATH"
	;;
	unmount)
		if [ $ROOT_FS_TYPE != 'btrfs' ]; then
			umount $TEMP_MOUNT_PATH
			rmdir $TEMP_MOUNT_PATH
		fi
		echo "Unmounted LXD storage $ARG_POOL from $TEMP_MOUNT_PATH"
	;;
	list)
		$ARG_SELF mount --pool=$ARG_POOL
		btrfs subvolume list -t --sort=path $TEMP_MOUNT_PATH
		df -h $TEMP_MOUNT_PATH
		$ARG_SELF unmount --pool=$ARG_POOL
		if [ -n "$IMAGEFILE_PATH" ]; then filesize $IMAGEFILE_PATH; fi
		lxc storage info default | head -n6 | tail -n2 | sed -e 's/^[[:space:]]*//'
		btrfs filesystem show $ARG_POOL
	;;
	size)
		$ARG_SELF mount --pool=$ARG_POOL
		lxc storage info default | head -n6 | tail -n2 | sed -e 's/^[[:space:]]*//'
		btrfs filesystem show $ARG_POOL
		$ARG_SELF unmount --pool=$ARG_POOL
	;;
	defrag)
		$ARG_SELF mount --pool=$ARG_POOL
		echo "Defragmenting path $ARG_PATH on BTRFS pool $ARG_POOL.."
		btrfs filesystem defragment -r $TEMP_MOUNT_PATH/$ARG_PATH 2>/dev/null || true
		$ARG_SELF unmount --pool=$ARG_POOL
		echo "Done."
	;;
	fstrim)
		echo "Trimming BTRFS pool $ARG_POOL.."
		$ARG_SELF mount --pool=$ARG_POOL
		fstrim -v $TEMP_MOUNT_PATH
		sync $TEMP_MOUNT_PATH
		$ARG_SELF unmount --pool=$ARG_POOL
		if [ -n "$IMAGEFILE_PATH" ]; then
			echo "Waiting for BTRFS filesystem sync-to-disk.."
			sleep 10
			echo "Trimming loop image file.."
			fstrim -v $(dirname $IMAGEFILE_PATH)
		fi
		echo "Done."
	;;
esac