@caozeal/dsh-next-git 0.1.5 → 0.1.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@caozeal/dsh-next-git",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "description": "DSH Git 管理插件：分支切换、变更暂存、提交/推送/拉取、AI 提交信息、提交图与 diff 查看",
   "type": "module",
   "main": "index.js",
@@ -16,6 +16,8 @@
     "src",
     "lib/client.js",
     "cordis.patch.yml",
+    "screenshots.json",
+    "assets/screenshots",
     "locale/*.json",
     "icon.svg",
     "icon.png"

package/screenshots.json

@@ -0,0 +1,8 @@
+[
+  "assets/screenshots/01-changes.png",
+  "assets/screenshots/02-commit-graph.png",
+  "assets/screenshots/03-word-level-diff.png",
+  "assets/screenshots/04-branches.png",
+  "assets/screenshots/05-graph-multibranch.png",
+  "assets/screenshots/06-fullscreen.png"
+]

package/src/ai-commit.js

@@ -2,8 +2,12 @@
  * git-manager — AI 提交信息生成。
  *
  * 流程：收集「已暂存 diff（为空则工作区 diff）→ 最近提交风格 → 调用默认模型的
- * llm.stream」→ 清理成一行提交信息。任何一步不可用都退回确定性文案（fallback），
- * 让面板始终能拿到一个可编辑的建议。
+ * llm.stream」→ 清理成「标题 + 可选正文」。任何一步不可用都退回确定性文案
+ * （fallback），让面板始终能拿到一条可编辑的建议。
+ *
+ * 系统提示词可由仓库自定义（git config dsh.commitPrompt），只替换「怎么写」的规则，
+ * 不改变「给模型看什么数据」——diff/文件/最近提交这条装配逻辑始终由本模块掌握，
+ * 避免用户把模板写坏后模型看不到 diff 而胡乱编造。
  */
 import { GitError } from './git.js'
 
@@ -11,11 +15,13 @@ const MAX_DIFF_CHARS = 12000
 /** 单次生成的 token 预算：reasoning 模型会先消耗思考 token，200 太小容易截断。 */
 const MAX_OUTPUT_TOKENS = 600
 
-const SYSTEM_PROMPT = [
+export const DEFAULT_SYSTEM_PROMPT = [
   '你是一个 Git 提交信息生成器。',
-  '只输出一行提交信息，不要解释、不要代码块、不要引号、不要句号结尾。',
-  '使用 Conventional Commits 风格（如 feat:/fix:/refactor:/docs:/chore:），',
-  '不超过 72 个字符。跟随仓库最近提交的语言：最近提交多为中文就用中文，否则用英文。',
+  '第一行是标题：Conventional Commits 风格（feat:/fix:/refactor:/docs:/chore: 等），不超过 72 个字符，不以句号结尾。',
+  '如果这次改动需要说明动机、影响面或取舍，空一行后写正文，正文每行不超过 72 个字符，可以分段。',
+  '改动很小、一句话能说清时不要硬凑正文。',
+  '不要解释、不要代码块、不要引号。',
+  '跟随仓库最近提交的语言：最近提交多为中文就用中文，否则用英文。',
 ].join('')
 
 /** 从 diff 与最近提交构造 user prompt。 */
@@ -48,6 +54,44 @@ export function sanitizeSubject(text) {
   return subject
 }
 
+/** 去掉整行的代码块围栏，保留内容。 */
+function stripFences(text) {
+  return String(text ?? '').replace(/^[ \t]*```[a-zA-Z0-9_-]*[ \t]*$/gm, '')
+}
+
+/**
+ * 清理模型输出的正文部分：去围栏、去首尾空行、去「正文：」这类前缀、限长。
+ * 不做逐行 trim——正文里的缩进（列表、代码片段）要保留。
+ * @param {string} text
+ * @returns {string}
+ */
+export function sanitizeBody(text) {
+  let value = stripFences(text).replace(/\r\n/g, '\n')
+  const lines = value.split('\n').map((line) => line.replace(/[ \t]+$/, ''))
+  while (lines.length > 0 && lines[0].trim() === '') lines.shift()
+  while (lines.length > 0 && lines[lines.length - 1].trim() === '') lines.pop()
+  let body = lines.join('\n').trim()
+  body = body.replace(/^(commit body|提交正文|正文)\s*[:：]\s*/i, '')
+  if (body.length > 2000) body = body.slice(0, 2000).trim()
+  return body
+}
+
+/**
+ * 把模型输出拆成「标题 + 正文」。
+ * 模型偶尔只给标题，也可能给「标题 / 空行 / 正文」，两种都要能吃下。
+ * @param {string} text
+ * @returns {{subject: string, body: string}}
+ */
+export function parseCommitMessage(text) {
+  const cleaned = stripFences(text).replace(/\r\n/g, '\n')
+  const lines = cleaned.split('\n')
+  let index = 0
+  while (index < lines.length && lines[index].trim() === '') index += 1
+  const subjectLine = index < lines.length ? lines[index] : ''
+  const rest = lines.slice(index + 1).join('\n')
+  return { subject: sanitizeSubject(subjectLine), body: sanitizeBody(rest) }
+}
+
 /** 最近提交是否以中文为主。 */
 function looksChinese(subjects) {
   const joined = subjects.join('')
@@ -104,8 +148,9 @@ export function fallbackSubject({ files, stagedCount, Chinese }) {
  *   signal?: AbortSignal,
  *   llm?: unknown,
  *   selection?: {provider?: string, model?: string, reasoningEffort?: string},
+ *   systemPrompt?: string,
  * }} options
- * @returns {Promise<{message: string, model: string|null, fallback: boolean}>}
+ * @returns {Promise<{message: string, subject: string, body: string, model: string|null, fallback: boolean}>}
  */
 export async function generateCommitMessage(options) {
   const {
@@ -117,6 +162,7 @@ export async function generateCommitMessage(options) {
     selection,
     files = [],
     stagedCount = 0,
+    systemPrompt = '',
   } = options
 
   const staged = await runGit(['diff', '--cached', '--no-color'], { cwd, signal, allowFailure: true })
@@ -144,13 +190,12 @@ export async function generateCommitMessage(options) {
 
   const canUseLlm = Boolean(llm && selection && selection.provider && selection.model && diff.trim())
   if (!canUseLlm) {
-    return {
-      message: fallbackSubject({ files: changedFiles, stagedCount, Chinese: chinese }),
-      model,
-      fallback: true,
-    }
+    const subject = fallbackSubject({ files: changedFiles, stagedCount, Chinese: chinese })
+    return { message: subject, subject, body: '', model, fallback: true }
   }
 
+  // 用户自定义提示词：只覆盖「怎么写」的规则，装配数据仍由 buildPrompt 负责。
+  const customPrompt = typeof systemPrompt === 'string' ? systemPrompt.trim().slice(0, 4000) : ''
   const prompt = buildPrompt({ diff, files: changedFiles, recent, hint })
   let text = ''
   try {
@@ -158,7 +203,7 @@ export async function generateCommitMessage(options) {
       provider: selection.provider,
       model: selection.model,
       reasoningEffort: selection.reasoningEffort,
-      system: SYSTEM_PROMPT,
+      system: customPrompt === '' ? DEFAULT_SYSTEM_PROMPT : customPrompt,
       messages: [{ role: 'user', content: [{ type: 'text', text: prompt }] }],
       maxTokens: MAX_OUTPUT_TOKENS,
       temperature: 0.2,
@@ -182,13 +227,11 @@ export async function generateCommitMessage(options) {
     throw new GitError(`模型生成失败：${message}`, { code: 'ai-unavailable' })
   }
 
-  const subject = sanitizeSubject(text)
-  if (!looksLikeSubject(subject)) {
-    return {
-      message: fallbackSubject({ files: changedFiles, stagedCount, Chinese: chinese }),
-      model,
-      fallback: true,
-    }
+  const parsed = parseCommitMessage(text)
+  if (!looksLikeSubject(parsed.subject)) {
+    const subject = fallbackSubject({ files: changedFiles, stagedCount, Chinese: chinese })
+    return { message: subject, subject, body: '', model, fallback: true }
   }
-  return { message: subject, model, fallback: false }
-}
+  const message = parsed.body === '' ? parsed.subject : parsed.subject + '\n\n' + parsed.body
+  return { message, subject: parsed.subject, body: parsed.body, model, fallback: false }
+}

package/src/ai-conflict.js

@@ -0,0 +1,159 @@
+/**
+ * git-manager — AI 冲突解决。
+ *
+ * 冲突文件有三份 stage：共同祖先（:1:）、当前分支（:2:）、合入分支（:3:）。
+ * 这里把它们装配成 prompt，让模型输出「解决后的完整文件」。
+ *
+ * 本模块只做清理与校验，**不写盘**：真正的写入由 operations 的 applyConflict 完成，
+ * 而且写之前会再校验一次是否残留冲突标记。这样即使模型胡说，也不会把
+ * `<<<<<<<` 直接 stage 进仓库。
+ */
+import { GitError } from './git.js'
+
+/** 单份版本最多送给模型多少字符；再大就不是「看懂了再合」，而是碰运气。 */
+export const MAX_SIDE_CHARS = 40000
+const MAX_OUTPUT_TOKENS = 1600
+
+export const DEFAULT_SYSTEM_PROMPT = [
+  '你是一个 Git 合并冲突解决器。',
+  '用户会给出冲突文件的三份内容：base（共同祖先）、ours（当前分支）、theirs（合入分支）。',
+  '请输出解决冲突后的完整文件内容，遵守：',
+  '- 保留双方各自有价值的改动；同一处语义等价的重复只保留一份。',
+  '- 版本号、依赖版本这类冲突取更新/更高的一方，并与文件其余部分保持一致。',
+  '- 只输出文件内容本身：不要解释、不要 Markdown 围栏、不要行号、不要「以下是」这类开场白。',
+  '- 输出必须能直接编译或解析：括号、缩进、import 与文件其余部分自洽。',
+  '- 拿不准时选择让整个文件保持自洽的一侧，不要把两边拼贴成互相矛盾的代码。',
+  '- 不要臆造两边都没有的新逻辑。',
+].join('\n')
+
+/** 冲突标记只认行首的 7 个 < 或 >（`=======` 可能合法出现在 Markdown 里，不查）。 */
+export function hasConflictMarkers(text) {
+  return String(text ?? '')
+    .split('\n')
+    .some((line) => /^(<{7}|>{7})(\s|$)/.test(line))
+}
+
+/** 取最长的围栏代码块；模型常常在代码块前后加一句解释。 */
+export function extractFenced(text) {
+  const source = String(text ?? '')
+  const re = /^[ \t]*```[^\n]*\n([\s\S]*?)^[ \t]*```[ \t]*$/gm
+  let match
+  let best = null
+  while ((match = re.exec(source)) !== null) {
+    const block = match[1]
+    if (best === null || block.length > best.length) best = block
+  }
+  return best
+}
+
+/**
+ * 清理模型输出为文件内容。
+ * 只剥围栏和首尾空行，绝不逐行 trim —— 缩进就是语义。
+ * @param {string} text
+ * @returns {string}
+ */
+export function sanitizeResolution(text) {
+  const fenced = extractFenced(text)
+  let value = fenced !== null ? fenced : String(text ?? '')
+  value = value.replace(/\r\n/g, '\n')
+  if (fenced === null) {
+    // 没有围栏时，模型可能仍写了整行 ``` 围栏
+    value = value.replace(/^[ \t]*```[a-zA-Z0-9_-]*[ \t]*$/gm, '')
+  }
+  return value.replace(/\n{0,2}$/, '\n')
+}
+
+/**
+ * 校验模型给出的解决结果。不合格就抛错，**不要**用一个坏结果去覆盖用户文件。
+ * @param {string} suggested
+ */
+export function validateResolution(suggested) {
+  if (typeof suggested !== 'string') {
+    throw new GitError('模型没有返回内容', { code: 'ai-unavailable' })
+  }
+  if (suggested.trim() === '') {
+    throw new GitError('模型返回了空文件，未采用', { code: 'ai-unavailable' })
+  }
+  if (hasConflictMarkers(suggested)) {
+    throw new GitError('模型输出仍包含冲突标记，未采用', { code: 'ai-unavailable' })
+  }
+}
+
+function capSide(label, text) {
+  const value = String(text ?? '')
+  if (value.length <= MAX_SIDE_CHARS) return value
+  return value.slice(0, MAX_SIDE_CHARS) + `\n…（${label} 超过 ${MAX_SIDE_CHARS} 字符，已截断）`
+}
+
+/** 构造 user prompt。 */
+export function buildConflictPrompt({ path, base, ours, theirs, branch, incoming, hint }) {
+  const parts = ['文件：' + path]
+  if (branch) parts.push('当前分支（ours）：' + branch)
+  if (incoming) parts.push('合入分支（theirs）：' + incoming)
+  if (hint) parts.push('用户补充说明：' + hint)
+  parts.push('')
+  if (base !== null && base !== undefined) {
+    parts.push('=== base（共同祖先）===', capSide('base', base), '')
+  } else {
+    parts.push('=== base ===', '（这个文件没有共同祖先版本，是 add/add 冲突）', '')
+  }
+  parts.push('=== ours（当前分支）===', capSide('ours', ours), '')
+  parts.push('=== theirs（合入分支）===', capSide('theirs', theirs))
+  return parts.join('\n')
+}
+
+/**
+ * 让模型给出一份解决结果。
+ * @param {{
+ *   path: string, base: string|null, ours: string, theirs: string,
+ *   branch?: string|null, incoming?: string|null, hint?: string,
+ *   llm?: unknown, selection?: {provider?: string, model?: string, reasoningEffort?: string},
+ *   signal?: AbortSignal, systemPrompt?: string,
+ * }} options
+ * @returns {Promise<{suggested: string, model: string}>}
+ */
+export async function resolveConflict(options) {
+  const { llm, selection, signal } = options
+  if (!llm || !selection || !selection.provider || !selection.model) {
+    throw new GitError('没有可用的默认模型，无法生成解决建议', { code: 'ai-unavailable' })
+  }
+  const model = selection.provider + '/' + selection.model
+  const systemPrompt = typeof options.systemPrompt === 'string' && options.systemPrompt.trim() !== ''
+    ? options.systemPrompt.trim().slice(0, 4000)
+    : DEFAULT_SYSTEM_PROMPT
+  const prompt = buildConflictPrompt(options)
+
+  let text = ''
+  try {
+    const stream = llm.stream({
+      provider: selection.provider,
+      model: selection.model,
+      reasoningEffort: selection.reasoningEffort,
+      system: systemPrompt,
+      messages: [{ role: 'user', content: [{ type: 'text', text: prompt }] }],
+      maxTokens: MAX_OUTPUT_TOKENS,
+      temperature: 0.1,
+      signal,
+    })
+    for await (const chunk of stream) {
+      if (chunk.type === 'text-delta') text += chunk.text
+      if (chunk.type === 'finish') {
+        const kind = chunk.reason && chunk.reason.kind
+        if (kind === 'error' || kind === 'aborted') {
+          const failure = chunk.reason.failure
+          throw new GitError('模型生成失败：' + (failure && failure.message ? failure.message : kind), {
+            code: 'ai-unavailable',
+          })
+        }
+      }
+    }
+  } catch (error) {
+    if (error instanceof GitError) throw error
+    const message = error && error.message ? error.message : String(error)
+    throw new GitError('模型生成失败：' + message, { code: 'ai-unavailable' })
+  }
+
+  const suggested = sanitizeResolution(text)
+  validateResolution(suggested)
+  return { suggested, model }
+}