kimaki 0.4.10 → 0.4.13

package/src/escape-backticks.test.ts

@@ -1,6 +1,6 @@
 import { test, expect } from 'vitest'
 import { Lexer } from 'marked'
-import { escapeBackticksInCodeBlocks } from './discordBot.js'
+import { escapeBackticksInCodeBlocks, splitMarkdownForDiscord } from './discordBot.js'
 
 
 
@@ -144,3 +144,304 @@ const x = \\\`hello\\\`
 "
 `)
 })
+
+test('splitMarkdownForDiscord returns single chunk for short content', () => {
+  const result = splitMarkdownForDiscord({
+    content: 'Hello world',
+    maxLength: 100,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "Hello world",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord splits at line boundaries', () => {
+  const result = splitMarkdownForDiscord({
+    content: 'Line 1\nLine 2\nLine 3\nLine 4',
+    maxLength: 15,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "Line 1
+    Line 2
+    ",
+      "Line 3
+    Line 4",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord preserves code blocks when not split', () => {
+  const result = splitMarkdownForDiscord({
+    content: '```js\nconst x = 1\n```',
+    maxLength: 100,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "\`\`\`js
+    const x = 1
+    \`\`\`",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord adds closing and opening fences when splitting code block', () => {
+  const result = splitMarkdownForDiscord({
+    content: '```js\nline1\nline2\nline3\nline4\n```',
+    maxLength: 20,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "\`\`\`js
+    line1
+    line2
+    \`\`\`
+    ",
+      "\`\`\`js
+    line3
+    line4
+    \`\`\`
+    ",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles code block with language', () => {
+  const result = splitMarkdownForDiscord({
+    content: '```typescript\nconst a = 1\nconst b = 2\n```',
+    maxLength: 30,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "\`\`\`typescript
+    const a = 1
+    \`\`\`
+    ",
+      "\`\`\`typescript
+    const b = 2
+    \`\`\`
+    ",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles mixed content with code blocks', () => {
+  const result = splitMarkdownForDiscord({
+    content: 'Text before\n```js\ncode\n```\nText after',
+    maxLength: 25,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "Text before
+    \`\`\`js
+    code
+    \`\`\`
+    ",
+      "Text after",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles code block without language', () => {
+  const result = splitMarkdownForDiscord({
+    content: '```\nline1\nline2\n```',
+    maxLength: 12,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "\`\`\`
+    line1
+    \`\`\`
+    ",
+      "\`\`\`
+    line2
+    \`\`\`
+    ",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles multiple consecutive code blocks', () => {
+  const result = splitMarkdownForDiscord({
+    content: '```js\nfoo\n```\n```py\nbar\n```',
+    maxLength: 20,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "\`\`\`js
+    foo
+    \`\`\`
+    \`\`\`py
+    \`\`\`
+    ",
+      "\`\`\`py
+    bar
+    \`\`\`
+    ",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles empty code block', () => {
+  const result = splitMarkdownForDiscord({
+    content: 'before\n```\n```\nafter',
+    maxLength: 50,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "before
+    \`\`\`
+    \`\`\`
+    after",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles content exactly at maxLength', () => {
+  const result = splitMarkdownForDiscord({
+    content: '12345678901234567890',
+    maxLength: 20,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "12345678901234567890",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles code block only', () => {
+  const result = splitMarkdownForDiscord({
+    content: '```ts\nconst x = 1\n```',
+    maxLength: 15,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "\`\`\`ts
+    \`\`\`
+    ",
+      "\`\`\`ts
+    const x = 1
+    \`\`\`
+    ",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles code block at start with text after', () => {
+  const result = splitMarkdownForDiscord({
+    content: '```js\ncode\n```\nSome text after',
+    maxLength: 20,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "\`\`\`js
+    code
+    \`\`\`
+    ",
+      "Some text after",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles text before code block at end', () => {
+  const result = splitMarkdownForDiscord({
+    content: 'Some text before\n```js\ncode\n```',
+    maxLength: 25,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "Some text before
+    \`\`\`js
+    \`\`\`
+    ",
+      "\`\`\`js
+    code
+    \`\`\`
+    ",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles very long line inside code block', () => {
+  const result = splitMarkdownForDiscord({
+    content: '```js\nshort\nveryverylonglinethatexceedsmaxlength\nshort\n```',
+    maxLength: 25,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "\`\`\`js
+    short
+    \`\`\`
+    ",
+      "\`\`\`js
+    veryverylonglinethatexceedsmaxlength
+    \`\`\`
+    ",
+      "\`\`\`js
+    short
+    \`\`\`
+    ",
+    ]
+  `)
+})
+
+test('splitMarkdownForDiscord handles realistic long markdown with code block', () => {
+  const content = `Here is some explanation text before the code.
+
+\`\`\`typescript
+export function calculateTotal(items: Item[]): number {
+  let total = 0
+  for (const item of items) {
+    total += item.price * item.quantity
+  }
+  return total
+}
+
+export function formatCurrency(amount: number): string {
+  return new Intl.NumberFormat('en-US', {
+    style: 'currency',
+    currency: 'USD',
+  }).format(amount)
+}
+\`\`\`
+
+And here is some text after the code block.`
+
+  const result = splitMarkdownForDiscord({
+    content,
+    maxLength: 200,
+  })
+  expect(result).toMatchInlineSnapshot(`
+    [
+      "Here is some explanation text before the code.
+
+    \`\`\`typescript
+    export function calculateTotal(items: Item[]): number {
+      let total = 0
+      for (const item of items) {
+    \`\`\`
+    ",
+      "\`\`\`typescript
+        total += item.price * item.quantity
+      }
+      return total
+    }
+
+    export function formatCurrency(amount: number): string {
+      return new Intl.NumberFormat('en-US', {
+        style: 'currency',
+    \`\`\`
+    ",
+      "\`\`\`typescript
+        currency: 'USD',
+      }).format(amount)
+    }
+    \`\`\`
+
+
+    And here is some text after the code block.",
+    ]
+  `)
+})

package/src/tools.ts

@@ -15,7 +15,10 @@ import { formatDistanceToNow } from 'date-fns'
 
 import { ShareMarkdown } from './markdown.js'
 import pc from 'picocolors'
-import { initializeOpencodeForDirectory } from './discordBot.js'
+import {
+  initializeOpencodeForDirectory,
+  OPENCODE_SYSTEM_MESSAGE,
+} from './discordBot.js'
 
 export async function getTools({
   onMessageCompleted,
@@ -72,10 +75,10 @@ export async function getTools({
         getClient()
           .session.prompt({
             path: { id: sessionId },
-
             body: {
               parts: [{ type: 'text', text: message }],
               model: sessionModel,
+              system: OPENCODE_SYSTEM_MESSAGE,
             },
           })
           .then(async (response) => {
@@ -149,7 +152,7 @@ export async function getTools({
               path: { id: session.data.id },
               body: {
                 parts: [{ type: 'text', text: message }],
-                // model,
+                system: OPENCODE_SYSTEM_MESSAGE,
               },
             })
             .then(async (response) => {