@kkelly-offical/kkcode 0.1.2

package/src/review/diff-parser.mjs

@@ -0,0 +1,36 @@
+export function parseUnifiedDiff(diffText) {
+  const lines = diffText.split(/\r?\n/)
+  const files = []
+  let current = null
+
+  for (const line of lines) {
+    if (line.startsWith("diff --git ")) {
+      if (current) files.push(current)
+      const parts = line.split(" ")
+      const bPath = parts[3] || ""
+      current = {
+        path: bPath.replace(/^b\//, ""),
+        added: 0,
+        removed: 0,
+        rawLines: [],
+        addedLines: []
+      }
+      continue
+    }
+    if (!current) continue
+    current.rawLines.push(line)
+    if (line.startsWith("+") && !line.startsWith("+++")) {
+      current.added += 1
+      current.addedLines.push(line.slice(1))
+    }
+    if (line.startsWith("-") && !line.startsWith("---")) {
+      current.removed += 1
+    }
+  }
+  if (current) files.push(current)
+  return files.filter((file) => file.path.length > 0)
+}
+
+export function previewLines(file, limit = 80) {
+  return file.rawLines.slice(0, limit)
+}

package/src/review/rejection-queue.mjs

@@ -0,0 +1,62 @@
+import { ensureProjectRoot, reviewRejectionQueuePath } from "../storage/paths.mjs"
+import { readJson, writeJson } from "../storage/json-store.mjs"
+
+function now() {
+  return Date.now()
+}
+
+function defaults() {
+  return {
+    updatedAt: now(),
+    entries: []
+  }
+}
+
+async function load(cwd = process.cwd()) {
+  await ensureProjectRoot(cwd)
+  return readJson(reviewRejectionQueuePath(cwd), defaults())
+}
+
+async function save(data, cwd = process.cwd()) {
+  data.updatedAt = now()
+  await writeJson(reviewRejectionQueuePath(cwd), data)
+}
+
+export async function enqueueRejection(entry, cwd = process.cwd()) {
+  const data = await load(cwd)
+  data.entries.push({
+    id: `rej_${now()}_${Math.random().toString(36).slice(2, 8)}`,
+    createdAt: now(),
+    consumed: false,
+    ...entry
+  })
+  await save(data, cwd)
+  return data.entries[data.entries.length - 1]
+}
+
+export async function listRejections(cwd = process.cwd()) {
+  const data = await load(cwd)
+  return data.entries.sort((a, b) => b.createdAt - a.createdAt)
+}
+
+export async function pendingRejections(cwd = process.cwd()) {
+  const data = await load(cwd)
+  return data.entries.filter((entry) => !entry.consumed)
+}
+
+export async function markRejectionsConsumed(ids, sessionId, cwd = process.cwd()) {
+  if (!ids.length) return
+  const data = await load(cwd)
+  const set = new Set(ids)
+  for (const entry of data.entries) {
+    if (!set.has(entry.id)) continue
+    entry.consumed = true
+    entry.consumedAt = now()
+    entry.consumedBy = sessionId
+  }
+  await save(data, cwd)
+}
+
+export async function clearRejections(cwd = process.cwd()) {
+  await save(defaults(), cwd)
+}

package/src/review/review-store.mjs

@@ -0,0 +1,21 @@
+import { ensureProjectRoot, reviewStorePath } from "../storage/paths.mjs"
+import { readJson, writeJson } from "../storage/json-store.mjs"
+
+export function defaultReviewState() {
+  return {
+    createdAt: Date.now(),
+    sessionId: null,
+    currentIndex: 0,
+    files: []
+  }
+}
+
+export async function readReviewState(cwd = process.cwd()) {
+  await ensureProjectRoot(cwd)
+  return readJson(reviewStorePath(cwd), defaultReviewState())
+}
+
+export async function writeReviewState(state, cwd = process.cwd()) {
+  await ensureProjectRoot(cwd)
+  await writeJson(reviewStorePath(cwd), state)
+}

package/src/review/risk-score.mjs

@@ -0,0 +1,61 @@
+const DEFAULT_PATH_RISK = ["config", "auth", "permission", "migration", "infra", "script"]
+const COMMAND_RISK_RE = /\b(curl|wget|powershell|pwsh|bash|sh|chmod|sudo|eval|exec|rm\s+-rf)\b/i
+
+const DEFAULT_WEIGHTS = {
+  sensitive_path: 4,
+  large_change: 3,
+  medium_change: 2,
+  small_change: 1,
+  executable_script: 2,
+  command_pattern: 3
+}
+
+export function scoreRisk(file, options = {}) {
+  const w = { ...DEFAULT_WEIGHTS, ...options.weights }
+  const sensitiveKeys = options.sensitive_paths || DEFAULT_PATH_RISK
+
+  let score = 1
+  const reasons = []
+  const lowerPath = file.path.toLowerCase()
+
+  for (const key of sensitiveKeys) {
+    if (lowerPath.includes(key)) {
+      score += w.sensitive_path
+      reasons.push(`path contains "${key}"`)
+    }
+  }
+
+  const changed = file.added + file.removed
+  if (changed > 200) {
+    score += w.large_change
+    reasons.push("large change size (>200 lines)")
+  } else if (changed > 80) {
+    score += w.medium_change
+    reasons.push("medium change size (>80 lines)")
+  } else if (changed > 30) {
+    score += w.small_change
+    reasons.push("noticeable change size (>30 lines)")
+  }
+
+  if (/\.(sh|ps1|bat|cmd)$/i.test(file.path)) {
+    score += w.executable_script
+    reasons.push("executable script file")
+  }
+
+  if (file.addedLines.some((line) => COMMAND_RISK_RE.test(line))) {
+    score += w.command_pattern
+    reasons.push("contains executable command patterns")
+  }
+
+  return { score, reasons }
+}
+
+export function sortReviewFiles(files, sortMode) {
+  if (sortMode === "file_order") {
+    return [...files].sort((a, b) => a.path.localeCompare(b.path))
+  }
+  if (sortMode === "time_order") {
+    return [...files]
+  }
+  return [...files].sort((a, b) => b.riskScore - a.riskScore || a.path.localeCompare(b.path))
+}

package/src/rules/load-rules.mjs

@@ -0,0 +1,64 @@
+import path from "node:path"
+import { access, readdir, readFile } from "node:fs/promises"
+
+async function exists(file) {
+  try {
+    await access(file)
+    return true
+  } catch {
+    return false
+  }
+}
+
+async function readRuleDir(dir, scope) {
+  if (!(await exists(dir))) return []
+  const entries = await readdir(dir, { withFileTypes: true })
+  const files = entries
+    .filter((entry) => entry.isFile() && entry.name.toLowerCase().endsWith(".md"))
+    .map((entry) => entry.name)
+    .sort((a, b) => a.localeCompare(b))
+  const blocks = []
+  for (const file of files) {
+    const target = path.join(dir, file)
+    const content = (await readFile(target, "utf8")).trim()
+    if (!content) continue
+    blocks.push({
+      scope,
+      file: target,
+      content
+    })
+  }
+  return blocks
+}
+
+async function readSingleRuleFile(filePath, scope) {
+  if (!(await exists(filePath))) return []
+  const content = (await readFile(filePath, "utf8")).trim()
+  if (!content) return []
+  return [{ scope, file: filePath, content }]
+}
+
+export async function loadRuleBlocks(cwd = process.cwd()) {
+  const userHome = process.env.USERPROFILE || process.env.HOME || cwd
+  const userRuleFile = path.join(userHome, ".kkcode", "rule.md")
+  const userRulesDir = path.join(userHome, ".kkcode", "rules")
+  const projectRuleFile = path.join(cwd, ".kkcode", "rule.md")
+  const projectRulesDir = path.join(cwd, ".kkcode", "rules")
+  const [globalSingle, globalDir, projectSingle, projectDir] = await Promise.all([
+    readSingleRuleFile(userRuleFile, "global"),
+    readRuleDir(userRulesDir, "global"),
+    readSingleRuleFile(projectRuleFile, "project"),
+    readRuleDir(projectRulesDir, "project")
+  ])
+  return [...globalSingle, ...globalDir, ...projectSingle, ...projectDir]
+}
+
+export async function renderRulesPrompt(cwd = process.cwd()) {
+  const blocks = await loadRuleBlocks(cwd)
+  if (!blocks.length) return ""
+  return blocks
+    .map((block) => {
+      return [`<rule scope="${block.scope}" source="${block.file}">`, block.content, `</rule>`].join("\n")
+    })
+    .join("\n\n")
+}

package/src/runtime.mjs

@@ -0,0 +1 @@
+export { resolveMode, newSessionId, executeTurn } from "./session/engine.mjs"

package/src/session/checkpoint.mjs

@@ -0,0 +1,239 @@
+import path from "node:path"
+import { mkdir, readdir } from "node:fs/promises"
+import { readJson, writeJson } from "../storage/json-store.mjs"
+import { userRootDir } from "../storage/paths.mjs"
+import { isGitRepo } from "../util/git.mjs"
+import { gitSnapshotTool } from "../tool/git-auto.mjs"
+import { listGhostCommits, getLatestGhostCommit } from "../storage/ghost-commit-store.mjs"
+
+function checkpointDir(sessionId) {
+  return path.join(userRootDir(), "checkpoints", sessionId)
+}
+
+function checkpointFile(sessionId, name) {
+  return path.join(checkpointDir(sessionId), `${name}.json`)
+}
+
+function latestFile(sessionId) {
+  return checkpointFile(sessionId, "latest")
+}
+
+export async function saveCheckpoint(sessionId, data) {
+  const dir = checkpointDir(sessionId)
+  await mkdir(dir, { recursive: true })
+  const checkpoint = {
+    sessionId,
+    savedAt: Date.now(),
+    ...data
+  }
+  await writeJson(latestFile(sessionId), checkpoint)
+  const numbered = checkpointFile(sessionId, `cp_${data.iteration || 0}`)
+  await writeJson(numbered, checkpoint)
+  return checkpoint
+}
+
+export async function loadCheckpoint(sessionId, name = "latest") {
+  const file = name === "latest" ? latestFile(sessionId) : checkpointFile(sessionId, name)
+  return readJson(file, null)
+}
+
+export async function listCheckpoints(sessionId) {
+  const dir = checkpointDir(sessionId)
+  const files = await readdir(dir, { withFileTypes: true }).catch(() => [])
+  return files
+    .filter((entry) => entry.isFile() && entry.name.endsWith(".json"))
+    .map((entry) => entry.name.replace(/\.json$/, ""))
+    .sort()
+}
+
+// ============================================================================
+// Git Snapshot Integration - AI Agent 自动 Git 快照功能
+// ============================================================================
+
+/**
+ * 在 AI 修改前自动创建 Git 快照
+ * 
+ * @param {string} sessionId - 会话ID
+ * @param {string} cwd - 工作目录
+ * @param {Object} config - 配置对象
+ * @param {Object} options - 选项
+ * @param {string} [options.reason] - 快照原因
+ * @returns {Promise<{ok: boolean, snapshot?: Object, skipped?: boolean, reason?: string}>}
+ */
+export async function autoSnapshotBeforeEdit(sessionId, cwd, config = {}, options = {}) {
+  // 检查 Git 自动化是否启用
+  if (config.git_auto?.enabled === false) {
+    return { ok: true, skipped: true, reason: "git_auto_disabled" }
+  }
+
+  // 检查自动快照是否启用
+  if (config.git_auto?.auto_snapshot === false) {
+    return { ok: true, skipped: true, reason: "auto_snapshot_disabled" }
+  }
+
+  // 检查是否是 Git 仓库
+  if (!(await isGitRepo(cwd))) {
+    return { ok: true, skipped: true, reason: "not_a_git_repo" }
+  }
+
+  try {
+    const result = await gitSnapshotTool.execute(
+      {
+        auto: true,
+        message: options.reason || `Auto snapshot before AI edit (session: ${sessionId})`
+      },
+      { cwd, sessionId, config }
+    )
+
+    if (result.ok) {
+      return {
+        ok: true,
+        snapshot: result.snapshot,
+        skipped: false
+      }
+    } else {
+      return {
+        ok: false,
+        skipped: true,
+        reason: result.message || "snapshot_failed"
+      }
+    }
+  } catch (error) {
+    return {
+      ok: false,
+      skipped: true,
+      reason: error.message
+    }
+  }
+}
+
+/**
+ * 获取会话的 Git 快照历史
+ * 
+ * @param {string} sessionId - 会话ID
+ * @param {string} cwd - 工作目录
+ * @returns {Promise<Array<Object>>}
+ */
+export async function getSessionSnapshots(sessionId, cwd) {
+  if (!(await isGitRepo(cwd))) {
+    return []
+  }
+
+  const snapshots = await listGhostCommits(cwd)
+  // 过滤出当前会话的快照
+  return snapshots.filter(s => 
+    s.message?.includes(`session: ${sessionId}`) ||
+    s.message?.includes("Auto snapshot")
+  )
+}
+
+/**
+ * 恢复到会话的最近一次快照
+ * 
+ * @param {string} sessionId - 会话ID
+ * @param {string} cwd - 工作目录
+ * @returns {Promise<{ok: boolean, message?: string, error?: string}>}
+ */
+export async function restoreLastSessionSnapshot(sessionId, cwd) {
+  if (!(await isGitRepo(cwd))) {
+    return { ok: false, error: "Not a git repository" }
+  }
+
+  const snapshots = await getSessionSnapshots(sessionId, cwd)
+  if (snapshots.length === 0) {
+    return { ok: false, error: "No snapshots found for this session" }
+  }
+
+  const latest = snapshots[0]
+  const { gitRestoreTool } = await import("../tool/git-auto.mjs")
+
+  const result = await gitRestoreTool.execute(
+    { snapshot_id: latest.id },
+    { cwd, sessionId }
+  )
+
+  return result
+}
+
+/**
+ * Checkpoint Manager - 统一的管理器
+ * 
+ * 协调 JSON checkpoint 和 Git snapshot 两种机制：
+ * - JSON checkpoint: 保存会话状态（内存中的数据）
+ * - Git snapshot: 保存工作目录状态（文件系统状态）
+ */
+export class CheckpointManager {
+  constructor(sessionId, cwd, config = {}) {
+    this.sessionId = sessionId
+    this.cwd = cwd
+    this.config = config
+    this.lastSnapshotId = null
+  }
+
+  /**
+   * 在修改前创建检查点（自动决定使用哪种机制）
+   */
+  async beforeEdit(reason = "AI edit") {
+    const results = {
+      jsonCheckpoint: null,
+      gitSnapshot: null
+    }
+
+    // 1. 创建 JSON checkpoint（如果配置启用）
+    if (this.config.checkpoint?.enabled !== false) {
+      // 这里可以扩展保存更多会话状态
+      results.jsonCheckpoint = await saveCheckpoint(this.sessionId, {
+        type: "pre_edit",
+        reason,
+        timestamp: Date.now()
+      })
+    }
+
+    // 2. 创建 Git snapshot（如果配置启用）
+    if (this.config.git_auto?.enabled !== false && this.config.git_auto?.auto_snapshot !== false) {
+      const snapshotResult = await autoSnapshotBeforeEdit(
+        this.sessionId,
+        this.cwd,
+        this.config,
+        { reason }
+      )
+      
+      if (snapshotResult.ok && !snapshotResult.skipped) {
+        results.gitSnapshot = snapshotResult.snapshot
+        this.lastSnapshotId = snapshotResult.snapshot.id
+      }
+    }
+
+    return results
+  }
+
+  /**
+   * 恢复到最近一次检查点
+   */
+  async restore() {
+    if (this.lastSnapshotId) {
+      const { gitRestoreTool } = await import("../tool/git-auto.mjs")
+      return await gitRestoreTool.execute(
+        { snapshot_id: this.lastSnapshotId },
+        { cwd: this.cwd, sessionId: this.sessionId }
+      )
+    }
+
+    // 如果没有快照ID，尝试恢复到最近一次会话快照
+    return await restoreLastSessionSnapshot(this.sessionId, this.cwd)
+  }
+
+  /**
+   * 获取当前会话的所有快照
+   */
+  async listSnapshots() {
+    return await getSessionSnapshots(this.sessionId, this.cwd)
+  }
+}
+
+/**
+ * 创建 CheckpointManager 实例的工厂函数
+ */
+export function createCheckpointManager(sessionId, cwd, config) {
+  return new CheckpointManager(sessionId, cwd, config)
+}