From 92efdb08484cb163d4f7167c7cceacbfd87049e2 Mon Sep 17 00:00:00 2001
From: Valentin Lab <valentin.lab@kalysto.org>
Date: Tue, 2 Apr 2024 16:29:27 +0200
Subject: [PATCH] new: [0km] add ``vps-subscribe {add,rm} CHANNEL TOPICS``
 actions

---
 bin/0km | 127 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 127 insertions(+)

diff --git a/bin/0km b/bin/0km
index 4f13f5b..8bb3393 100755
--- a/bin/0km
+++ b/bin/0km
@@ -656,6 +656,92 @@ EOF
 
 }
 
+NTFY_TOPIC_FILE="/etc/ntfy/topics.yml"
+subscribe:ntfy:exists() {
+    local vps="$1"
+    if ! out=$(echo "[ -f \"$NTFY_TOPIC_FILE\" ] && echo ok || true" | \
+                   ssh:run "root@$vps" -- bash); then
+        err "Unable to check for existence of '$NTFY_TOPIC_FILE'."
+    fi
+    if [ -z "$out" ]; then
+        err "File '$NTFY_TOPIC_FILE' not found on $vps."
+        return 1
+    fi
+}
+
+
+ntfy:rm() {
+    local channel="$1" topic="$2" vps="$3"
+    subscribe:ntfy:exists "$vps" || return 1
+    if ! out=$(echo "yq -i 'del(.[\"$channel\"][] | select(. == \"$TOPIC\"))' \"$NTFY_TOPIC_FILE\"" | \
+                   ssh:run "root@$vps" -- bash); then
+        err "Failed to remove channel '$channel' from '$NTFY_TOPIC_FILE'."
+        return 1
+    fi
+    info "Channel '$channel' removed from '$NTFY_TOPIC_FILE' on $vps."
+    ssh:run "root@$vps" -- cat "$NTFY_TOPIC_FILE"
+}
+
+ntfy:add() {
+    local vps="$1"
+    read-0 channel topic || {
+        err "Couldn't read CHANNEL and TOPIC arguments."
+        return 1
+    }
+    vps_connection_check "$vps" </dev/null || return 1
+    subscribe:ntfy:exists "$vps" || return 1
+    if ! out=$(echo "yq '. | has(\"$channel\")' \"$NTFY_TOPIC_FILE\"" | \
+                   ssh:run "root@$vps" -- bash); then
+        err "Failed to check if channel '$channel' with topic '$topic' is already in '$NTFY_TOPIC_FILE'."
+        return 1
+    fi
+    if [ "$out" != "true" ]; then
+        ## Channel does not exist
+        if ! out=$(echo "yq -i '.[\"$channel\"] = []' \"$NTFY_TOPIC_FILE\"" | \
+                       ssh:run "root@$vps" -- bash); then
+            err "Failed to create a new channel '$channel' entry in '$NTFY_TOPIC_FILE'."
+            return 1
+        fi
+    else
+        ## Channel exists
+        if ! out=$(echo "yq '.[\"$channel\"] | any_c(. == \"$topic\")' \"$NTFY_TOPIC_FILE\"" | \
+                       ssh:run "root@$vps" -- bash); then
+            err "Failed to check if channel '$channel' with topic '$topic' is already in '$NTFY_TOPIC_FILE'."
+            return 1
+        fi
+        if [ "$out" == "true" ]; then
+            info "Channel '$channel' with topic '$topic' already exists in '$NTFY_TOPIC_FILE'."
+            return 0
+        fi
+    fi
+    if ! out=$(echo "yq -i '.[\"$channel\"] += [\"$topic\"]' \"$NTFY_TOPIC_FILE\"" | \
+                   ssh:run "root@$vps" -- bash); then
+        err "Failed to add channel '$channel' with topic '$topic' to '$NTFY_TOPIC_FILE'."
+        return 1
+    fi
+    info "Channel '$channel' added with topic '$topic' to '$NTFY_TOPIC_FILE' on $vps."
+}
+
+NTFY_SERVER="https://ntfy.0k.io"
+subscribe:add() {
+    local vps="$1"
+    read-0 channel topic || {
+        err "Couldn't read CHANNEL and TOPIC arguments."
+        return 1
+    }
+    vps_connection_check "$vps" </dev/null || return 1
+    ntfy:add "$channel" "$topic" "$vps"
+}
+
+subscribe:rm() {
+    local vps="$1"
+    read-0 channel topic || {
+        err "Couldn't read CHANNEL and TOPIC arguments."
+        return 1
+    }
+    vps_connection_check "$vps" </dev/null || return 1
+    ntfy:rm "$channel" "$topic" "$vps"
+}
 vps_backup_recover() {
     local vps="$1" admin server id path rtype force type
 
@@ -1594,4 +1680,45 @@ graph:def:load_avg() {
 }
 
 
+
+
+cmdline.spec.gnu vps-subscribe
+cmdline.spec::cmd:vps-subscribe:run() {
+    :
+}
+cmdline.spec.gnu add
+cmdline.spec:vps-subscribe:cmd:add:run() {
+
+    : :posarg: CHANNEL         'Channel which will be sent to given topic'
+    : :posarg: TOPIC           'Ntfy topic to recieve messages of given channel
+                               (format: "[MYSERVER:]MYTOPICS"
+                                 Examples: "ntfy.0k.io:main,storage,alerts",
+                                 "main{1,3,7}"
+                               )'
+    : :posarg: [VPS...]        'Target host(s) to get stats'
+
+    printf "%s\0" "$CHANNEL" "$TOPIC" |
+        vps_mux subscribe:add  "${VPS[@]}"
+
+}
+
+cmdline.spec.gnu rm
+cmdline.spec:vps-subscribe:cmd:rm:run() {
+
+    : :posarg: CHANNEL         'Channel which will be sent to given topic'
+    : :posarg: TOPIC           'Ntfy topic to recieve messages of given channel
+                               (format: "[MYSERVER:]MYTOPICS"
+                                 Examples: "ntfy.0k.io:main,storage,alerts",
+                                 "main{1,3,7}"
+                               )'
+    : :posarg: [VPS...]        'Target host(s) to get stats'
+
+
+    printf "%s\0" "$CHANNEL" "$TOPIC" |
+        vps_mux subscribe:rm "${VPS[@]}"
+
+}
+
+
+
 cmdline::parse "$@"