thincoder 0.7.8 → 0.8.1

package/src/tui/startup.mjs

@@ -0,0 +1,113 @@
+import { listSlots } from "../session.mjs"
+import { sliceByWidth } from "./render.mjs"
+import { ansi, C } from "./ansi.mjs"
+
+/** 启动画面 + 会话恢复 + 后台索引。
+ *  从 index.mjs 抽出。
+ *  ctx: { agent, state, opts, pushLine, pushLabel, render, startWizard } */
+export function showStartup(ctx) {
+  const { agent, state, opts, pushLine, pushLabel, render, startWizard } = ctx
+
+  // 启动画面
+  if (!agent.provider.apiKey) {
+    pushLabel(`Welcome to ThinCoder!`, ansi.bold + C.tool)
+    pushLine("No API key configured yet — entering initial setup (Esc to skip anytime)", C.text)
+    startWizard()
+  } else {
+    pushLine(`Welcome to ThinCoder. Provider: ${agent.activeProvider} / ${agent.provider.model}`, C.dim)
+  }
+  pushLine(`Tools: ${agent.tools.map((t) => t.name).join(", ")}`, C.dim)
+
+  // 恢复上次会话：重建对话区显示 (tool 结果行省略，保持清爽）
+  if (opts.restored?.display?.length) {
+    // 用户视角的恢复：display 是退出前对话区的原样快照，所见即所得
+    state.lines = [...opts.restored.display.map((l) => ({ text: l.text, color: l.color })), ...state.lines]
+    pushLabel(`── Restored previous session; /new for a fresh session ──`, C.warn)
+  } else if (opts.restored?.history?.length) {
+    // 重建对话区：user/assistant 消息逐条展示，tool 结果行只保留首行摘要
+    for (let i = 0; i < opts.restored.history.length; i++) {
+      const m = opts.restored.history[i]
+      if (m.role === "user") {
+        if (typeof m.content === "string" && m.content.startsWith("[System reminder:")) continue
+        pushLabel(`❯ You:`, ansi.bold + C.user)
+        if (typeof m.content === "string" && m.content) pushLine(m.content, C.text)
+      } else if (m.role === "assistant") {
+        pushLabel(`❯ ThinCoder:`, ansi.bold + C.assistant)
+        if (typeof m.content === "string" && m.content) pushLine(m.content, C.text)
+        for (const tc of m.tool_calls ?? []) {
+          // 找到下一条对应的 tool 结果，显示首行摘要
+          const toolResult = opts.restored.history[i + 1]
+          const hasResult = toolResult?.role === "tool" && toolResult?.tool_call_id === tc.id
+          const summary = hasResult ? " → " + sliceByWidth(String(toolResult.content).split("\n")[0], 80) : ""
+          pushLine(`  [tool] ${tc.function?.name ?? "?"}${summary}`, C.tool)
+        }
+      }
+      // tool 消息本身不单独渲染——已在 assistant 的 tool_calls 后以摘要形式展示
+    }
+    pushLabel(`── Restored previous session (${opts.restored.history.length} messages); /new for a fresh session ──`, C.warn)
+  }
+
+  // 有归档槽位时给个提示
+  if (listSlots(agent.cwd).length > 0) {
+    pushLine("Tip: archived sessions available — /session to view/switch", C.dim)
+  }
+  render()
+}
+
+/** 后台索引 (进界面后再跑，不阻塞启动）；进度走底部状态栏，不往对话区塞行。
+ *  优先用 git diff 增量（快），git 不可用或首次运行时退到全量扫描。 */
+export async function backgroundIndex(ctx) {
+  const { agent, state, render } = ctx
+  const { codeSync, docSync, gitSync } = await import("../memory.mjs")
+  const cwd = agent.cwd
+  let codeFiles = 0, docFiles = 0
+
+  state.status = "Indexing..."
+  render()
+
+  const gitRes = await gitSync(agent.memory, cwd, {
+    onProgress: (p) => {
+      if (p.phase === "index" && p.current % 5 === 0) {
+        state.status = `Indexing... ${p.current}/${p.total}`
+        render()
+      }
+    }
+  })
+
+  if (gitRes !== null) {
+    // git 增量成功，直接统计
+    codeFiles = agent.memory.db.prepare(`SELECT COUNT(DISTINCT path) AS n FROM code_chunks`).get()?.n ?? 0
+    docFiles = agent.memory.db.prepare(`SELECT COUNT(DISTINCT path) AS n FROM doc_chunks`).get()?.n ?? 0
+  } else {
+    // 退到全量扫描（codeSync 和 docSync 并行——读写不同表，SQLite WAL 天然支持）
+    const [codeRes, docRes] = await Promise.allSettled([
+      codeSync(agent.memory, cwd, {
+        onProgress: (p) => {
+          if (p.phase === "index" && p.current % 30 === 0) {
+            state.status = `Indexing code... ${p.current}/${p.total}`
+            render()
+          }
+        }
+      }),
+      docSync(agent.memory, cwd, {
+        onProgress: (p) => {
+          if (p.phase === "index" && p.current % 10 === 0) {
+            state.status = `Indexing docs... ${p.current}/${p.total}`
+            render()
+          }
+        }
+      }),
+    ])
+    if (codeRes.status === "fulfilled") {
+      codeFiles = agent.memory.db.prepare(`SELECT COUNT(DISTINCT path) AS n FROM code_chunks`).get()?.n ?? 0
+    }
+    if (docRes.status === "fulfilled") {
+      docFiles = agent.memory.db.prepare(`SELECT COUNT(DISTINCT path) AS n FROM doc_chunks`).get()?.n ?? 0
+    }
+  }
+
+  state.status = codeFiles || docFiles
+    ? `Ready — idx code ${codeFiles} doc ${docFiles}`
+    : "Ready"
+  render()
+}

package/src/tui/wizard.mjs

@@ -0,0 +1,168 @@
+/**
+ * wizard.mjs — 首次启动 Config 向导
+ * 从 index.mjs 抽出：provider 选择 → 逐项输入 (name/baseURL/model/key/embedkey) → 持久化 → 接模型选择器。
+ * 通过 ctx 对象访问 startTUI 闭包中的共享状态与 UI 函数。
+ * ctx: { agent, state, pushLine, pushLabel, render, persistRaw, openModelPicker }
+ */
+
+import { PROVIDER_PRESETS as PRESETS } from "../config.mjs"
+import { ansi, C } from "./ansi.mjs"
+
+/**
+ * 创建向导控制器。
+ * 返回 { startWizard, renderWizard, wizardChooseProvider, wizardSubmitText, cancelWizard, finishWizard }
+ */
+export function createWizard(ctx) {
+  const { agent, state, pushLine, pushLabel, render, persistRaw } = ctx
+
+  /** 菜单步的候选项：已有 provider (no key 的标注）+ 未添加的预设 + 自定义 */
+  function wizardProviderItems() {
+    const items = []
+    for (const p of agent.providers) {
+      items.push({ kind: "existing", name: p.name, baseURL: p.baseURL, model: p.model, label: `${p.name} (added${p.apiKey ? "" : ", no key"})` })
+    }
+    for (const [name, p] of Object.entries(PRESETS)) {
+      if (!agent.providers.some((x) => x.name === name)) {
+        items.push({ kind: "preset", name, baseURL: p.baseURL, model: p.model, label: `${name} (${p.desc})` })
+      }
+    }
+    items.push({ kind: "custom", name: null, label: "Custom endpoint…" })
+    return items
+  }
+
+  /** 文本步骤定义：提示语 + 校验 (通过返回 true，否则返回错误文案） */
+  const WIZARD_STEPS = {
+    name: {
+      prompt: "Name this provider (alphanumeric/-/_ e.g. my-openai)",
+      validate: (v) =>
+        (/^[\w-]+$/.test(v) && !agent.providers.some((p) => p.name === v)) || "Name must be alphanumeric/-/_ and unique",
+    },
+    baseURL: {
+      prompt: "Enter baseURL (e.g. https://api.openai.com/v1)",
+      validate: (v) => /^https?:\/\/.+/.test(v) || "baseURL must start with http(s)://",
+    },
+    model: {
+      prompt: "Enter model name (e.g. gpt-4o)",
+      validate: (v) => v.length > 0 || "Model name required",
+    },
+    key: {
+      prompt: "Enter API key",
+      validate: (v) => v.length > 0 || "Key must not be empty",
+    },
+    embedkey: {
+      prompt: "Optional: embedding API key (SiliconFlow, for memory vector search; press Enter to skip)",
+      validate: () => true, // 可跳过
+    },
+  }
+  const WIZARD_NEXT = { name: "baseURL", baseURL: "model", model: "key", key: "embedkey", embedkey: null }
+
+  function startWizard() {
+    state.wizard = { step: "provider", index: 0, scroll: 0, selectedLine: 0, fields: {}, error: null, lines: [] }
+    renderWizard()
+  }
+
+  function renderWizard() {
+    const w = state.wizard
+    if (!w) return
+    const lines = []
+    if (w.step === "provider") {
+      lines.push({ text: " Choose a model provider:", color: C.text })
+      wizardProviderItems().forEach((it, i) => {
+        if (i === w.index) w.selectedLine = lines.length
+        lines.push({
+          text: `${i === w.index ? " ▸ " : "   "}${it.label}`,
+          color: i === w.index ? ansi.bold + C.text : C.dim,
+        })
+      })
+    } else {
+      const f = w.fields
+      if (f.name) lines.push({ text: ` Provider:  ${f.name}`, color: C.dim })
+      if (f.baseURL) lines.push({ text: ` baseURL: ${f.baseURL}`, color: C.dim })
+      if (f.model) lines.push({ text: ` Model:   ${f.model}`, color: C.dim })
+      lines.push({ text: ` ❯ ${WIZARD_STEPS[w.step].prompt}`, color: ansi.bold + C.text })
+      lines.push({ text: " (type in input box below)", color: C.dim })
+      w.selectedLine = 0
+    }
+    if (w.error) lines.push({ text: ` ${w.error}`, color: C.error })
+    w.lines = lines
+    render()
+  }
+
+  function wizardChooseProvider(item) {
+    const w = state.wizard
+    if (item.kind === "custom") {
+      w.step = "name"
+    } else {
+      w.fields = { name: item.name, baseURL: item.baseURL, model: item.model }
+      w.step = "key"
+    }
+    renderWizard()
+  }
+
+  function wizardSubmitText() {
+    const w = state.wizard
+    const value = state.input.join("").trim()
+    const ok = WIZARD_STEPS[w.step].validate(value)
+    if (ok !== true) {
+      w.error = ok
+      renderWizard()
+      return
+    }
+    w.error = null
+    state.input = []
+    state.cursor = 0
+    w.fields[w.step === "key" ? "key" : w.step] = w.step === "baseURL" ? value.replace(/\/+$/, "") : value
+    const next = WIZARD_NEXT[w.step]
+    if (next) {
+      w.step = next
+      renderWizard()
+    } else {
+      finishWizard().catch((e) => pushLine(`[error] ${e.message}`, C.error))
+    }
+  }
+
+  function cancelWizard() {
+    state.wizard = null
+    pushLine("Skipped initial setup. Add a provider anytime via /provider add, set its key via /provider key.", C.dim)
+    render()
+  }
+
+  /** 向导完成：写入 provider (有则更新）、设为激活、持久化，然后接模型选择器 */
+  async function finishWizard() {
+    const f = state.wizard.fields
+    state.wizard = null
+    const existing = agent.providers.find((p) => p.name === f.name)
+    if (existing) Object.assign(existing, { baseURL: f.baseURL, model: f.model, apiKey: f.key })
+    else agent.providers.push({ name: f.name, baseURL: f.baseURL, model: f.model, apiKey: f.key })
+    agent.activeProvider = f.name
+    agent.provider = { ...agent.providers.find((p) => p.name === f.name) }
+    if (agent.config?.agent?.compactThresholdAuto) {
+      const { resolveCompactThreshold } = await import("../config.mjs")
+      agent.config.agent.compactThreshold = resolveCompactThreshold(null, f.model).value
+    }
+    await persistRaw((raw) => {
+      raw.providers = agent.providers
+      raw.activeProvider = f.name
+    })
+    agent.config.activeProvider = f.name
+    pushLabel(`❯ Setup`, ansi.bold + C.tool)
+    pushLine(`Setup complete: ${f.name} / ${f.model} (saved to config)`, C.tool)
+    // embedding key：配了就启用向量检索，没配提示事后通道
+    if (f.embedkey) {
+      agent.config.embedding ??= {}
+      agent.config.embedding.apiKey = f.embedkey
+      await persistRaw((raw) => { raw.embedding = { ...(raw.embedding ?? {}), apiKey: f.embedkey } })
+      if (agent.memory && !agent.memory.embedder) {
+        const { createEmbedder } = await import("../embedding.mjs")
+        agent.memory.embedder = createEmbedder(agent.config.embedding)
+      }
+      pushLine(`Vector search enabled (${agent.config.embedding.model ?? "BAAI/bge-m3"})`, C.tool)
+    } else {
+      pushLine(`Vector search disabled (memory falls back to text-only search). Run /config embedkey <key> to enable.`, C.dim)
+    }
+    pushLine(`Select model (Esc to keep ${f.model})`, C.dim)
+    ctx.openModelPicker().catch((e) => pushLine(`[error] ${e.message}`, C.error))
+  }
+
+  return { startWizard, renderWizard, wizardChooseProvider, wizardSubmitText, cancelWizard, finishWizard }
+}

package/src/tui-render.mjs

@@ -0,0 +1,4 @@
+/**
+ * tui-render.mjs — 终端显示工具（重新导出）
+ */
+export { charWidth, stringWidth, sliceByWidth, formatTables, layoutInput, sanitizeDisplay, wrapText } from "./tui/render.mjs"