kimaki 0.4.24 → 0.4.26

package/dist/commands/model.js

@@ -0,0 +1,313 @@
+// /model command - Set the preferred model for this channel or session.
+import { ChatInputCommandInteraction, StringSelectMenuInteraction, StringSelectMenuBuilder, ActionRowBuilder, ChannelType, } from 'discord.js';
+import crypto from 'node:crypto';
+import { getDatabase, setChannelModel, setSessionModel, runModelMigrations } from '../database.js';
+import { initializeOpencodeForDirectory } from '../opencode.js';
+import { resolveTextChannel, getKimakiMetadata } from '../discord-utils.js';
+import { abortAndRetrySession } from '../session-handler.js';
+import { createLogger } from '../logger.js';
+const modelLogger = createLogger('MODEL');
+// Store context by hash to avoid customId length limits (Discord max: 100 chars)
+const pendingModelContexts = new Map();
+/**
+ * Handle the /model slash command.
+ * Shows a select menu with available providers.
+ */
+export async function handleModelCommand({ interaction, appId, }) {
+    modelLogger.log('[MODEL] handleModelCommand called');
+    // Defer reply immediately to avoid 3-second timeout
+    await interaction.deferReply({ ephemeral: true });
+    modelLogger.log('[MODEL] Deferred reply');
+    // Ensure migrations are run
+    runModelMigrations();
+    const channel = interaction.channel;
+    if (!channel) {
+        await interaction.editReply({
+            content: 'This command can only be used in a channel',
+        });
+        return;
+    }
+    // Determine if we're in a thread or text channel
+    const isThread = [
+        ChannelType.PublicThread,
+        ChannelType.PrivateThread,
+        ChannelType.AnnouncementThread,
+    ].includes(channel.type);
+    let projectDirectory;
+    let channelAppId;
+    let targetChannelId;
+    let sessionId;
+    if (isThread) {
+        const thread = channel;
+        const textChannel = await resolveTextChannel(thread);
+        const metadata = getKimakiMetadata(textChannel);
+        projectDirectory = metadata.projectDirectory;
+        channelAppId = metadata.channelAppId;
+        targetChannelId = textChannel?.id || channel.id;
+        // Get session ID for this thread
+        const row = getDatabase()
+            .prepare('SELECT session_id FROM thread_sessions WHERE thread_id = ?')
+            .get(thread.id);
+        sessionId = row?.session_id;
+    }
+    else if (channel.type === ChannelType.GuildText) {
+        const textChannel = channel;
+        const metadata = getKimakiMetadata(textChannel);
+        projectDirectory = metadata.projectDirectory;
+        channelAppId = metadata.channelAppId;
+        targetChannelId = channel.id;
+    }
+    else {
+        await interaction.editReply({
+            content: 'This command can only be used in text channels or threads',
+        });
+        return;
+    }
+    if (channelAppId && channelAppId !== appId) {
+        await interaction.editReply({
+            content: 'This channel is not configured for this bot',
+        });
+        return;
+    }
+    if (!projectDirectory) {
+        await interaction.editReply({
+            content: 'This channel is not configured with a project directory',
+        });
+        return;
+    }
+    try {
+        const getClient = await initializeOpencodeForDirectory(projectDirectory);
+        const providersResponse = await getClient().provider.list({
+            query: { directory: projectDirectory },
+        });
+        if (!providersResponse.data) {
+            await interaction.editReply({
+                content: 'Failed to fetch providers',
+            });
+            return;
+        }
+        const { all: allProviders, connected } = providersResponse.data;
+        // Filter to only connected providers (have credentials)
+        const availableProviders = allProviders.filter((p) => {
+            return connected.includes(p.id);
+        });
+        if (availableProviders.length === 0) {
+            await interaction.editReply({
+                content: 'No providers with credentials found. Use `/connect` in OpenCode TUI to add provider credentials.',
+            });
+            return;
+        }
+        // Store context with a short hash key to avoid customId length limits
+        const context = {
+            dir: projectDirectory,
+            channelId: targetChannelId,
+            sessionId: sessionId,
+            isThread: isThread,
+            thread: isThread ? channel : undefined,
+        };
+        const contextHash = crypto.randomBytes(8).toString('hex');
+        pendingModelContexts.set(contextHash, context);
+        const options = availableProviders.slice(0, 25).map((provider) => {
+            const modelCount = Object.keys(provider.models || {}).length;
+            return {
+                label: provider.name.slice(0, 100),
+                value: provider.id,
+                description: `${modelCount} model${modelCount !== 1 ? 's' : ''} available`.slice(0, 100),
+            };
+        });
+        const selectMenu = new StringSelectMenuBuilder()
+            .setCustomId(`model_provider:${contextHash}`)
+            .setPlaceholder('Select a provider')
+            .addOptions(options);
+        const actionRow = new ActionRowBuilder().addComponents(selectMenu);
+        await interaction.editReply({
+            content: '**Set Model Preference**\nSelect a provider:',
+            components: [actionRow],
+        });
+    }
+    catch (error) {
+        modelLogger.error('Error loading providers:', error);
+        await interaction.editReply({
+            content: `Failed to load providers: ${error instanceof Error ? error.message : 'Unknown error'}`,
+        });
+    }
+}
+/**
+ * Handle the provider select menu interaction.
+ * Shows a second select menu with models for the chosen provider.
+ */
+export async function handleProviderSelectMenu(interaction) {
+    const customId = interaction.customId;
+    if (!customId.startsWith('model_provider:')) {
+        return;
+    }
+    // Defer update immediately to avoid timeout
+    await interaction.deferUpdate();
+    const contextHash = customId.replace('model_provider:', '');
+    const context = pendingModelContexts.get(contextHash);
+    if (!context) {
+        await interaction.editReply({
+            content: 'Selection expired. Please run /model again.',
+            components: [],
+        });
+        return;
+    }
+    const selectedProviderId = interaction.values[0];
+    if (!selectedProviderId) {
+        await interaction.editReply({
+            content: 'No provider selected',
+            components: [],
+        });
+        return;
+    }
+    try {
+        const getClient = await initializeOpencodeForDirectory(context.dir);
+        const providersResponse = await getClient().provider.list({
+            query: { directory: context.dir },
+        });
+        if (!providersResponse.data) {
+            await interaction.editReply({
+                content: 'Failed to fetch providers',
+                components: [],
+            });
+            return;
+        }
+        const provider = providersResponse.data.all.find((p) => p.id === selectedProviderId);
+        if (!provider) {
+            await interaction.editReply({
+                content: 'Provider not found',
+                components: [],
+            });
+            return;
+        }
+        const models = Object.entries(provider.models || {})
+            .map(([modelId, model]) => ({
+            id: modelId,
+            name: model.name,
+            releaseDate: model.release_date,
+        }))
+            // Sort by release date descending (most recent first)
+            .sort((a, b) => {
+            const dateA = a.releaseDate ? new Date(a.releaseDate).getTime() : 0;
+            const dateB = b.releaseDate ? new Date(b.releaseDate).getTime() : 0;
+            return dateB - dateA;
+        });
+        if (models.length === 0) {
+            await interaction.editReply({
+                content: `No models available for ${provider.name}`,
+                components: [],
+            });
+            return;
+        }
+        // Take first 25 models (most recent since sorted descending)
+        const recentModels = models.slice(0, 25);
+        // Update context with provider info and reuse the same hash
+        context.providerId = selectedProviderId;
+        context.providerName = provider.name;
+        pendingModelContexts.set(contextHash, context);
+        const options = recentModels.map((model) => {
+            const dateStr = model.releaseDate
+                ? new Date(model.releaseDate).toLocaleDateString()
+                : 'Unknown date';
+            return {
+                label: model.name.slice(0, 100),
+                value: model.id,
+                description: dateStr.slice(0, 100),
+            };
+        });
+        const selectMenu = new StringSelectMenuBuilder()
+            .setCustomId(`model_select:${contextHash}`)
+            .setPlaceholder('Select a model')
+            .addOptions(options);
+        const actionRow = new ActionRowBuilder().addComponents(selectMenu);
+        await interaction.editReply({
+            content: `**Set Model Preference**\nProvider: **${provider.name}**\nSelect a model:`,
+            components: [actionRow],
+        });
+    }
+    catch (error) {
+        modelLogger.error('Error loading models:', error);
+        await interaction.editReply({
+            content: `Failed to load models: ${error instanceof Error ? error.message : 'Unknown error'}`,
+            components: [],
+        });
+    }
+}
+/**
+ * Handle the model select menu interaction.
+ * Stores the model preference in the database.
+ */
+export async function handleModelSelectMenu(interaction) {
+    const customId = interaction.customId;
+    if (!customId.startsWith('model_select:')) {
+        return;
+    }
+    // Defer update immediately
+    await interaction.deferUpdate();
+    const contextHash = customId.replace('model_select:', '');
+    const context = pendingModelContexts.get(contextHash);
+    if (!context || !context.providerId || !context.providerName) {
+        await interaction.editReply({
+            content: 'Selection expired. Please run /model again.',
+            components: [],
+        });
+        return;
+    }
+    const selectedModelId = interaction.values[0];
+    if (!selectedModelId) {
+        await interaction.editReply({
+            content: 'No model selected',
+            components: [],
+        });
+        return;
+    }
+    // Build full model ID: provider_id/model_id
+    const fullModelId = `${context.providerId}/${selectedModelId}`;
+    try {
+        // Store in appropriate table based on context
+        if (context.isThread && context.sessionId) {
+            // Store for session
+            setSessionModel(context.sessionId, fullModelId);
+            modelLogger.log(`Set model ${fullModelId} for session ${context.sessionId}`);
+            // Check if there's a running request and abort+retry with new model
+            let retried = false;
+            if (context.thread) {
+                retried = await abortAndRetrySession({
+                    sessionId: context.sessionId,
+                    thread: context.thread,
+                    projectDirectory: context.dir,
+                });
+            }
+            if (retried) {
+                await interaction.editReply({
+                    content: `Model changed for this session:\n**${context.providerName}** / **${selectedModelId}**\n\n\`${fullModelId}\`\n\n_Retrying current request with new model..._`,
+                    components: [],
+                });
+            }
+            else {
+                await interaction.editReply({
+                    content: `Model preference set for this session:\n**${context.providerName}** / **${selectedModelId}**\n\n\`${fullModelId}\``,
+                    components: [],
+                });
+            }
+        }
+        else {
+            // Store for channel
+            setChannelModel(context.channelId, fullModelId);
+            modelLogger.log(`Set model ${fullModelId} for channel ${context.channelId}`);
+            await interaction.editReply({
+                content: `Model preference set for this channel:\n**${context.providerName}** / **${selectedModelId}**\n\n\`${fullModelId}\`\n\nAll new sessions in this channel will use this model.`,
+                components: [],
+            });
+        }
+        // Clean up the context from memory
+        pendingModelContexts.delete(contextHash);
+    }
+    catch (error) {
+        modelLogger.error('Error saving model preference:', error);
+        await interaction.editReply({
+            content: `Failed to save model preference: ${error instanceof Error ? error.message : 'Unknown error'}`,
+            components: [],
+        });
+    }
+}

package/dist/commands/permissions.js

@@ -0,0 +1,126 @@
+// Permission commands - /accept, /accept-always, /reject
+import { ChannelType } from 'discord.js';
+import { initializeOpencodeForDirectory } from '../opencode.js';
+import { pendingPermissions } from '../session-handler.js';
+import { SILENT_MESSAGE_FLAGS } from '../discord-utils.js';
+import { createLogger } from '../logger.js';
+const logger = createLogger('PERMISSIONS');
+export async function handleAcceptCommand({ command, }) {
+    const scope = command.commandName === 'accept-always' ? 'always' : 'once';
+    const channel = command.channel;
+    if (!channel) {
+        await command.reply({
+            content: 'This command can only be used in a channel',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    const isThread = [
+        ChannelType.PublicThread,
+        ChannelType.PrivateThread,
+        ChannelType.AnnouncementThread,
+    ].includes(channel.type);
+    if (!isThread) {
+        await command.reply({
+            content: 'This command can only be used in a thread with an active session',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    const pending = pendingPermissions.get(channel.id);
+    if (!pending) {
+        await command.reply({
+            content: 'No pending permission request in this thread',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    try {
+        const getClient = await initializeOpencodeForDirectory(pending.directory);
+        await getClient().postSessionIdPermissionsPermissionId({
+            path: {
+                id: pending.permission.sessionID,
+                permissionID: pending.permission.id,
+            },
+            body: {
+                response: scope,
+            },
+        });
+        pendingPermissions.delete(channel.id);
+        const msg = scope === 'always'
+            ? `✅ Permission **accepted** (auto-approve similar requests)`
+            : `✅ Permission **accepted**`;
+        await command.reply({ content: msg, flags: SILENT_MESSAGE_FLAGS });
+        logger.log(`Permission ${pending.permission.id} accepted with scope: ${scope}`);
+    }
+    catch (error) {
+        logger.error('[ACCEPT] Error:', error);
+        await command.reply({
+            content: `Failed to accept permission: ${error instanceof Error ? error.message : 'Unknown error'}`,
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+    }
+}
+export async function handleRejectCommand({ command, }) {
+    const channel = command.channel;
+    if (!channel) {
+        await command.reply({
+            content: 'This command can only be used in a channel',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    const isThread = [
+        ChannelType.PublicThread,
+        ChannelType.PrivateThread,
+        ChannelType.AnnouncementThread,
+    ].includes(channel.type);
+    if (!isThread) {
+        await command.reply({
+            content: 'This command can only be used in a thread with an active session',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    const pending = pendingPermissions.get(channel.id);
+    if (!pending) {
+        await command.reply({
+            content: 'No pending permission request in this thread',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    try {
+        const getClient = await initializeOpencodeForDirectory(pending.directory);
+        await getClient().postSessionIdPermissionsPermissionId({
+            path: {
+                id: pending.permission.sessionID,
+                permissionID: pending.permission.id,
+            },
+            body: {
+                response: 'reject',
+            },
+        });
+        pendingPermissions.delete(channel.id);
+        await command.reply({
+            content: `❌ Permission **rejected**`,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        logger.log(`Permission ${pending.permission.id} rejected`);
+    }
+    catch (error) {
+        logger.error('[REJECT] Error:', error);
+        await command.reply({
+            content: `Failed to reject permission: ${error instanceof Error ? error.message : 'Unknown error'}`,
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+    }
+}

package/dist/commands/queue.js

@@ -0,0 +1,129 @@
+// Queue commands - /queue, /clear-queue
+import { ChannelType } from 'discord.js';
+import { getDatabase } from '../database.js';
+import { resolveTextChannel, getKimakiMetadata, sendThreadMessage, SILENT_MESSAGE_FLAGS, } from '../discord-utils.js';
+import { handleOpencodeSession, abortControllers, addToQueue, getQueueLength, clearQueue, } from '../session-handler.js';
+import { createLogger } from '../logger.js';
+const logger = createLogger('QUEUE');
+export async function handleQueueCommand({ command, }) {
+    const message = command.options.getString('message', true);
+    const channel = command.channel;
+    if (!channel) {
+        await command.reply({
+            content: 'This command can only be used in a channel',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    const isThread = [
+        ChannelType.PublicThread,
+        ChannelType.PrivateThread,
+        ChannelType.AnnouncementThread,
+    ].includes(channel.type);
+    if (!isThread) {
+        await command.reply({
+            content: 'This command can only be used in a thread with an active session',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    const row = getDatabase()
+        .prepare('SELECT session_id FROM thread_sessions WHERE thread_id = ?')
+        .get(channel.id);
+    if (!row?.session_id) {
+        await command.reply({
+            content: 'No active session in this thread. Send a message directly instead.',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    // Check if there's an active request running
+    const hasActiveRequest = abortControllers.has(row.session_id);
+    if (!hasActiveRequest) {
+        // No active request, send immediately
+        const textChannel = await resolveTextChannel(channel);
+        const { projectDirectory } = getKimakiMetadata(textChannel);
+        if (!projectDirectory) {
+            await command.reply({
+                content: 'Could not determine project directory',
+                ephemeral: true,
+                flags: SILENT_MESSAGE_FLAGS,
+            });
+            return;
+        }
+        await command.reply({
+            content: `» **${command.user.displayName}:** ${message.slice(0, 100)}${message.length > 100 ? '...' : ''}`,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        logger.log(`[QUEUE] No active request, sending immediately in thread ${channel.id}`);
+        handleOpencodeSession({
+            prompt: message,
+            thread: channel,
+            projectDirectory,
+            channelId: textChannel?.id || channel.id,
+        }).catch(async (e) => {
+            logger.error(`[QUEUE] Failed to send message:`, e);
+            const errorMsg = e instanceof Error ? e.message : String(e);
+            await sendThreadMessage(channel, `✗ Failed: ${errorMsg.slice(0, 200)}`);
+        });
+        return;
+    }
+    // Add to queue
+    const queuePosition = addToQueue({
+        threadId: channel.id,
+        message: {
+            prompt: message,
+            userId: command.user.id,
+            username: command.user.displayName,
+            queuedAt: Date.now(),
+        },
+    });
+    await command.reply({
+        content: `✅ Message queued (position: ${queuePosition}). Will be sent after current response.`,
+        ephemeral: true,
+        flags: SILENT_MESSAGE_FLAGS,
+    });
+    logger.log(`[QUEUE] User ${command.user.displayName} queued message in thread ${channel.id}`);
+}
+export async function handleClearQueueCommand({ command, }) {
+    const channel = command.channel;
+    if (!channel) {
+        await command.reply({
+            content: 'This command can only be used in a channel',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    const isThread = [
+        ChannelType.PublicThread,
+        ChannelType.PrivateThread,
+        ChannelType.AnnouncementThread,
+    ].includes(channel.type);
+    if (!isThread) {
+        await command.reply({
+            content: 'This command can only be used in a thread',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    const queueLength = getQueueLength(channel.id);
+    if (queueLength === 0) {
+        await command.reply({
+            content: 'No messages in queue',
+            ephemeral: true,
+            flags: SILENT_MESSAGE_FLAGS,
+        });
+        return;
+    }
+    clearQueue(channel.id);
+    await command.reply({
+        content: `🗑 Cleared ${queueLength} queued message${queueLength > 1 ? 's' : ''}`,
+        flags: SILENT_MESSAGE_FLAGS,
+    });
+    logger.log(`[QUEUE] User ${command.user.displayName} cleared queue in thread ${channel.id}`);
+}