index.js

import { Input, Telegraf } from "telegraf";
import { message } from "telegraf/filters";
import { generateCodeImage } from "./lib/carbon.js";
import { getImage } from "./lib/mods.js";
import express from "express";
import morgan from "morgan";

const logger = { info: console.log };

const defaultAvatar = "https://avatars.githubusercontent.com/u/124599?v=4";

const bot = new Telegraf(process.env.BOT_TOKEN);

const port = process.env.PORT || 3000;
const webhookDomain =
  process.env.DETA_SPACE_APP_HOSTNAME ||
  process.env.WEBHOOK_DOMAIN ||
  "https://example.com";

process.once("SIGINT", () => bot.stop("SIGINT"));
process.once("SIGTERM", () => bot.stop("SIGTERM"));

bot.start((ctx) => ctx.reply("Welcome to Code Beautifier Bot"));
bot.help((ctx) => ctx.reply("Send me a code to beautify"));

const loaderMessages = [
  "🔅 Processing your code...",
  "🔆 Please wait a moment...",
  "☀️ This may take a while...",
  "🌟 Generating image...",
  "⭐️ Almost done...",
];

let loaderIndex = 0;

bot.on(message("text"), async (ctx) => {
  const botUserName = "@" + bot.botInfo.username;

  const isPrivate = ctx.message.chat.type === "private";
  if (!isPrivate) {
    if (
      !ctx.message.text.includes(botUserName) &&
      !ctx.message.text.includes("/format")
    )
      return;
  }

  let msg = "";
  let _user = {};

  // check is the message is a reply
  if (ctx.message.reply_to_message) {
    logger.info(`Reply message recieved from ${ctx.message.from.username}`);
    // if (!ctx.message.reply_to_message.from.is_bot) return;
    if (!ctx.message.reply_to_message.text) return;
    msg = ctx.message.reply_to_message.text
      .replace(botUserName, "")
      .replace("/format", "");
    _user = ctx.message.reply_to_message.from;
  } else {
    logger.info(`Message recieved from ${ctx.message.from.username}`);
    if (!ctx.message.text) return;
    msg = ctx.message.text.replace(botUserName, "").replace("/format", "");
    _user = ctx.message.from;
  }

  if (!msg) return;
  const _msg = msg.trim().toLowerCase();
  if (_msg.length < 3) return;

  const codeImage = generateCodeImage(msg); // Long task - Takes 10-15 seconds

  let loader = null;
  let loaderT = null;

  if (isPrivate) {
    loader = await ctx.reply("🔅 Nice");
    loaderT = setInterval(async () => {
      await ctx.telegram.editMessageText(
        ctx.message.chat.id,
        loader.message_id,
        null,
        loaderMessages[0]
      );
      loaderMessages.push(loaderMessages.shift());
    }, 4000);
    setTimeout(() => {
      clearInterval(loaderT);
    }, 1000 * 60);
  }
  const avatar = getAvatar(ctx, _user.id);

  const image = await getImage(await codeImage, {
    name: (_user.first_name || "") + " " + (_user.last_name || ""),
    username: _user.username ? `@${_user.username}` : "Telegram User",
    avatar: await avatar,
  });

  if (isPrivate) {
    clearInterval(loaderT);
    // delete loader message
    loader &&
      (await ctx.telegram.deleteMessage(
        ctx.message.chat.id,
        loader.message_id
      ));
  }

  ctx.replyWithPhoto(Input.fromBuffer(image, "code.png"), {
    caption: "Here is your code 🎉",
  });
});

const app = express();
app.use(morgan(":method :url :status :http-version :response-time "));
app.get("/", (req, res) => {
  res.send("Its Working!");
});

app.use(await bot.createWebhook({ domain: webhookDomain }));
app.listen(port, () => console.log("Server listening on port", port));

async function getAvatar(ctx, id) {
  try {
    const userAvatar = await ctx.telegram.getUserProfilePhotos(id);
    const avatarId = userAvatar.photos[0][0].file_id;
    const avatar = await ctx.telegram.getFileLink(avatarId);
    return avatar.toString() || defaultAvatar;
  } catch (error) {
    return defaultAvatar;
  }
}