flavor-code 1.3.23 → 1.4.0-beta.10

package/dist/astgraph/db.mjs

@@ -1,237 +0,0 @@
-// astgraph graph database — SQLite-backed node/edge graph (codegraph-inspired).
-// Uses node:sqlite with WAL mode so concurrent readers never block on a writer.
-
-import { mkdirSync } from "node:fs";
-import { dirname } from "node:path";
-import { DatabaseSync } from "node:sqlite";
-
-const SCHEMA = `
-CREATE TABLE IF NOT EXISTS schema_versions (
-    version INTEGER PRIMARY KEY,
-    applied_at INTEGER NOT NULL,
-    description TEXT
-);
-
-CREATE TABLE IF NOT EXISTS files (
-    path TEXT PRIMARY KEY,
-    content_hash TEXT NOT NULL,
-    language TEXT NOT NULL,
-    size INTEGER NOT NULL,
-    modified_at INTEGER NOT NULL DEFAULT 0,
-    indexed_at INTEGER NOT NULL DEFAULT 0,
-    node_count INTEGER NOT NULL DEFAULT 0
-);
-
-CREATE TABLE IF NOT EXISTS nodes (
-    id TEXT PRIMARY KEY,
-    kind TEXT NOT NULL,
-    name TEXT NOT NULL,
-    qualified_name TEXT NOT NULL,
-    file_path TEXT NOT NULL,
-    language TEXT NOT NULL,
-    start_line INTEGER NOT NULL,
-    end_line INTEGER NOT NULL,
-    signature TEXT,
-    docstring TEXT,
-    is_exported INTEGER NOT NULL DEFAULT 0,
-    is_async INTEGER NOT NULL DEFAULT 0
-);
-
-CREATE TABLE IF NOT EXISTS edges (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    source TEXT NOT NULL,
-    target TEXT NOT NULL,
-    kind TEXT NOT NULL,
-    line INTEGER,
-    col INTEGER,
-    FOREIGN KEY (source) REFERENCES nodes(id) ON DELETE CASCADE,
-    FOREIGN KEY (target) REFERENCES nodes(id) ON DELETE CASCADE
-);
-
-CREATE TABLE IF NOT EXISTS unresolved_refs (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    from_node_id TEXT,
-    reference_name TEXT NOT NULL,
-    reference_kind TEXT NOT NULL,
-    line INTEGER NOT NULL,
-    col INTEGER NOT NULL,
-    file_path TEXT NOT NULL DEFAULT '',
-    module_specifier TEXT,
-    status TEXT NOT NULL DEFAULT 'pending',
-    FOREIGN KEY (from_node_id) REFERENCES nodes(id) ON DELETE CASCADE
-);
-
-CREATE INDEX IF NOT EXISTS idx_nodes_name ON nodes(name);
-CREATE INDEX IF NOT EXISTS idx_nodes_lower_name ON nodes(lower(name));
-CREATE INDEX IF NOT EXISTS idx_nodes_file_path ON nodes(file_path);
-CREATE INDEX IF NOT EXISTS idx_nodes_kind ON nodes(kind);
-CREATE INDEX IF NOT EXISTS idx_edges_source_kind ON edges(source, kind);
-CREATE INDEX IF NOT EXISTS idx_edges_target_kind ON edges(target, kind);
-CREATE UNIQUE INDEX IF NOT EXISTS idx_edges_identity
-  ON edges(source, target, kind, IFNULL(line, -1), IFNULL(col, -1));
-CREATE INDEX IF NOT EXISTS idx_unresolved_from ON unresolved_refs(from_node_id);
-CREATE INDEX IF NOT EXISTS idx_unresolved_name ON unresolved_refs(reference_name);
-
-CREATE VIRTUAL TABLE IF NOT EXISTS nodes_fts USING fts5(
-    name,
-    qualified_name,
-    signature,
-    docstring,
-    content='nodes',
-    content_rowid='rowid'
-);
-
-CREATE TRIGGER IF NOT EXISTS nodes_ai AFTER INSERT ON nodes BEGIN
-    INSERT INTO nodes_fts(rowid, name, qualified_name, signature, docstring)
-    VALUES (NEW.rowid, NEW.name, NEW.qualified_name, NEW.signature, NEW.docstring);
-END;
-
-CREATE TRIGGER IF NOT EXISTS nodes_ad AFTER DELETE ON nodes BEGIN
-    INSERT INTO nodes_fts(nodes_fts, rowid, name, qualified_name, signature, docstring)
-    VALUES ('delete', OLD.rowid, OLD.name, OLD.qualified_name, OLD.signature, OLD.docstring);
-END;
-
-CREATE TRIGGER IF NOT EXISTS nodes_au AFTER UPDATE ON nodes BEGIN
-    INSERT INTO nodes_fts(nodes_fts, rowid, name, qualified_name, signature, docstring)
-    VALUES ('delete', OLD.rowid, OLD.name, OLD.qualified_name, OLD.signature, OLD.docstring);
-    INSERT INTO nodes_fts(rowid, name, qualified_name, signature, docstring)
-    VALUES (NEW.rowid, NEW.name, NEW.qualified_name, NEW.signature, NEW.docstring);
-END;
-
-CREATE TABLE IF NOT EXISTS project_metadata (
-    key TEXT PRIMARY KEY,
-    value TEXT NOT NULL,
-    updated_at INTEGER NOT NULL
-);
-`;
-
-/** Open (creating when absent) the graph database at `path`. Returns a DatabaseSync. */
-export function openDb(path) {
-  if (path !== ":memory:") mkdirSync(dirname(path), { recursive: true });
-  const db = new DatabaseSync(path);
-  db.exec("PRAGMA journal_mode = WAL");
-  db.exec("PRAGMA foreign_keys = ON");
-  db.exec("PRAGMA synchronous = NORMAL");
-  db.exec(SCHEMA);
-  const version = db.prepare("SELECT COUNT(*) AS count FROM schema_versions").get();
-  if (version.count === 0) {
-    db.prepare("INSERT INTO schema_versions(version, applied_at, description) VALUES (?, ?, ?)")
-      .run(1, Date.now(), "Initial astgraph schema");
-  }
-  return db;
-}
-
-/** Insert or replace the tracking record for one source file. */
-export function upsertFileRecord(db, record) {
-  db.prepare(`
-    INSERT INTO files(path, content_hash, language, size, modified_at, indexed_at, node_count)
-    VALUES (?, ?, ?, ?, ?, ?, ?)
-    ON CONFLICT(path) DO UPDATE SET
-      content_hash = excluded.content_hash,
-      language = excluded.language,
-      size = excluded.size,
-      modified_at = excluded.modified_at,
-      indexed_at = excluded.indexed_at,
-      node_count = excluded.node_count
-  `).run(
-    record.path, record.contentHash, record.language, record.size,
-    record.modifiedAt ?? Date.now(), record.indexedAt ?? Date.now(), record.nodeCount ?? 0,
-  );
-}
-
-/** Read the tracking record for one source file, or undefined when absent. */
-export function getFileRecord(db, path) {
-  const row = db.prepare("SELECT * FROM files WHERE path = ?").get(path);
-  if (row === undefined) return undefined;
-  return {
-    path: row.path, contentHash: row.content_hash, language: row.language,
-    size: row.size, modifiedAt: row.modified_at, indexedAt: row.indexed_at,
-    nodeCount: row.node_count,
-  };
-}
-
-/** Delete a file record; nodes cascade (and with them edges + refs + FTS rows). */
-export function deleteFileRecord(db, path) {
-  db.prepare("DELETE FROM nodes WHERE file_path = ?").run(path);
-  db.prepare("DELETE FROM files WHERE path = ?").run(path);
-}
-
-/**
- * Replace every node belonging to a file. Runs inside one transaction; the FTS
- * triggers keep nodes_fts in sync.
- */
-export function replaceNodes(db, filePath, nodes) {
-  const insert = db.prepare(`
-    INSERT OR REPLACE INTO nodes(
-      id, kind, name, qualified_name, file_path, language,
-      start_line, end_line, signature, docstring, is_exported, is_async
-    ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-  `);
-  const clear = db.prepare("DELETE FROM nodes WHERE file_path = ?");
-  db.exec("BEGIN");
-  try {
-    clear.run(filePath);
-    for (const node of nodes) {
-      insert.run(
-        node.id, node.kind, node.name, node.qualifiedName, node.filePath, node.language,
-        node.startLine, node.endLine, node.signature ?? null, node.docstring ?? null,
-        node.isExported ? 1 : 0, node.isAsync ? 1 : 0,
-      );
-    }
-    db.exec("COMMIT");
-  } catch (error) {
-    db.exec("ROLLBACK");
-    throw error;
-  }
-}
-
-/** Insert one edge; duplicate (source, target, kind, line, col) rows are ignored. */
-export function insertEdge(db, edge) {
-  db.prepare(`
-    INSERT OR IGNORE INTO edges(source, target, kind, line, col) VALUES (?, ?, ?, ?, ?)
-  `).run(edge.source, edge.target, edge.kind, edge.line ?? null, edge.col ?? null);
-}
-
-/** Delete every edge whose source OR target belongs to a file. */
-export function deleteEdgesForFile(db, filePath) {
-  db.prepare(`
-    DELETE FROM edges WHERE source IN (SELECT id FROM nodes WHERE file_path = ?)
-       OR target IN (SELECT id FROM nodes WHERE file_path = ?)
-  `).run(filePath, filePath);
-}
-
-/** Insert one unresolved cross-file reference (status 'pending'). */
-export function insertUnresolvedRef(db, ref) {
-  db.prepare(`
-    INSERT INTO unresolved_refs(from_node_id, reference_name, reference_kind, line, col, file_path, module_specifier, status)
-    VALUES (?, ?, ?, ?, ?, ?, ?, 'pending')
-  `).run(ref.fromNodeId, ref.referenceName, ref.referenceKind, ref.line, ref.col, ref.filePath, ref.moduleSpecifier ?? null);
-}
-
-/** Delete every unresolved reference originating from a file. */
-export function deleteUnresolvedRefsForFile(db, filePath) {
-  db.prepare("DELETE FROM unresolved_refs WHERE file_path = ?").run(filePath);
-}
-
-/** Summary counters used by /ast status. */
-export function stats(db) {
-  const files = db.prepare("SELECT COUNT(*) AS count FROM files").get().count;
-  const nodes = db.prepare("SELECT COUNT(*) AS count FROM nodes").get().count;
-  const edges = db.prepare("SELECT COUNT(*) AS count FROM edges").get().count;
-  const unresolved = db.prepare("SELECT COUNT(*) AS count FROM unresolved_refs WHERE status = 'pending'").get().count;
-  return { files, nodes, edges, unresolved };
-}
-
-/** Store a provenance metadata value. */
-export function setMetadata(db, key, value) {
-  db.prepare(`
-    INSERT INTO project_metadata(key, value, updated_at) VALUES (?, ?, ?)
-    ON CONFLICT(key) DO UPDATE SET value = excluded.value, updated_at = excluded.updated_at
-  `).run(key, value, Date.now());
-}
-
-/** Read a provenance metadata value, or undefined when absent. */
-export function getMetadata(db, key) {
-  const row = db.prepare("SELECT value FROM project_metadata WHERE key = ?").get(key);
-  return row?.value;
-}

package/dist/astgraph/extract.mjs

@@ -1,239 +0,0 @@
-// astgraph TypeScript/JavaScript structural extractor.
-// Walks the tree-sitter syntax tree and produces:
-//   nodes — functions, classes, methods, interfaces, type aliases, variables
-//   edges — intra-file calls / extends / implements
-//   refs  — cross-file references (imports + unresolved calls/heritage) resolved later
-//
-// Lines are 1-based in all outputs so they match editor/Read tool coordinates.
-
-import { parseSource } from "./grammars.mjs";
-
-const DECLARATION_KINDS = new Map([
-  ["function_declaration", "function"],
-  ["class_declaration", "class"],
-  ["interface_declaration", "interface"],
-  ["type_alias_declaration", "type"],
-]);
-
-/** Extract the doc text from a `/** ... */` or `// ...` comment node. */
-function docstringFrom(commentNode) {
-  if (commentNode === null || commentNode.type !== "comment") return undefined;
-  const text = commentNode.text;
-  if (text.startsWith("/**") || text.startsWith("/*!")) {
-    return text
-      .replace(/^\/\*+[!*]?/, "")
-      .replace(/\*+\/$/, "")
-      .split("\n")
-      .map((line) => line.replace(/^\s*\*+\s?/, "").trim())
-      .filter((line) => line.length > 0)
-      .join("\n") || undefined;
-  }
-  if (text.startsWith("//")) return text.replace(/^\/\/+\s?/, "").trim() || undefined;
-  return undefined;
-}
-
-function firstLine(text, limit = 240) {
-  const line = text.split("\n")[0].trim();
-  return line.length > limit ? `${line.slice(0, limit)}…` : line;
-}
-
-function signatureOf(node) {
-  const text = node.text;
-  const brace = text.indexOf("{");
-  const head = brace >= 0 ? text.slice(0, brace) : text.split("\n")[0];
-  return firstLine(head.replace(/\s+/g, " "));
-}
-
-function nameOf(node) {
-  const nameNode = node.childForFieldName("name");
-  return nameNode === null ? undefined : nameNode.text;
-}
-
-function isAsync(node) {
-  return node.children.some((child) => child.type === "async");
-}
-
-export async function extract(source, filePath, language) {
-  const tree = await parseSource(source, language);
-
-  /** @type {Array<object>} */
-  const nodes = [];
-  /** @type {Array<object>} */
-  const edges = [];
-  /** @type {Array<object>} */
-  const refs = [];
-  /** Map local symbol name → node id (methods registered as name only for call matching). */
-  const localNames = new Map();
-  /** Map imported binding name → module specifier. */
-  const imports = new Map();
-  /** Map declaration AST node id → node record, for enclosing-scope lookup.
-   *  Keyed by numeric node.id because web-tree-sitter returns fresh node object
-   *  instances on every navigation (parent/childForFieldName), so object identity
-   *  cannot be used as a Map key. */
-  const scopeByNodeId = new Map();
-
-  const makeNode = (astNode, kind, name, qualifiedName, exported, extra = {}) => {
-    // The docstring lives on the top-level statement's preceding sibling (comments
-    // attach to program/export_statement, not to the inner declaration node).
-    const docSource = extra.docSource ?? astNode;
-    const record = {
-      id: `${filePath}#${qualifiedName}`,
-      kind,
-      name,
-      qualifiedName,
-      filePath,
-      language,
-      startLine: astNode.startPosition.row + 1,
-      endLine: astNode.endPosition.row + 1,
-      signature: signatureOf(astNode),
-      docstring: docstringFrom(docSource.previousSibling),
-      isExported: exported,
-      isAsync: extra.isAsync ?? false,
-      ...extra,
-    };
-    delete record.docSource;
-    nodes.push(record);
-    scopeByNodeId.set(astNode.id, record);
-    if (!localNames.has(name)) localNames.set(name, record.id);
-    return record;
-  };
-
-  const collectDeclaration = (statement, exported) => {
-    const inner = statement.type === "export_statement" ? statement.namedChildren.find((child) => DECLARATION_KINDS.has(child.type) || child.type === "lexical_declaration" || child.type === "abstract_class_declaration") : statement;
-    const target = inner ?? statement;
-    const kind = DECLARATION_KINDS.get(target.type);
-    if (kind !== undefined) {
-      const name = nameOf(target);
-      if (name !== undefined) {
-        // Comments attach to the top-level statement (export_statement / declaration),
-        // so the docstring is read from the statement's preceding sibling.
-        makeNode(target, kind, name, name, exported, {
-          ...(kind === "function" ? { isAsync: isAsync(target) } : {}),
-          docSource: statement,
-        });
-        if (target.type === "class_declaration" || target.type === "abstract_class_declaration") collectClassMembers(target, exported);
-        collectHeritage(target, exported);
-      }
-      return;
-    }
-    if (target.type === "lexical_declaration") {
-      for (const declarator of target.namedChildren.filter((child) => child.type === "variable_declarator")) {
-        const name = declarator.childForFieldName("name")?.text;
-        const value = declarator.childForFieldName("value");
-        if (name === undefined || value === null) continue;
-        const isFunction = value.type === "arrow_function" || value.type === "function_expression" || value.type === "function";
-        makeNode(declarator, isFunction ? "function" : "variable", name, name, exported, {
-          isAsync: isFunction && value.children.some((child) => child.type === "async"),
-          docSource: statement,
-        });
-      }
-    }
-  };
-
-  const collectClassMembers = (classNode, classExported) => {
-    const className = nameOf(classNode);
-    const body = classNode.childForFieldName("body");
-    if (body === null) return;
-    for (const member of body.namedChildren) {
-      if (member.type !== "method_definition") continue;
-      const name = member.childForFieldName("name")?.text;
-      if (name === undefined) continue;
-      const qualified = `${className}.${name}`;
-      makeNode(member, "method", name, qualified, classExported, { isAsync: isAsync(member) });
-    }
-  };
-
-  const collectHeritage = (classNode, _exported) => {
-    const heritage = classNode.namedChildren.find((child) => child.type === "class_heritage");
-    if (heritage === undefined) return;
-    const sourceName = nameOf(classNode);
-    if (sourceName === undefined) return;
-    const sourceId = `${filePath}#${sourceName}`;
-    for (const clause of heritage.namedChildren) {
-      const kind = clause.type === "extends_clause" ? "extends" : clause.type === "implements_clause" ? "implements" : undefined;
-      if (kind === undefined) continue;
-      for (const typeNode of clause.namedChildren.filter((child) => child.type === "type_identifier" || child.type === "identifier")) {
-        const typeName = typeNode.text;
-        if (localNames.has(typeName)) {
-          edges.push({ source: sourceId, target: localNames.get(typeName), kind, line: typeNode.startPosition.row + 1, col: typeNode.startPosition.column });
-        } else {
-          refs.push({
-            fromNodeId: sourceId, referenceName: typeName, referenceKind: kind,
-            line: typeNode.startPosition.row + 1, col: typeNode.startPosition.column,
-            filePath, moduleSpecifier: undefined,
-          });
-        }
-      }
-    }
-  };
-
-  // Pass 1 — top-level declarations and imports.
-  for (const statement of tree.rootNode.namedChildren) {
-    if (statement.type === "import_statement") {
-      const sourceNode = statement.childForFieldName("source");
-      const moduleSpecifier = sourceNode === null ? undefined : sourceNode.text.replace(/^["']|["']$/g, "");
-      for (const clause of statement.namedChildren.filter((child) => child.type === "import_clause")) {
-        for (const named of clause.namedChildren.filter((child) => child.type === "named_imports")) {
-          for (const specifier of named.namedChildren.filter((child) => child.type === "import_specifier")) {
-            const nameNode = specifier.childForFieldName("name");
-            if (nameNode === null) continue;
-            const importedName = nameNode.text;
-            imports.set(importedName, moduleSpecifier);
-            refs.push({
-              fromNodeId: null, referenceName: importedName, referenceKind: "import",
-              line: nameNode.startPosition.row + 1, col: nameNode.startPosition.column,
-              filePath, moduleSpecifier,
-            });
-          }
-        }
-        const defaultName = clause.namedChildren.find((child) => child.type === "identifier");
-        if (defaultName !== undefined) {
-          imports.set(defaultName.text, moduleSpecifier);
-          refs.push({
-            fromNodeId: null, referenceName: defaultName.text, referenceKind: "import",
-            line: defaultName.startPosition.row + 1, col: defaultName.startPosition.column,
-            filePath, moduleSpecifier,
-          });
-        }
-      }
-      continue;
-    }
-    collectDeclaration(statement, statement.type === "export_statement");
-  }
-
-  // Pass 2 — call expressions anywhere in the file.
-  const enclosingScope = (node) => {
-    let cursor = node.parent;
-    while (cursor !== null) {
-      const record = scopeByNodeId.get(cursor.id);
-      if (record !== undefined) return record;
-      cursor = cursor.parent;
-    }
-    return undefined;
-  };
-
-  const visitCalls = (node) => {
-    if (node.type === "call_expression") {
-      const callee = node.childForFieldName("function");
-      if (callee !== null && callee.type === "identifier") {
-        const name = callee.text;
-        const scope = enclosingScope(node);
-        if (scope !== undefined) {
-          if (localNames.has(name) && localNames.get(name) !== scope.id) {
-            edges.push({ source: scope.id, target: localNames.get(name), kind: "calls", line: callee.startPosition.row + 1, col: callee.startPosition.column });
-          } else if (!localNames.has(name)) {
-            refs.push({
-              fromNodeId: scope.id, referenceName: name, referenceKind: "call",
-              line: callee.startPosition.row + 1, col: callee.startPosition.column,
-              filePath, moduleSpecifier: imports.get(name),
-            });
-          }
-        }
-      }
-    }
-    for (const child of node.namedChildren) visitCalls(child);
-  };
-  visitCalls(tree.rootNode);
-
-  return { nodes, edges, refs, imports: Object.fromEntries(imports) };
-}

package/dist/astgraph/flavor-plugin.json

@@ -1,25 +0,0 @@
-{
-  "name": "astgraph",
-  "version": "0.1.0",
-  "apiVersion": "1",
-  "main": "index.js",
-  "permissions": [],
-  "contributes": {
-    "commands": [
-      { "name": "ast" }
-    ],
-    "tools": [
-      { "name": "ast_search" },
-      { "name": "ast_callers" },
-      { "name": "ast_callees" },
-      { "name": "ast_impact" },
-      { "name": "ast_context" }
-    ],
-    "skillRoots": [],
-    "modelAdapters": [],
-    "hooks": [
-      { "name": "SessionStart" },
-      { "name": "PostToolUse" }
-    ]
-  }
-}

package/dist/astgraph/grammars.mjs

@@ -1,68 +0,0 @@
-// astgraph grammar loading — web-tree-sitter WASM runtime, vendored under ./vendor.
-// Resolves the vendor directory relative to this module so it works both from the
-// source tree (src/init/astgraph) and the copied plugin root (.flavor/plugins/astgraph).
-
-import { dirname, join } from "node:path";
-import { fileURLToPath } from "node:url";
-
-import { Language, Parser } from "./vendor/tree-sitter.js";
-
-const MODULE_DIR = dirname(fileURLToPath(import.meta.url));
-export const VENDOR_DIR = join(MODULE_DIR, "vendor");
-
-const GRAMMAR_FILES = {
-  typescript: "tree-sitter-typescript.wasm",
-  tsx: "tree-sitter-tsx.wasm",
-  javascript: "tree-sitter-javascript.wasm",
-  jsx: "tree-sitter-javascript.wasm",
-};
-
-export const SUPPORTED_LANGUAGES = Object.keys(GRAMMAR_FILES);
-
-export function languageForFile(path) {
-  const lower = path.toLowerCase();
-  if (lower.endsWith(".tsx")) return "tsx";
-  if (lower.endsWith(".ts") || lower.endsWith(".mts") || lower.endsWith(".cts")) return "typescript";
-  if (lower.endsWith(".jsx")) return "jsx";
-  if (lower.endsWith(".js") || lower.endsWith(".mjs") || lower.endsWith(".cjs")) return "javascript";
-  return undefined;
-}
-
-let initialized = false;
-let initPromise;
-const languageCache = new Map();
-const parserCache = new Map();
-
-/** Initialize the WASM runtime once. Safe to call repeatedly. */
-export async function initGrammars() {
-  if (initialized) return;
-  initPromise ??= Parser.init({ locateFile: (name) => join(VENDOR_DIR, name) }).then(() => {
-    initialized = true;
-  });
-  await initPromise;
-}
-
-/** Load (and cache) a grammar for one of the supported languages. */
-export async function loadLanguage(language) {
-  const file = GRAMMAR_FILES[language];
-  if (file === undefined) throw new Error(`astgraph: unsupported language "${language}"`);
-  await initGrammars();
-  let languageInstance = languageCache.get(language);
-  if (languageInstance === undefined) {
-    languageInstance = await Language.load(join(VENDOR_DIR, file));
-    languageCache.set(language, languageInstance);
-  }
-  return languageInstance;
-}
-
-/** Parse source text and return the syntax tree. */
-export async function parseSource(source, language) {
-  const languageInstance = await loadLanguage(language);
-  let parser = parserCache.get(language);
-  if (parser === undefined) {
-    parser = new Parser();
-    parser.setLanguage(languageInstance);
-    parserCache.set(language, parser);
-  }
-  return parser.parse(source);
-}