feat: implement buy and sell handlers, add keyboard configurations, and enhance message handling in TelegramAdapter

Salasifaoui · Salasifaoui · commit f71af39bd8f2 · 2024-11-23T14:16:39.000+01:00
diff --git a/src/adapters/controllers/buy.handler.ts b/src/adapters/controllers/buy.handler.ts
@@ -0,0 +1,21 @@
+import { Context } from "grammy";
+import { closeKeyboard } from "../controllers/listKeyboard";
+
+export const buyHandler = async (
+    context: Context,
+    amount?: number
+  ) => {
+    if(!amount){
+        return context.reply("Please enter the amount of the cryptocurrency you want to buy", {
+            
+            reply_markup: closeKeyboard,
+            });
+    }
+    
+    const message = `You have successfully bought ${amount} of the cryptocurrency`;
+    return context.reply(message, {
+        reply_markup: closeKeyboard,
+    });
+
+
+  };
diff --git a/src/adapters/controllers/callback.handler.ts b/src/adapters/controllers/callback.handler.ts
@@ -0,0 +1,73 @@
+import { Context } from "grammy";
+import { CallbackQuery } from "grammy/types";
+import {
+  backKeyboard,
+  menuKeyboard,
+  editKeyboard,
+  settingKeyboard,
+  walletKeyboard,
+} from "./listKeyboard";
+import { buyHandler } from "./buy.handler";
+import { selHandler } from "./sel.handler";
+import { deleteHandler } from "./delete.handler";
+import { screenHandler } from "./screen.handler";
+import { settingsMessage } from "../utils/messages";
+import { BaseBot } from "src/bots/@base";
+
+export const callbackQueryHandler = async (bot: BaseBot, context: Context) => {
+  try {
+    if (!context.update.callback_query) {
+      return;
+    }
+    const callbackQuery: CallbackQuery = context.update.callback_query;
+
+    const { data, message } = callbackQuery;
+
+    if (!message || !message.text || context.chat?.type !== "private") {
+      return;
+    }
+
+    const messageId = message.message_id;
+
+    if (data === "complete") {
+      await screenHandler(
+        context,
+        messageId,
+        message.text + "Edited",
+        backKeyboard,
+      );
+      await context.answerCallbackQuery();
+    } else if (data === "delete") {
+      await deleteHandler(context, messageId);
+      await context.answerCallbackQuery();
+    } else if (data === "refresh") {
+      await context.answerCallbackQuery("Refreshing...");
+    } else if (data === "edit") {
+      await screenHandler(context, messageId, "Choose an option", editKeyboard);
+      await context.answerCallbackQuery;
+    } else if (data === "back") {
+      await screenHandler(
+        context,
+        messageId,
+        "Choose a cryptocurrency",
+        menuKeyboard,
+      );
+      await context.answerCallbackQuery();
+    } else if (data === "close") {
+      await deleteHandler(context, messageId);
+      await context.answerCallbackQuery();
+    } else if (data === "buy_custom") {
+      await buyHandler(context, 10);
+      await context.answerCallbackQuery();
+    } else if (data === "sell_custom") {
+      await selHandler(context, 10, 10);
+      await context.answerCallbackQuery();
+    } else if (data === "settings") {
+      await screenHandler(context, messageId, settingsMessage, settingKeyboard);
+      await context.answerCallbackQuery();
+    } else if (data === "wallet_view") {
+      await screenHandler(context, messageId, "Wallet", walletKeyboard);
+      await context.answerCallbackQuery();
+    }
+  } catch (error) {}
+};
diff --git a/src/adapters/controllers/delete.handler.ts b/src/adapters/controllers/delete.handler.ts
@@ -0,0 +1,9 @@
+import { Context } from "grammy";
+
+export const deleteHandler = async (context: Context, messageId: number) => {
+  if (!context.chat?.id) {
+    return;
+  }
+  const chatId = context.chat.id;
+  await context.api.deleteMessage(chatId, messageId);
+};
diff --git a/src/adapters/controllers/listKeyboard.ts b/src/adapters/controllers/listKeyboard.ts
@@ -0,0 +1,73 @@
+import { InlineKeyboard } from "grammy";
+
+export const menuKeyboard = new InlineKeyboard()
+  .text("Buy 💰", "buy_custom")
+  .text("Sell ⚡️", "sell_custom")
+  .text("Wallet", "wallet_view")
+  .row()
+  .text("Payout 📈", "payout_custom")
+  .text("Edit", "edit")
+  .row()
+  .text("Settings", "settings")
+  .row()
+  .text("Refresh", "refresh")
+  .row();
+  
+
+export const editKeyboard = new InlineKeyboard()
+  .text("Save", "complete")
+  .text("Delete", "delete")
+  .text("crypto", "crypto")
+  .row();
+
+
+export const backKeyboard = new InlineKeyboard().text("Back", "back");
+export const closeKeyboard = new InlineKeyboard().text("Close", "close");
+
+export const settingKeyboard = new InlineKeyboard()
+  .text("Change Address", "change_address")
+  .text("Change Security", "change_security")
+  .row()
+  .text("Change Currency", "change_currency")
+  .text("Change Language", "change_language")
+  .row()
+  .text("Change Notification", "change_notification")
+  .text("Change Theme", "change_theme")
+  .row()
+  .text("Change Timezone", "change_timezone")
+  .text("Change Privacy", "change_privacy")
+  .row()
+  .text("Change Backup", "change_backup")
+  .text("Change Delete", "change_delete")
+  .row()
+  .text("Back", "back")
+  .text("Close", "close")
+  .row();
+
+export const walletKeyboard = new InlineKeyboard()
+  .text("Deposit", "deposit")
+  .text("Withdraw", "withdraw")
+  .row()
+  .text("History", "history")
+  .text("Settings", "settings")
+  .row()
+  .text("Back", "back")
+  .text("Close", "close")
+  .row();
+
+  // export const auditInlineKeyboard = (chain: any, address: string) => {
+  //   const auditUrl = `https://gopluslabs.io/token-security/${chain.id}/${address}`
+  //   const scanUrl = `${chain.scan}${address}`
+  //   const swapUrl = `https://app.1inch.io/#/${chain.id}/simple/swap/${chain.coin}/${address}`
+  //   const chartUrl = `https://${
+  //     chain.name === 'BSC' ? 'poocoin.app/tokens' : 'coingecko.com/en/coins'
+  //   }/${address}`
+  
+  //   return new InlineKeyboard()
+  //     .url('Audit', auditUrl)
+  //     .url('Contract', scanUrl)
+  //     .row()
+  //     .url('Buy', swapUrl)
+  //     .url('Chart', chartUrl)
+  // }
+
diff --git a/src/adapters/controllers/message.handler.ts b/src/adapters/controllers/message.handler.ts
@@ -0,0 +1,41 @@
+import { Context } from "grammy";
+import { IMessage, ISender } from "../../types";
+import { BaseBot } from "src/bots/@base";
+
+export const messageHandler = async (bot: BaseBot, ctx: Context) => {
+  if (!bot) {
+    return;
+  }
+  console.log("first messageHandler bot", bot);
+  if (!ctx.message) {
+    return;
+  }
+  console.log("messageHandler chatId", ctx.message.chat.id);
+  const sender: ISender = {
+    id: ctx.message.chat.id.toString(),
+    name: ctx.message.from.username || ctx.message.from.first_name,
+    isDirect: true,
+  };
+  const message: IMessage = {
+    chatId: ctx.message.chat.id.toString(), // TODO
+    text: ctx.message.text || "",
+  };
+  if (message.text) {
+    bot.onMessage(
+      sender,
+      message,
+      (reply: any) => {
+        ctx.reply(reply, {
+          parse_mode: "Markdown",
+        });
+
+        return reply;
+      },
+      (isTyping: any) => {
+        if (isTyping) {
+          ctx.replyWithChatAction("typing");
+        }
+      },
+    );
+  }
+};
diff --git a/src/adapters/controllers/screen.handler.ts b/src/adapters/controllers/screen.handler.ts
@@ -0,0 +1,25 @@
+import { Context } from "grammy";
+
+export const screenHandler = async (
+  context: Context,
+  messageId: number,
+  message: string,
+  keyboard?: any,
+) => {
+  if (!context.chat?.id) {
+    return;
+  }
+
+  const chatId = context.chat.id;
+  await context.api.editMessageText(
+    chatId,
+    messageId,
+    `
+    👋 Hello, ${context.from?.first_name}
+    ${message}
+    `,
+    {
+      reply_markup: keyboard,
+    },
+  );
+};
diff --git a/src/adapters/controllers/sel.handler.ts b/src/adapters/controllers/sel.handler.ts
@@ -0,0 +1,22 @@
+import { Context } from "grammy";
+import { closeKeyboard } from "../controllers/listKeyboard";
+
+export const selHandler = async (
+  context: Context,
+  amount?: number,
+  percent?: number,
+) => {
+  if (!amount || !percent) {
+    return context.reply(
+      "Please enter the amount and the percentage of the cryptocurrency you want to sell",
+      {
+        reply_markup: closeKeyboard,
+      },
+    );
+  }
+
+  const message = `You have successfully sold ${amount} of the cryptocurrency at ${percent}% profit`;
+  return context.reply(message, {
+    reply_markup: closeKeyboard,
+  });
+};
diff --git a/src/adapters/telegram.ts b/src/adapters/telegram.ts
@@ -1,8 +1,10 @@
-import { Bot } from "grammy";
-
-import { IMessage, ISender } from "../types";
+import { Bot, Context } from "grammy";
 import { BaseAdapter } from "./base";
 import { BaseBot } from "src/bots/@base";
+import { messageHandler } from "./controllers/message.handler";
+import { callbackQueryHandler } from "./controllers/callback.handler";
+import { menuKeyboard } from "./controllers/listKeyboard";
+import { welcomeMessage } from "./utils/messages";
 
 type TelegramAdapterOptions = {
   token?: string;
@@ -27,7 +29,7 @@ export class TelegramAdapter extends BaseAdapter {
         "No access token provided through environment. Please PERMANENTLY store the access token in an the environment variable.",
       );
     }
-  
+
     this.token = token;
   }
 
@@ -41,47 +43,34 @@ export class TelegramAdapter extends BaseAdapter {
         description: command.description,
       });
     });
+
     this.client.api.setMyCommands(commands);
 
     return this;
   }
 
   async init({ autoJoin }: { autoJoin?: boolean } = { autoJoin: true }) {
+    this.client.command("start", async (ctx, next) => {
+      // this middleware does not handle deep-links
+      if (ctx.match) return next();
+      await ctx.reply(welcomeMessage, {
+        parse_mode: "HTML",
+        reply_markup: menuKeyboard,
+      });
+    });
+
+    this.client.on(
+      "callback_query:data",
+      async function onCallbackQuery(ctx: Context) {
+        // pass the bot instance to the callbackQueryHandler
+        // const bot = this.bot;
+        // console.log("first onCallbackQuery bot", bot);
+        callbackQueryHandler(this.bot, ctx);
+      },
+    );
 
     this.client.on("message", async (ctx) => {
-      console.log("first chatId", ctx.message.chat.id);
-      const sender: ISender = {
-        id: ctx.message.chat.id.toString(),
-        name: ctx.message.from.username || ctx.message.from.first_name,
-        isDirect: true,
-      };
-      const message: IMessage = {
-        chatId: ctx.message.chat.id.toString(), // TODO
-        text: ctx.message.text || "",
-      };
-
-      if (!this.bot) {
-        return;
-      }
-
-      if (message.text) {
-        this.bot.onMessage(
-          sender,
-          message,
-          (reply) => {
-            ctx.reply(reply, {
-              parse_mode: "Markdown",
-            });
-
-            return reply;
-          },
-          (isTyping) => {
-            if (isTyping) {
-              ctx.replyWithChatAction("typing");
-            }
-          },
-        );
-      }
+      messageHandler(this.bot, ctx);
     });
 
     this.client.start();
diff --git a/src/adapters/utils/messages.ts b/src/adapters/utils/messages.ts
@@ -0,0 +1,9 @@
+export const welcomeMessage = `
+  Welcome to the Crypto Bot.
+  You can use the following commands:
+  /start - Start the bot
+  `;
+
+export const settingsMessage = `
+  ⚙️ This is the bot settings menu. To continue, choose the action you want to perform.
+  `;
