kimaki 0.0.3 → 0.1.2

package/src/tools.ts

@@ -0,0 +1,422 @@
+import { tool } from 'ai'
+import { z } from 'zod'
+import { spawn, type ChildProcess } from 'node:child_process'
+import net from 'node:net'
+import {
+  createOpencodeClient,
+  type OpencodeClient,
+  type AssistantMessage,
+  type Provider,
+} from '@opencode-ai/sdk'
+import { createLogger } from './logger.js'
+
+const toolsLogger = createLogger('TOOLS')
+import { formatDistanceToNow } from 'date-fns'
+
+import { ShareMarkdown } from './markdown.js'
+import pc from 'picocolors'
+import { initializeOpencodeForDirectory } from './discordBot.js'
+
+export async function getTools({
+  onMessageCompleted,
+  directory,
+}: {
+  directory: string
+  onMessageCompleted?: (params: {
+    sessionId: string
+    messageId: string
+    data?: { info: AssistantMessage }
+    error?: unknown
+    markdown?: string
+  }) => void
+}) {
+  const getClient = await initializeOpencodeForDirectory(directory)
+  const client = getClient()
+
+  const markdownRenderer = new ShareMarkdown(client)
+
+  const providersResponse = await client.config.providers({})
+  const providers: Provider[] = providersResponse.data?.providers || []
+
+  // Helper: get last assistant model for a session (non-summary)
+  const getSessionModel = async (
+    sessionId: string,
+  ): Promise<{ providerID: string; modelID: string } | undefined> => {
+    const res = await getClient().session.messages({ path: { id: sessionId } })
+    const data = res.data
+    if (!data || data.length === 0) return undefined
+    for (let i = data.length - 1; i >= 0; i--) {
+      const info = data?.[i]?.info
+      if (info?.role === 'assistant') {
+        const ai = info as AssistantMessage
+        if (!ai.summary && ai.providerID && ai.modelID) {
+          return { providerID: ai.providerID, modelID: ai.modelID }
+        }
+      }
+    }
+    return undefined
+  }
+
+  const tools = {
+    submitMessage: tool({
+      description:
+        'Submit a message to an existing chat session. Does not wait for the message to complete',
+      inputSchema: z.object({
+        sessionId: z.string().describe('The session ID to send message to'),
+        message: z.string().describe('The message text to send'),
+      }),
+      execute: async ({ sessionId, message }) => {
+        const sessionModel = await getSessionModel(sessionId)
+
+        // do not await
+        getClient()
+          .session.prompt({
+            path: { id: sessionId },
+
+            body: {
+              parts: [{ type: 'text', text: message }],
+              model: sessionModel,
+            },
+          })
+          .then(async (response) => {
+            const markdown = await markdownRenderer.generate({
+              sessionID: sessionId,
+              lastAssistantOnly: true,
+            })
+            onMessageCompleted?.({
+              sessionId,
+              messageId: '',
+              data: response.data,
+              markdown,
+            })
+          })
+          .catch((error) => {
+            onMessageCompleted?.({
+              sessionId,
+              messageId: '',
+              error,
+            })
+          })
+        return {
+          success: true,
+          sessionId,
+          directive: 'Tell user that message has been sent successfully',
+        }
+      },
+    }),
+
+    createNewChat: tool({
+      description:
+        'Start a new chat session with an initial message. Does not wait for the message to complete',
+      inputSchema: z.object({
+        message: z
+          .string()
+          .describe('The initial message to start the chat with'),
+        title: z.string().optional().describe('Optional title for the session'),
+        model: z
+          .object({
+            providerId: z
+              .string()
+              .describe('The provider ID (e.g., "anthropic", "openai")'),
+            modelId: z
+              .string()
+              .describe(
+                'The model ID (e.g., "claude-opus-4-20250514", "gpt-5")',
+              ),
+          })
+          .optional()
+          .describe('Optional model to use for this session'),
+      }),
+      execute: async ({ message, title, model }) => {
+        if (!message.trim()) {
+          throw new Error(`message must be a non empty string`)
+        }
+
+        try {
+          const session = await getClient().session.create({
+            body: {
+              title: title || message.slice(0, 50),
+            },
+          })
+
+          if (!session.data) {
+            throw new Error('Failed to create session')
+          }
+
+          // do not await
+          getClient()
+            .session.prompt({
+              path: { id: session.data.id },
+              body: {
+                parts: [{ type: 'text', text: message }],
+              },
+            })
+            .then(async (response) => {
+              const markdown = await markdownRenderer.generate({
+                sessionID: session.data.id,
+                lastAssistantOnly: true,
+              })
+              onMessageCompleted?.({
+                sessionId: session.data.id,
+                messageId: '',
+                data: response.data,
+                markdown,
+              })
+            })
+            .catch((error) => {
+              onMessageCompleted?.({
+                sessionId: session.data.id,
+                messageId: '',
+                error,
+              })
+            })
+
+          return {
+            success: true,
+            sessionId: session.data.id,
+            title: session.data.title,
+          }
+        } catch (error) {
+          return {
+            success: false,
+            error:
+              error instanceof Error
+                ? error.message
+                : 'Failed to create chat session',
+          }
+        }
+      },
+    }),
+
+    listChats: tool({
+      description:
+        'Get a list of available chat sessions sorted by most recent',
+      inputSchema: z.object({}),
+      execute: async () => {
+        toolsLogger.log(`Listing opencode sessions`)
+        const sessions = await getClient().session.list()
+
+        if (!sessions.data) {
+          return { success: false, error: 'No sessions found' }
+        }
+
+        const sortedSessions = [...sessions.data]
+          .sort((a, b) => {
+            return b.time.updated - a.time.updated
+          })
+          .slice(0, 20)
+
+        const sessionList = sortedSessions.map(async (session) => {
+          const finishedAt = session.time.updated
+          const status = await (async () => {
+            if (session.revert) return 'error'
+            const messagesResponse = await getClient().session.messages({
+              path: { id: session.id },
+            })
+            const messages = messagesResponse.data || []
+            const lastMessage = messages[messages.length - 1]
+            if (
+              lastMessage?.info.role === 'assistant' &&
+              !lastMessage.info.time.completed
+            ) {
+              return 'in_progress'
+            }
+            return 'finished'
+          })()
+
+          return {
+            id: session.id,
+            folder: session.directory,
+            status,
+            finishedAt: formatDistanceToNow(new Date(finishedAt), {
+              addSuffix: true,
+            }),
+            title: session.title,
+            prompt: session.title,
+          }
+        })
+
+        const resolvedList = await Promise.all(sessionList)
+
+        return {
+          success: true,
+          sessions: resolvedList,
+        }
+      },
+    }),
+
+    searchFiles: tool({
+      description: 'Search for files in a folder',
+      inputSchema: z.object({
+        folder: z
+          .string()
+          .optional()
+          .describe(
+            'The folder path to search in, optional. only use if user specifically asks for it',
+          ),
+        query: z.string().describe('The search query for files'),
+      }),
+      execute: async ({ folder, query }) => {
+        const results = await getClient().find.files({
+          query: {
+            query,
+            directory: folder,
+          },
+        })
+
+        return {
+          success: true,
+          files: results.data || [],
+        }
+      },
+    }),
+
+    readSessionMessages: tool({
+      description: 'Read messages from a chat session',
+      inputSchema: z.object({
+        sessionId: z.string().describe('The session ID to read messages from'),
+        lastAssistantOnly: z
+          .boolean()
+          .optional()
+          .describe('Only read the last assistant message'),
+      }),
+      execute: async ({ sessionId, lastAssistantOnly = false }) => {
+        if (lastAssistantOnly) {
+          const messages = await getClient().session.messages({
+            path: { id: sessionId },
+          })
+
+          if (!messages.data) {
+            return { success: false, error: 'No messages found' }
+          }
+
+          const assistantMessages = messages.data.filter(
+            (m) => m.info.role === 'assistant',
+          )
+
+          if (assistantMessages.length === 0) {
+            return {
+              success: false,
+              error: 'No assistant messages found',
+            }
+          }
+
+          const lastMessage = assistantMessages[assistantMessages.length - 1]
+          const status =
+            'completed' in lastMessage!.info.time &&
+            lastMessage!.info.time.completed
+              ? 'completed'
+              : 'in_progress'
+
+          const markdown = await markdownRenderer.generate({
+            sessionID: sessionId,
+            lastAssistantOnly: true,
+          })
+
+          return {
+            success: true,
+            markdown,
+            status,
+          }
+        } else {
+          const markdown = await markdownRenderer.generate({
+            sessionID: sessionId,
+          })
+
+          const messages = await getClient().session.messages({
+            path: { id: sessionId },
+          })
+          const lastMessage = messages.data?.[messages.data.length - 1]
+          const status =
+            lastMessage?.info.role === 'assistant' &&
+            lastMessage?.info.time &&
+            'completed' in lastMessage.info.time &&
+            !lastMessage.info.time.completed
+              ? 'in_progress'
+              : 'completed'
+
+          return {
+            success: true,
+            markdown,
+            status,
+          }
+        }
+      },
+    }),
+
+    abortChat: tool({
+      description: 'Abort/stop an in-progress chat session',
+      inputSchema: z.object({
+        sessionId: z.string().describe('The session ID to abort'),
+      }),
+      execute: async ({ sessionId }) => {
+        try {
+          const result = await getClient().session.abort({
+            path: { id: sessionId },
+          })
+
+          if (!result.data) {
+            return {
+              success: false,
+              error: 'Failed to abort session',
+            }
+          }
+
+          return {
+            success: true,
+            sessionId,
+            message: 'Session aborted successfully',
+          }
+        } catch (error) {
+          return {
+            success: false,
+            error:
+              error instanceof Error ? error.message : 'Unknown error occurred',
+          }
+        }
+      },
+    }),
+
+    getModels: tool({
+      description: 'Get all available AI models from all providers',
+      inputSchema: z.object({}),
+      execute: async () => {
+        try {
+          const providersResponse = await getClient().config.providers({})
+          const providers: Provider[] = providersResponse.data?.providers || []
+
+          const models: Array<{ providerId: string; modelId: string }> = []
+
+          providers.forEach((provider) => {
+            if (provider.models && typeof provider.models === 'object') {
+              Object.entries(provider.models).forEach(([modelId, model]) => {
+                models.push({
+                  providerId: provider.id,
+                  modelId: modelId,
+                })
+              })
+            }
+          })
+
+          return {
+            success: true,
+            models,
+            totalCount: models.length,
+          }
+        } catch (error) {
+          return {
+            success: false,
+            error:
+              error instanceof Error ? error.message : 'Failed to fetch models',
+            models: [],
+          }
+        }
+      },
+    }),
+  }
+
+  return {
+    tools,
+    providers,
+  }
+}

package/src/utils.ts

@@ -0,0 +1,73 @@
+import { PermissionsBitField } from 'discord.js'
+
+type GenerateInstallUrlOptions = {
+  clientId: string
+  permissions?: bigint[]
+  scopes?: string[]
+  guildId?: string
+  disableGuildSelect?: boolean
+}
+
+export function generateBotInstallUrl({
+  clientId,
+  permissions = [
+    PermissionsBitField.Flags.ViewChannel,
+    PermissionsBitField.Flags.ManageChannels,
+    PermissionsBitField.Flags.SendMessages,
+    PermissionsBitField.Flags.SendMessagesInThreads,
+    PermissionsBitField.Flags.CreatePublicThreads,
+    PermissionsBitField.Flags.ManageThreads,
+    PermissionsBitField.Flags.ReadMessageHistory,
+    PermissionsBitField.Flags.AddReactions,
+    PermissionsBitField.Flags.ManageMessages,
+    PermissionsBitField.Flags.UseExternalEmojis,
+    PermissionsBitField.Flags.AttachFiles,
+    PermissionsBitField.Flags.Connect,
+    PermissionsBitField.Flags.Speak,
+  ],
+  scopes = ['bot'],
+  guildId,
+  disableGuildSelect = false,
+}: GenerateInstallUrlOptions): string {
+  const permissionsBitField = new PermissionsBitField(permissions)
+  const permissionsValue = permissionsBitField.bitfield.toString()
+
+  const url = new URL('https://discord.com/api/oauth2/authorize')
+  url.searchParams.set('client_id', clientId)
+  url.searchParams.set('permissions', permissionsValue)
+  url.searchParams.set('scope', scopes.join(' '))
+
+  if (guildId) {
+    url.searchParams.set('guild_id', guildId)
+  }
+
+  if (disableGuildSelect) {
+    url.searchParams.set('disable_guild_select', 'true')
+  }
+
+  return url.toString()
+}
+
+function getRequiredBotPermissions(): bigint[] {
+  return [
+    PermissionsBitField.Flags.ViewChannel,
+    PermissionsBitField.Flags.ManageChannels,
+    PermissionsBitField.Flags.SendMessages,
+    PermissionsBitField.Flags.SendMessagesInThreads,
+    PermissionsBitField.Flags.CreatePublicThreads,
+    PermissionsBitField.Flags.ManageThreads,
+    PermissionsBitField.Flags.ReadMessageHistory,
+    PermissionsBitField.Flags.AddReactions,
+    PermissionsBitField.Flags.ManageMessages,
+    PermissionsBitField.Flags.UseExternalEmojis,
+    PermissionsBitField.Flags.AttachFiles,
+    PermissionsBitField.Flags.Connect,
+    PermissionsBitField.Flags.Speak,
+  ]
+}
+
+function getPermissionNames(): string[] {
+  const permissions = getRequiredBotPermissions()
+  const permissionsBitField = new PermissionsBitField(permissions)
+  return permissionsBitField.toArray()
+}

package/src/voice.ts

@@ -0,0 +1,42 @@
+import { openai } from '@ai-sdk/openai'
+import { experimental_transcribe as transcribe } from 'ai'
+import { createLogger } from './logger.js'
+
+const voiceLogger = createLogger('VOICE')
+
+export async function transcribeAudio({
+  audio,
+  prompt,
+  language,
+  temperature,
+}: {
+  audio: Buffer | Uint8Array | ArrayBuffer | string
+  prompt?: string
+  language?: string
+  temperature?: number
+}): Promise<string> {
+  try {
+    const result = await transcribe({
+      model: openai.transcription('whisper-1'),
+      audio,
+      ...(prompt || language || temperature !== undefined
+        ? {
+            providerOptions: {
+              openai: {
+                ...(prompt && { prompt }),
+                ...(language && { language }),
+                ...(temperature !== undefined && { temperature }),
+              },
+            },
+          }
+        : {}),
+    })
+
+    return result.text
+  } catch (error) {
+    voiceLogger.error('Failed to transcribe audio:', error)
+    throw new Error(
+      `Audio transcription failed: ${error instanceof Error ? error.message : 'Unknown error'}`,
+    )
+  }
+}

package/src/worker-types.ts

@@ -0,0 +1,60 @@
+import type { Tool as AITool } from 'ai'
+
+// Messages sent from main thread to worker
+export type WorkerInMessage =
+  | {
+      type: 'init'
+      directory: string // Project directory for tools
+      systemMessage?: string
+      guildId: string
+      channelId: string
+    }
+  | {
+      type: 'sendRealtimeInput'
+      audio?: {
+        mimeType: string
+        data: string // base64
+      }
+      audioStreamEnd?: boolean
+    }
+  | {
+      type: 'sendTextInput'
+      text: string
+    }
+  | {
+      type: 'interrupt'
+    }
+  | {
+      type: 'stop'
+    }
+
+// Messages sent from worker to main thread via parentPort
+export type WorkerOutMessage =
+  | {
+      type: 'assistantOpusPacket'
+      packet: ArrayBuffer // Opus encoded audio packet
+    }
+  | {
+      type: 'assistantStartSpeaking'
+    }
+  | {
+      type: 'assistantStopSpeaking'
+    }
+  | {
+      type: 'assistantInterruptSpeaking'
+    }
+  | {
+      type: 'toolCallCompleted'
+      sessionId: string
+      messageId: string
+      data?: any
+      error?: any
+      markdown?: string
+    }
+  | {
+      type: 'error'
+      error: string
+    }
+  | {
+      type: 'ready'
+    }

package/src/xml.test.ts

@@ -0,0 +1,37 @@
+import { describe, test, expect } from 'vitest'
+import { extractNonXmlContent } from './xml.js'
+
+describe('extractNonXmlContent', () => {
+  test('removes xml tags and returns only text content', () => {
+    const xml = 'Hello <tag>content</tag> world <nested><inner>deep</inner></nested> end'
+    expect(extractNonXmlContent(xml)).toMatchInlineSnapshot(`
+      "Hello
+      world
+      end"
+    `)
+  })
+
+  test('handles multiple text segments', () => {
+    const xml = 'Start <a>tag1</a> middle <b>tag2</b> finish'
+    expect(extractNonXmlContent(xml)).toMatchInlineSnapshot(`
+      "Start
+      middle
+      finish"
+    `)
+  })
+
+  test('handles only xml without text', () => {
+    const xml = '<root><child>content</child></root>'
+    expect(extractNonXmlContent(xml)).toMatchInlineSnapshot(`""`)
+  })
+
+  test('handles only text without xml', () => {
+    const xml = 'Just plain text'
+    expect(extractNonXmlContent(xml)).toMatchInlineSnapshot(`"Just plain text"`)
+  })
+
+  test('handles empty string', () => {
+    const xml = ''
+    expect(extractNonXmlContent(xml)).toMatchInlineSnapshot(`""`)
+  })
+})