mirror of
https://github.com/Zygo/bees.git
synced 2025-05-17 21:35:45 +02:00
47805253e6
The service starter wasn't able to pass options to the new getopt parser. This commit fixes it.
130 lines
3.1 KiB
Bash
Executable File
130 lines
3.1 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
## Helpful functions
|
|
INFO(){ echo "INFO:" "$@"; }
|
|
ERRO(){ echo "ERROR:" "$@"; exit 1; }
|
|
YN(){ [[ "$1" =~ (1|Y|y) ]]; }
|
|
|
|
## Global vars
|
|
export BEESHOME BEESSTATUS
|
|
export WORK_DIR CONFIG_DIR
|
|
export CONFIG_FILE
|
|
export UUID AL16M
|
|
|
|
readonly AL16M="$((16*1024*1024))"
|
|
readonly CONFIG_DIR=@PREFIX@/etc/bees/
|
|
|
|
## Pre checks
|
|
{
|
|
[ ! -d "$CONFIG_DIR" ] && ERRO "Missing: $CONFIG_DIR"
|
|
[ "$UID" == "0" ] || ERRO "Must be run as root"
|
|
}
|
|
|
|
command -v @LIBEXEC_PREFIX@/bees &> /dev/null || ERRO "Missing 'bees' agent"
|
|
|
|
## Parse args
|
|
ARGUMENTS=()
|
|
while [ $# -gt 0 ]; do
|
|
case "$1" in
|
|
-*)
|
|
ARGUMENTS+=($1)
|
|
;;
|
|
*)
|
|
if [ -z "$UUID" ]; then
|
|
UUID="$1"
|
|
else
|
|
ERRO "Only one filesystem may be supplied"
|
|
fi
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
case "$UUID" in
|
|
*-*-*-*-*)
|
|
FILE_CONFIG=""
|
|
for file in "$CONFIG_DIR"/*.conf; do
|
|
[ ! -f "$file" ] && continue
|
|
if grep -q "$UUID" "$file"; then
|
|
INFO "Find $UUID in $file, use as conf"
|
|
FILE_CONFIG="$file"
|
|
fi
|
|
done
|
|
[ ! -f "$FILE_CONFIG" ] && ERRO "No config for $UUID"
|
|
source "$FILE_CONFIG"
|
|
;;
|
|
*)
|
|
echo "beesd [options] <btrfs_uuid>"
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
WORK_DIR="${WORK_DIR:-/run/bees/}"
|
|
MNT_DIR="${MNT_DIR:-$WORK_DIR/mnt/$UUID}"
|
|
BEESHOME="${BEESHOME:-$MNT_DIR/.beeshome}"
|
|
BEESSTATUS="${BEESSTATUS:-$WORK_DIR/$UUID.status}"
|
|
DB_SIZE="${DB_SIZE:-$((64*AL16M))}"
|
|
LOG_SHORT_PATH="${LOG_SHORT_PATH:-N}"
|
|
|
|
INFO "Check: Disk exists"
|
|
if [ ! -b "/dev/disk/by-uuid/$UUID" ]; then
|
|
ERRO "Missing disk: /dev/disk/by-uuid/$UUID"
|
|
fi
|
|
|
|
is_btrfs(){ [ "$(blkid -s TYPE -o value "$1")" == "btrfs" ]; }
|
|
|
|
INFO "Check: Disk with btrfs"
|
|
if ! is_btrfs "/dev/disk/by-uuid/$UUID"; then
|
|
ERRO "Disk not contain btrfs: /dev/disk/by-uuid/$UUID"
|
|
fi
|
|
|
|
INFO "WORK DIR: $WORK_DIR"
|
|
mkdir -p "$WORK_DIR" || exit 1
|
|
|
|
INFO "MOUNT DIR: $MNT_DIR"
|
|
mkdir -p "$MNT_DIR" || exit 1
|
|
|
|
umount_w(){ mountpoint -q "$1" && umount -l "$1"; }
|
|
force_umount(){ umount_w "$MNT_DIR"; }
|
|
trap force_umount SIGINT SIGTERM EXIT
|
|
|
|
mount -osubvolid=5 /dev/disk/by-uuid/$UUID "$MNT_DIR" || exit 1
|
|
|
|
if [ ! -d "$BEESHOME" ]; then
|
|
INFO "Create subvol $BEESHOME for store bees data"
|
|
btrfs sub cre "$BEESHOME"
|
|
else
|
|
btrfs sub show "$BEESHOME" &> /dev/null || ERRO "$BEESHOME MUST BE A SUBVOL!"
|
|
fi
|
|
|
|
# Check DB size
|
|
{
|
|
DB_PATH="$BEESHOME/beeshash.dat"
|
|
touch "$DB_PATH"
|
|
OLD_SIZE="$(du -b "$DB_PATH" | sed 's/\t/ /g' | cut -d' ' -f1)"
|
|
NEW_SIZE="$DB_SIZE"
|
|
if (( "$NEW_SIZE"%AL16M > 0 )); then
|
|
ERRO "DB_SIZE Must be multiple of 16M"
|
|
fi
|
|
if (( "$OLD_SIZE" != "$NEW_SIZE" )); then
|
|
INFO "Resize db: $OLD_SIZE -> $NEW_SIZE"
|
|
[ -f "$BEESHOME/beescrawl.$UUID.dat" ] && rm "$BEESHOME/beescrawl.$UUID.dat"
|
|
truncate -s $NEW_SIZE $DB_PATH
|
|
fi
|
|
chmod 700 "$DB_PATH"
|
|
}
|
|
|
|
MNT_DIR="${MNT_DIR//\/\//\/}"
|
|
|
|
filter_path(){
|
|
if YN $LOG_SHORT_PATH; then
|
|
sed -e "s#$MNT_DIR##g"
|
|
else
|
|
cat
|
|
fi
|
|
}
|
|
|
|
@LIBEXEC_PREFIX@/bees ${ARGUMENTS[@]} $OPTIONS "$MNT_DIR" 3>&1 2>&1 | filter_path
|
|
|
|
exit 0
|