made and documented the telegram bot (HYPE)

This commit is contained in:
Dhravya 2024-06-23 20:04:09 -05:00
parent fff8b7abd7
commit 51dd5ec9dd
7 changed files with 307 additions and 7 deletions

View file

@ -1,3 +1,7 @@
import { db } from "@/server/db";
import { storedContent, users } from "@/server/db/schema";
import { cipher, decipher } from "@/server/encrypt";
import { eq } from "drizzle-orm";
import { Bot, webhookCallback } from "grammy";
import { User } from "grammy/types";
@ -12,17 +16,96 @@ const token = process.env.TELEGRAM_BOT_TOKEN;
const bot = new Bot(token);
const getUserByTelegramId = async (telegramId: string) => {
return await db.query.users
.findFirst({
where: eq(users.telegramId, telegramId),
})
.execute();
};
bot.command("start", async (ctx) => {
const user: User = (await ctx.getAuthor()).user;
const cipherd = cipher(user.id.toString());
await ctx.reply(
`Welcome to Supermemory bot, ${user.first_name}. I am here to help you remember things better.`,
`Welcome to Supermemory bot. I am here to help you remember things better. Click here to create and link your accont: http://localhost:3000/signin?telegramUser=${cipherd}`,
);
});
bot.on("message", async (ctx) => {
await ctx.reply(
"Hi there! This is Supermemory bot. I am here to help you remember things better.",
const user: User = (await ctx.getAuthor()).user;
const cipherd = cipher(user.id.toString());
const dbUser = await getUserByTelegramId(user.id.toString());
if (!dbUser) {
await ctx.reply(
`Welcome to Supermemory bot. I am here to help you remember things better. Click here to create and link your accont: http://localhost:3000/signin?telegramUser=${cipherd}`,
);
return;
}
const message = await ctx.reply("I'm thinking...");
const response = await fetch(
`${process.env.BACKEND_BASE_URL}/api/autoChatOrAdd?query=${ctx.message.text}&user=${dbUser.id}`,
{
method: "POST",
headers: {
Authorization: "Bearer " + process.env.BACKEND_SECURITY_KEY,
"Content-Type": "application/json",
},
body: JSON.stringify({
// TODO: we can use the conversations API to get the last 5 messages
// get chatHistory from this conversation.
// Basically the last 5 messages between the user and the assistant.
// In ths form of [{role: 'user' | 'assistant', content: string}]
// https://grammy.dev/plugins/conversations
chatHistory: [],
}),
},
);
if (response.status !== 200) {
console.log("Failed to get response from backend");
console.log(response.status);
console.log(await response.text());
await ctx.reply(
"Sorry, I am not able to process your request at the moment.",
);
return;
}
const data = (await response.json()) as {
status: string;
response: string;
contentAdded: {
type: string;
content: string;
url: string;
};
};
// TODO: we might want to enrich this data with more information
if (data.contentAdded) {
await db
.insert(storedContent)
.values({
content: data.contentAdded.content,
title: `${data.contentAdded.content.slice(0, 30)}... (Added from chatbot)`,
description: "",
url: data.contentAdded.url,
baseUrl: data.contentAdded.url,
image: "",
savedAt: new Date(),
userId: dbUser.id,
type: data.contentAdded.type,
})
.returning({ id: storedContent.id });
}
await ctx.api.editMessageText(ctx.chat.id, message.message_id, data.response);
});
export const POST = webhookCallback(bot, "std/http");