ethagent 3.0.2 → 3.1.0

package/src/chat/chatTurnOrchestrator.ts

@@ -1,5 +1,3 @@
-import fs from 'node:fs/promises'
-import path from 'node:path'
 import type { Message, Provider } from '../providers/contracts.js'
 import { directToolUsesForUserText } from '../runtime/toolIntent.js'
 import { toPermissionMode, type SessionMode } from '../runtime/sessionMode.js'
@@ -18,6 +16,11 @@ import {
   type TurnCheckpoint,
 } from './chatScreenUtils.js'
 import { collapseImagePathsToRefs, userTextToContentBlocks } from '../utils/images.js'
+import { buildFileMentionContextMessages } from './chatTurnContext.js'
+import {
+  finalizeStreamingRowsById,
+  updateStreamingRows,
+} from './chatTurnRows.js'
 
 type MutableRef<T> = { current: T }
 
@@ -473,106 +476,6 @@ async function handleEvent(ev: TurnEvent, ctx: EventHandlerContext): Promise<voi
   }
 }
 
-function updateStreamingRows(
-  rows: MessageRow[],
-  assistantId: string | null,
-  thinkingRowId: string | null,
-  assistantText: string | null,
-  thinkingText: string | null,
-): MessageRow[] {
-  let next: MessageRow[] | null = null
-  if (assistantId && assistantText !== null) {
-    const index = findRowIndexById(rows, assistantId)
-    const row = rows[index]
-    if (row?.role === 'assistant') {
-      next = next ?? rows.slice()
-      next[index] = { ...row, content: assistantText, liveTail: '' }
-    }
-  }
-  const source = next ?? rows
-  if (thinkingRowId && thinkingText !== null) {
-    const index = findRowIndexById(source, thinkingRowId)
-    const row = source[index]
-    if (row?.role === 'thinking') {
-      next = next ?? rows.slice()
-      next[index] = { ...row, content: thinkingText, liveTail: '' }
-    }
-  }
-  return next ?? rows
-}
-
-function finalizeStreamingRowsById(
-  rows: MessageRow[],
-  assistantId: string | null,
-  thinkingRowId: string | null,
-  assistantText: string,
-  thinkingText: string,
-): MessageRow[] {
-  let next: MessageRow[] | null = null
-  if (assistantId) {
-    const index = findRowIndexById(rows, assistantId)
-    const row = rows[index]
-    if (row?.role === 'assistant') {
-      next = next ?? rows.slice()
-      next[index] = { ...row, content: assistantText || row.content, liveTail: undefined, streaming: false }
-    }
-  }
-  const source = next ?? rows
-  if (thinkingRowId) {
-    const index = findRowIndexById(source, thinkingRowId)
-    const row = source[index]
-    if (row?.role === 'thinking') {
-      next = next ?? rows.slice()
-      next[index] = { ...row, content: thinkingText || row.content, liveTail: undefined, streaming: false, showCursor: false }
-    }
-  }
-  return next ?? rows
-}
-
-function findRowIndexById(rows: MessageRow[], id: string): number {
-  for (let index = rows.length - 1; index >= 0; index -= 1) {
-    if (rows[index]?.id === id) return index
-  }
-  return -1
-}
-
-async function buildFileMentionContextMessages(
-  userText: string,
-  cwd: string,
-): Promise<Message[]> {
-  const mentions = extractFileMentions(userText)
-  if (mentions.length === 0) return []
-
-  const lines: string[] = []
-  for (const mention of mentions) {
-    const resolved = path.resolve(cwd, mention)
-    const rel = path.relative(cwd, resolved)
-    if (rel.startsWith('..') || path.isAbsolute(rel)) {
-      lines.push(
-        `@${mention} -> outside current workspace; do not use unless the user changes directory or names an allowed path.`,
-      )
-      continue
-    }
-    try {
-      const stats = await fs.stat(resolved)
-      lines.push(`@${mention} -> ${mention} (${stats.isDirectory() ? 'directory' : 'file'})`)
-    } catch {
-      lines.push(`@${mention} -> unresolved`)
-    }
-  }
-
-  return [
-    {
-      role: 'user',
-      content: [
-        'Resolved file mentions for this request:',
-        ...lines,
-        'Treat these mentions as authoritative filenames from the user request. Read referenced context files when needed, and edit only the file requested by the user or the target file you have inspected.',
-      ].join('\n'),
-    },
-  ]
-}
-
 const PRIVATE_SKILLS_INDEX_BUDGET = 2048
 
 export async function buildSkillsIndexMessage(
@@ -664,16 +567,6 @@ export async function buildIdentityContinuityContextMessages(
   }
 }
 
-function extractFileMentions(text: string): string[] {
-  const mentions = new Set<string>()
-  for (const match of text.matchAll(/@([^\s]+)/g)) {
-    const raw = match[1]?.replace(/[),.;:!?]+$/g, '')
-    if (!raw || raw.length === 0) continue
-    mentions.add(raw.replace(/\\/g, '/'))
-  }
-  return [...mentions]
-}
-
 function buildWorkingMessages(
   context: Pick<
     TurnOrchestratorContext,

package/src/chat/chatTurnRows.ts

@@ -0,0 +1,64 @@
+import type { MessageRow } from './MessageList.js'
+
+export function updateStreamingRows(
+  rows: MessageRow[],
+  assistantId: string | null,
+  thinkingRowId: string | null,
+  assistantText: string | null,
+  thinkingText: string | null,
+): MessageRow[] {
+  let next: MessageRow[] | null = null
+  if (assistantId && assistantText !== null) {
+    const index = findRowIndexById(rows, assistantId)
+    const row = rows[index]
+    if (row?.role === 'assistant') {
+      next = next ?? rows.slice()
+      next[index] = { ...row, content: assistantText, liveTail: '' }
+    }
+  }
+  const source = next ?? rows
+  if (thinkingRowId && thinkingText !== null) {
+    const index = findRowIndexById(source, thinkingRowId)
+    const row = source[index]
+    if (row?.role === 'thinking') {
+      next = next ?? rows.slice()
+      next[index] = { ...row, content: thinkingText, liveTail: '' }
+    }
+  }
+  return next ?? rows
+}
+
+export function finalizeStreamingRowsById(
+  rows: MessageRow[],
+  assistantId: string | null,
+  thinkingRowId: string | null,
+  assistantText: string,
+  thinkingText: string,
+): MessageRow[] {
+  let next: MessageRow[] | null = null
+  if (assistantId) {
+    const index = findRowIndexById(rows, assistantId)
+    const row = rows[index]
+    if (row?.role === 'assistant') {
+      next = next ?? rows.slice()
+      next[index] = { ...row, content: assistantText || row.content, liveTail: undefined, streaming: false }
+    }
+  }
+  const source = next ?? rows
+  if (thinkingRowId) {
+    const index = findRowIndexById(source, thinkingRowId)
+    const row = source[index]
+    if (row?.role === 'thinking') {
+      next = next ?? rows.slice()
+      next[index] = { ...row, content: thinkingText || row.content, liveTail: undefined, streaming: false, showCursor: false }
+    }
+  }
+  return next ?? rows
+}
+
+function findRowIndexById(rows: MessageRow[], id: string): number {
+  for (let index = rows.length - 1; index >= 0; index -= 1) {
+    if (rows[index]?.id === id) return index
+  }
+  return -1
+}

package/src/chat/commands.ts

@@ -3,10 +3,7 @@ import { getConfigPath, localProviderBaseUrlFor, saveConfig } from '../storage/c
 import { detectLlamaCpp } from '../models/llamacpp.js'
 import { detectSpec } from '../models/runtimeDetection.js'
 import { hasKey } from '../storage/secrets.js'
-import {
-  clearIdentity,
-  getIdentityStatus,
-} from '../storage/identity.js'
+import { getIdentityStatus } from '../storage/identity.js'
 import { discoverProviderModels, type ModelCatalogResult } from '../models/catalog.js'
 import { getLocalHfCacheDir, loadLocalHfModels } from '../models/huggingface.js'
 import { copyToClipboard } from '../utils/clipboard.js'
@@ -21,6 +18,8 @@ import type { ContextUsage } from '../runtime/compaction.js'
 import { formatModelDisplayName } from '../models/modelDisplay.js'
 import { providerDisplayName } from '../models/modelPickerOptions.js'
 import type { McpManager } from '../mcp/manager.js'
+import { runHuggingFace, runIdentity, runMcp } from './slashCommandHandlers.js'
+import { renderStatus } from './slashCommandViews.js'
 
 export type IdentityRequestAction =
   | 'manage'
@@ -355,161 +354,6 @@ const COMMANDS: CommandSpec[] = [
   },
 ]
 
-async function runHuggingFace(args: string, ctx: SlashContext): Promise<SlashResult> {
-  const tokens = args.trim().split(/\s+/).filter(Boolean)
-  const sub = tokens[0]?.toLowerCase() ?? ''
-
-  if (!sub || sub === 'installed') {
-    const installed = await loadLocalHfModels()
-    if (installed.length === 0) {
-      return {
-        kind: 'note',
-        variant: 'dim',
-        text: 'No local model files downloaded. Press Alt+P and choose "Add Local Model File".',
-      }
-    }
-    const lines = installed.map(model => {
-      const marker = model.id === ctx.config.model && ctx.config.provider === 'llamacpp' ? '*' : ' '
-      const displayName = formatModelDisplayName('llamacpp', model.id, { displayName: model.displayName, maxLength: 64 })
-      return `${marker} ${displayName}  ${formatBytes(model.sizeBytes)}  ${model.risk}`
-    })
-    return { kind: 'note', text: ['installed Hugging Face models:', ...lines].join('\n') }
-  }
-
-  if (sub === 'download' || sub === 'model') {
-    const link = tokens.slice(1).join(' ')
-    ctx.onModelPickerRequest()
-    return {
-      kind: 'note',
-      variant: 'dim',
-      text: link
-        ? `Alt+P opened. Choose "Add Local Model File" and paste: ${link}`
-        : 'Alt+P opened. Choose "Add Local Model File" and paste the model URL or repo ID.',
-    }
-  }
-
-  return {
-    kind: 'note',
-    variant: 'error',
-    text: 'usage: /hf [installed|download <huggingface.co link or repo id>]',
-  }
-}
-
-async function runMcp(args: string, ctx: SlashContext): Promise<SlashResult> {
-  if (!ctx.mcp) {
-    return { kind: 'note', variant: 'error', text: 'MCP runtime is not available in this session.' }
-  }
-
-  const tokens = args.trim().split(/\s+/).filter(Boolean)
-  const sub = tokens[0]?.toLowerCase() ?? ''
-  if (!sub || sub === 'status' || sub === 'list') {
-    return { kind: 'note', text: ctx.mcp.renderStatus(), variant: 'info' }
-  }
-
-  try {
-    if (sub === 'approve') {
-      const name = tokens.slice(1).join(' ')
-      if (!name) return { kind: 'note', variant: 'error', text: 'usage: /mcp approve <server>' }
-      return { kind: 'note', text: await ctx.mcp.approveServer(name), variant: 'dim' }
-    }
-    if (sub === 'reject') {
-      const name = tokens.slice(1).join(' ')
-      if (!name) return { kind: 'note', variant: 'error', text: 'usage: /mcp reject <server>' }
-      return { kind: 'note', text: await ctx.mcp.rejectServer(name), variant: 'dim' }
-    }
-    if (sub === 'reconnect') {
-      const name = tokens.slice(1).join(' ') || undefined
-      return { kind: 'note', text: await ctx.mcp.reconnect(name), variant: 'dim' }
-    }
-    if (sub === 'enable' || sub === 'disable') {
-      const name = tokens.slice(1).join(' ')
-      if (!name) return { kind: 'note', variant: 'error', text: `usage: /mcp ${sub} <server>` }
-      return { kind: 'note', text: await ctx.mcp.setEnabled(name, sub === 'enable'), variant: 'dim' }
-    }
-    if (sub === 'add-json') {
-      const project = tokens[1] === '--project'
-      const nameIndex = project ? 2 : 1
-      const name = tokens[nameIndex]
-      if (!name) return { kind: 'note', variant: 'error', text: 'usage: /mcp add-json [--project] <name> <json>' }
-      const jsonStart = nthTokenStart(args, nameIndex + 1)
-      const json = jsonStart >= 0 ? args.slice(jsonStart).trim() : ''
-      if (!json) return { kind: 'note', variant: 'error', text: 'usage: /mcp add-json [--project] <name> <json>' }
-      return { kind: 'note', text: await ctx.mcp.addJson(name, json, project ? 'project' : 'user'), variant: 'dim' }
-    }
-  } catch (err: unknown) {
-    return { kind: 'note', variant: 'error', text: `MCP failed: ${(err as Error).message}` }
-  }
-
-  return {
-    kind: 'note',
-    variant: 'error',
-    text: 'usage: /mcp [status|approve <server>|reject <server>|reconnect [server]|enable <server>|disable <server>|add-json [--project] <name> <json>]',
-  }
-}
-
-async function runIdentity(args: string, ctx: SlashContext): Promise<SlashResult> {
-  const tokens = args.trim().split(/\s+/).filter(Boolean)
-  const sub = tokens[0]?.toLowerCase() ?? ''
-  const rest = tokens.slice(1)
-
-  if (!sub) {
-    ctx.onIdentityRequest('manage')
-    return { kind: 'handled' }
-  }
-
-  if (sub === 'status') {
-    const status = await getIdentityStatus(ctx.config)
-    if (!status) {
-      return {
-        kind: 'note',
-        variant: 'dim',
-        text: 'No Ethereum identity set. Run /identity create to make one.',
-      }
-    }
-    const lines = [
-      `address    ${status.address}`,
-      `created    ${status.createdAt}`,
-      `backend    ${status.backend}`,
-    ]
-    if (status.source) lines.push(`source     ${status.source}`)
-    if (status.agentId) lines.push(`token      #${status.agentId}`)
-    return { kind: 'note', text: lines.join('\n') }
-  }
-
-  if (sub === 'create') {
-    ctx.onIdentityRequest('create')
-    return { kind: 'handled' }
-  }
-
-  if (sub === 'load') {
-    ctx.onIdentityRequest('load')
-    return { kind: 'handled' }
-  }
-
-  if (sub === 'remove') {
-    if (rest[0] !== 'confirm') {
-      return {
-        kind: 'note',
-        variant: 'error',
-        text: 'Remove deletes local identity metadata and any legacy stored key. Re-run with: /identity remove confirm',
-      }
-    }
-    const status = await getIdentityStatus(ctx.config)
-    if (!status) {
-      return { kind: 'note', variant: 'dim', text: 'No Ethereum identity to remove.' }
-    }
-    const next = await clearIdentity(ctx.config)
-    ctx.onReplaceConfig(next)
-    return { kind: 'note', text: `Removed identity ${status.address}.`, variant: 'dim' }
-  }
-
-  return {
-    kind: 'note',
-    variant: 'error',
-    text: 'usage: /identity [status|create|load|remove confirm]',
-  }
-}
-
 function renderHelp(): string {
   const visibleCommands = COMMANDS.filter(c => !c.hidden)
   const maxName = Math.max(...visibleCommands.map(c => commandLabel(c).length))
@@ -530,25 +374,6 @@ function commandLabel(cmd: CommandSpec): string {
   return `/${cmd.name} (${cmd.aliases.map(a => `/${a}`).join(', ')})`
 }
 
-function renderStatus(ctx: SlashContext): string {
-  const elapsedMs = Date.now() - ctx.startedAt
-  const minutes = Math.floor(elapsedMs / 60000)
-  const seconds = Math.floor((elapsedMs % 60000) / 1000)
-  const elapsed = minutes > 0 ? `${minutes}m${seconds.toString().padStart(2, '0')}s` : `${seconds}s`
-  const displayModel = formatModelDisplayName(ctx.config.provider, ctx.config.model, { maxLength: 72 })
-  return [
-    `provider   ${providerDisplayName(ctx.config.provider)}`,
-    `model      ${displayModel}`,
-    `cwd        ${ctx.cwd}`,
-    `session    ${ctx.sessionId.slice(0, 8)}`,
-    'state      active',
-    `turns      ${ctx.turns}`,
-    `tokens     ~${ctx.approxTokens}`,
-    `context    ${ctx.contextUsage.percent}% (~${ctx.contextUsage.usedTokens}/${ctx.contextUsage.windowTokens}, ${ctx.contextUsage.source})`,
-    `elapsed    ${elapsed}`,
-  ].join('\n')
-}
-
 function renderContext(ctx: SlashContext): string {
   const usage = ctx.contextUsage
   const free = Math.max(0, usage.windowTokens - usage.usedTokens)

package/src/chat/continuityEditReview.ts

@@ -0,0 +1,42 @@
+import { splitFileChangeResult } from '../tools/fileDiff.js'
+import type { ContinuityEditReviewState } from './views/ContinuityEditReviewView.js'
+
+export function privateContinuityEditReviewFromToolResult(
+  name: string,
+  input: Record<string, unknown>,
+  result: { ok: boolean; summary: string; content: string },
+): ContinuityEditReviewState | null {
+  if (name !== 'propose_private_continuity_edit' || !result.ok) return null
+  const file = normalizePrivateContinuityFile(input.file)
+  if (!file) return null
+  const parsed = splitFileChangeResult(result.content)
+  const filePath = extractReviewFilePath(parsed.content)
+  if (!filePath) return null
+  return {
+    file,
+    filePath,
+    summary: result.summary,
+    ...(parsed.diff ? { diff: parsed.diff } : {}),
+  }
+}
+
+function normalizePrivateContinuityFile(value: unknown): ContinuityEditReviewState['file'] | null {
+  if (typeof value !== 'string') return null
+  if (/^soul\.md$/i.test(value.trim())) return 'SOUL.md'
+  if (/^memory\.md$/i.test(value.trim())) return 'MEMORY.md'
+  return null
+}
+
+function extractReviewFilePath(content: string): string | null {
+  for (const line of content.split(/\r?\n/)) {
+    const review = line.match(/^(?:[-*]\s+)?review file:\s*(.+)$/i)
+    if (review?.[1]?.trim()) return cleanReviewFilePath(review[1])
+    const updated = line.match(/^(?:[-*]\s+)?updated local private continuity file\s+(.+)$/i)
+    if (updated?.[1]?.trim()) return cleanReviewFilePath(updated[1])
+  }
+  return null
+}
+
+function cleanReviewFilePath(value: string): string {
+  return value.trim().replace(/^`+|`+$/g, '').trim()
+}

package/src/chat/input/ChatInput.tsx

@@ -1,5 +1,3 @@
-import fs from 'node:fs/promises'
-import path from 'node:path'
 import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react'
 import { Box, Text, useStdout } from 'ink'
 import { theme } from '../../ui/theme.js'
@@ -19,11 +17,6 @@ import {
   type ChatBuffer,
   type FileMentionToken,
 } from './chatInputState.js'
-import {
-  getVisibleVisualLineWindow,
-  getVisualLineIndex,
-  getVisualLines,
-} from './textCursor.js'
 import {
   countPastedTextLineBreaks,
   expandPastedTextRefs,
@@ -39,6 +32,16 @@ import {
   pruneImageRefs,
   type ImageRef,
 } from './imageRefs.js'
+import { inputWrapWidth, renderWithCursor } from './inputRendering.js'
+import {
+  isFallbackPasteInput,
+  isSoftBreak,
+  listFileMentionSuggestions,
+  summarizeQueuedMessage,
+  type FileMentionSuggestion,
+} from './chatInputHelpers.js'
+
+export { inputWrapWidth, renderWithCursor } from './inputRendering.js'
 
 type PromptInputProps = {
   onSubmit: (value: string) => void
@@ -64,11 +67,6 @@ const PASTE_FLUSH_LIMIT = 4096
 const MIN_INPUT_VIEWPORT_LINES = 3
 const PROMPT_FOOTER_LINES = 5
 const MAX_INLINE_PASTE_LINES = 2
-const STACK_HORIZONTAL_PADDING = 2
-const INPUT_BORDER_WIDTH = 2
-const INPUT_HORIZONTAL_PADDING = 4
-const PROMPT_PREFIX_WIDTH = 2
-
 export const ChatInput: React.FC<PromptInputProps> = ({
   onSubmit,
   history,
@@ -627,135 +625,3 @@ export const ChatInput: React.FC<PromptInputProps> = ({
     </Box>
   )
 }
-
-function isSoftBreak(key: { return: boolean; meta?: boolean; shift?: boolean }): boolean {
-  return key.return && Boolean(key.meta || key.shift)
-}
-
-type RenderedVisualLine = {
-  visualLineIndex: number
-  node: React.ReactNode
-}
-
-type RenderedInputViewport = {
-  lines: RenderedVisualLine[]
-  hiddenAbove: number
-  hiddenBelow: number
-  visibleLineCount: number
-}
-
-export function renderWithCursor(
-  value: string,
-  cursor: number,
-  showCursor: boolean,
-  wrapWidth: number,
-  maxVisibleLines: number,
-): RenderedInputViewport {
-  const lines = getVisualLines(value, wrapWidth)
-  const cursorLine = getVisualLineIndex(lines, cursor)
-  const window = getVisibleVisualLineWindow(lines.length, cursorLine, maxVisibleLines)
-  const visibleLines = lines.slice(window.start, window.end)
-
-  if (!showCursor) {
-    return {
-      lines: visibleLines.map((line, i) => ({
-        visualLineIndex: window.start + i,
-        node: (
-          <Text color={theme.text} wrap="wrap">
-            {value.slice(line.start, line.end) || ' '}
-          </Text>
-        ),
-      })),
-      hiddenAbove: window.start,
-      hiddenBelow: lines.length - window.end,
-      visibleLineCount: Math.max(1, visibleLines.length),
-    }
-  }
-
-  return {
-    lines: visibleLines.map((line, i) => {
-      const visualLineIndex = window.start + i
-      const text = value.slice(line.start, line.end)
-      if (visualLineIndex !== cursorLine) {
-        return {
-          visualLineIndex,
-          node: <Text color={theme.text} wrap="wrap">{text || ' '}</Text>,
-        }
-      }
-      const column = Math.max(0, Math.min(cursor - line.start, text.length))
-      const before = text.slice(0, column)
-      const atChar = text[column] ?? ' '
-      const after = text.slice(column + 1)
-      return {
-        visualLineIndex,
-        node: (
-          <Text color={theme.text} wrap="wrap">
-            {before}
-            <Text backgroundColor={theme.accentPeriwinkle} color="#0c0c1f">{atChar}</Text>
-            {after}
-          </Text>
-        ),
-      }
-    }),
-    hiddenAbove: window.start,
-    hiddenBelow: lines.length - window.end,
-    visibleLineCount: Math.max(1, visibleLines.length),
-  }
-}
-
-export function inputWrapWidth(columns: number): number {
-  const fixedChromeWidth =
-    STACK_HORIZONTAL_PADDING
-    + INPUT_BORDER_WIDTH
-    + INPUT_HORIZONTAL_PADDING
-    + PROMPT_PREFIX_WIDTH
-  return Math.max(1, Math.floor(columns) - fixedChromeWidth)
-}
-
-function isFallbackPasteInput(input: string): boolean {
-  if (!input) return false
-  return input.length > LARGE_PASTE_THRESHOLD
-    || countPastedTextLineBreaks(normalizePastedText(input)) > MAX_INLINE_PASTE_LINES
-}
-
-function summarizeQueuedMessage(text: string): string {
-  const normalized = text.replace(/\s+/g, ' ').trim()
-  if (!normalized) return ''
-  if (normalized.length <= 72) return normalized
-  return `${normalized.slice(0, 69)}...`
-}
-
-type FileMentionSuggestion = {
-  path: string
-  hint: string
-}
-
-async function listFileMentionSuggestions(
-  cwd: string,
-  mention: FileMentionToken,
-): Promise<FileMentionSuggestion[]> {
-  const query = mention.query.replace(/\\/g, '/')
-  const lastSlash = query.lastIndexOf('/')
-  const queryDir = lastSlash >= 0 ? query.slice(0, lastSlash + 1) : ''
-  const basenameQuery = lastSlash >= 0 ? query.slice(lastSlash + 1).toLowerCase() : query.toLowerCase()
-  const baseDir = path.resolve(cwd, queryDir || '.')
-
-  let entries: Array<{ name: string; isFile: () => boolean }>
-  try {
-    entries = await fs.readdir(baseDir, { withFileTypes: true })
-  } catch {
-    return []
-  }
-
-  return entries
-    .filter(entry => entry.isFile() && entry.name.toLowerCase().startsWith(basenameQuery))
-    .sort((left, right) => left.name.localeCompare(right.name))
-    .slice(0, 32)
-    .map(entry => {
-      const relative = (queryDir + entry.name).replace(/\\/g, '/')
-      return {
-        path: relative,
-        hint: path.extname(entry.name).slice(1) || 'file',
-      }
-    })
-}

package/src/chat/input/chatInputHelpers.ts

@@ -0,0 +1,62 @@
+import fs from 'node:fs/promises'
+import path from 'node:path'
+import {
+  countPastedTextLineBreaks,
+  LARGE_PASTE_THRESHOLD,
+  normalizePastedText,
+} from './chatPaste.js'
+import type { FileMentionToken } from './chatInputState.js'
+
+const MAX_INLINE_PASTE_LINES = 2
+
+export function isSoftBreak(key: { return: boolean; meta?: boolean; shift?: boolean }): boolean {
+  return key.return && Boolean(key.meta || key.shift)
+}
+
+export function isFallbackPasteInput(input: string): boolean {
+  if (!input) return false
+  return input.length > LARGE_PASTE_THRESHOLD
+    || countPastedTextLineBreaks(normalizePastedText(input)) > MAX_INLINE_PASTE_LINES
+}
+
+export function summarizeQueuedMessage(text: string): string {
+  const normalized = text.replace(/\s+/g, ' ').trim()
+  if (!normalized) return ''
+  if (normalized.length <= 72) return normalized
+  return `${normalized.slice(0, 69)}...`
+}
+
+export type FileMentionSuggestion = {
+  path: string
+  hint: string
+}
+
+export async function listFileMentionSuggestions(
+  cwd: string,
+  mention: FileMentionToken,
+): Promise<FileMentionSuggestion[]> {
+  const query = mention.query.replace(/\\/g, '/')
+  const lastSlash = query.lastIndexOf('/')
+  const queryDir = lastSlash >= 0 ? query.slice(0, lastSlash + 1) : ''
+  const basenameQuery = lastSlash >= 0 ? query.slice(lastSlash + 1).toLowerCase() : query.toLowerCase()
+  const baseDir = path.resolve(cwd, queryDir || '.')
+
+  let entries: Array<{ name: string; isFile: () => boolean }>
+  try {
+    entries = await fs.readdir(baseDir, { withFileTypes: true })
+  } catch {
+    return []
+  }
+
+  return entries
+    .filter(entry => entry.isFile() && entry.name.toLowerCase().startsWith(basenameQuery))
+    .sort((left, right) => left.name.localeCompare(right.name))
+    .slice(0, 32)
+    .map(entry => {
+      const relative = (queryDir + entry.name).replace(/\\/g, '/')
+      return {
+        path: relative,
+        hint: path.extname(entry.name).slice(1) || 'file',
+      }
+    })
+}