🐛 fix: add pinned stories chunking for files over 50mb

chupapee · Jun 12, 2024 · 6e12b84 · 6e12b84
1 parent da17129
commit 6e12b84
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 14 deletions.
diff --git a/src/controllers/download-stories.ts b/src/controllers/download-stories.ts
@@ -2,10 +2,10 @@ import { Userbot } from 'config/userbot';
 import { timeout } from 'lib';
 import { Api } from 'telegram';
 
-export type StoriesModel = typeof mapStories;
+export type StoriesModel = ReturnType<typeof mapStories>;
 
 export async function downloadStories(
-  stories: ReturnType<StoriesModel>,
+  stories: StoriesModel,
   storiesType: 'active' | 'pinned'
 ) {
   const client = await Userbot.getInstance();

diff --git a/src/controllers/send-stories.ts b/src/controllers/send-stories.ts
@@ -86,7 +86,7 @@ async function sendActiveStories({ stories, task }: SendStoriesArgs) {
           album.map((x) => ({
             media: { source: x.buffer! },
             type: x.mediaType,
-            caption: 'Active stories',
+            caption: x.caption ?? 'Active stories',
           }))
         );
       }
@@ -172,14 +172,18 @@ async function sendPinnedStories({ stories, task }: SendStoriesArgs) {
       .catch(() => null);
 
     if (uploadableStories.length > 0) {
-      await bot.telegram.sendMediaGroup(
-        task.chatId,
-        uploadableStories.map((x) => ({
-          media: { source: x.buffer! },
-          type: x.mediaType,
-          caption: 'Pinned stories',
-        }))
-      );
+      const chunkedList = chunkMediafiles(uploadableStories);
+
+      for (const album of chunkedList) {
+        await bot.telegram.sendMediaGroup(
+          task.chatId,
+          album.map((x) => ({
+            media: { source: x.buffer! },
+            type: x.mediaType,
+            caption: x.caption ?? 'Pinned stories',
+          }))
+        );
+      }
     } else {
       await bot.telegram.sendMessage(
         task.chatId,

diff --git a/src/lib/helpers.ts b/src/lib/helpers.ts
@@ -1,13 +1,18 @@
 import { StoriesModel } from 'controllers/download-stories';
 
+const MAX_STORIES_SIZE = 45;
+
 export const timeout = (sec: number): Promise<null> =>
   new Promise((ok) => setTimeout(ok, sec));
 
-export function chunkMediafiles(files: ReturnType<StoriesModel>) {
+export function chunkMediafiles(files: StoriesModel) {
   return files.reduce(
-    (acc: Array<ReturnType<StoriesModel>>, curr) => {
+    (acc: Array<StoriesModel>, curr) => {
       const tempAccWithCurr = [...acc[acc.length - 1], curr];
-      if (tempAccWithCurr.length === 10 || sumOfSizes(tempAccWithCurr) >= 50) {
+      if (
+        tempAccWithCurr.length === 10 ||
+        sumOfSizes(tempAccWithCurr) >= MAX_STORIES_SIZE
+      ) {
         acc.push([curr]);
         return acc;
       }