mcp-code-context 2.2.0 → 2.4.0

package/src/ast/writers/dartWriter.ts

@@ -1,648 +0,0 @@
-/**
- * dartWriter.ts — Dart Surgical Writer
- *
- * Uses the same brace-counting + regex engine as dartCompressor.ts
- * to locate symbol boundaries, then performs line-splice operations.
- *
- * Dart code structure is brace-delimited like most C-style languages,
- * so we reuse the proven boundary detection from the read side.
- *
- * Supports: .dart
- */
-
-// ─── Types ──────────────────────────────────────────────────────────
-
-export interface WriteResult {
-  success: boolean;
-  newContent: string;
-  symbolsAffected: string[];
-  error?: string;
-}
-
-export type InsertPosition = "before" | "after" | "inside_start" | "inside_end";
-
-// ─── Regex Patterns ─────────────────────────────────────────────────
-
-const DECLARATION_REGEX =
-  /^(abstract\s+)?(class|mixin|extension|enum)\s+(\w+)/;
-
-const IMPORT_REGEX = /^\s*import\s+['"]/;
-
-// ─── Replace Symbol ─────────────────────────────────────────────────
-
-/**
- * Replace the full text of a named symbol in a Dart file.
- * When className is provided, scopes the search to within that class only.
- */
-export function replaceDartSymbol(
-  content: string,
-  symbolName: string,
-  newContent: string,
-  className?: string,
-): WriteResult {
-  // Normalize CRLF to LF for consistent processing
-  content = content.replace(/\r\n/g, "\n");
-  newContent = newContent.replace(/\r\n/g, "\n");
-  
-  const lines = content.split("\n");
-  const range = findSymbolRange(lines, symbolName, className);
-
-  if (!range) {
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: className
-        ? `Symbol "${symbolName}" not found in class "${className}" in Dart file`
-        : `Symbol "${symbolName}" not found in Dart file`,
-    };
-  }
-
-  const { start, end } = range;
-
-  // Detect indentation of original
-  const originalIndent = getIndent(lines[start]);
-  const reindented = reindentCode(newContent.trim(), originalIndent);
-
-  // Splice
-  const before = lines.slice(0, start);
-  const after = lines.slice(end + 1);
-  const result = [...before, reindented, ...after].join("\n");
-
-  // Basic validation: check brace balance
-  if (!hasBraceBalance(result)) {
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: `Replacement produced unbalanced braces. Aborting.`,
-    };
-  }
-
-  return {
-    success: true,
-    newContent: result,
-    symbolsAffected: [symbolName],
-  };
-}
-
-// ─── Insert Code ────────────────────────────────────────────────────
-
-/**
- * Insert new code at a precise location relative to an anchor symbol.
- * When className is provided, scopes the anchor search to within that class only.
- */
-export function insertDartCode(
-  content: string,
-  code: string,
-  anchorSymbol: string | null,
-  position: InsertPosition,
-  className?: string,
-): WriteResult {
-  // Normalize CRLF to LF for consistent processing
-  content = content.replace(/\r\n/g, "\n");
-  code = code.replace(/\r\n/g, "\n");
-  
-  const lines = content.split("\n");
-
-  if (!anchorSymbol) {
-    // No anchor — append at end of file
-    lines.push("", code.trim(), "");
-    return {
-      success: true,
-      newContent: lines.join("\n"),
-      symbolsAffected: [],
-    };
-  }
-
-  const range = findSymbolRange(lines, anchorSymbol, className);
-  if (!range) {
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: className
-        ? `Anchor symbol "${anchorSymbol}" not found in class "${className}"`
-        : `Anchor symbol "${anchorSymbol}" not found`,
-    };
-  }
-
-  const { start, end } = range;
-  const indent = getIndent(lines[start]);
-
-  switch (position) {
-    case "before": {
-      const reindented = reindentCode(code.trim(), indent);
-      lines.splice(start, 0, reindented, "");
-      break;
-    }
-    case "after": {
-      const reindented = reindentCode(code.trim(), indent);
-      lines.splice(end + 1, 0, "", reindented);
-      break;
-    }
-    case "inside_start": {
-      // Find opening brace
-      const braceIdx = findBraceLine(lines, start);
-      if (braceIdx === -1) {
-        return {
-          success: false,
-          newContent: content,
-          symbolsAffected: [],
-          error: `Cannot find opening brace for "${anchorSymbol}"`,
-        };
-      }
-      const innerIndent = indent + "  "; // Dart convention: 2-space indent
-      const innerCode = reindentCode(code.trim(), innerIndent);
-      lines.splice(braceIdx + 1, 0, innerCode, "");
-      break;
-    }
-    case "inside_end": {
-      const innerIndent = indent + "  ";
-      const innerCode = reindentCode(code.trim(), innerIndent);
-      lines.splice(end, 0, "", innerCode);
-      break;
-    }
-  }
-
-  return {
-    success: true,
-    newContent: lines.join("\n"),
-    symbolsAffected: [anchorSymbol],
-  };
-}
-
-// ─── Rename Symbol ──────────────────────────────────────────────────
-
-/**
- * Rename all occurrences of a symbol within a Dart file.
- */
-export function renameDartSymbol(
-  content: string,
-  oldName: string,
-  newName: string,
-): WriteResult {
-  // Normalize CRLF to LF for consistent processing
-  content = content.replace(/\r\n/g, "\n");
-  
-  const lines = content.split("\n");
-  const range = findSymbolRange(lines, oldName);
-
-  if (!range) {
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: `Symbol "${oldName}" not found for renaming`,
-    };
-  }
-
-  // Whole-word replacement
-  const escaped = oldName.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
-  const regex = new RegExp(`\\b${escaped}\\b`, "g");
-  const result = content.replace(regex, newName);
-
-  return {
-    success: true,
-    newContent: result,
-    symbolsAffected: [oldName],
-  };
-}
-
-// ─── Remove Symbol ──────────────────────────────────────────────────
-
-/**
- * Remove a named symbol from a Dart file.
- * When className is provided, scopes the search to within that class only.
- */
-export function removeDartSymbol(
-  content: string,
-  symbolName: string,
-  className?: string,
-): WriteResult {
-  // Normalize CRLF to LF for consistent processing
-  content = content.replace(/\r\n/g, "\n");
-  
-  const lines = content.split("\n");
-  const range = findSymbolRange(lines, symbolName, className);
-
-  if (!range) {
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: className
-        ? `Symbol "${symbolName}" not found in class "${className}" for removal`
-        : `Symbol "${symbolName}" not found for removal`,
-    };
-  }
-
-  lines.splice(range.start, range.end - range.start + 1);
-
-  let result = lines.join("\n");
-  result = result.replace(/\n{3,}/g, "\n\n");
-
-  return {
-    success: true,
-    newContent: result,
-    symbolsAffected: [symbolName],
-  };
-}
-
-// ─── Symbol Finder ──────────────────────────────────────────────────
-
-interface SymbolRange {
-  start: number; // 0-indexed line
-  end: number;   // 0-indexed line (inclusive)
-}
-
-/**
- * Find the start and end line indices of a named symbol in a Dart file.
- * Includes leading doc comments and annotations.
- * When className is provided, scopes method search to within that class only.
- */
-function findSymbolRange(
-  lines: string[],
-  symbolName: string,
-  className?: string,
-): SymbolRange | null {
-  // If className is provided, scope the search to within that class
-  if (className) {
-    for (let i = 0; i < lines.length; i++) {
-      const trimmed = lines[i].trim();
-      const declMatch = DECLARATION_REGEX.exec(trimmed);
-      if (declMatch && declMatch[3] === className) {
-        // Found the class — determine body boundaries
-        const classEnd = findBlockEnd(lines, i);
-
-        // Search only within the class body for the method
-        for (let j = i + 1; j < classEnd; j++) {
-          const bodyTrimmed = lines[j].trim();
-          const escaped = symbolName.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
-          const methodMatch = bodyTrimmed.match(
-            new RegExp(`\\b${escaped}\\s*[<(]`),
-          );
-          if (
-            methodMatch &&
-            !DECLARATION_REGEX.test(bodyTrimmed) &&
-            !IMPORT_REGEX.test(bodyTrimmed) &&
-            (bodyTrimmed.includes("(") || bodyTrimmed.includes("<"))
-          ) {
-            if (isMethodOrConstructor(bodyTrimmed) || isTopLevelFunction(bodyTrimmed)) {
-              const start = findDocStart(lines, j);
-              const end = findBlockEnd(lines, j);
-              return { start, end };
-            }
-          }
-        }
-        return null; // Method not found within class
-      }
-    }
-    return null; // Class not found
-  }
-
-  // Original behavior when no className is provided
-  for (let i = 0; i < lines.length; i++) {
-    const trimmed = lines[i].trim();
-
-    // Match class-level declarations
-    const declMatch = DECLARATION_REGEX.exec(trimmed);
-    if (declMatch && declMatch[3] === symbolName) {
-      const start = findDocStart(lines, i);
-      const end = findBlockEnd(lines, i);
-      return { start, end };
-    }
-
-    // Match top-level functions
-    if (isTopLevelFunction(trimmed)) {
-      const funcName = extractFunctionName(trimmed);
-      if (funcName === symbolName) {
-        const start = findDocStart(lines, i);
-        const end = findBlockEnd(lines, i);
-        return { start, end };
-      }
-    }
-
-    // Match methods inside classes
-    const escaped = symbolName.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
-    const methodMatch = trimmed.match(
-      new RegExp(`\\b${escaped}\\s*[<(]`),
-    );
-    if (
-      methodMatch &&
-      !DECLARATION_REGEX.test(trimmed) &&
-      !IMPORT_REGEX.test(trimmed) &&
-      (trimmed.includes("(") || trimmed.includes("<"))
-    ) {
-      if (isMethodOrConstructor(trimmed) || isTopLevelFunction(trimmed)) {
-        const start = findDocStart(lines, i);
-        const end = findBlockEnd(lines, i);
-        return { start, end };
-      }
-    }
-  }
-
-  return null;
-}
-
-/**
- * Walk backward from a declaration to find leading doc comments/annotations.
- */
-function findDocStart(lines: string[], declLine: number): number {
-  let start = declLine;
-  for (let k = declLine - 1; k >= 0; k--) {
-    const t = lines[k].trim();
-    if (t.startsWith("///") || t.startsWith("@") || t === "") {
-      if (t !== "") start = k;
-    } else {
-      break;
-    }
-  }
-  return start;
-}
-
-/**
- * Find the end of a block (matching closing brace).
- */
-function findBlockEnd(lines: string[], startIdx: number): number {
-  const line = lines[startIdx].trim();
-
-  // Single-line declaration (no body)
-  if (line.endsWith(";") && !line.includes("{")) {
-    return startIdx;
-  }
-
-  // Find opening brace
-  let braceLineIdx = startIdx;
-  while (
-    braceLineIdx < lines.length &&
-    !lines[braceLineIdx].includes("{")
-  ) {
-    braceLineIdx++;
-  }
-
-  if (braceLineIdx >= lines.length) return lines.length - 1;
-
-  const result = findMatchingBraceStateAware(lines, braceLineIdx);
-  return result !== -1 ? result : lines.length - 1;
-}
-
-/**
- * Advanced brace matcher that ignores strings and comments.
- * Returns the line index of the matching closing brace.
- * Starts counting from the FIRST brace that appears after a closing parenthesis
- * to avoid counting braces in method parameters (e.g., {int? cart}).
- */
-function findMatchingBraceStateAware(lines: string[], startLine: number): number {
-  let depth = 0;
-  let inString = false;
-  let stringDelim = "";
-  let inTripleQuote = false;
-  let inComment = false;
-  let inMultiLineComment = false;
-  let foundMethodBodyBrace = false;
-  let lastCloseParen = -1;
-
-  for (let i = startLine; i < lines.length; i++) {
-    const line = lines[i];
-    for (let j = 0; j < line.length; j++) {
-      const char = line[j];
-      const prev = j > 0 ? line[j - 1] : "";
-      const next = j < line.length - 1 ? line[j + 1] : "";
-
-      // Handle Escape
-      if (prev === "\\" && (inString || inTripleQuote)) continue;
-
-      // Handle Comments
-      if (!inString && !inTripleQuote) {
-        if (!inMultiLineComment && char === "/" && next === "/") {
-          inComment = true; // Skip rest of line
-          break;
-        }
-        if (!inMultiLineComment && char === "/" && next === "*") {
-          inMultiLineComment = true;
-          j++;
-          continue;
-        }
-        if (inMultiLineComment && char === "*" && next === "/") {
-          inMultiLineComment = false;
-          j++;
-          continue;
-        }
-      }
-
-      if (inComment || inMultiLineComment) continue;
-
-      // Handle Multi-line Triple Quotes (''' or """)
-      if (!inString && !inMultiLineComment) {
-        const isTriple = line.slice(j, j + 3) === "'''" || line.slice(j, j + 3) === '"""';
-        if (isTriple) {
-          if (!inTripleQuote) {
-            inTripleQuote = true;
-            stringDelim = line.slice(j, j + 3);
-            j += 2;
-            continue;
-          } else if (line.slice(j, j + 3) === stringDelim) {
-            inTripleQuote = false;
-            stringDelim = "";
-            j += 2;
-            continue;
-          }
-        }
-      }
-
-      if (inTripleQuote) continue;
-
-      // Handle Basic Strings (' or ")
-      if (!inMultiLineComment && (char === "'" || char === '"')) {
-        if (!inString) {
-          inString = true;
-          stringDelim = char;
-        } else if (char === stringDelim) {
-          inString = false;
-          stringDelim = "";
-        }
-        continue;
-      }
-
-      if (inString) continue;
-
-      // Track closing parenthesis to find where method signature ends
-      if (char === ")") {
-        lastCloseParen = j;
-      }
-
-      // Real Braces
-      if (char === "{") {
-        // Only start counting after we've seen a closing paren (method signature end)
-        if (!foundMethodBodyBrace && lastCloseParen !== -1) {
-          foundMethodBodyBrace = true;
-          depth = 1; // This is the opening brace of the method body
-        } else if (foundMethodBodyBrace) {
-          depth++;
-        }
-        // Ignore braces before the method body (e.g., in parameters)
-      }
-      else if (char === "}") {
-        if (foundMethodBodyBrace) {
-          depth--;
-          if (depth === 0) return i;
-        }
-      }
-    }
-    inComment = false; // Reset line comment
-  }
-
-  return -1;
-}
-
-// ─── Utility Functions ──────────────────────────────────────────────
-
-function isTopLevelFunction(trimmed: string): boolean {
-  // Matches: void foo(), Future<T> bar(), static Type baz()
-  // But NOT: class/mixin/extension/enum declarations
-  if (DECLARATION_REGEX.test(trimmed)) return false;
-  if (IMPORT_REGEX.test(trimmed)) return false;
-  if (trimmed.startsWith("//") || trimmed.startsWith("///")) return false;
-  if (trimmed.startsWith("@")) return false;
-
-  // Look for pattern: [returnType] functionName( or functionName<
-  return /^(?:(?:static|final|const|async|Future|Stream|void|int|double|String|bool|List|Map|Set|dynamic|var)\s+)*\w+\s*[<(]/.test(
-    trimmed,
-  );
-}
-
-function extractFunctionName(trimmed: string): string {
-  // Remove return type and modifiers, extract the name before ( or <
-  const match = trimmed.match(/(\w+)\s*[<(]/);
-  return match ? match[1] : "";
-}
-
-function isMethodOrConstructor(trimmed: string): boolean {
-  // Similar to top-level function but may be indented
-  if (DECLARATION_REGEX.test(trimmed)) return false;
-  if (trimmed.startsWith("//") || trimmed.startsWith("///")) return false;
-  if (trimmed.startsWith("@")) return false;
-
-  return /^(?:(?:static|final|const|async|override|abstract|Future|Stream|void|int|double|String|bool|List|Map|Set|dynamic|var)\s+)*\w+\s*[<(]/.test(
-    trimmed,
-  );
-}
-
-function getIndent(line: string): string {
-  const match = line.match(/^(\s*)/);
-  return match ? match[1] : "";
-}
-
-function reindentCode(code: string, targetIndent: string): string {
-  // Normalize CRLF to LF
-  code = code.replace(/\r\n/g, "\n");
-  
-  const lines = code.split("\n");
-  if (lines.length === 0) return code;
-
-  let minIndent = Infinity;
-  for (const line of lines) {
-    if (line.trim() === "") continue;
-    const indent = line.match(/^(\s*)/)?.[1]?.length ?? 0;
-    if (indent < minIndent) minIndent = indent;
-  }
-
-  if (minIndent === Infinity) minIndent = 0;
-
-  return lines
-    .map((line) => {
-      if (line.trim() === "") return "";
-      return targetIndent + line.slice(minIndent);
-    })
-    .join("\n");
-}
-
-function findBraceLine(lines: string[], startIdx: number): number {
-  for (let i = startIdx; i < lines.length; i++) {
-    if (lines[i].includes("{")) return i;
-  }
-  return -1;
-}
-
-function hasBraceBalance(content: string): boolean {
-  // Normalize CRLF to LF for consistent processing
-  content = content.replace(/\r\n/g, "\n");
-  
-  const lines = content.split("\n");
-  let depth = 0;
-  let inString = false;
-  let stringDelim = "";
-  let inTripleQuote = false;
-  let inComment = false;
-  let inMultiLineComment = false;
-
-  for (let i = 0; i < lines.length; i++) {
-    const line = lines[i];
-    for (let j = 0; j < line.length; j++) {
-      const char = line[j];
-      const prev = j > 0 ? line[j - 1] : "";
-      const next = j < line.length - 1 ? line[j + 1] : "";
-
-      if (prev === "\\" && (inString || inTripleQuote)) continue;
-
-      if (!inString && !inTripleQuote) {
-        if (!inMultiLineComment && char === "/" && next === "/") {
-          inComment = true;
-          break;
-        }
-        if (!inMultiLineComment && char === "/" && next === "*") {
-          inMultiLineComment = true;
-          j++;
-          continue;
-        }
-        if (inMultiLineComment && char === "*" && next === "/") {
-          inMultiLineComment = false;
-          j++;
-          continue;
-        }
-      }
-
-      if (inComment || inMultiLineComment) continue;
-
-      if (!inString && !inMultiLineComment) {
-        const isTriple = line.slice(j, j + 3) === "'''" || line.slice(j, j + 3) === '"""';
-        if (isTriple) {
-          if (!inTripleQuote) {
-            inTripleQuote = true;
-            stringDelim = line.slice(j, j + 3);
-            j += 2;
-            continue;
-          } else if (line.slice(j, j + 3) === stringDelim) {
-            inTripleQuote = false;
-            stringDelim = "";
-            j += 2;
-            continue;
-          }
-        }
-      }
-      if (inTripleQuote) continue;
-
-      if (!inMultiLineComment && (char === "'" || char === '"')) {
-        if (!inString) {
-          inString = true;
-          stringDelim = char;
-        } else if (char === stringDelim) {
-          inString = false;
-          stringDelim = "";
-        }
-        continue;
-      }
-      if (inString) continue;
-
-      if (char === "{") depth++;
-      else if (char === "}") {
-        depth--;
-        if (depth < 0) return false;
-      }
-    }
-    inComment = false;
-  }
-
-  return depth === 0;
-}