#!/bin/bash
set -euo pipefail

zfs-clean () {
    if [ -z "$1" ]; then
        exit 1
    elif [ "$1" == "upgrades" ]; then
        if [ -z "$2" ]; then
            zfs list -t snapshot | grep 'apt-' | sed 's| .*$||g' | head -n -5 | xargs -d '\n' -I {} sudo zfs destroy {}
        else
            zfs list -t snapshot | grep "${2}" | sed 's| .*$||g' | head -n -5 | xargs -d '\n' -I {} sudo zfs destroy {}
        fi
    fi
}