tss/aptly/snaper at main · hadret/tss · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
#!/usr/bin/env bash
set -euo pipefail

# Check for required dependencies
for dep in aptly gum; do
  command -v "$dep" >/dev/null 2>&1 || {
    printf "Missing dependency: %s :(\n" "$dep" >&2
    exit 1
  }
done

# It's a simple script, print a simple help please
usage() {
  printf '%s\n' "Usage: ${0##*/} [-a] [-c] [-d] [-h] [-l] [-s]

FLAGS
    -a  List all snapshots
    -c  Create new snapshot(s)
    -d  Drop snapshot(s)
    -h  Print help
    -l  List snapshots per chosen distro
    -s  Show snapshot details" 1>&2
  exit 3
}

# Pick a distro(s)
pick_me_pick_me() {
  mapfile -t distros < <(gum choose --no-limit "bookworm" "trixie" "unstable")
}

# Create snapshot
create_snap() {
  pick_me_pick_me

  for distro in "${distros[@]}"; do
    # Ensure gum choose did its job
    [ -z "$distro" ] && echo "No distro was picked, exiting..." && exit 1

    # Catch already existing snapshot for the distro
    printf '\nPreparing snapshot for %s...\n' "${distro}-main"
    mapfile -t snaps < <(aptly snapshot list -raw | grep "$distro")
    printf '\nHere are current snaps for %s:\n' "$distro"
    printf ' %s\n' "${snaps[@]}"

    # New snapshot suggested name
    new_snap=$distro-$(date +%Y%m%d)-1

    # Confirm whether new name is OK, based on pre-existing snapshots
    gum confirm "Create $new_snap?" && snap_choice=true || snap_choice=false

    # Ask for a new name when not happy with the default
    if [ "$snap_choice" = false ]; then
      new_snap=$(gum input --value="$new_snap" --header="Provide name for the new snap")
    fi

    # Create new snapshot
    printf '\nCreating new snap %s...\n' "$new_snap"
    aptly snapshot create "$new_snap" from repo "${distro}-main"
    printf '%s created successfully!\n\n' "$new_snap"
  done
}

drop_snap() {
  pick_me_pick_me

  for distro in "${distros[@]}"; do
    # Ensure gum choose did its job
    [ -z "$distro" ] && echo "No distro was picked, exiting..." && exit 1

    # Grab currently published snapshot
    current=$(aptly publish list -json \
              | jq -r --arg distro "$distro" '.[] | select(.Distribution == $distro) | .Sources[].Name')

    # Fetch all available snapshots and remove currently set one
    mapfile -t snaps < <(aptly snapshot list -raw | grep "^$distro" | grep -v "$current")

    # Print message if no snapshots available to drop
    if [ -z "${snaps[*]}" ]; then
      echo "No snapshots available to drop for $distro!"
    # Pick & drop'em otherwise
    else
      snaps_were_made=$(gum choose --header="Drop snapshots for $distro:" --no-limit "${snaps[@]}")

      # Check whether anything got selected for removal
      if [ -z "$snaps_were_made" ]; then
        echo "No snapshot selected for dropping for $distro!"
      # Remove all the things
      else
        echo "$snaps_were_made" | xargs -n1 aptly snapshot drop
      fi
    fi
  done
}

# Show snapshot details
show_snap() {
  aptly snapshot show "$(gum choose $(aptly snapshot list -raw))"
}

# List available snaps per distro
list_snaps() {
  pick_me_pick_me

  for distro in "${distros[@]}"; do
    # Ensure gum choose did its job
    [ -z "$distro" ] && echo "No distro was picked, exiting..." && exit 1

    aptly snapshot list -raw | grep "$distro"
  done
}

# List all available snaps
all_snaps() {
  aptly snapshot list
}

case $# in
0) create_snap ;;
1)
    case "$1" in
    -c) create_snap ;;
    -d | -r) drop_snap ;;
    -s) show_snap ;;
    -l) list_snaps ;;
    -a) all_snaps ;;
    -h) usage ;;
    *) usage ;;
    esac
    ;;
*) usage ;;
esac