feat: add telegram notification to docker image upgrade

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/quickadapter/user_data/config-template.json b/quickadapter/user_data/config-template.json
index aa32a1d170f81b004839410b1fc4b6b00087b260..c486b4e19c3cff562f70d2fad65923f11f082b5f 100644 (file)
--- a/quickadapter/user_data/config-template.json
+++ b/quickadapter/user_data/config-template.json
@@ -155,11 +155,11 @@
         // "1d"
       ],
       "label_period_candles": 24,
-      "label_metric": "seuclidean",
-      // "label_weights": [
-      //   0.5,
-      //   0.5
-      // ],
+      "label_metric": "euclidean",
+      "label_weights": [
+        0.6,
+        0.4
+      ],
       "include_shifted_candles": 6,
       "DI_threshold": 10,
       "weight_factor": 0.9,

diff --git a/scripts/docker-upgrade.sh b/scripts/docker-upgrade.sh
index f3b09f33c3fc518b3bcb3f1a1c93b3e0b20657d6..eec89eab0f14e08a723dcb7762a35d80a7a62485 100755 (executable)
--- a/scripts/docker-upgrade.sh
+++ b/scripts/docker-upgrade.sh
@@ -1,6 +1,7 @@
 #!/usr/bin/env sh
 set -eu
 
+FREQTRADE_CONFIG="user_data/config.json"
 LOCAL_DOCKER_IMAGE="reforcexy-freqtrade"
 REMOTE_DOCKER_IMAGE="freqtradeorg/freqtrade:stable_freqairl"
 
@@ -8,13 +9,122 @@ echo_timestamped() {
   command echo "$(date +"%Y-%m-%d %H:%M:%S") - $*"
 }
 
+LOCKFILE="/tmp/docker-upgrade.lock"
+if [ -f "$LOCKFILE" ]; then
+  echo_timestamped "Error: already running"
+  exit 1
+fi
+trap 'rm -f "$LOCKFILE"' EXIT
+touch "$LOCKFILE"
+
+jsonc_to_json() {
+  awk '
+    BEGIN{in_str=0;esc=0;in_block=0;have_prev=0}
+    {
+      line=$0; len=length(line); i=1; out="";
+      sub(/\r$/, "", line)
+      while(i<=len){
+        c=substr(line,i,1);
+        nextc = (i<len)?substr(line,i+1,1):"\n";
+        if(in_block){
+          if(c=="*" && nextc=="/"){ in_block=0; i+=2; }
+          else { i++; }
+          continue;
+        }
+        if(!in_str){
+          if(c=="/" && nextc=="/"){ break; }
+          if(c=="/" && nextc=="*"){ in_block=1; i+=2; continue; }
+          if(c=="\""){ in_str=1; out=out c; i++; continue; }
+          out=out c; i++;
+        } else {
+          out=out c;
+          if(esc){ esc=0; }
+          else if(c=="\\"){ esc=1; }
+          else if(c=="\""){ in_str=0; }
+          i++;
+        }
+      }
+      sub(/[[:space:]]+$/, "", out)
+      if (out ~ /^[[:space:]]*$/) next
+      cur = out
+      if (have_prev){
+        sub(/[[:space:]]*,[[:space:]]*([}\]])[[:space:]]*$/, "\\1", prev)
+        if (prev ~ /,[[:space:]]*$/ && cur ~ /^[[:space:]]*[}\]]/) {
+          sub(/,[[:space:]]*$/, "", prev)
+        }
+        key = (cur ~ /^[[:space:]]*"[^"]+"[[:space:]]*:/)
+        openval = (cur ~ /^[[:space:]]*[{[]/)
+        strval = (cur ~ /^[[:space:]]*"/) && !(key)
+        numval = (cur ~ /^[[:space:]]*-?[0-9]/)
+        boolnull = (cur ~ /^[[:space:]]*(true|false|null)\b/)
+        prev_value_end = (prev ~ /[}\]][[:space:]]*$/) || (prev ~ /"[[:space:]]*$/) || (prev ~ /-?[0-9]+([.][0-9]+)?([eE][+-]?[0-9]+)?[[:space:]]*$/) || (prev ~ /(true|false|null)[[:space:]]*$/)
+        if (prev_value_end && (key || openval || strval || numval || boolnull)) {
+          prev = prev ","
+        }
+        print prev
+      }
+      prev = cur
+      have_prev=1
+    }
+    END {
+      if (have_prev){
+        sub(/[[:space:]]*,[[:space:]]*([}\]])[[:space:]]*$/, "\\1", prev)
+        sub(/,[[:space:]]*$/, "", prev)
+        print prev
+      }
+    }
+  ' "$1" | jq -c '.'
+}
+
+send_telegram_message() {
+    if [ -z "${FREQTRADE_CONFIG_JSON:-}" ]; then
+      FREQTRADE_CONFIG_JSON=$(jsonc_to_json "$FREQTRADE_CONFIG" 2>/dev/null || echo "")
+    fi
+    printf '%s' "$FREQTRADE_CONFIG_JSON" | jq empty 2>/dev/null || { echo_timestamped "Error: invalid config JSON"; exit 1; }
+
+    freqtrade_telegram_enabled=$(printf '%s' "$FREQTRADE_CONFIG_JSON" | jq -r '.telegram.enabled // "false"' 2>/dev/null || echo "false")
+    if [ "$freqtrade_telegram_enabled" = "false" ]; then
+        return 0
+    fi
+
+    telegram_message="$1"
+    if [ -z "$telegram_message" ]; then
+        echo_timestamped "Error: message variable is empty"
+        return 1
+    fi
+
+    freqtrade_telegram_token=$(printf '%s' "$FREQTRADE_CONFIG_JSON" | jq -r '.telegram.token // ""' 2>/dev/null || echo "")
+    freqtrade_telegram_chat_id=$(printf '%s' "$FREQTRADE_CONFIG_JSON" | jq -r '.telegram.chat_id // ""' 2>/dev/null || echo "")
+    if [ -n "$freqtrade_telegram_token" ] && [ -n "$freqtrade_telegram_chat_id" ]; then
+      curl_error=$(command curl -s -X POST \
+        --data-urlencode "text=${telegram_message}" \
+        --data-urlencode "parse_mode=markdown" \
+        --data "chat_id=$freqtrade_telegram_chat_id" \
+        "https://api.telegram.org/bot${freqtrade_telegram_token}/sendMessage" 2>&1 1>/dev/null)
+        if [ $? -ne 0 ]; then
+            echo_timestamped "Error: failed to send telegram notification: $curl_error"
+            return 1
+        fi
+    fi
+}
+
+if ! command -v jq >/dev/null 2>&1; then
+  echo_timestamped "Error: jq not found in PATH"
+  exit 1
+fi
+
 if ! command -v docker >/dev/null 2>&1; then
   echo_timestamped "Error: docker not found in PATH"
   exit 1
 fi
 
 if [ ! -f "docker-compose.yml" ] && [ ! -f "docker-compose.yaml" ]; then
-  echo_timestamped "Error: docker-compose.yml or docker-compose.yaml not found in current directory"
+  echo_timestamped "Error: docker-compose.yml or docker-compose.yaml file not found in current directory"
+  exit 1
+fi
+
+if [ ! -f "$FREQTRADE_CONFIG" ]; then
+  echo_timestamped "Error: $FREQTRADE_CONFIG file not found from current directory"
   exit 1
 fi
 
@@ -29,13 +139,17 @@ remote_digest=$(command docker image inspect --format='{{.Id}}' "$REMOTE_DOCKER_
 rebuild_local_image=false
 if [ "$local_digest" != "$remote_digest" ]; then
   rebuild_local_image=true
-  echo_timestamped "Info: docker image ${REMOTE_DOCKER_IMAGE} was updated ($local_digest -> $remote_digest)"
+  message="docker image ${REMOTE_DOCKER_IMAGE} was updated ($local_digest -> $remote_digest)"
+  echo_timestamped "Info: $message"
+  send_telegram_message "$message"
 else
   echo_timestamped "Info: docker image ${REMOTE_DOCKER_IMAGE} is up to date"
 fi
 
 if [ "$rebuild_local_image" = true ]; then
-  echo_timestamped "Info: rebuilding and restarting docker image ${LOCAL_DOCKER_IMAGE}"
+  message="rebuilding and restarting docker image ${LOCAL_DOCKER_IMAGE}"
+  echo_timestamped "Info: $message"
+  send_telegram_message "$message"
   if ! command docker compose --progress quiet down; then
     echo_timestamped "Error: docker compose down failed"
     exit 1
@@ -47,7 +161,9 @@ if [ "$rebuild_local_image" = true ]; then
     echo_timestamped "Error: docker compose up failed"
     exit 1
   fi
-  echo_timestamped "Info: rebuilt and restarted docker image ${LOCAL_DOCKER_IMAGE}"
+  message="rebuilt and restarted docker image ${LOCAL_DOCKER_IMAGE}"
+  echo_timestamped "Info: $message"
+  send_telegram_message "$message"
 else
   echo_timestamped "Info: no rebuild and restart needed for docker image ${LOCAL_DOCKER_IMAGE}"
 fi