kimaki 0.4.20 → 0.4.22

package/src/discordBot.ts

@@ -46,6 +46,7 @@ import * as prism from 'prism-media'
 import dedent from 'string-dedent'
 import { transcribeAudio } from './voice.js'
 import { extractTagsArrays, extractNonXmlContent } from './xml.js'
+import { formatMarkdownTables } from './format-tables.js'
 import prettyMilliseconds from 'pretty-ms'
 import type { Session } from '@google/genai'
 import { createLogger } from './logger.js'
@@ -61,7 +62,6 @@ type ParsedCommand = {
 } | {
   isCommand: false
 }
-
 function parseSlashCommand(text: string): ParsedCommand {
   const trimmed = text.trim()
   if (!trimmed.startsWith('/')) {
@@ -84,6 +84,32 @@ The user cannot see bash tool outputs. If there is important information in bash
 
 Your current OpenCode session ID is: ${sessionId}
 
+## permissions
+
+Only users with these Discord permissions can send messages to the bot:
+- Server Owner
+- Administrator permission
+- Manage Server permission
+- "Kimaki" role (case-insensitive)
+
+## changing the model
+
+To change the model used by OpenCode, edit the project's \`opencode.json\` config file and set the \`model\` field:
+
+\`\`\`json
+{
+  "model": "anthropic/claude-sonnet-4-20250514"
+}
+\`\`\`
+
+Examples:
+- \`"anthropic/claude-sonnet-4-20250514"\` - Claude Sonnet 4
+- \`"anthropic/claude-opus-4-20250514"\` - Claude Opus 4
+- \`"openai/gpt-4o"\` - GPT-4o
+- \`"google/gemini-2.5-pro"\` - Gemini 2.5 Pro
+
+Format is \`provider/model-name\`. You can also set \`small_model\` for tasks like title generation.
+
 ## uploading files to discord
 
 To upload files to the Discord thread (images, screenshots, long files that would clutter the chat), run:
@@ -731,6 +757,7 @@ async function sendThreadMessage(
 ): Promise<Message> {
   const MAX_LENGTH = 2000
 
+  content = formatMarkdownTables(content)
   content = escapeBackticksInCodeBlocks(content)
 
   const chunks = splitMarkdownForDiscord({ content, maxLength: MAX_LENGTH })
@@ -805,7 +832,6 @@ async function processVoiceAttachment({
     `Detected audio attachment: ${audioAttachment.name} (${audioAttachment.contentType})`,
   )
 
-  await message.react('⏳')
   await sendThreadMessage(thread, '🎤 Transcribing voice message...')
 
   const audioResponse = await fetch(audioAttachment.url)
@@ -1068,9 +1094,9 @@ function escapeInlineCode(text: string): string {
     .replace(/\|\|/g, '\\|\\|') // Double pipes (spoiler syntax)
 }
 
-function resolveTextChannel(
+async function resolveTextChannel(
   channel: TextChannel | ThreadChannel | null | undefined,
-): TextChannel | null {
+): Promise<TextChannel | null> {
   if (!channel) {
     return null
   }
@@ -1084,9 +1110,12 @@ function resolveTextChannel(
     channel.type === ChannelType.PrivateThread ||
     channel.type === ChannelType.AnnouncementThread
   ) {
-    const parent = channel.parent
-    if (parent?.type === ChannelType.GuildText) {
-      return parent as TextChannel
+    const parentId = channel.parentId
+    if (parentId) {
+      const parent = await channel.guild.channels.fetch(parentId)
+      if (parent?.type === ChannelType.GuildText) {
+        return parent as TextChannel
+      }
     }
   }
 
@@ -1259,38 +1288,65 @@ function getToolSummaryText(part: Part): string {
   if (part.type !== 'tool') return ''
 
   if (part.tool === 'edit') {
+    const filePath = (part.state.input?.filePath as string) || ''
     const newString = (part.state.input?.newString as string) || ''
     const oldString = (part.state.input?.oldString as string) || ''
     const added = newString.split('\n').length
     const removed = oldString.split('\n').length
-    return `(+${added}-${removed})`
+    const fileName = filePath.split('/').pop() || ''
+    return fileName ? `*${fileName}* (+${added}-${removed})` : `(+${added}-${removed})`
   }
 
   if (part.tool === 'write') {
+    const filePath = (part.state.input?.filePath as string) || ''
     const content = (part.state.input?.content as string) || ''
     const lines = content.split('\n').length
-    return `(${lines} line${lines === 1 ? '' : 's'})`
+    const fileName = filePath.split('/').pop() || ''
+    return fileName ? `*${fileName}* (${lines} line${lines === 1 ? '' : 's'})` : `(${lines} line${lines === 1 ? '' : 's'})`
   }
 
   if (part.tool === 'webfetch') {
     const url = (part.state.input?.url as string) || ''
     const urlWithoutProtocol = url.replace(/^https?:\/\//, '')
-    return urlWithoutProtocol ? `(${urlWithoutProtocol})` : ''
+    return urlWithoutProtocol ? `*${urlWithoutProtocol}*` : ''
   }
 
-  if (
-    part.tool === 'bash' ||
-    part.tool === 'read' ||
-    part.tool === 'list' ||
-    part.tool === 'glob' ||
-    part.tool === 'grep' ||
-    part.tool === 'task' ||
-    part.tool === 'todoread' ||
-    part.tool === 'todowrite'
-  ) {
+  if (part.tool === 'read') {
+    const filePath = (part.state.input?.filePath as string) || ''
+    const fileName = filePath.split('/').pop() || ''
+    return fileName ? `*${fileName}*` : ''
+  }
+
+  if (part.tool === 'list') {
+    const path = (part.state.input?.path as string) || ''
+    const dirName = path.split('/').pop() || path
+    return dirName ? `*${dirName}*` : ''
+  }
+
+  if (part.tool === 'glob') {
+    const pattern = (part.state.input?.pattern as string) || ''
+    return pattern ? `*${pattern}*` : ''
+  }
+
+  if (part.tool === 'grep') {
+    const pattern = (part.state.input?.pattern as string) || ''
+    return pattern ? `*${pattern}*` : ''
+  }
+
+  if (part.tool === 'bash' || part.tool === 'todoread' || part.tool === 'todowrite') {
     return ''
   }
 
+  if (part.tool === 'task') {
+    const description = (part.state.input?.description as string) || ''
+    return description ? `_${description}_` : ''
+  }
+
+  if (part.tool === 'skill') {
+    const name = (part.state.input?.name as string) || ''
+    return name ? `_${name}_` : ''
+  }
+
   if (!part.state.input) return ''
 
   const inputFields = Object.entries(part.state.input)
@@ -1314,19 +1370,12 @@ function formatTodoList(part: Part): string {
       content: string
       status: 'pending' | 'in_progress' | 'completed' | 'cancelled'
     }[]) || []
-  if (todos.length === 0) return ''
-  return todos
-    .map((todo, i) => {
-      const num = `${i + 1}.`
-      if (todo.status === 'in_progress') {
-        return `${num} **${todo.content}**`
-      }
-      if (todo.status === 'completed' || todo.status === 'cancelled') {
-        return `${num} ~~${todo.content}~~`
-      }
-      return `${num} ${todo.content}`
-    })
-    .join('\n')
+  const activeIndex = todos.findIndex((todo) => {
+    return todo.status === 'in_progress'
+  })
+  const activeTodo = todos[activeIndex]
+  if (activeIndex === -1 || !activeTodo) return ''
+  return `${activeIndex + 1}. **${activeTodo.content}**`
 }
 
 function formatPart(part: Part): string {
@@ -1372,17 +1421,16 @@ function formatPart(part: Part): string {
       toolTitle = part.state.error || 'error'
     } else if (part.tool === 'bash') {
       const command = (part.state.input?.command as string) || ''
+      const description = (part.state.input?.description as string) || ''
       const isSingleLine = !command.includes('\n')
-      const hasBackticks = command.includes('`')
-      if (isSingleLine && command.length <= 120 && !hasBackticks) {
+      const hasUnderscores = command.includes('_')
+      if (isSingleLine && !hasUnderscores && command.length <= 50) {
         toolTitle = `_${command}_`
-      } else {
-        toolTitle = stateTitle ? `_${stateTitle}_` : ''
+      } else if (description) {
+        toolTitle = `_${description}_`
+      } else if (stateTitle) {
+        toolTitle = `_${stateTitle}_`
       }
-    } else if (part.tool === 'edit' || part.tool === 'write') {
-      const filePath = (part.state.input?.filePath as string) || ''
-      const fileName = filePath.split('/').pop() || filePath
-      toolTitle = fileName ? `_${fileName}_` : ''
     } else if (stateTitle) {
       toolTitle = `_${stateTitle}_`
     }
@@ -1541,6 +1589,51 @@ async function handleOpencodeSession({
   let usedModel: string | undefined
   let usedProviderID: string | undefined
   let tokensUsedInSession = 0
+  let lastDisplayedContextPercentage = 0
+  let modelContextLimit: number | undefined
+
+  let typingInterval: NodeJS.Timeout | null = null
+
+  function startTyping(): () => void {
+    if (abortController.signal.aborted) {
+      discordLogger.log(`Not starting typing, already aborted`)
+      return () => {}
+    }
+    if (typingInterval) {
+      clearInterval(typingInterval)
+      typingInterval = null
+    }
+
+    thread.sendTyping().catch((e) => {
+      discordLogger.log(`Failed to send initial typing: ${e}`)
+    })
+
+    typingInterval = setInterval(() => {
+      thread.sendTyping().catch((e) => {
+        discordLogger.log(`Failed to send periodic typing: ${e}`)
+      })
+    }, 8000)
+
+    if (!abortController.signal.aborted) {
+      abortController.signal.addEventListener(
+        'abort',
+        () => {
+          if (typingInterval) {
+            clearInterval(typingInterval)
+            typingInterval = null
+          }
+        },
+        { once: true },
+      )
+    }
+
+    return () => {
+      if (typingInterval) {
+        clearInterval(typingInterval)
+        typingInterval = null
+      }
+    }
+  }
 
   const sendPartMessage = async (part: Part) => {
     const content = formatPart(part) + '\n\n'
@@ -1570,58 +1663,6 @@ async function handleOpencodeSession({
   }
 
   const eventHandler = async () => {
-    // Local typing function for this session
-    // Outer-scoped interval for typing notifications. Only one at a time.
-    let typingInterval: NodeJS.Timeout | null = null
-
-    function startTyping(thread: ThreadChannel): () => void {
-      if (abortController.signal.aborted) {
-        discordLogger.log(`Not starting typing, already aborted`)
-        return () => {}
-      }
-      // Clear any previous typing interval
-      if (typingInterval) {
-        clearInterval(typingInterval)
-        typingInterval = null
-      }
-
-      // Send initial typing
-      thread.sendTyping().catch((e) => {
-        discordLogger.log(`Failed to send initial typing: ${e}`)
-      })
-
-      // Set up interval to send typing every 8 seconds
-      typingInterval = setInterval(() => {
-        thread.sendTyping().catch((e) => {
-          discordLogger.log(`Failed to send periodic typing: ${e}`)
-        })
-      }, 8000)
-
-      // Only add listener if not already aborted
-      if (!abortController.signal.aborted) {
-        abortController.signal.addEventListener(
-          'abort',
-          () => {
-            if (typingInterval) {
-              clearInterval(typingInterval)
-              typingInterval = null
-            }
-          },
-          {
-            once: true,
-          },
-        )
-      }
-
-      // Return stop function
-      return () => {
-        if (typingInterval) {
-          clearInterval(typingInterval)
-          typingInterval = null
-        }
-      }
-    }
-
     try {
       let assistantMessageId: string | undefined
 
@@ -1645,6 +1686,30 @@ async function handleOpencodeSession({
             assistantMessageId = msg.id
             usedModel = msg.modelID
             usedProviderID = msg.providerID
+
+            if (tokensUsedInSession > 0 && usedProviderID && usedModel) {
+              if (!modelContextLimit) {
+                try {
+                  const providersResponse = await getClient().provider.list({ query: { directory } })
+                  const provider = providersResponse.data?.all?.find((p) => p.id === usedProviderID)
+                  const model = provider?.models?.[usedModel]
+                  if (model?.limit?.context) {
+                    modelContextLimit = model.limit.context
+                  }
+                } catch (e) {
+                  sessionLogger.error('Failed to fetch provider info for context limit:', e)
+                }
+              }
+
+              if (modelContextLimit) {
+                const currentPercentage = Math.floor((tokensUsedInSession / modelContextLimit) * 100)
+                const thresholdCrossed = Math.floor(currentPercentage / 10) * 10
+                if (thresholdCrossed > lastDisplayedContextPercentage && thresholdCrossed >= 10) {
+                  lastDisplayedContextPercentage = thresholdCrossed
+                  await sendThreadMessage(thread, `◼︎ context usage ${currentPercentage}%`)
+                }
+              }
+            }
           }
         } else if (event.type === 'message.part.updated') {
           const part = event.properties.part
@@ -1672,7 +1737,7 @@ async function handleOpencodeSession({
 
           // Start typing on step-start
           if (part.type === 'step-start') {
-            stopTyping = startTyping(thread)
+            stopTyping = startTyping()
           }
 
           // Send tool parts immediately when they start running
@@ -1698,7 +1763,7 @@ async function handleOpencodeSession({
             // start typing in a moment, so that if the session finished, because step-finish is at the end of the message, we do not show typing status
             setTimeout(() => {
               if (abortController.signal.aborted) return
-              stopTyping = startTyping(thread)
+              stopTyping = startTyping()
             }, 300)
           }
         } else if (event.type === 'session.error') {
@@ -1847,13 +1912,7 @@ async function handleOpencodeSession({
       return
     }
 
-    if (originalMessage) {
-      try {
-        await originalMessage.react('⏳')
-      } catch (e) {
-        discordLogger.log(`Could not add processing reaction:`, e)
-      }
-    }
+    stopTyping = startTyping()
 
     let response: { data?: unknown; error?: unknown; response: Response }
     if (parsedCommand?.isCommand) {
@@ -2074,14 +2133,20 @@ export async function startDiscordBot({
         }
       }
 
-      // Check if user is authoritative (server owner or has admin permissions)
+      // Check if user is authoritative (server owner, admin, manage server, or has Kimaki role)
       if (message.guild && message.member) {
         const isOwner = message.member.id === message.guild.ownerId
         const isAdmin = message.member.permissions.has(
           PermissionsBitField.Flags.Administrator,
         )
+        const canManageServer = message.member.permissions.has(
+          PermissionsBitField.Flags.ManageGuild,
+        )
+        const hasKimakiRole = message.member.roles.cache.some(
+          (role) => role.name.toLowerCase() === 'kimaki',
+        )
 
-        if (!isOwner && !isAdmin) {
+        if (!isOwner && !isAdmin && !canManageServer && !hasKimakiRole) {
           return
         }
       }
@@ -2297,11 +2362,8 @@ export async function startDiscordBot({
 
             // Get the channel's project directory from its topic
             let projectDirectory: string | undefined
-            if (
-              interaction.channel &&
-              interaction.channel.type === ChannelType.GuildText
-            ) {
-              const textChannel = resolveTextChannel(
+            if (interaction.channel) {
+              const textChannel = await resolveTextChannel(
                 interaction.channel as TextChannel | ThreadChannel | null,
               )
               if (textChannel) {
@@ -2383,11 +2445,8 @@ export async function startDiscordBot({
 
               // Get the channel's project directory from its topic
               let projectDirectory: string | undefined
-              if (
-                interaction.channel &&
-                interaction.channel.type === ChannelType.GuildText
-              ) {
-                const textChannel = resolveTextChannel(
+              if (interaction.channel) {
+                const textChannel = await resolveTextChannel(
                   interaction.channel as TextChannel | ThreadChannel | null,
                 )
                 if (textChannel) {
@@ -2751,7 +2810,7 @@ export async function startDiscordBot({
               if (partsToRender.length > 0) {
                 const combinedContent = partsToRender
                   .map((p) => p.content)
-                  .join('\n\n')
+                  .join('\n')
 
                 const discordMessage = await sendThreadMessage(
                   thread,
@@ -2856,7 +2915,7 @@ export async function startDiscordBot({
                 `Failed to create channels: ${error instanceof Error ? error.message : 'Unknown error'}`,
               )
             }
-          } else if (command.commandName === 'add-new-project') {
+          } else if (command.commandName === 'create-new-project') {
             await command.deferReply({ ephemeral: false })
 
             const projectName = command.options.getString('name', true)
@@ -3091,7 +3150,7 @@ export async function startDiscordBot({
               return
             }
 
-            const textChannel = resolveTextChannel(channel as ThreadChannel)
+            const textChannel = await resolveTextChannel(channel as ThreadChannel)
             const { projectDirectory: directory } = getKimakiMetadata(textChannel)
 
             if (!directory) {
@@ -3162,7 +3221,7 @@ export async function startDiscordBot({
               return
             }
 
-            const textChannel = resolveTextChannel(channel as ThreadChannel)
+            const textChannel = await resolveTextChannel(channel as ThreadChannel)
             const { projectDirectory: directory } = getKimakiMetadata(textChannel)
 
             if (!directory) {
@@ -3270,15 +3329,20 @@ export async function startDiscordBot({
       const member = newState.member || oldState.member
       if (!member) return
 
-      // Check if user is admin or server owner
+      // Check if user is admin, server owner, can manage server, or has Kimaki role
       const guild = newState.guild || oldState.guild
       const isOwner = member.id === guild.ownerId
       const isAdmin = member.permissions.has(
         PermissionsBitField.Flags.Administrator,
       )
+      const canManageServer = member.permissions.has(
+        PermissionsBitField.Flags.ManageGuild,
+      )
+      const hasKimakiRole = member.roles.cache.some(
+        (role) => role.name.toLowerCase() === 'kimaki',
+      )
 
-      if (!isOwner && !isAdmin) {
-        // Not an admin user, ignore
+      if (!isOwner && !isAdmin && !canManageServer && !hasKimakiRole) {
         return
       }
 
@@ -3304,7 +3368,9 @@ export async function startDiscordBot({
             if (m.id === member.id || m.user.bot) return false
             return (
               m.id === guild.ownerId ||
-              m.permissions.has(PermissionsBitField.Flags.Administrator)
+              m.permissions.has(PermissionsBitField.Flags.Administrator) ||
+              m.permissions.has(PermissionsBitField.Flags.ManageGuild) ||
+              m.roles.cache.some((role) => role.name.toLowerCase() === 'kimaki')
             )
           })
 
@@ -3349,7 +3415,9 @@ export async function startDiscordBot({
               if (m.id === member.id || m.user.bot) return false
               return (
                 m.id === guild.ownerId ||
-                m.permissions.has(PermissionsBitField.Flags.Administrator)
+                m.permissions.has(PermissionsBitField.Flags.Administrator) ||
+                m.permissions.has(PermissionsBitField.Flags.ManageGuild) ||
+                m.roles.cache.some((role) => role.name.toLowerCase() === 'kimaki')
               )
             })