tlg-wotd: generate fance image

big thanks to @soerface
2026-03-16 18:21:07 +01:00 · 2024-09-22 00:27:51 +02:00
parent 09c631b4e5
commit 9b0b4ca08c
1 changed files with 110 additions and 12 deletions
--- a/configs/telegram-bots/tlg-wotd.nix
+++ b/configs/telegram-bots/tlg-wotd.nix
@@ -4,19 +4,117 @@
  config,
  ...
 }: {
-  niveum.telegramBots.tlg-wotd = {
+  systemd.services.telegram-bot-tlg-wotd = {
    enable = true;
-    time = "9:30";
-    chatIds = ["@tlgwotd"];
-    tokenFile = config.age.secrets.telegram-token-kmein.path;
-    command = toString (pkgs.writers.writeDash "tlg-wotd" ''
-      ${pkgs.curl}/bin/curl -sSL http://stephanus.tlg.uci.edu/Iris/Wotd \
-      | ${pkgs.recode}/bin/recode html..utf8 \
-      | ${pkgs.jq}/bin/jq -r '
-        "*\(.word)* '\'''\(.definition | sub("<.*>"; "") | rtrimstr(" "))'\'''\n\nFirst occurrence: \(.firstOccurrence)\nNumber of occurrences: \(.totalOccurrences)"
-      '
-    '');
-    parseMode = "Markdown";
+    wantedBy = ["multi-user.target"];
+    wants = ["network-online.target"];
+    startAt = "9:30";
+    path = [ pkgs.jq pkgs.curl pkgs.recode pkgs.deno pkgs.imagemagick pkgs.gawk pkgs.gnugrep pkgs.coreutils ];
+    environment = {
+      NPM_CONFIG_CACHE = "/tmp";
+      DENO_DIR = "/tmp";
+    };
+    script = ''
+      set -efux
+
+      chat_id=@tlgwotd
+
+      export TOKEN="$(cat "$CREDENTIALS_DIRECTORY/token")"
+
+      json_data=$(curl -sSL http://stephanus.tlg.uci.edu/Iris/Wotd | recode html..utf8)
+
+      word=$(echo "$json_data" | jq -r '.word')
+      compact_word=$(echo "$word" | sed 's/,.*$//')
+      definition=$(echo "$json_data" | jq -r '.definition | sub("<.*>"; "") | rtrimstr(" ")')
+      first_occurrence=$(echo "$json_data" | jq -r '.firstOccurrence')
+      total_occurrences=$(echo "$json_data" | jq -r '.totalOccurrences')
+
+      transliteration=$(deno run ${pkgs.writeText "translit.ts" ''
+        import grc from "npm:greek-transliteration";
+        console.log(grc.transliterate(Deno.args.join(" ")));
+      ''} "$compact_word")
+
+
+      photo_path=/tmp/output.png
+
+      hex_to_rgb() {
+          hex="$1"
+          r=$(printf "%d" "0x$(echo "$hex" | cut -c2-3)")
+          g=$(printf "%d" "0x$(echo "$hex" | cut -c4-5)")
+          b=$(printf "%d" "0x$(echo "$hex" | cut -c6-7)")
+          echo "$r $g $b"
+      }
+
+      calculate_luminance() {
+          r="$1"
+          g="$2"
+          b="$3"
+
+          r_l=$(echo "$r" | awk '{print ($1 / 255 <= 0.03928) ? $1 / 255 / 12.92 : (($1 / 255 + 0.055) / 1.055)^2.4}')
+          g_l=$(echo "$g" | awk '{print ($1 / 255 <= 0.03928) ? $1 / 255 / 12.92 : (($1 / 255 + 0.055) / 1.055)^2.4}')
+          b_l=$(echo "$b" | awk '{print ($1 / 255 <= 0.03928) ? $1 / 255 / 12.92 : (($1 / 255 + 0.055) / 1.055)^2.4}')
+
+          echo "$r_l $g_l $b_l" | awk '{print 0.2126*$1 + 0.7152*$2 + 0.0722*$3}'
+      }
+
+
+      hex_color="#$(echo "$compact_word" | md5sum | cut -c 1-6)"
+      if echo "$hex_color" | grep -qE '^#[0-9A-Fa-f]{6}$'; then
+          set -- $(hex_to_rgb "$hex_color")
+          r="$1"
+          g="$2"
+          b="$3"
+      fi
+
+    luminance=$(calculate_luminance "$r" "$g" "$b")
+
+    threshold="0.1"
+    echo "$r $g $b"
+    if [ "$(echo "$luminance" | awk -v threshold="$threshold" '{print ($1 > threshold)}')" -eq 1 ]; then
+        color1="black"
+        color2="#333"
+    else
+        color1="white"
+        color2=lightgrey
+    fi
+
+    magick -size 1400x846 \
+        xc:"$hex_color" \
+        -font "${pkgs.gentium}/share/fonts/truetype/GentiumBookPlus-Bold.ttf" \
+        -fill "$color1" \
+        -pointsize 150 -gravity west \
+        -annotate +100-160 "$compact_word" \
+        -font "${pkgs.gentium}/share/fonts/truetype/GentiumBookPlus-Regular.ttf" \
+        -fill "$color2" \
+        -pointsize 60 -gravity west \
+        -annotate +100+00 "/$transliteration/" \
+        -fill "$color1" \
+        -annotate +100+100 "‘$definition’" \
+        -fill "$color2" \
+        -pointsize 40 -gravity southwest \
+        -annotate +100+60 "t.me/TLGWotD" \
+        -pointsize 40 -gravity southeast \
+        -annotate +100+60 "$(date -I)" \
+        "$photo_path"
+
+    ls -la /tmp
+
+    curl -X POST "https://api.telegram.org/bot$TOKEN/sendPhoto" \
+         -F "chat_id=\"$chat_id\"" \
+         -F "photo=@$photo_path" \
+         -F parse_mode=Markdown \
+         -F caption="*$word* ‘$definition’
+
+    First occurrence (century): $first_occurrence
+    Number of occurrences (in all Ancient Greek texts): $total_occurrences"
+    '';
+    serviceConfig = {
+      Type = "oneshot";
+      DynamicUser = true;
+      StateDirectory = "tlgwotd";
+      PrivateTmp = true;
+      LoadCredential = "token:${config.age.secrets.telegram-token-kmein.path}";
+    };
  };

  niveum.passport.services = [