feat: Add the KOL profile to the display

sukbearai · Jan 7, 2025 · 1b2ace5 · 1b2ace5
1 parent daa633d
commit 1b2ace5
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/src/scenes/kol.ts b/src/scenes/kol.ts
@@ -1,3 +1,4 @@
+import type { TwitterProfile } from '../twitter/manager'
 import { Scenes } from 'telegraf'
 import { startTwitterAuth } from '~/src/twitter/'
 import { cleanMarkdown } from '../utils/markdown'
@@ -12,7 +13,25 @@ kolScene.on('text', async (ctx) => {
   const twitterManager = await startTwitterAuth()
   const msg = ctx.message.text
 
+  const profile = (await twitterManager.fetchProfile(msg)) as TwitterProfile
   const content = await twitterManager.handleTwitterKolList(msg, 'kol')
+
+  if (profile?.screenName) {
+    await ctx.replyWithMediaGroup([
+      {
+        type: 'photo',
+        media: profile.banner,
+        caption: `
+<b>${profile.screenName}</b>
+<code>
+${profile.bio}
+</code>
+`.trim(),
+        parse_mode: 'HTML',
+      },
+    ])
+  }
+
   await ctx.reply(cleanMarkdown(content))
   await ctx.scene.leave()
 })
diff --git a/src/twitter/manager.ts b/src/twitter/manager.ts
@@ -9,7 +9,7 @@ import { writeTweetsTxt } from '~/src/utils/file'
 import { consola } from '~/src/utils/log'
 import { RequestQueue } from './queue'
 
-interface TwitterProfile {
+export interface TwitterProfile {
   id: string
   username: string
   screenName: string
@@ -268,7 +268,7 @@ export class TwitterManager {
     }
   }
 
-  async fetchProfile(username: string): Promise<TwitterProfile | null> {
+  async fetchProfile(username: string): Promise<TwitterProfile | string> {
     try {
       const profile = await this.requestQueue.add(async () => {
         const profile = await this.twitterClient.getProfile(username)
@@ -286,7 +286,7 @@ export class TwitterManager {
     } catch (error) {
       console.error('Error fetching Twitter profile:', error)
 
-      return null
+      return 'error'
     }
   }