@brianli/kimaki 0.4.72-brianli.1

package/dist/commands/fork.js

@@ -0,0 +1,217 @@
+// /fork command - Fork the session from a past user message.
+import { ChatInputCommandInteraction, StringSelectMenuInteraction, StringSelectMenuBuilder, ActionRowBuilder, ChannelType, ThreadAutoArchiveDuration, MessageFlags, } from 'discord.js';
+import { getThreadSession, setThreadSession, setPartMessagesBatch, } from '../database.js';
+import { initializeOpencodeForDirectory } from '../opencode.js';
+import { resolveWorkingDirectory, resolveTextChannel, sendThreadMessage, } from '../discord-utils.js';
+import { collectLastAssistantParts } from '../message-formatting.js';
+import { createLogger, LogPrefix } from '../logger.js';
+import * as errore from 'errore';
+const sessionLogger = createLogger(LogPrefix.SESSION);
+const forkLogger = createLogger(LogPrefix.FORK);
+export async function handleForkCommand(interaction) {
+    const channel = interaction.channel;
+    if (!channel) {
+        await interaction.reply({
+            content: 'This command can only be used in a channel',
+            flags: MessageFlags.Ephemeral,
+        });
+        return;
+    }
+    const isThread = [
+        ChannelType.PublicThread,
+        ChannelType.PrivateThread,
+        ChannelType.AnnouncementThread,
+    ].includes(channel.type);
+    if (!isThread) {
+        await interaction.reply({
+            content: 'This command can only be used in a thread with an active session',
+            flags: MessageFlags.Ephemeral,
+        });
+        return;
+    }
+    const resolved = await resolveWorkingDirectory({
+        channel: channel,
+    });
+    if (!resolved) {
+        await interaction.reply({
+            content: 'Could not determine project directory for this channel',
+            flags: MessageFlags.Ephemeral,
+        });
+        return;
+    }
+    const { projectDirectory } = resolved;
+    const sessionId = await getThreadSession(channel.id);
+    if (!sessionId) {
+        await interaction.reply({
+            content: 'No active session in this thread',
+            flags: MessageFlags.Ephemeral,
+        });
+        return;
+    }
+    // Defer reply before API calls to avoid 3-second timeout
+    await interaction.deferReply({ flags: MessageFlags.Ephemeral });
+    const getClient = await initializeOpencodeForDirectory(projectDirectory);
+    if (getClient instanceof Error) {
+        await interaction.editReply({
+            content: `Failed to load messages: ${getClient.message}`,
+        });
+        return;
+    }
+    try {
+        const messagesResponse = await getClient().session.messages({
+            sessionID: sessionId,
+        });
+        if (!messagesResponse.data) {
+            await interaction.editReply({
+                content: 'Failed to fetch session messages',
+            });
+            return;
+        }
+        const userMessages = messagesResponse.data.filter((m) => m.info.role === 'user');
+        if (userMessages.length === 0) {
+            await interaction.editReply({
+                content: 'No user messages found in this session',
+            });
+            return;
+        }
+        const recentMessages = userMessages.slice(-25);
+        // Filter out synthetic parts (branch context, memory reminders, etc.)
+        // injected by the opencode plugin — they clutter the dropdown preview.
+        const options = recentMessages
+            .map((m, index) => {
+            const textPart = m.parts.find((p) => p.type === 'text' && !p.synthetic);
+            if (!textPart?.text) {
+                return null;
+            }
+            const preview = textPart.text.slice(0, 80);
+            const label = `${index + 1}. ${preview}${preview.length >= 80 ? '...' : ''}`;
+            return {
+                label: label.slice(0, 100),
+                value: m.info.id,
+                description: new Date(m.info.time.created)
+                    .toLocaleString()
+                    .slice(0, 50),
+            };
+        })
+            .filter((o) => o !== null);
+        const selectMenu = new StringSelectMenuBuilder()
+            // Discord component custom_id max length is 100 chars.
+            // Avoid embedding long directory paths (or base64 of them) in the custom ID.
+            // handleForkSelectMenu resolves the directory from the current thread instead.
+            .setCustomId(`fork_select:${sessionId}`)
+            .setPlaceholder('Select a message to fork from')
+            .addOptions(options);
+        const actionRow = new ActionRowBuilder().addComponents(selectMenu);
+        await interaction.editReply({
+            content: '**Fork Session**\nSelect the user message to fork from. The forked session will continue as if you had not sent that message:',
+            components: [actionRow],
+        });
+    }
+    catch (error) {
+        forkLogger.error('Error loading messages:', error);
+        await interaction.editReply({
+            content: `Failed to load messages: ${error instanceof Error ? error.message : 'Unknown error'}`,
+        });
+    }
+}
+export async function handleForkSelectMenu(interaction) {
+    const customId = interaction.customId;
+    if (!customId.startsWith('fork_select:')) {
+        return;
+    }
+    const [, sessionId] = customId.split(':');
+    if (!sessionId) {
+        await interaction.reply({
+            content: 'Invalid selection data',
+            flags: MessageFlags.Ephemeral,
+        });
+        return;
+    }
+    const selectedMessageId = interaction.values[0];
+    if (!selectedMessageId) {
+        await interaction.reply({
+            content: 'No message selected',
+            flags: MessageFlags.Ephemeral,
+        });
+        return;
+    }
+    await interaction.deferReply({ ephemeral: false });
+    const threadChannel = interaction.channel;
+    if (!threadChannel) {
+        await interaction.editReply('Could not access thread channel');
+        return;
+    }
+    const resolved = await resolveWorkingDirectory({
+        channel: threadChannel,
+    });
+    if (!resolved) {
+        await interaction.editReply('Could not determine project directory for this channel');
+        return;
+    }
+    const { projectDirectory } = resolved;
+    const getClient = await initializeOpencodeForDirectory(projectDirectory);
+    if (getClient instanceof Error) {
+        await interaction.editReply(`Failed to fork session: ${getClient.message}`);
+        return;
+    }
+    try {
+        const forkResponse = await getClient().session.fork({
+            sessionID: sessionId,
+            messageID: selectedMessageId,
+        });
+        if (!forkResponse.data) {
+            await interaction.editReply('Failed to fork session');
+            return;
+        }
+        const forkedSession = forkResponse.data;
+        const parentChannel = interaction.channel;
+        if (!parentChannel ||
+            ![
+                ChannelType.PublicThread,
+                ChannelType.PrivateThread,
+                ChannelType.AnnouncementThread,
+            ].includes(parentChannel.type)) {
+            await interaction.editReply('Could not access parent channel');
+            return;
+        }
+        const textChannel = await resolveTextChannel(parentChannel);
+        if (!textChannel) {
+            await interaction.editReply('Could not resolve parent text channel');
+            return;
+        }
+        const thread = await textChannel.threads.create({
+            name: `Fork: ${forkedSession.title}`.slice(0, 100),
+            autoArchiveDuration: ThreadAutoArchiveDuration.OneDay,
+            reason: `Forked from session ${sessionId}`,
+        });
+        // Add user to thread so it appears in their sidebar
+        await thread.members.add(interaction.user.id);
+        await setThreadSession(thread.id, forkedSession.id);
+        sessionLogger.log(`Created forked session ${forkedSession.id} in thread ${thread.id}`);
+        await sendThreadMessage(thread, `**Forked session created!**\nFrom: \`${sessionId}\`\nNew session: \`${forkedSession.id}\``);
+        // Fetch and display the last assistant messages from the forked session
+        const messagesResponse = await getClient().session.messages({
+            sessionID: forkedSession.id,
+        });
+        if (messagesResponse.data) {
+            const { partIds, content } = collectLastAssistantParts({
+                messages: messagesResponse.data,
+            });
+            if (content.trim()) {
+                const discordMessage = await sendThreadMessage(thread, content);
+                // Store part-message mappings atomically
+                await setPartMessagesBatch(partIds.map((partId) => ({
+                    partId,
+                    messageId: discordMessage.id,
+                    threadId: thread.id,
+                })));
+            }
+        }
+        await sendThreadMessage(thread, `You can now continue the conversation from this point.`);
+        await interaction.editReply(`Session forked! Continue in ${thread.toString()}`);
+    }
+    catch (error) {
+        forkLogger.error('Error forking session:', error);
+        await interaction.editReply(`Failed to fork session: ${error instanceof Error ? error.message : 'Unknown error'}`);
+    }
+}

package/dist/commands/gemini-apikey.js

@@ -0,0 +1,70 @@
+// Transcription API key button, slash command, and modal handlers.
+// Auto-detects provider from key prefix: sk-* = OpenAI, otherwise Gemini.
+import { ActionRowBuilder, ModalBuilder, TextInputBuilder, TextInputStyle, MessageFlags, } from 'discord.js';
+import { setGeminiApiKey, setOpenAIApiKey } from '../database.js';
+function buildTranscriptionApiKeyModal(appId) {
+    const modal = new ModalBuilder()
+        .setCustomId(`transcription_apikey_modal:${appId}`)
+        .setTitle('Transcription API Key');
+    const apiKeyInput = new TextInputBuilder()
+        .setCustomId('apikey')
+        .setLabel('OpenAI or Gemini API Key')
+        .setPlaceholder('sk-... or AIza...')
+        .setStyle(TextInputStyle.Short)
+        .setRequired(true);
+    const actionRow = new ActionRowBuilder().addComponents(apiKeyInput);
+    modal.addComponents(actionRow);
+    return modal;
+}
+export async function handleTranscriptionApiKeyButton(interaction) {
+    if (!interaction.customId.startsWith('transcription_apikey:'))
+        return;
+    const appId = interaction.customId
+        .slice('transcription_apikey:'.length)
+        .trim();
+    if (!appId) {
+        await interaction.reply({
+            content: 'Missing app id for API key setup.',
+            flags: MessageFlags.Ephemeral,
+        });
+        return;
+    }
+    await interaction.showModal(buildTranscriptionApiKeyModal(appId));
+}
+export async function handleTranscriptionApiKeyCommand({ interaction, appId, }) {
+    await interaction.showModal(buildTranscriptionApiKeyModal(appId));
+}
+export async function handleTranscriptionApiKeyModalSubmit(interaction) {
+    if (!interaction.customId.startsWith('transcription_apikey_modal:'))
+        return;
+    const appId = interaction.customId
+        .slice('transcription_apikey_modal:'.length)
+        .trim();
+    await interaction.deferReply({ flags: MessageFlags.Ephemeral });
+    if (!appId) {
+        await interaction.editReply({
+            content: 'Missing app id for API key setup.',
+        });
+        return;
+    }
+    const apiKey = interaction.fields.getTextInputValue('apikey').trim();
+    if (!apiKey) {
+        await interaction.editReply({
+            content: 'API key is required.',
+        });
+        return;
+    }
+    // Auto-detect provider from key prefix
+    if (apiKey.startsWith('sk-')) {
+        await setOpenAIApiKey(appId, apiKey);
+        await interaction.editReply({
+            content: 'OpenAI API key saved. Voice messages will be transcribed with OpenAI.',
+        });
+    }
+    else {
+        await setGeminiApiKey(appId, apiKey);
+        await interaction.editReply({
+            content: 'Gemini API key saved. Voice messages will be transcribed with Gemini.',
+        });
+    }
+}