#!/bin/bash
# Copyright (C) 2016 Ian Kelling

# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

[[ $EUID == 0 ]] || exec sudo -E "$BASH_SOURCE" "$@"

x="$(readlink -f "$BASH_SOURCE")"; source "${x%/*}/bash-trace"
cd ${x%/*}

usage() {
    cat <<EOF
Usage: ${0##*/} [-h|--help]
reinstall chboot to /boot subvols, for chboot updates.

We install to /boot in case there is an issue booting and only the /boot
vol is readily available. For the bootstrap subvol, this is the normal
case.
EOF
    exit $1
}
case $1 in
    -h|--help) usage ;;
esac


e() { echo "$@"; "$@"; }

boot_dev=$(mount | sed -rn "s#^(\S+) on /boot .*#\1#p")
mount_point=$(mktemp -d)
e mount -o subvolid=0 $boot_dev $mount_point

shopt -s nullglob
for dir in $mount_point/*; do
    btrfs subvol show $dir &>/dev/null || continue
    if [[ -e $dir/boot ]]; then
        dir=$dir/boot
    fi
    e install -m 755 -o root -g root bash-trace $dir
    e install -m 755 -o root -g root chboot $dir
done
e umount $mount_point
e rmdir $mount_point