ai-spec-dev 0.1.0 → 0.14.1

package/core/key-store.ts

@@ -0,0 +1,49 @@
+import * as fs from "fs-extra";
+import * as path from "path";
+import * as os from "os";
+
+const KEY_STORE_FILE = path.join(os.homedir(), ".ai-spec-keys.json");
+
+type KeyStore = Record<string, string>;
+
+async function readStore(): Promise<KeyStore> {
+  try {
+    if (await fs.pathExists(KEY_STORE_FILE)) {
+      return await fs.readJson(KEY_STORE_FILE);
+    }
+  } catch {
+    // ignore corrupt file
+  }
+  return {};
+}
+
+async function writeStore(store: KeyStore): Promise<void> {
+  await fs.writeJson(KEY_STORE_FILE, store, { spaces: 2 });
+  // Restrict permissions to owner only (600)
+  await fs.chmod(KEY_STORE_FILE, 0o600);
+}
+
+export async function getSavedKey(provider: string): Promise<string | undefined> {
+  const store = await readStore();
+  return store[provider] || undefined;
+}
+
+export async function saveKey(provider: string, key: string): Promise<void> {
+  const store = await readStore();
+  store[provider] = key;
+  await writeStore(store);
+}
+
+export async function clearAllKeys(): Promise<void> {
+  if (await fs.pathExists(KEY_STORE_FILE)) {
+    await fs.remove(KEY_STORE_FILE);
+  }
+}
+
+export async function clearKey(provider: string): Promise<void> {
+  const store = await readStore();
+  delete store[provider];
+  await writeStore(store);
+}
+
+export { KEY_STORE_FILE };

package/core/knowledge-memory.ts

@@ -0,0 +1,171 @@
+import chalk from "chalk";
+import * as fs from "fs-extra";
+import * as path from "path";
+import { AIProvider } from "./spec-generator";
+import { CONSTITUTION_FILE } from "./constitution-generator";
+import { parseConstitutionStats } from "../prompts/consolidate.prompt";
+
+// ─── Types ──────────────────────────────────────────────────────────────────────
+
+export interface ReviewIssue {
+  /** Short description of the issue */
+  description: string;
+  /** Which file or area */
+  location?: string;
+  /** Category: bug / pattern / style / security / performance */
+  category: string;
+}
+
+// ─── Extract Issues from Review ─────────────────────────────────────────────────
+
+/**
+ * Parse review text to extract issues from the "⚠️ 问题" section.
+ * Returns up to 10 issues (to keep constitution append manageable).
+ */
+export function extractIssuesFromReview(reviewText: string): ReviewIssue[] {
+  const issues: ReviewIssue[] = [];
+
+  // Find the issues section (between ⚠️ and 💡 or 📊)
+  const issuesMatch = reviewText.match(
+    /## ⚠[^\n]*\n([\s\S]*?)(?=## [💡📊]|\n*$)/i
+  );
+  if (!issuesMatch) return issues;
+
+  const section = issuesMatch[1];
+  const lines = section.split("\n");
+
+  for (const line of lines) {
+    const trimmed = line.trim();
+    if (!trimmed || trimmed.startsWith("#")) continue;
+
+    // Match list items: "- issue text" or "1. issue text" or "**Category**: text"
+    const itemMatch = trimmed.match(/^[-*\d]+[.)]?\s*(.+)/);
+    if (itemMatch) {
+      const desc = itemMatch[1].replace(/\*\*/g, "").trim();
+      if (desc.length > 10) {
+        issues.push({
+          description: desc.slice(0, 200),
+          category: categorizeIssue(desc),
+        });
+      }
+    }
+  }
+
+  return issues.slice(0, 10);
+}
+
+function categorizeIssue(text: string): string {
+  const lower = text.toLowerCase();
+  if (lower.includes("security") || lower.includes("auth") || lower.includes("注入") || lower.includes("xss") || lower.includes("sql")) return "security";
+  if (lower.includes("performance") || lower.includes("慢") || lower.includes("性能") || lower.includes("n+1")) return "performance";
+  if (lower.includes("error") || lower.includes("异常") || lower.includes("crash") || lower.includes("bug")) return "bug";
+  if (lower.includes("pattern") || lower.includes("模式") || lower.includes("convention") || lower.includes("命名")) return "pattern";
+  return "general";
+}
+
+// ─── Append to Constitution ────────────────────────────────────────────────────
+
+const MEMORY_SECTION_HEADER = "\n\n## 9. 积累教训 (Accumulated Lessons)\n";
+const MEMORY_SECTION_MARKER = "## 9. 积累教训";
+
+/**
+ * Append review issues to the project constitution as accumulated lessons.
+ * Creates the section if it doesn't exist; appends if it does.
+ * Deduplicates by checking if a similar lesson already exists.
+ */
+export async function appendLessonsToConstitution(
+  projectRoot: string,
+  issues: ReviewIssue[]
+): Promise<void> {
+  if (issues.length === 0) return;
+
+  const constitutionPath = path.join(projectRoot, CONSTITUTION_FILE);
+  let content = "";
+  try {
+    content = await fs.readFile(constitutionPath, "utf-8");
+  } catch {
+    console.log(chalk.gray("  No constitution file — skipping knowledge memory."));
+    return;
+  }
+
+  // Check if section 9 already exists
+  const hasMemorySection = content.includes(MEMORY_SECTION_MARKER);
+
+  // Build new lesson entries with date stamp
+  const date = new Date().toISOString().slice(0, 10);
+  const newEntries: string[] = [];
+
+  for (const issue of issues) {
+    // Simple dedup: check if a similar line already exists (case-insensitive substring)
+    const normalized = issue.description.toLowerCase().slice(0, 50);
+    if (content.toLowerCase().includes(normalized)) continue;
+
+    const badge = issue.category === "security" ? "🔒" :
+                  issue.category === "performance" ? "⚡" :
+                  issue.category === "bug" ? "🐛" :
+                  issue.category === "pattern" ? "📐" : "📝";
+    newEntries.push(`- ${badge} **[${date}]** ${issue.description}`);
+  }
+
+  if (newEntries.length === 0) {
+    console.log(chalk.gray("  No new lessons to add (all deduplicated)."));
+    return;
+  }
+
+  let updatedContent: string;
+  if (hasMemorySection) {
+    // Append to existing section — find the end of section 9
+    // Strategy: find "## 9." then insert before the next "## " or EOF
+    const sectionStart = content.indexOf(MEMORY_SECTION_MARKER);
+    const afterHeader = sectionStart + MEMORY_SECTION_HEADER.length;
+    // Find next section header after section 9
+    const nextSectionMatch = content.slice(afterHeader).match(/\n## \d/);
+    const insertPos = nextSectionMatch
+      ? afterHeader + nextSectionMatch.index!
+      : content.length;
+    updatedContent =
+      content.slice(0, insertPos) +
+      newEntries.join("\n") + "\n" +
+      content.slice(insertPos);
+  } else {
+    // Append new section at the end
+    updatedContent = content + MEMORY_SECTION_HEADER + newEntries.join("\n") + "\n";
+  }
+
+  await fs.writeFile(constitutionPath, updatedContent, "utf-8");
+  console.log(chalk.green(`  ✔ ${newEntries.length} lesson(s) appended to constitution (§9).`));
+
+  // Warn when §9 is getting long
+  const stats = parseConstitutionStats(updatedContent);
+  if (stats.lessonCount >= 8) {
+    console.log(
+      chalk.yellow(
+        `  ⚠ §9 now has ${stats.lessonCount} accumulated lessons. Run \`ai-spec init --consolidate\` to prune and rebase.`
+      )
+    );
+  }
+}
+
+/**
+ * Full knowledge memory flow: extract issues from review → append to constitution.
+ */
+export async function accumulateReviewKnowledge(
+  provider: AIProvider,
+  projectRoot: string,
+  reviewText: string
+): Promise<void> {
+  console.log(chalk.blue("\n─── Knowledge Memory ────────────────────────────"));
+
+  const issues = extractIssuesFromReview(reviewText);
+  if (issues.length === 0) {
+    console.log(chalk.gray("  No actionable issues found in review. Skipping."));
+    return;
+  }
+
+  console.log(chalk.gray(`  Extracted ${issues.length} issue(s) from review:`));
+  for (const issue of issues) {
+    console.log(chalk.gray(`    - [${issue.category}] ${issue.description.slice(0, 80)}`));
+  }
+
+  await appendLessonsToConstitution(projectRoot, issues);
+}