#!/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.enc
        echo $dst/$latest_filename.zfs.zst.enc
    }
}

enccmd() {
    ~stargrave/bin/cmenctool 4</home/stargrave/.cm/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.enc
        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:t:r
        [[ $latest_dst != $latest_filename ]]
        setopt PIPE_FAIL
        set -x
        zfs send -Rwv -i ${latest_dst:gs#%#/} $latest | zstdmt | enccmd > \
            $dst/$latest_filename.zfs.zst.enc
        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