This repository has been archived on 2024-07-25. You can view files and clone it, but cannot push or open issues or pull requests.
yandexgpt_tg_bot/main.mjs

67 lines
1.5 KiB
JavaScript
Raw Normal View History

2023-06-28 14:45:04 +03:00
import TelegramBot from "node-telegram-bot-api";
import {
readerTaskEither as rte,
readonlyArray,
semigroup,
string,
} from "fp-ts";
import { flow, pipe } from "fp-ts/lib/function.js";
import { describeArticle } from "./api.mjs";
import config from "./config.mjs";
const bot = new TelegramBot(config.telegramBotToken, {
polling: true,
});
2023-07-01 00:32:58 +03:00
console.log("The telegram bot listens for updates");
2023-06-28 14:45:04 +03:00
bot.on("channel_post", async (msg) => {
const link = extractMessageLink(msg);
2023-07-01 00:32:58 +03:00
if (!link) {
console.log("cannot find link in msg:", msg);
return;
}
2023-06-28 14:45:04 +03:00
pipe(
2023-07-01 00:32:58 +03:00
describeArticle(link),
2023-06-28 14:45:04 +03:00
rte.map(
flow(
readonlyArray.foldMap({
...string.Monoid,
...pipe(string.Semigroup, semigroup.intercalate("\n")),
})((row) => `- ${row}`),
string.trimLeft,
),
),
rte.match(
(err) => console.log({ err }),
(res) =>
bot.editMessageText(msg.text + "\n\nYandexGPT:\n\n" + res, {
chat_id: msg.chat.id,
message_id: msg.message_id,
}),
),
)(config)();
});
2023-06-29 00:51:36 +03:00
// TODO: use option instead
2023-06-28 14:45:04 +03:00
function extractMessageLink(msg) {
2023-07-01 00:32:58 +03:00
return getTextLink(msg) || getMessageUrl(msg);
}
const RE_URL =
/(http|https):\/\/([\w_-]+(?:(?:\.[\w_-]+)+))([\w.,@?^=%&:\/~+#-]*[\w@?^=%&\/~+#-])/;
function getMessageUrl(msg) {
return RE_URL.exec(msg.text)[0];
}
function getTextLink(msg) {
2023-07-01 13:37:06 +03:00
const textLink = (msg.entities ?? []).find(isTextLink);
return textLink ? textLink.url : null;
2023-06-28 14:45:04 +03:00
}
2023-07-01 00:32:58 +03:00
function isTextLink(msgEntity) {
2023-06-28 14:45:04 +03:00
return msgEntity.type === "text_link";
}