#!/usr/bin/env zsh

setopt ERR_EXIT EXTENDED_GLOB

usage() {
    cat <<EOF
Usage: $0 DATASET@SNAPPREFIX snap-initial
       $0 DATASET@SNAPPREFIX snap
       $0 DATASET@SNAPPREFIX sync-initial DST
       $0 DATASET@SNAPPREFIX sync DST
       $0 DATASET@SNAPPREFIX clean COUNT
       FIFO=1
EOF
    exit 1
}

mk_fifo() {
    [[ -z $FIFO ]] || {
        mkfifo $dst/$latest_filename.zfs.zst.age
        echo $dst/$latest_filename.zfs.zst.age
    }
}

enccmd=(~stargrave/bin/age -R /home/stargrave/.age/general.pub)

[[ $# -ge 2 ]] || usage

what=$1
action=$2

zmodload -F zsh/datetime b:strftime
now=$(strftime %Y%m%d%H%M)
[[ $what =~ / ]] && depth=1 || depth=${#${(s./.)what}}
snaps=$(zfs list -t snap -d $depth -o name -H | grep $what | sort -nr)
snaps=(${(f)snaps})
latest=${snaps[1]}
latest_filename=${latest:gs#/#%}
dst=$3

case $action in
    snap-initial)
        zfs snap -r $what-$now
        print $what-$now
        ;;
    snap)
        [[ -n $latest ]]
        [[ "$what-$now" != $latest ]]
        zfs snap -r $what-$now
        print $what-$now
        ;;
    sync-initial)
        [[ -d $dst ]] || usage
        mk_fifo
        setopt PIPE_FAIL
        zfs send -Rwv $latest | zstdmt | $enccmd > $dst/$latest_filename.zfs.zst.age
        sync
        touch $dst/$latest_filename.from
        ;;
    sync)
        [[ -d $dst ]] || usage
        mk_fifo
        latest_dst=($dst/*.from~$dst/.*(.Onn[1]))
        [[ $latest_dst ]]
        latest_dst=${latest_dst[1]}
        latest_dst=${${latest_dst##*/}%.from}
        [[ $latest_dst != $latest_filename ]]
        setopt PIPE_FAIL
        set -x
        zfs send -Rwv -i ${latest_dst:gs#%#/} $latest | zstdmt | $enccmd > \
            $dst/$latest_filename.zfs.zst.age
        set +x
        sync
        print $latest_dst > $dst/$latest_filename.from
        ;;
    clean)
        count=$3
        [[ -n $count ]] || usage
        count=$(( $count + 1 ))
        for snap in $snaps ; do
            count=$(( $count - 1 ))
            [[ $count -le 0 ]] || continue
            zfs destroy -Rv $snap
        done
        ;;
    *)
        usage
        ;;
esac