chore: test

ubq-testing · Sep 11, 2024 · 9095add · 9095add
1 parent 77e392d
commit 9095add
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 8 deletions.
diff --git a/src/adapters/supabase/helpers/chats.ts b/src/adapters/supabase/helpers/chats.ts
@@ -17,6 +17,26 @@ export class Chats extends Super {
         super(supabase, context);
     }
 
+    async userSnapshot(chatId: number, userIds: number[]) {
+        const { error } = await this.supabase.from("chats").upsert({ chatId, userIds })
+        if (error) {
+            this.context.logger.error("Failed to save chat users", { chatId, userIds, er: error });
+        } else {
+            this.context.logger.info("Successfully saved chat users", { chatId, userIds });
+        }
+    }
+
+    async getChatUsers(chatId: number) {
+        const { data, error } = (await this.supabase.from("chats").select("userIds").eq("chatId", chatId).single())
+        if (error || !data) {
+            this.context.logger.error("No chat users found", { chatId });
+        } else {
+            this.context.logger.info("Successfully fetched chat users", { chatId });
+        }
+
+        return data;
+    }
+
     async updateChatStatus(status: string, taskNodeId?: string, chatId?: number) {
         if (!taskNodeId && !chatId) {
             this.context.logger.error("No taskNodeId or chatId provided to update chat status");

diff --git a/src/bot/mtproto-api/workrooms.ts b/src/bot/mtproto-api/workrooms.ts
@@ -111,9 +111,6 @@ export async function closeChat(context: Context<"issues.closed", SupportedEvent
             throw new Error("Failed to fetch chat participants");
         }
 
-        console.log("chatParticipants", chatParticipants);
-        console.log("fetchChat users:", fetchChat.users);
-
         // archive it
         await mtProto.client.invoke(
             new mtProto.api.folders.EditPeerFolders({
@@ -136,6 +133,8 @@ export async function closeChat(context: Context<"issues.closed", SupportedEvent
                 return participant.userId;
             });
 
+            await chats.userSnapshot(chat.chatId, userIDs.map((id) => id.toJSNumber()));
+
             for (let i = 0; i < userIDs.length; i++) {
                 if (userIDs[i].toJSNumber() === context.config.botId) {
                     continue;
@@ -197,8 +196,6 @@ export async function reopenChat(context: Context<"issues.reopened", SupportedEv
     chatFull = fetchChat.fullChat as Api.ChatFull
     participants = chatFull.participants as Api.ChatParticipantsForbidden;
 
-    console.log("participants", participants);
-
     const chatCreator = participants.selfParticipant?.userId;
     if (!chatCreator) {
         throw new Error("Failed to get chat creator");
@@ -224,11 +221,26 @@ export async function reopenChat(context: Context<"issues.reopened", SupportedEv
 
     await chats.updateChatStatus("reopened", payload.issue.node_id);
 
-    // Now we must re-invite all previously banned users
-    // can we get a list of banned users from the chat object?
+    const users = await chats.getChatUsers(chat.chatId);
 
+    if (!users) {
+        throw new Error("Failed to get chat users");
+    }
 
+    const { userIds } = users;
 
-    return { status: 200, reason: "chat_reopened" };
+    for (let i = 0; i < userIds.length; i++) {
+        if (userIds[i] === context.config.botId) {
+            continue;
+        }
+        await mtProto.client.invoke(
+            new mtProto.api.messages.AddChatUser({
+                chatId: chat.chatId,
+                userId: userIds[i].userId,
+                fwdLimit: 50,
+            })
+        );
+    }
 
+    return { status: 200, reason: "chat_reopened" };
 }