mcp-code-context 2.2.0 → 2.3.1

package/src/ast/writers/tsWriter.ts

@@ -1,418 +0,0 @@
-/**
- * tsWriter.ts — TypeScript / JavaScript Surgical Writer
- *
- * Uses ts-morph to perform AST-aware code modifications on TypeScript
- * and JavaScript files. This is the most robust writer because ts-morph
- * provides a full read-write AST with source manipulation capabilities.
- *
- * Supports: .ts, .tsx, .js, .jsx, .mts, .mjs, .cts, .cjs
- */
-
-import {
-  Project,
-  type SourceFile,
-  type ClassDeclaration,
-  SyntaxKind,
-  IndentationText,
-} from "ts-morph";
-
-import * as path from "node:path";
-
-// ─── Types ──────────────────────────────────────────────────────────
-
-export interface WriteResult {
-  success: boolean;
-  newContent: string;
-  symbolsAffected: string[];
-  error?: string;
-}
-
-// ─── Singleton Project ──────────────────────────────────────────────
-
-let _project: Project | null = null;
-
-function getProject(): Project {
-  if (!_project) {
-    _project = new Project({
-      compilerOptions: {
-        allowJs: true,
-        jsx: 2, // React
-      },
-      useInMemoryFileSystem: true,
-    });
-  }
-  return _project;
-}
-
-function createSourceFile(content: string, filePath: string): SourceFile {
-  const project = getProject();
-  const ext = path.extname(filePath);
-  const tempName = `write_${Date.now()}_${Math.random().toString(36).slice(2)}${ext}`;
-  return project.createSourceFile(tempName, content, { overwrite: true });
-}
-
-function cleanupSourceFile(sf: SourceFile): void {
-  try {
-    sf.getProject().removeSourceFile(sf);
-  } catch {
-    // Ignore cleanup errors
-  }
-}
-
-// ─── Replace Symbol ─────────────────────────────────────────────────
-
-/**
- * Replace the full text of a named symbol in a TypeScript/JavaScript file.
- * The newContent should be the complete replacement (signature + body).
- * When className is provided, scopes the search to within that class only.
- */
-export function replaceTSSymbol(
-  content: string,
-  symbolName: string,
-  newContent: string,
-  filePath: string,
-  className?: string,
-): WriteResult {
-  const sf = createSourceFile(content, filePath);
-
-  try {
-    const node = findNode(sf, symbolName, className);
-    if (!node) {
-      return {
-        success: false,
-        newContent: content,
-        symbolsAffected: [],
-        error: className
-          ? `Symbol "${symbolName}" not found in class "${className}"`
-          : `Symbol "${symbolName}" not found in file`,
-      };
-    }
-
-    // Replace the node's full text with the new content
-    node.replaceWithText(newContent.trim());
-
-    const result = sf.getFullText();
-
-    // Validate that the result is parseable
-    const validationSf = createSourceFile(result, filePath);
-    const diagnostics = validationSf.getPreEmitDiagnostics();
-    const hasSyntaxErrors = diagnostics.some(
-      (d) => d.getCategory() === 1, // DiagnosticCategory.Error
-    );
-    cleanupSourceFile(validationSf);
-
-    if (hasSyntaxErrors) {
-      // Still return the result but mark potential syntax issues
-      // We don't reject because some "errors" are just type errors, not syntax
-    }
-
-    return {
-      success: true,
-      newContent: result,
-      symbolsAffected: [symbolName],
-    };
-  } catch (error: unknown) {
-    const msg = error instanceof Error ? error.message : String(error);
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: `Failed to replace symbol "${symbolName}": ${msg}`,
-    };
-  } finally {
-    cleanupSourceFile(sf);
-  }
-}
-
-// ─── Insert Code ────────────────────────────────────────────────────
-
-export type InsertPosition = "before" | "after" | "inside_start" | "inside_end";
-
-/**
- * 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 insertTSCode(
-  content: string,
-  code: string,
-  anchorSymbol: string | null,
-  position: InsertPosition,
-  filePath: string,
-  className?: string,
-): WriteResult {
-  const sf = createSourceFile(content, filePath);
-
-  try {
-    if (!anchorSymbol) {
-      // No anchor — append at end of file
-      sf.addStatements(code.trim());
-      return {
-        success: true,
-        newContent: sf.getFullText(),
-        symbolsAffected: [],
-      };
-    }
-
-    const node = findNode(sf, anchorSymbol, className);
-    if (!node) {
-      return {
-        success: false,
-        newContent: content,
-        symbolsAffected: [],
-        error: className
-          ? `Anchor symbol "${anchorSymbol}" not found in class "${className}"`
-          : `Anchor symbol "${anchorSymbol}" not found`,
-      };
-    }
-
-    const trimmedCode = code.trim();
-
-    switch (position) {
-      case "before": {
-        const startPos = node.getStart();
-        sf.insertText(startPos, trimmedCode + "\n\n");
-        break;
-      }
-      case "after": {
-        const endPos = node.getEnd();
-        sf.insertText(endPos, "\n\n" + trimmedCode);
-        break;
-      }
-      case "inside_start": {
-        // Only valid for classes/interfaces
-        const cls =
-          sf.getClass(anchorSymbol) || sf.getInterface(anchorSymbol);
-        if (!cls) {
-          return {
-            success: false,
-            newContent: content,
-            symbolsAffected: [],
-            error: `"inside_start" requires a class or interface. "${anchorSymbol}" is not one.`,
-          };
-        }
-        if ("insertMember" in cls) {
-          (cls as ClassDeclaration).insertMember(0, trimmedCode);
-        }
-        break;
-      }
-      case "inside_end": {
-        const cls2 =
-          sf.getClass(anchorSymbol) || sf.getInterface(anchorSymbol);
-        if (!cls2) {
-          return {
-            success: false,
-            newContent: content,
-            symbolsAffected: [],
-            error: `"inside_end" requires a class or interface. "${anchorSymbol}" is not one.`,
-          };
-        }
-        if ("addMember" in cls2) {
-          (cls2 as ClassDeclaration).addMember(trimmedCode);
-        }
-        break;
-      }
-    }
-
-    return {
-      success: true,
-      newContent: sf.getFullText(),
-      symbolsAffected: [anchorSymbol],
-    };
-  } catch (error: unknown) {
-    const msg = error instanceof Error ? error.message : String(error);
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: `Failed to insert code: ${msg}`,
-    };
-  } finally {
-    cleanupSourceFile(sf);
-  }
-}
-
-// ─── Rename Symbol ──────────────────────────────────────────────────
-
-/**
- * Rename all occurrences of a symbol within a single TypeScript/JavaScript file.
- * This performs a smart rename that understands scoping.
- */
-export function renameTSSymbol(
-  content: string,
-  oldName: string,
-  newName: string,
-  filePath: string,
-): WriteResult {
-  const sf = createSourceFile(content, filePath);
-
-  try {
-    const node = findNode(sf, oldName);
-    if (!node) {
-      return {
-        success: false,
-        newContent: content,
-        symbolsAffected: [],
-        error: `Symbol "${oldName}" not found for renaming`,
-      };
-    }
-
-    // Use regex-based replacement scoped to identifiers
-    // ts-morph rename in single-file mode: find all references within the file
-    const fullText = sf.getFullText();
-
-    // Smart rename: replace whole-word occurrences of oldName with newName
-    // This is not a naive replace — it respects word boundaries
-    const escaped = oldName.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
-    const regex = new RegExp(`\\b${escaped}\\b`, "g");
-    const newText = fullText.replace(regex, newName);
-
-    // Validate result is parseable
-    const validationSf = createSourceFile(newText, filePath);
-    cleanupSourceFile(validationSf);
-
-    return {
-      success: true,
-      newContent: newText,
-      symbolsAffected: [oldName],
-    };
-  } catch (error: unknown) {
-    const msg = error instanceof Error ? error.message : String(error);
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: `Failed to rename "${oldName}" to "${newName}": ${msg}`,
-    };
-  } finally {
-    cleanupSourceFile(sf);
-  }
-}
-
-// ─── Remove Symbol ──────────────────────────────────────────────────
-
-/**
- * Remove a named symbol from a TypeScript/JavaScript file.
- * Returns the file content without the symbol.
- * When className is provided, scopes the search to within that class only.
- */
-export function removeTSSymbol(
-  content: string,
-  symbolName: string,
-  filePath: string,
-  className?: string,
-): WriteResult {
-  const sf = createSourceFile(content, filePath);
-
-  try {
-    const node = findNode(sf, symbolName, className);
-    if (!node) {
-      return {
-        success: false,
-        newContent: content,
-        symbolsAffected: [],
-        error: className
-          ? `Symbol "${symbolName}" not found in class "${className}" for removal`
-          : `Symbol "${symbolName}" not found for removal`,
-      };
-    }
-
-    node.remove();
-
-    // Clean up double blank lines left behind
-    let result = sf.getFullText();
-    result = result.replace(/\n{3,}/g, "\n\n");
-
-    return {
-      success: true,
-      newContent: result,
-      symbolsAffected: [symbolName],
-    };
-  } catch (error: unknown) {
-    const msg = error instanceof Error ? error.message : String(error);
-    return {
-      success: false,
-      newContent: content,
-      symbolsAffected: [],
-      error: `Failed to remove symbol "${symbolName}": ${msg}`,
-    };
-  } finally {
-    cleanupSourceFile(sf);
-  }
-}
-
-// ─── Node Finder ────────────────────────────────────────────────────
-
-/**
- * Find a named declaration node in a source file.
- * Searches: functions, classes, interfaces, type aliases, enums,
- * class methods, getters, setters, and variable declarations.
- * When className is provided, scopes method/property search to within that class only.
- */
-function findNode(sf: SourceFile, symbolName: string, className?: string) {
-  // If className is provided, scope search to within that class
-  if (className) {
-    const cls = sf.getClass(className);
-    if (!cls) return null;
-
-    const method = cls.getMethod(symbolName);
-    if (method) return method;
-
-    const getter = cls.getGetAccessor(symbolName);
-    if (getter) return getter;
-
-    const setter = cls.getSetAccessor(symbolName);
-    if (setter) return setter;
-
-    const prop = cls.getProperty(symbolName);
-    if (prop) return prop;
-
-    return null;
-  }
-
-  // Top-level function
-  const func = sf.getFunction(symbolName);
-  if (func) return func;
-
-  // Class
-  const cls = sf.getClass(symbolName);
-  if (cls) return cls;
-
-  // Interface
-  const iface = sf.getInterface(symbolName);
-  if (iface) return iface;
-
-  // Type alias
-  const typeAlias = sf.getTypeAlias(symbolName);
-  if (typeAlias) return typeAlias;
-
-  // Enum
-  const enumDecl = sf.getEnum(symbolName);
-  if (enumDecl) return enumDecl;
-
-  // Methods inside classes
-  for (const c of sf.getClasses()) {
-    const method = c.getMethod(symbolName);
-    if (method) return method;
-
-    const getter = c.getGetAccessor(symbolName);
-    if (getter) return getter;
-
-    const setter = c.getSetAccessor(symbolName);
-    if (setter) return setter;
-
-    // Properties
-    const prop = c.getProperty(symbolName);
-    if (prop) return prop;
-  }
-
-  // Variable declarations (const, let, var — including arrow functions)
-  for (const varStmt of sf.getVariableStatements()) {
-    for (const decl of varStmt.getDeclarations()) {
-      if (decl.getName() === symbolName) {
-        return varStmt;
-      }
-    }
-  }
-
-  return null;
-}