new: [send] add cron hourly for disk_usage and load_average_max

10 months ago · 71ace8b78f
7 changed files with 218 additions and 21 deletions
--- a/bin/myc-install
+++ b/bin/myc-install
@ -145,6 +145,40 @@ pip install ovh ||
 docker pull docker.0k.io/php:7.4-myc && docker tag docker.0k.io/php:7.4-myc myc_frontend &&
 docker pull docker.0k.io/cron:jessie && docker tag docker.0k.io/cron:jessie myc_cron &&

+## Copy Ntfy key to root
+umask 077
+cat /opt/apps/myc-manage/etc/ssh/ntfy-key | gpg -c --batch --yes --passphrase 'uniquepass' > /root/.ssh/ntfy-key || >&2 echo "Error while copying ntfy key to root"
+
+## Request token to ntfy server and add to config file 
+ssh-keyscan -H core-01.0k.io >> /root/.ssh/known_hosts || >&2 echo "Error while adding ntfy server to known_hosts"
+cred=$(ssh -i /root/.ssh/ntfy-key ntfy@core-01.0k.io request-token) || >&2 echo "Error while requesting token to ntfy server"
+login_ntfy=$(printf "%s" "${cred%$'\n'*}")
+password_ntfy=$(printf "%s" "${cred#$'\n'*}")
+
+## if the config file exist and LOGIN PASSWORD ARE already in we do nothing
+if [ -z "$login_ntfy"] || [[ "$login_ntfy" == *$'\n'*]]; then 
+    echo "Error: couldn’t infer credential from ntfy server" >&2;
+    printf "%s" "$cred" | sed -r 's/^ |/g' >&2; 
+    exit
+fi
+
+config_file="/etc/ntfy/ntfy.conf"
+mkdir -p "${config_file%/*}"
+if [ -f "$config_file" ] || touch $config_file || {
+    echo "Error: couldn’t create config file $config_file" >&2;
+    exit 1
+}; then
+    if grep -qE '^LOGIN=' "$config_file"; then
+        sed -i "s/^LOGIN=.*/LOGIN='$login'/" "$config_file"
+    else
+        echo "LOGIN='$login'" >> "$config_file"
+    fi
+    if grep -qE '^PASSWORD=' "$config_file"; then
+        sed -i "s/^PASSWORD=.*/PASSWORD='$password'/" "$config_file"
+    else
+        echo "PASSWORD='$password'" >> "$config_file"
+    fi
+fi

 ## Marker to probe if this script finished it's job
 echo "done" > /var/run/myc-installer.0k.io.state
--- a/bin/myc-update
+++ b/bin/myc-update
@ -113,6 +113,42 @@ sed -ri 's/^# (export LS_OPTIONS=.--color=auto.)/\1/;
 EOF
 fi

+Wrap -d "Update authorization to send to ntfy server " <<'EOF' || exit 1
+mkdir -p /root/.ssh 
+umask 077
+cat /opt/apps/myc-manage/etc/ssh/ntfy-key | gpg -c --batch --yes --passphrase 'uniquepass' > /root/.ssh/ntfy-key || >&2 echo "Error while copying ntfy key to root"
+
+## Request token to ntfy server and add to config file 
+ssh-keyscan -H core-01.0k.io >> /root/.ssh/known_hosts || >&2 echo "Error while adding ntfy server to known_hosts"
+cred=$(ssh -i /root/.ssh/ntfy-key ntfy@core-01.0k.io request-token) || >&2 echo "Error while requesting token to ntfy server"
+login_ntfy=$(printf "%s" "${cred%$'\n'*}")
+password_ntfy=$(printf "%s" "${cred#$'\n'*}")
+
+## if the config file exist and LOGIN PASSWORD ARE already in we do nothing
+if [ -z "$login_ntfy"] || [[ "$login_ntfy" == *$'\n'*]]; then 
+    echo "Error: couldn’t infer credential from ntfy server" >&2;
+    printf "%s" "$cred" | sed -r 's/^ |/g' >&2; 
+    exit
+fi
+
+config_file="/etc/ntfy/ntfy.conf"
+mkdir -p "${config_file%/*}"
+if [ -f "$config_file" ] || touch $config_file || {
+    echo "Error: couldn’t create config file $config_file" >&2;
+    exit 1
+}; then
+    if grep -qE '^LOGIN=' "$config_file"; then
+        sed -i "s/^LOGIN=.*/LOGIN='$login'/" "$config_file"
+    else
+        echo "LOGIN='$login'" >> "$config_file"
+    fi
+    if grep -qE '^PASSWORD=' "$config_file"; then
+        sed -i "s/^PASSWORD=.*/PASSWORD='$password'/" "$config_file"
+    else
+        echo "PASSWORD='$password'" >> "$config_file"
+    fi
+fi
+EOF

 for keyfile in {/root,/home/debian}/.ssh/authorized_keys; do
    [ -e "$keyfile" ] || continue
--- a/bin/send
+++ b/bin/send
@ -5,35 +5,89 @@
 if [[ "$UID" == "0" ]]; then
    NTFY_CONFIG_FILE="/etc/ntfy/ntfy.conf"
 else
-    NTFY_CONFIG_FILE="$HOME/.config/ntfy/ntfy.conf"
+    NTFY_CONFIG_FILE=~/.config/ntfy/ntfy.conf
 fi

+SERVER="https://ntfy.0k.io/"
+
 if ! [ -e "$NTFY_CONFIG_FILE" ]; then
    mkdir -p "${NTFY_CONFIG_FILE%/*}"
    ## default option to change if needed
-    echo 'SERVER="https://ntfy.0k.io/"' > "$NTFY_CONFIG_FILE"
-else
-    source "$NTFY_CONFIG_FILE"
-
-    for var in TOKEN SERVER; do
-        if ! [ -v "$var" ]; then
-            echo "Error: missing $var in $NTFY_CONFIG_FILE"
-            exit 1
-        fi
-    done
+    echo "SERVER=$SERVER" > "$NTFY_CONFIG_FILE"
+elif ! grep -q "^SERVER=" "$NTFY_CONFIG_FILE"; then
+    echo "SERVER=$SERVER" >> "$NTFY_CONFIG_FILE"
 fi

+source "$NTFY_CONFIG_FILE"
+
+for var in SERVER LOGIN PASSWORD; do
+    if ! [ -v "$var" ]; then
+        echo "Error: missing $var in $NTFY_CONFIG_FILE"
+        exit 1
+    fi
+done
+
+
 exname=${0##*/}
-usage="Usage: $exname CHANNEL MESSAGE"
+channel="main" 
+
+usage="Usage: $exname [-c CHANNEL] [-t TITLE ] MESSAGE 
+----------------------------------------------
+--- Send MESSAGE with TITLE to the specified CHANNEL. ---
+----------------------------------------------
+If no CHANNEL is provided, the message will be sent to the default channel
+Default CHANNEL is format as follow : ConfiguredLOGIN_${default_channel}
+If no TITLE is provided, the message will be sent with the hostname as title."
+
+while [[ $# -gt 0 ]]; do
+    arg="$1"
+    shift 
+    case "$arg" in
+        -h|--help)
+            echo "$usage"
+            exit 0
+            ;;
+        -c|--channel)
+            channel="$1"
+            [ -z "$channel" ] || {
+                echo "Error: no argument for channel option." >&2
+                echo "$usage" >&2
+                exit 1
+            }
+            shift 
+            ;;
+        -t|--title)
+            title="$1"
+            [ -z "$title" ] || {
+                echo "Error: no argument for title option." >&2
+                echo "$usage" >&2
+                exit 1
+            }
+            shift 
+            ;;
+        *)  
+            [ -z "$message" ] && { message="$arg"; continue; }
+            echo "Error : Unexpected positional argument '$arg'." >&2
+            echo "$usage" >&2
+            exit 1
+            ;;
+    esac
+done

-if [ "$#" -ne 2 ]; then
+[ -n "$message" ] || {
+    echo "Error: missing message." >&2
    echo "$usage" >&2
    exit 1
-fi
+}

-channel="$1"
-message="$2"
+curl_opts=(
+    -s 
+    -u "$LOGIN:$PASSWORD"
+    -d "$message"
+)

-curl -s -H "Authorization: Bearer $TOKEN" \
-     -d "$message" "$SERVER/$channel" > /dev/null
+if [ -n "$title" ]; then
+    curl_opts+=(-H "Title: [$(hostname)] $title")
+fi

+curl "${curl_opts[@]}" "$SERVER/${LOGIN}_$channel" > /dev/null
--- a/bin/vps
+++ b/bin/vps
@ -2615,4 +2615,73 @@ cmdline.spec:monujo:cmd:set-version:run() {
 }


+
+cmdline.spec::cmd:check:run() {
+    :
+}
+
+cmdline.spec.gnu check
+cmdline.spec:check:cmd:backup:run() {
+
+    : :optfla: --notify,-n      "Send result through notify"
+    : :optval: --service,-s      "The backup service name (defaults to 'rsync-backup')"
+
+
+    ## Check on daily bases if backup exist in config and when is the last backup done :  
+    ## ALERT if backup is set and last backup is older than 24h 
+    
+    local STATE_FILE="/var/run/myc-manage/backup.state"
+    mkdir -p "${STATE_FILE%/*}"
+
+    service=${opt_service:-rsync-backup}
+    project_name=$(compose:project_name) || exit 1
+
+    ## check if service exists in compose.yml
+    if ! compose:service:exists "$project_name" "$service"; then
+        warn "no service ${DARKYELLOW}$service${NORMAL}. Ignoring."
+        return 0
+    fi
+
+    last_backup_datetime=$(
+        cat /srv/datastore/data/cron/var/log/cron/*rsync-backup_script{_*,}.log | grep "total size is" | sort | tail -n 1 | cut -f -2 -d " ")
+    last_backup_ts=$(date -d "$last_backup_datetime" +%s)
+    max_ts=$(date -d "24 hours ago" +%s)
+
+    state="ok"
+    if [ "$last_backup_ts" -lt "$max_ts" ]; then
+        state="ko"
+    fi
+
+    if [ -z "$opt_notify" ]; then
+        if [ "$state" == "ok" ]; then
+            info "Everything is ${GREEN}ok${NORMAL}. (last backup: ${DARKCYAN}$last_backup_datetime${NORMAL})"
+            return 0
+        fi
+        warn "Last backup older than 1 day. (last backup: ${DARKCYAN}$last_backup_datetime${NORMAL})"
+        return 1
+    fi
+
+    ## notify
+
+    last_state=$(cat "$STATE_FILE" 2>/dev/null) || true 
+    if [ "$state" == "$last_state" ]; then
+       [ "$state" == "ko" ] || return 0
+       is_old=$(find "$STATE_FILE" -type f -mtime +2) || return 1
+       [ -n "$is_old" ] || return 0
+    fi
+   
+    echo "$state" > "$STATE_FILE"
+
+    message="[$(hostname)]: WARNING no backup done in the last 24h (No backup since $days days and $hours hours)"
+
+    timestamp=$(date +%s)
+    time_difference=$((timestamp - last_backup_ts))
+    days=$((time_difference / 86400))
+    hours=$((time_difference % 86400 / 3600))
+    message="WARNING: no backup done in the last 24h (No backup since $days days and $hours hours)"
+    send -t "ALERT Backup" "$message"
+        
+}
+
+
 cmdline::parse "$@"
--- a/etc/cron.d/check-backup
+++ b/etc/cron.d/check-backup
@ -0,0 +1,4 @@
+SHELL=/bin/bash
+PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin
+
+12 * * * *   root   vps check backup -n 2>&1 | logger -t stats
--- a/etc/ssh/ntfy-key
+++ b/etc/ssh/ntfy-key