#!/bin/bash

set -eu

function snap {
    unsnap

    znap-list | xargs -n1 znap-mount /a
    mkdir -p /a/boot
    mount --bind /boot /a/boot
}

function unsnap {
    umount /a/boot || true
    rmdir /a/boot || true
    znap-umount-all /a
}

if [ "$(id -u)" -ne 0 ]; then
    echo "Must be root"
    exit 1
fi

if [ "${1:-}" = "off" ]; then
    unsnap
elif [ "${1:-}" = "on" ]; then
    snap
else
    echo "Usage: $0 on|off"
    exit 1
fi