niveum/modules/telegram-bot.nix

{
  lib,
  config,
  pkgs,
  ...
}:
with lib; let
  cfg = config.niveum.bots;

  botService = name: bot:
    nameValuePair "bot-${name}" {
      enable = bot.enable;
      startAt = bot.time;
      serviceConfig = {
        Type = "oneshot";
        LoadCredential = lib.optionals (bot.telegram.enable) [
          "telegram-token:${bot.telegram.tokenFile}"
        ] ++ lib.optionals (bot.mastodon.enable) [
          "mastodon-token:${bot.mastodon.tokenFile}"
        ] ++ lib.optionals (bot.matrix.enable) [
          "matrix-token:${bot.matrix.tokenFile}"
        ];
      };
      wants = ["network-online.target"];
      script = ''
        QUOTE=$(${bot.command})
        if [ -n "$QUOTE" ]; then
          echo $QUOTE >&2

          ${lib.optionalString (bot.matrix.enable) ''
            export MATRIX_TOKEN="$(cat "$CREDENTIALS_DIRECTORY/matrix-token")"
            export JSON_PAYLOAD=$(${pkgs.jq}/bin/jq -n --arg msgtype "m.text" --arg body "$QUOTE" '{msgtype: $msgtype, body: $body}')
            ${strings.concatStringsSep "\n" (map (chatId: ''
              ${pkgs.curl}/bin/curl -X POST "https://${bot.matrix.homeserver}/_matrix/client/r0/rooms/${chatId}/send/m.room.message" \
                -d "$JSON_PAYLOAD" \
                -H "Authorization: Bearer $MATRIX_TOKEN" \
                -H "Content-Type: application/json"
            '') bot.matrix.chatIds)}
          ''}

          ${lib.optionalString (bot.mastodon.enable) ''
            export MASTODON_TOKEN="$(cat "$CREDENTIALS_DIRECTORY/mastodon-token")"
            ${pkgs.curl}/bin/curl -X POST "https://${bot.mastodon.homeserver}/api/v1/statuses" \
              -H "Authorization: Bearer $MASTODON_TOKEN" \
              -d status="$QUOTE" \
              -d "language=${bot.mastodon.language}" \
              -d "visibility=public"
          ''}

          ${lib.optionalString (bot.telegram.enable) ''
            export TELEGRAM_TOKEN="$(cat "$CREDENTIALS_DIRECTORY/telegram-token")"
            ${strings.concatStringsSep "\n" (map (chatId: ''
              ${pkgs.curl}/bin/curl -X POST "https://api.telegram.org/bot''${TELEGRAM_TOKEN}/sendMessage" \
                -d chat_id="${chatId}" \
                -d text="$QUOTE" ${
                lib.strings.optionalString (bot.telegram.parseMode != null)
                "-d parse_mode=${bot.telegram.parseMode}"
              } | ${pkgs.jq}/bin/jq -e .ok
            '')
            bot.telegram.chatIds)}
          ''}
        fi
      '';
    };
in {
  options.niveum.bots = mkOption {
    type = types.attrsOf (types.submodule {
      options = {
        enable = mkEnableOption "Mastodon and Telegram bot";
        time = mkOption {type = types.str;};
        command = mkOption {type = types.str;};
        matrix = mkOption {
          default = {};
          type = types.submodule {
            options = {
              enable = mkEnableOption "Posting to Matrix";
              tokenFile = mkOption {type = types.path;};
              homeserver = mkOption {
                type = types.str;
              };
              chatIds = mkOption {
                type = types.listOf types.str;
              };
            };
          };
        };
        mastodon = mkOption {
          default = {};
          type = types.submodule {
            options = {
              enable = mkEnableOption "Posting to Mastodon";
              language = mkOption {
                type = types.str;
                default = "en";
              };
              tokenFile = mkOption {type = types.path;};
              homeserver = mkOption {
                type = types.str;
                default = "social.krebsco.de";
              };
            };
          };
        };
        telegram = mkOption {
          default = {};
          type = types.submodule {
            options = {
              enable = mkEnableOption "Posting to Telegram";
              tokenFile = mkOption {type = types.path;};
              chatIds = mkOption {
                type = types.listOf (types.strMatching "-?[0-9]+|@[A-Za-z0-9]+");
              };
              parseMode = mkOption {
                type = types.nullOr (types.enum ["HTML" "Markdown"]);
                default = null;
              };
            };
          };
        };
      };
    });
    default = {};
  };

  config = {systemd.services = attrsets.mapAttrs' botService cfg;};
}