#!/bin/bash

# archypr:summary=Create or restore system snapshots with snapper
# archypr:args=<create|restore>
# archypr:requires-sudo=true

set -e

COMMAND="$1"
ARCHYPR_PATH=${ARCHYPR_PATH:-$HOME/.local/share/omarchy}

if [[ -z $COMMAND ]]; then
  echo "Usage: archypr-snapshot <create|restore>" >&2
  exit 1
fi

if ! command -v snapper &>/dev/null; then
  exit 127 # archypr-update can use this to just ignore if snapper is not available
fi

case "$COMMAND" in
create)
  DESC="$(archypr-ersion)"

  echo -e "\e[32mCreate system snapshot\e[0m"

  # Get existing snapper config names from CSV output
  mapfile -t CONFIGS < <(sudo snapper --csvout list-configs | awk -F, 'NR>1 {print $1}')

  for config in "${CONFIGS[@]}"; do
    sudo snapper -c "$config" create -c number -d "$DESC"
    sudo snapper -c "$config" cleanup number
  done
  echo
  ;;
restore)
  sudo limine-snapper-restore
  ;;
esac