@geminixiang/mama 0.2.0-beta.0 → 0.2.0-beta.2

package/dist/adapters/slack/bot.js

@@ -4,6 +4,7 @@ import { appendFileSync, existsSync, mkdirSync, readFileSync } from "fs";
 import { readFile } from "fs/promises";
 import { basename, join } from "path";
 import * as log from "../../log.js";
+import { PRODUCT_NAME, formatForceStopped, formatNothingRunning } from "../../ui-copy.js";
 import { createSlackAdapters } from "./context.js";
 // ============================================================================
 // Exponential backoff utility for Slack API calls
@@ -125,6 +126,21 @@ export class SlackBot {
             return result.ts;
         });
     }
+    async postEphemeral(channel, user, text) {
+        return withRetry(async () => {
+            await this.webClient.chat.postEphemeral({ channel, user, text });
+        });
+    }
+    async openDirectMessage(userId) {
+        return withRetry(async () => {
+            const result = await this.webClient.conversations.open({ users: userId });
+            const channelId = result.channel?.id;
+            if (!channelId) {
+                throw new Error(`Failed to open DM for user ${userId}`);
+            }
+            return channelId;
+        });
+    }
     async updateMessage(channel, ts, text) {
         return withRetry(async () => {
             await this.webClient.chat.update({ channel, ts, text });
@@ -236,14 +252,30 @@ export class SlackBot {
      * Returns true if enqueued, false if queue is full (max 5).
      */
     enqueueEvent(event) {
-        const queue = this.getQueue(event.channel);
+        const conversationId = event.conversationId;
+        const queue = this.getQueue(conversationId);
         if (queue.size() >= 5) {
-            log.logWarning(`Event queue full for ${event.channel}, discarding: ${event.text.substring(0, 50)}`);
+            log.logWarning(`Event queue full for ${conversationId}, discarding: ${event.text.substring(0, 50)}`);
             return false;
         }
-        log.logInfo(`Enqueueing event for ${event.channel}: ${event.text.substring(0, 50)}`);
+        log.logInfo(`Enqueueing event for ${conversationId}: ${event.text.substring(0, 50)}`);
         queue.enqueue(() => {
-            const adapters = createSlackAdapters(event, this, true);
+            const slackEvent = {
+                type: event.type,
+                conversationId,
+                conversationKind: event.conversationKind,
+                channel: conversationId,
+                ts: event.ts,
+                thread_ts: event.thread_ts,
+                user: event.user,
+                text: event.text,
+                attachments: event.attachments?.map((attachment) => ({
+                    original: attachment.name,
+                    localPath: attachment.localPath,
+                })),
+                sessionKey: event.sessionKey,
+            };
+            const adapters = createSlackAdapters(slackEvent, this, true);
             return this.handler.handleEvent(event, this, adapters, true);
         });
         return true;
@@ -267,12 +299,12 @@ export class SlackBot {
                 type: "section",
                 text: {
                     type: "mrkdwn",
-                    text: "*Pi Agent*\nWelcome back! Start a new task or check on running work.",
+                    text: `*${PRODUCT_NAME}*\nStart a new task or check on running work.`,
                 },
                 accessory: {
                     type: "image",
                     image_url: "https://media1.tenor.com/m/lfDATg4Bhc0AAAAC/happy-cat.gif",
-                    alt_text: "Pi Agent",
+                    alt_text: PRODUCT_NAME,
                 },
             },
         ];
@@ -364,11 +396,11 @@ export class SlackBot {
             for (const ev of periodicEvents) {
                 const channelLabel = ev.platform === "slack"
                     ? (() => {
-                        const channel = this.channels.get(ev.channelId);
-                        const channelName = channel ? `#${channel.name}` : ev.channelId;
+                        const channel = this.channels.get(ev.conversationId);
+                        const channelName = channel ? `#${channel.name}` : ev.conversationId;
                         return `${ev.platform}:${channelName}`;
                     })()
-                    : `${ev.platform}:${ev.channelId}`;
+                    : `${ev.platform}:${ev.conversationId}`;
                 const nextStr = ev.nextRun
                     ? new Date(ev.nextRun).toLocaleString("en-US", {
                         month: "short",
@@ -413,6 +445,115 @@ export class SlackBot {
         }
         return this.handler.isRunning(channelId) ? channelId : null;
     }
+    createDirectCommandAdapters(conversationId, userId, userName, text, ts) {
+        const message = {
+            id: ts,
+            sessionKey: conversationId,
+            conversationKind: "direct",
+            userId,
+            userName,
+            text,
+            attachments: [],
+        };
+        const responseCtx = {
+            respond: async (responseText) => {
+                const messageTs = await this.postMessage(conversationId, responseText);
+                this.logBotResponse(conversationId, responseText, messageTs);
+            },
+            replaceResponse: async (responseText) => {
+                const messageTs = await this.postMessage(conversationId, responseText);
+                this.logBotResponse(conversationId, responseText, messageTs);
+            },
+            respondInThread: async (responseText) => {
+                const messageTs = await this.postMessage(conversationId, responseText);
+                this.logBotResponse(conversationId, responseText, messageTs);
+            },
+            setTyping: async () => { },
+            setWorking: async () => { },
+            uploadFile: async (filePath, title) => {
+                await this.uploadFile(conversationId, filePath, title);
+            },
+            deleteResponse: async () => { },
+        };
+        return {
+            message,
+            responseCtx,
+            platform: this.getPlatformInfo(),
+        };
+    }
+    createSlashCommandBot(conversationId, threadTs) {
+        return {
+            start: async () => { },
+            postMessage: async (_channel, text) => {
+                if (threadTs) {
+                    return this.postInThread(conversationId, threadTs, text);
+                }
+                return this.postMessage(conversationId, text);
+            },
+            updateMessage: async (channel, ts, text) => {
+                await this.updateMessage(channel, ts, text);
+            },
+            enqueueEvent: (event) => this.enqueueEvent(event),
+            getPlatformInfo: () => this.getPlatformInfo(),
+        };
+    }
+    async routeSlashLoginCommand(payload) {
+        const commandSuffix = payload.text?.trim();
+        const commandText = commandSuffix ? `${payload.command} ${commandSuffix}` : payload.command;
+        const createdAt = new Date();
+        const eventTs = (createdAt.getTime() / 1000).toFixed(6);
+        const sourceChannelId = payload.channel_id;
+        const isDirectMessage = sourceChannelId.startsWith("D");
+        const targetChannelId = isDirectMessage
+            ? sourceChannelId
+            : await this.openDirectMessage(payload.user_id);
+        const userName = payload.user_name ?? this.getUser(payload.user_id)?.userName;
+        this.logToFile(targetChannelId, {
+            date: createdAt.toISOString(),
+            ts: eventTs,
+            user: payload.user_id,
+            userName,
+            text: commandText,
+            attachments: [],
+            isBot: false,
+        });
+        if (!isDirectMessage) {
+            await this.postEphemeral(sourceChannelId, payload.user_id, `我已私訊你 ${PRODUCT_NAME} 的登入連結，請到私訊完成設定。`);
+        }
+        const event = {
+            type: "dm",
+            conversationId: targetChannelId,
+            conversationKind: "direct",
+            ts: eventTs,
+            user: payload.user_id,
+            text: commandText,
+            attachments: [],
+            sessionKey: targetChannelId,
+        };
+        const adapters = this.createDirectCommandAdapters(targetChannelId, payload.user_id, userName, commandText, eventTs);
+        await this.handler.handleEvent(event, this, adapters, false);
+    }
+    async routeSlashNewCommand(payload) {
+        const conversationId = payload.channel_id;
+        if (!conversationId.startsWith("D")) {
+            await this.postEphemeral(conversationId, payload.user_id, `為了避免誤清除共享上下文，${payload.command} 目前只能在與 ${PRODUCT_NAME} 的私訊中使用。`);
+            return;
+        }
+        const createdAt = new Date();
+        const eventTs = (createdAt.getTime() / 1000).toFixed(6);
+        const userName = payload.user_name ?? this.getUser(payload.user_id)?.userName;
+        this.logToFile(conversationId, {
+            date: createdAt.toISOString(),
+            ts: eventTs,
+            user: payload.user_id,
+            userName,
+            text: payload.command,
+            attachments: [],
+            isBot: false,
+        });
+        const commandBot = this.createSlashCommandBot(conversationId);
+        await this.handler.handleNew(conversationId, conversationId, commandBot);
+    }
     setupEventHandlers() {
         // Channel @mentions
         this.socketClient.on("app_mention", ({ event, ack }) => {
@@ -427,6 +568,8 @@ export class SlackBot {
             const sessionKey = e.thread_ts ? `${e.channel}:${e.thread_ts}` : e.channel;
             const slackEvent = {
                 type: "mention",
+                conversationId: e.channel,
+                conversationKind: "shared",
                 channel: e.channel,
                 ts: e.ts,
                 thread_ts: e.thread_ts,
@@ -451,21 +594,15 @@ export class SlackBot {
                     this.handler.handleStop(stopTarget, e.channel, this);
                 }
                 else {
-                    this.postMessage(e.channel, "_Nothing running_");
+                    this.postMessage(e.channel, formatNothingRunning("slack"));
                 }
                 ack();
                 return;
             }
-            // SYNC: Check if busy (per-thread)
-            if (this.handler.isRunning(sessionKey)) {
-                this.postMessage(e.channel, "_Already working in this thread. Say `@mama stop` to cancel._");
-            }
-            else {
-                this.getQueue(sessionKey).enqueue(() => {
-                    const adapters = createSlackAdapters(slackEvent, this, false);
-                    return this.handler.handleEvent(slackEvent, this, adapters, false);
-                });
-            }
+            this.getQueue(sessionKey).enqueue(() => {
+                const adapters = createSlackAdapters(slackEvent, this, false);
+                return this.handler.handleEvent(slackEvent, this, adapters, false);
+            });
             ack();
         });
         // All messages (for logging) + DMs (for triggering)
@@ -485,6 +622,7 @@ export class SlackBot {
                 return;
             }
             const isDM = e.channel_type === "im";
+            const conversationKind = isDM ? "direct" : "shared";
             const isBotMention = e.text?.includes(`<@${this.botUserId}>`);
             // Skip channel @mentions - already handled by app_mention event
             if (!isDM && isBotMention) {
@@ -493,6 +631,8 @@ export class SlackBot {
             }
             const slackEvent = {
                 type: isDM ? "dm" : "mention",
+                conversationId: e.channel,
+                conversationKind,
                 channel: e.channel,
                 ts: e.ts,
                 thread_ts: e.thread_ts,
@@ -518,7 +658,7 @@ export class SlackBot {
                     this.handler.handleStop(stopTarget, e.channel, this);
                 }
                 else {
-                    this.postMessage(e.channel, "_Nothing running_");
+                    this.postMessage(e.channel, formatNothingRunning("slack"));
                 }
                 ack();
                 return;
@@ -532,23 +672,47 @@ export class SlackBot {
                         this.handler.handleStop(dmSessionKey, e.channel, this); // Don't await, don't queue
                     }
                     else {
-                        this.postMessage(e.channel, "_Nothing running_");
+                        this.postMessage(e.channel, formatNothingRunning("slack"));
                     }
                     ack();
                     return;
                 }
-                if (this.handler.isRunning(dmSessionKey)) {
-                    this.postMessage(e.channel, "_Already working. Say `stop` to cancel._");
-                }
-                else {
-                    this.getQueue(dmSessionKey).enqueue(() => {
-                        const adapters = createSlackAdapters(slackEvent, this, false);
-                        return this.handler.handleEvent(slackEvent, this, adapters, false);
-                    });
-                }
+                this.getQueue(dmSessionKey).enqueue(() => {
+                    const adapters = createSlackAdapters(slackEvent, this, false);
+                    return this.handler.handleEvent(slackEvent, this, adapters, false);
+                });
             }
             ack();
         });
+        this.socketClient.on("slash_commands", async ({ body, ack }) => {
+            const payload = body;
+            await ack();
+            if (!payload.command || !payload.channel_id || !payload.user_id) {
+                return;
+            }
+            const handlerPromise = payload.command === "/pi-login"
+                ? this.routeSlashLoginCommand({
+                    command: payload.command,
+                    text: payload.text,
+                    channel_id: payload.channel_id,
+                    user_id: payload.user_id,
+                    user_name: payload.user_name,
+                })
+                : payload.command === "/pi-new"
+                    ? this.routeSlashNewCommand({
+                        command: payload.command,
+                        channel_id: payload.channel_id,
+                        user_id: payload.user_id,
+                        user_name: payload.user_name,
+                    })
+                    : null;
+            if (!handlerPromise) {
+                return;
+            }
+            handlerPromise.catch((err) => {
+                log.logWarning("Slack slash command error", err instanceof Error ? err.message : String(err));
+            });
+        });
         // App Home tab
         this.socketClient.on("app_home_opened", ({ event, ack }) => {
             const e = event;
@@ -579,7 +743,7 @@ export class SlackBot {
             // Use handler's forceStop method
             this.handler.forceStop(sessionKey);
             // Notify in channel
-            await this.postMessage(channelId, `_🔴 Force stopped by ${userId}_`);
+            await this.postMessage(channelId, formatForceStopped("slack", userId ?? "unknown"));
             // Refresh home tab
             if (userId) {
                 this.webClient.views