tgbot.deno.ts

// The authors disclaim copyright to this source code (they are ashamed to
// admit they wrote it)

import unidecode from "npm:unidecode";
import { geohash } from "./geohash.deno.ts";
import { getImageForPoints, getUrlForPoints } from "./mapycz.deno.ts";

const token = Deno.env.get("TG_BOT_TOKEN");
const MAIN_CHAT_ID = parseInt(Deno.env.get("TG_MAIN_CHAT_ID")!);
const DOMAIN = Deno.env.get("DOMAIN");
const STICKER_SET_NAME = Deno.env.get("STICKER_SET_NAME");
const STICKER_SET_OWNER = parseInt(Deno.env.get("STICKER_SET_OWNER")!);

export const webhookPath = "/tg-webhook";

function genRandomToken(bytes: number) {
  return btoa(
    String.fromCharCode(...crypto.getRandomValues(new Uint8Array(bytes))),
  )
    .replaceAll("/", "_")
    .replaceAll("+", "-")
    .replaceAll("=", "");
}

const webhookUrlToken = genRandomToken(96);

async function tgCall(
  options: any,
  endpoint: string = "sendMessage",
): Promise<Response> {
  if (endpoint == "sendMessage") options.chat_id ??= MAIN_CHAT_ID;

  let req = await fetch(`https://api.telegram.org/bot${token}/${endpoint}`, {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
    },
    body: JSON.stringify(options),
  });
  try {
    let resp = await req.json();
    if (!resp.ok) {
      console.log("Req to", endpoint, "with", options, "failed:", resp);
    }
    return resp;
  } catch (e) {}
  return req;
}

async function domeny() {
  const resp = await fetch(
    "https://auctions-master.nic.cz/share/new_auctions.json",
  );
  const list = await resp.json();

  const x = list
    .filter(
      (a) =>
        a.auction_from.split("T")[0] === new Date().toISOString().split("T")[0],
    )
    .map((x) => x.item_title)
    .map((x) => x.replaceAll(/[_*[\\\]()~`>#+=|{}.!/-]/g, ($) => `\\${$}`));
  x.sort();
  x.sort((a, b) => a.length - b.length);
  while (x.length > 0) {
    const chunk = x.splice(0, 50);
    const webArchiveLinks = chunk.map(
      (l) => `[${l}](https://web.archive.org/web/${l})`,
    );
    await tgCall({
      text: webArchiveLinks.join("\n"),
      parse_mode: "MarkdownV2",
    });
  }
}

let tempDir = "";
const contentTypes = new Map<string, string>();
const runningProcesses = new Map<string, Deno.Process>();

const origin = { lat: 50.1005803, lon: 14.3954325 };
async function postGeohash() {
  const upcoming = new Date();
  upcoming.setHours(6);
  upcoming.setMinutes(Math.random() * 60);

  const now = new Date();
  if (upcoming.getTime() < now.getTime()) {
    upcoming.setDate(upcoming.getDate() + 1);
  }

  await new Promise((resolve) =>
    setTimeout(resolve, upcoming.getTime() - now.getTime())
  );

  const geoHash = await geohash(new Date(), origin);
  const text = `[ ](${
    getImageForPoints([
      origin,
      geoHash,
    ])
  })Today's geohash is at [${
    geoHash.lat
      .toFixed(5)
      .replace(".", "\\.")
  } ${
    geoHash.lon
      .toFixed(5)
      .replace(".", "\\.")
  }](${
    getUrlForPoints([
      geoHash,
    ])
  })\\.\nPlease refer to xkcd\\.com/426/ for further steps\\.`;

  await tgCall({
    text,
    parse_mode: "MarkdownV2",
  });

  await domeny();

  setTimeout(postGeohash, 1000 * 60 * 60 * 2);

  if (Math.random() < 0.03) {
    setTimeout(
      async () =>
        await tgCall({
          text: (
            await (
              await fetch(
                "https://%73%6e%65%64%6c-%75%7a-%6b%75%62%69%6b-%70%6f%6e%6f%7a%6b%75.%67%69%74%68%75%62.%69%6f",
              )
            ).text()
          )
            .replace(/\r|<[^>]+>/g, "")
            .replace(/\s{2,}/gm, "\n")
            .trim(),
        }),
      Math.random() * 600000 + 600000,
    );
  }
}

export async function init() {
  if (
    !token ||
    !DOMAIN ||
    isNaN(MAIN_CHAT_ID) ||
    !STICKER_SET_NAME ||
    isNaN(STICKER_SET_OWNER)
  ) {
    console.log(
      `TG_BOT_TOKEN: ${token}, TG_MAIN_CHAT_ID: ${MAIN_CHAT_ID}, DOMAIN: ${DOMAIN}, STICKER_SET_NAME: ${STICKER_SET_NAME}, STICKER_SET_OWNER: ${STICKER_SET_OWNER}`,
    );
    throw new Error(
      "TG_BOT_TOKEN, TG_MAIN_CHAT_ID, DOMAIN, STICKER_SET_NAME or STICKER_SET_OWNER is not set",
    );
  }

  tempDir = await Deno.makeTempDir();
  console.log("Using temp dir", tempDir);

  await tgCall(
    {
      url: `https://${DOMAIN}${webhookPath}`,
      secret_token: webhookUrlToken,
      allowed_updates: [
        "message",
        "callback_query",
        "inline_query",
        "edited_message",
      ],
    },
    "setWebhook",
  );

  await tgCall({
    text:
      "Babes wakeup, novy shitpost prave dropnul (nebo jenom matej restartoval vpsku)",
  });

  postGeohash();
}

export async function handleRequest(e: Deno.RequestEvent) {
  if (
    e.request.method.toUpperCase() !== "POST" ||
    e.request.headers.get("X-Telegram-Bot-Api-Secret-Token") !== webhookUrlToken
  ) {
    await e.respondWith(
      new Response("You shall not pass", {
        status: 401,
        headers: {
          "Content-Type": "text/plain",
        },
      }),
    );
    return;
  }

  const data = await e.request.json();

  await Promise.all([
    e.respondWith(
      new Response("processing", {
        status: 200,
        headers: {
          "Content-Type": "text/plain",
        },
      }),
    ),
    processTgUpdate(data),
  ]);
}

async function processTgUpdate(data: any) {
  if ("callback_query" in data) return await handleCallbackQuery(data);
  if ("inline_query" in data) return await handleInlineQuery(data);
  if ("edited_message" in data) {
    data.message = data.edited_message;
  }

  const text = data?.message?.text ?? data?.message?.caption;
  if (typeof text !== "string") return;

  const reactions = [
    { t: ["sex"], r: "🤨" },
    { t: ["rust", "růst"], r: "⚡" },
    { t: [/\bjs\b/i, "javascript"], r: "🕊" },
    { t: ["linux"], r: "🤓" },
  ];

  for (const { t, r } of reactions) {
    const trigger = t.some((x) => {
      if (typeof x === "string") return text.toLowerCase().includes(x);

      return x.exec(text) !== null;
    });

    if (!trigger) continue;

    await tgCall(
      {
        chat_id: data.message.chat.id,
        message_id: data.message.message_id,
        is_big: true,
        reaction: [
          {
            type: "emoji",
            emoji: r,
          },
        ],
      },
      "setMessageReaction",
    );
    break;
  }

  if (text.startsWith("/sh ") && data.message.chat.id === MAIN_CHAT_ID) {
    handleSh(data, text.slice(4));
  }

  if (text === "/kdo") {
    const reply_id = data.message.reply_to_message?.message_id;
    await tgCall(
      {
        chat_id: data.message.chat.id,
        message_id: data.message.message_id,
      },
      "deleteMessage",
    );
    if (reply_id !== undefined) {
      await tgCall(
        {
          chat_id: data.message.chat.id,
          reply_parameters: { message_id: reply_id },
          video:
            "BAACAgQAAxkDAANmZb5XjJUES6VCvJGtIRRrKMGwRpcAAq0SAAINl_BR5jVZOMRHxCI0BA",
        },
        "sendVideo",
      );
    }
  }

  if (text.includes("@yall") && data.message.chat.id === MAIN_CHAT_ID) {
    await tgCall({
      chat_id: data.message.chat.id,
      reply_to_message_id: data.message.message_id,
      parse_mode: "MarkdownV2",
      text: await Deno.readTextFile("./static/persistent/yall.txt"),
    });
  }

  if (text.toLowerCase().includes("balls")) {
    await tgCall(
      {
        chat_id: data.message.chat.id,
        video_note:
          "DQACAgQAAxkDAAM8ZWhhSjCXOdVCv7a8SkikjCDwEH4AAiQSAAKXPEhTuYZAGfYG_KwzBA",
      },
      "sendVideoNote",
    );
  }

  if (text.toLowerCase().includes("doslova")) {
    await tgCall(
      {
        chat_id: data.message.chat.id,
        sticker:
          "CAACAgQAAxUAAWYaZDro9kEe0mLkwvNEkBKbmBS6AAKLFAACwXbgUt-2B1-aBYwpNAQ",
      },
      "sendSticker",
    );
  }

  if (
    text.toLowerCase().includes("regiojet") ||
    text.toLowerCase().includes("php")
  ) {
    await tgCall(
      {
        chat_id: data.message.chat.id,
        message_id: data.message.message_id,
      },
      "deleteMessage",
    );
    await tgCall({
      chat_id: data.message.chat.id,
      text: `rule violation by ${data.message.from.first_name} detected`,
    });
  }

  if (text.toLowerCase().includes("zig")) {
    await tgCall({
      chat_id: data.message.chat.id,
      text: `Hail!`,
    });
  }

  if (
    text.toLowerCase().includes("gnu") &&
    text.toLowerCase().includes("linux")
  ) {
    const r1 = await tgCall({
      chat_id: data.message.chat.id,
      reply_to_message_id: data.message.message_id,
      text:
        `No, Richard, it's 'Linux', not 'GNU/Linux'. The most important contributions that the FSF made to Linux were the creation of the GPL and the GCC compiler. Those are fine and inspired products. GCC is a monumental achievement and has earned you, RMS, and the Free Software Foundation countless kudos and much appreciation.

Following are some reasons for you to mull over, including some already answered in your FAQ.

One guy, Linus Torvalds, used GCC to make his operating system (yes, Linux is an OS -- more on this later). He named it 'Linux' with a little help from his friends. Why doesn't he call it GNU/Linux? Because he wrote it, with more help from his friends, not you. You named your stuff, I named my stuff -- including the software I wrote using GCC -- and Linus named his stuff. The proper name is Linux because Linus Torvalds says so. Linus has spoken. Accept his authority. To do otherwise is to become a nag. You don't want to be known as a nag, do you?

(An operating system) != (a distribution). Linux is an operating system. By my definition, an operating system is that software which provides and limits access to hardware resources on a computer. That definition applies whereever you see Linux in use. However, Linux is usually distributed with a collection of utilities and applications to make it easily configurable as a desktop system, a server, a development box, or a graphics workstation, or whatever the user needs. In such a configuration, we have a Linux (based) distribution. Therein lies your strongest argument for the unwieldy title 'GNU/Linux' (when said bundled software is largely from the FSF). Go bug the distribution makers on that one. Take your beef to Red Hat, Mandrake, and Slackware. At least there you have an argument. Linux alone is an operating system that can be used in various applications without any GNU software whatsoever. Embedded applications come to mind as an obvious example.`,
    });

    if (r1.ok) {
      await new Promise((resolve) => setTimeout(resolve, 4000));
      await tgCall({
        chat_id: data.message.chat.id,
        reply_to_message_id: r1.result.message_id,
        text:
          `Next, even if we limit the GNU/Linux title to the GNU-based Linux distributions, we run into another obvious problem. XFree86 may well be more important to a particular Linux installation than the sum of all the GNU contributions. More properly, shouldn't the distribution be called XFree86/Linux? Or, at a minimum, XFree86/GNU/Linux? Of course, it would be rather arbitrary to draw the line there when many other fine contributions go unlisted. Yes, I know you've heard this one before. Get used to it. You'll keep hearing it until you can cleanly counter it.

You seem to like the lines-of-code metric. There are many lines of GNU code in a typical Linux distribution. You seem to suggest that (more LOC) == (more important). However, I submit to you that raw LOC numbers do not directly correlate with importance. I would suggest that clock cycles spent on code is a better metric. For example, if my system spends 90% of its time executing XFree86 code, XFree86 is probably the single most important collection of code on my system. Even if I loaded ten times as many lines of useless bloatware on my system and I never excuted that bloatware, it certainly isn't more important code than XFree86. Obviously, this metric isn't perfect either, but LOC really, really sucks. Please refrain from using it ever again in supporting any argument.

Last, I'd like to point out that we Linux and GNU users shouldn't be fighting among ourselves over naming other people's software. But what the heck, I'm in a bad mood now. I think I'm feeling sufficiently obnoxious to make the point that GCC is so very famous and, yes, so very useful only because Linux was developed. In a show of proper respect and gratitude, shouldn't you and everyone refer to GCC as 'the Linux compiler'? Or at least, 'Linux GCC'? Seriously, where would your masterpiece be without Linux? Languishing with the HURD?

If there is a moral buried in this rant, maybe it is this:

Be grateful for your abilities and your incredible success and your considerable fame. Continue to use that success and fame for good, not evil. Also, be especially grateful for Linux' huge contribution to that success. You, RMS, the Free Software Foundation, and GNU software have reached their current high profiles largely on the back of Linux. You have changed the world. Now, go forth and don't be a nag.`,
      });
    }
  }

  if (/\barch(?![ií][a-z])/i.exec(text) && data.message.from.id === 656461353) {
    await tgCall({
      chat_id: data.message.chat.id,
      reply_to_message_id: data.message.message_id,
      text: "Ano Mariane, my víme",
    });
  }

  if (text === "/inspect") {
    await tgCall({
      chat_id: data.message.chat.id,
      reply_to_message_id: data.message.message_id,
      parse_mode: "MarkdownV2",
      text: `\`\`\`\n${
        JSON.stringify(
          data.message.reply_to_message,
          null,
          2,
        )
      }\n\`\`\``,
    });
  }

  if (text.startsWith("/settype ") && data.message.chat.id === MAIN_CHAT_ID) {
    const [id, type] = text.slice(9).split(" ");
    if (contentTypes.has(id)) {
      contentTypes.set(id, type);
    }
  }

  if (text.startsWith("/logo ") && data.message.chat.id === MAIN_CHAT_ID) {
    await handleLogo(data, text.slice(6));
  }

  if (
    text.toLowerCase() === "sticker this" &&
    data.message.chat.id === MAIN_CHAT_ID
  ) {
    const result = await sticekrThis(data.message.reply_to_message);
    if (result !== null) {
      await tgCall({
        chat_id: data.message.chat.id,
        reply_to_message_id: data.message.message_id,
        text: `Not even your mom could make a sticker out of that (${result})`,
      });
    }
  }

  const manMatch = text.match(/^\s*man\s*([1-8])?\s*([a-z-_+.]+)\s*$/i);
  if (manMatch !== null) {
    let text = "https://man.archlinux.org/man/";
    text += manMatch[2];
    if (manMatch[1] !== undefined) text += "." + manMatch[1];
    let parse_mode: string | undefined = undefined;

    const response = await fetch(text + ".txt");
    if (response.status === 200) {
      const manText = await response.text();
      if (manText.length < 4000) {
        text = "```\n" + manText.replaceAll("```", "`´`") + "```";
        parse_mode = "MarkdownV2";
      }
    } else if (response.status === 404) {
      text = "";
      await tgCall(
        {
          chat_id: data.message.chat.id,
          message_id: data.message.message_id,
          is_big: true,
          reaction: [
            {
              type: "emoji",
              emoji: "🤷‍♂️",
            },
          ],
        },
        "setMessageReaction",
      );
    }

    if (text) {
      await tgCall({
        chat_id: data.message.chat.id,
        reply_to_message_id: data.message.message_id,
        text,
        parse_mode,
      });
    }
  }

  if (text.toLowerCase().includes("sus")) {
    await tgCall({
      chat_id: data.message.chat.id,
      text: "ඞ",
    });
  }
}

const decoder = new TextDecoder("utf8");

const LOGO_TEMPLATE = await Deno.readTextFile("./static/logo.svg");
const LOGO_RENDER_SIZE = 500;

async function generateLogos(text: string, filename: string) {
  const texted = LOGO_TEMPLATE.replace("TEMPLATETEXT", text.trim());
  await Deno.writeTextFile(`./static/persistent/logos/${filename}.svg`, texted);
  return (
    await Deno.run({
      cmd: [
        "inkscape",
        `./static/persistent/logos/${filename}.svg`,
        "-o",
        `./static/persistent/logos/${filename}.png`,
        "-w",
        LOGO_RENDER_SIZE.toString(),
      ],
      stderr: "null",
    }).status()
  ).code;
}

function slugify(text: string) {
  return (unidecode(text.trim()) as string)
    .replaceAll(" ", "-")
    .replaceAll(/[^a-z0-9_-]/gi, (x) => "0x" + x.charCodeAt(0).toString(16));
}

async function handleLogo(data: any, text: string) {
  const fn = `${slugify(text)}_${new Date().toISOString()}`;
  if ((await generateLogos(text, fn)) === 0) {
    await tgCall(
      {
        chat_id: data.message.chat.id,
        reply_to_message_id: data.message.message_id,
        photo: `https://${DOMAIN}/persistent/logos/${fn}.png`,
        caption: `https://${DOMAIN}/persistent/logos/${fn}.svg`,
      },
      "sendPhoto",
    );
  }
}

async function handleSh(data: any, cmd: string) {
  const id = genRandomToken(32);
  await Deno.writeFile(`${tempDir}/${id}.sh`, new TextEncoder().encode(cmd), {
    createNew: true,
  });

  const outFile = await Deno.open(`${tempDir}/${id}.out`, {
    write: true,
    createNew: true,
  });
  const proc = Deno.run({
    cmd: ["bash", `${tempDir}/${id}.sh`],
    stdout: outFile.rid,
    stderr: outFile.rid,
  });
  const raceResult = await Promise.race([
    proc.status(),
    new Promise<void>((resolve) => setTimeout(() => resolve(), 5000)),
  ]);

  if (raceResult !== undefined) {
    await reportProcessResult(
      outFile,
      id,
      data.message.message_id,
      raceResult.code,
    );
    return;
  }

  runningProcesses.set(id, proc);
  contentTypes.set(id, "application/octet-stream");

  const progressMessageResponse = await tgCall({
    reply_to_message_id: data.message.message_id,
    parse_mode: "MarkdownV2",
    text:
      `[Command is taking too long](https://${DOMAIN}/tgweb/${id})\\. Set Content\\-Type with \`/settype ${id} text/plain\``,
    reply_markup: {
      inline_keyboard: [
        [
          {
            text: "Kill process",
            callback_data: `kill:${id}`,
          },
        ],
      ],
    },
  });

  const status = await proc.status();
  runningProcesses.delete(id);

  await tgCall(
    {
      message_id: progressMessageResponse.result.message_id,
      reply_markup: {
        inline_keyboard: [],
      },
    },
    "editMessageReplyMarkup",
  );
  await reportProcessResult(
    outFile,
    id,
    progressMessageResponse.result.message_id,
    status.code,
  );
}

async function reportProcessResult(
  outFile: Deno.FsFile,
  id: string,
  reply_to_message_id: number,
  exitCode: number,
) {
  const outPath = `${tempDir}/${id}.out`;
  const stat = await outFile.stat();
  outFile.close();
  const fileProc = Deno.run({
    cmd: ["file", "-ib", outPath],
    stdout: "piped",
  });
  // need to await the status to not create zombie processes
  await fileProc.status();
  const mime = decoder.decode(await fileProc.output());
  contentTypes.set(id, mime);
  const isText = mime.startsWith("text/") ||
    mime.startsWith("application/json");
  let text;
  if (stat.size === 0) text = `No output \\(exit code ${exitCode}\\)\\.`;
  else if (isText && stat.size <= 5000) {
    let res = decoder
      .decode(await Deno.readFile(outPath))
      .replaceAll("\\", "\\\\")
      .replaceAll("`", "\\`");
    if (res.at(-1) !== "\n") res += "\n";
    text = "```\n" + res + "```";
    if (exitCode !== 0) {
      text = `[Exit code ${exitCode}](https://${DOMAIN}/tgweb/${id})\n` + text;
    }
  } else {
    text = "[" +
      (isText ? "Output too long" : "Binary output") +
      `](https://${DOMAIN}/tgweb/${id}) \\(exit code ${exitCode}, ${stat.size} bytes\\)\\. Set Content\\-Type with \`/settype ${id} mime/type\``;
  }

  await tgCall({
    reply_to_message_id,
    parse_mode: "MarkdownV2",
    text,
  });
}

async function handleCallbackQuery(data: any) {
  const cbData = data.callback_query.data;
  if (cbData.startsWith("kill:")) {
    const proc = runningProcesses.get(cbData.slice(5));
    if (proc === undefined) return;
    const killProc = Deno.run({
      cmd: ["rkill", "-9", proc.pid.toString()],
    });
    await killProc.status();
    return;
  }
}

export async function handleTgWeb(
  e: Deno.RequestEvent,
): Promise<Response | null> {
  const url = new URL(e.request.url);
  const path = url.pathname.slice(7);
  const ct = contentTypes.get(path);
  if (ct === undefined) {
    return new Response("Not found", {
      headers: { "Content-Type": "text/plain" },
      status: 404,
    });
  }

  const file = await Deno.open(`${tempDir}/${path}.out`);
  return new Response(file.readable, {
    headers: { "Content-Type": ct },
  });
}

let imageI = 0;
async function handleInlineQuery(data: any) {
  const { id: inline_query_id, query, from } = data.inline_query;
  console.log(
    `Logo from ${from.first_name} ${from.last_name} (@${from.username}): ${query}`,
  );
  const fn = `inline_${imageI++}_${slugify(query)}_${new Date().toISOString()}`;
  if ((await generateLogos(query, fn)) === 0) {
    await tgCall(
      {
        inline_query_id,
        results: [
          {
            type: "photo",
            id: "0",
            photo_url: `https://${DOMAIN}/persistent/logos/${fn}.png`,
            thumb_url: `https://${DOMAIN}/persistent/logos/${fn}.png`,
            photo_width: LOGO_RENDER_SIZE.toString(),
            photo_height: LOGO_RENDER_SIZE.toString(),
            title: "Sus?",
            caption: `https://${DOMAIN}/persistent/logos/${fn}.svg`,
          },
        ],
      },
      "answerInlineQuery",
    );
  }
}

async function sticekrThis(orig_msg: any): Promise<string | null> {
  if (!orig_msg) return "wtf";
  let file;
  if (Array.isArray(orig_msg.photo)) {
    file = orig_msg.photo.at(-1)?.file_id;
  } else if (orig_msg.document?.mime_type?.startsWith("image")) {
    file = orig_msg.document.file_id;
  }
  if (!file) return "not an image file duh";

  const data = await tgCall(
    {
      file_id: file,
    },
    "getFile",
  );
  if (!data.ok) return "telegrams a hoe2";

  const resp2 = await fetch(
    `https://api.telegram.org/file/bot${token}/${data.result.file_path}`,
  );
  if (!resp2.ok) return "telegram cdn is a hoe";

  const fileName = await Deno.makeTempFile();
  await Deno.writeFile(fileName, resp2.body!);
  const outFileName = fileName + ".webp";

  const cmd = new Deno.Command("convert", {
    args: [fileName, "-resize", "512x512", outFileName],
  });
  const res = await cmd.output();
  if (res.code !== 0) return "imagemagick is a hoe";
  const sticker = await Deno.readFile(outFileName);

  const body = new FormData();
  body.append("user_id", STICKER_SET_OWNER.toString());
  body.append("name", STICKER_SET_NAME);
  body.append(
    "sticker",
    JSON.stringify({ sticker: "attach://file", emoji_list: ["🤓"] }),
  );
  body.append("file", new Blob([sticker], { type: "image/webp" }), "file.webp");
  const resp3 = await fetch(
    `https://api.telegram.org/bot${token}/addStickerToSet`,
    {
      method: "POST",
      body,
    },
  );
  if (!resp3.ok) return "skill issue";

  const data4 = await tgCall(
    {
      name: STICKER_SET_NAME,
    },
    "getStickerSet",
  );
  if (!data4.ok) return "i ran out of error message ideas";
  const stickerId = data4.result.stickers.at(-1).file_id;
  if (!stickerId) return "i ran out of error message ideas the most";

  const resp5 = await tgCall(
    {
      chat_id: orig_msg.chat.id,
      sticker: stickerId,
    },
    "sendSticker",
  );

  if (!resp5.ok) return "actually it succeeded but i failed to send it";

  return null;
}