MCP ts-morph Refactoring Tools

import type { Project, SourceFile, Statement, SyntaxKind } from "ts-morph"; import { Node } from "ts-morph"; import logger from "../../utils/logger"; import type { DependencyClassification, NeededExternalImports } from "../types"; import { classifyDependencies } from "./classify-dependencies"; import { collectNeededExternalImports } from "./collect-external-imports"; import { createSourceFileIfNotExists } from "./create-source-file-if-not-exists"; import { ensureExportsInOriginalFile } from "./ensure-exports-in-original-file"; import { findTopLevelDeclarationByName } from "./find-declaration"; import { generateNewSourceFileContent, prepareDeclarationStrings, } from "./generate-content/generate-new-source-file-content"; import { getInternalDependencies } from "./internal-dependencies"; import { removeOriginalSymbol } from "./remove-original-symbol"; import { updateImportsInReferencingFiles } from "./update-imports-in-referencing-files"; import { updateTargetFile } from "./update-target-file"; import { calculateRequiredImportMap } from "./generate-content/build-new-file-import-section"; /** * Statement を取得し、必要なら export キーワードを追加して文字列を返す。 * isInternalOnly が true の場合は export キーワードを付けない。 */ function getPotentiallyExportedStatement( stmt: Statement, isInternalOnly: boolean, ): string { const stmtText = stmt.getText(); // デフォルトエクスポートの場合はそのまま返す if (Node.isExportable(stmt) && stmt.isDefaultExport()) { return stmtText; } // 内部でのみ使用される依存関係の場合は export しない if (isInternalOnly) { // 元々 export されていた場合は削除する if (Node.isExportable(stmt) && stmt.isExported()) { return stmtText.replace(/^export\s+/, ""); } return stmtText; } // それ以外の場合 (移動対象の宣言、または外部からも参照される依存関係) は export を確認・追加 let isExported = false; if (Node.isExportable(stmt)) { isExported = stmt.isExported(); } if (!isExported) { return `export ${stmtText}`; } return stmtText; } /** * シンボル移動に必要な情報を収集する。 * 元ファイル、移動対象の宣言、分類済み依存関係、外部インポート情報を返す。 */ async function gatherMovePrerequisites( project: Project, originalFilePath: string, symbolToMove: string, declarationKind?: SyntaxKind, ): Promise<{ originalSourceFile: SourceFile; declaration: Statement; classifiedDependencies: DependencyClassification[]; neededExternalImports: NeededExternalImports; }> { // --- ステップ 1: 元ファイルの取得 --- const originalSourceFile = project.getSourceFile(originalFilePath); if (!originalSourceFile) { throw new Error(`Original source file not found: ${originalFilePath}`); } logger.debug(`元のファイルを発見: ${originalFilePath}`); // --- ステップ 2: 移動対象シンボルの特定 --- const declaration = findTopLevelDeclarationByName( originalSourceFile, symbolToMove, declarationKind, ); if (!declaration) { throw new Error( `Symbol "${symbolToMove}" not found in ${originalFilePath}`, ); } logger.debug(`シンボルの宣言を発見: ${symbolToMove}`); // デフォルトエクスポートは対象外とするチェック let isDefaultExported = false; if ( Node.isFunctionDeclaration(declaration) || Node.isClassDeclaration(declaration) || Node.isInterfaceDeclaration(declaration) || Node.isEnumDeclaration(declaration) ) { isDefaultExported = declaration.isDefaultExport(); } if (isDefaultExported) { throw new Error( "Default exports cannot be moved using this function. Please refactor manually or use file moving tools.", ); } // --- ステップ 3: 内部依存関係の取得 --- const internalDependencies = getInternalDependencies(declaration); logger.debug(`${internalDependencies.length}個の内部依存関係を発見。`); // --- ステップ 4: 依存関係の分類 --- const classifiedDependencies = classifyDependencies( declaration, internalDependencies, ); // --- ステップ 5: 外部依存関係の収集 --- const allDepsToMove = [ declaration, ...classifiedDependencies.map((dep) => dep.statement), ]; const neededExternalImports = collectNeededExternalImports( allDepsToMove, originalSourceFile, ); logger.debug(`必要な外部インポートを${neededExternalImports.size}個収集。`); return { originalSourceFile, declaration, classifiedDependencies, neededExternalImports, }; } /** * 新しいファイルの内容を生成し、ファイルを作成または上書きする。 */ function generateAndCreateNewFile( project: Project, declaration: Statement, classifiedDependencies: DependencyClassification[], originalFilePath: string, newFilePath: string, neededExternalImports: NeededExternalImports, ): void { // --- ステップ 6: 新しいファイルの内容を生成 --- const newFileContent = generateNewSourceFileContent( declaration, classifiedDependencies, originalFilePath, newFilePath, neededExternalImports, ); logger.debug("新しいファイルの内容を生成。"); // --- ステップ 7: 新しいソースファイルを作成 (または上書き) --- createSourceFileIfNotExists(project, newFilePath, newFileContent); logger.debug(`ソースファイルを作成または更新: ${newFilePath}`); } /** * 参照元のインポートパス更新、元のファイルからのシンボル削除、元のファイルのインポート修正を行う。 */ async function updateReferencesAndOriginalFile( project: Project, originalSourceFile: SourceFile, declaration: Statement, classifiedDependencies: DependencyClassification[], originalFilePath: string, newFilePath: string, symbolToMove: string, ): Promise<void> { // --- ステップ 8: 参照元のインポート更新 --- await updateImportsInReferencingFiles( project, originalFilePath, newFilePath, symbolToMove, ); logger.debug("参照元ファイルのインポートを更新。 "); // --- ステップ 9: 元のファイルからシンボルと依存関係を削除 --- const dependenciesToRemoveDeclarations = classifiedDependencies .filter( ( dep: DependencyClassification, ): dep is Extract<DependencyClassification, { type: "moveToNewFile" }> => dep.type === "moveToNewFile", ) .map((dep) => dep.statement); const allDeclarationsToRemove = [ declaration, ...dependenciesToRemoveDeclarations, ]; logger.debug( `削除する宣言: ${allDeclarationsToRemove.map((d) => `"${d.getText().substring(0, 80).replaceAll("\n", " ")}..."`).join(", ")}`, ); removeOriginalSymbol(originalSourceFile, allDeclarationsToRemove); logger.debug(`削除後の元のファイルの内容:\n${originalSourceFile.getText()}`); logger.debug("元のファイルからシンボルと依存関係を削除。 "); // --- ステップ 10: 移動元ファイルのインポート修正 (fixMissingImports) --- logger.debug( `元のファイルの不足しているインポートを修正試行: ${originalFilePath}`, ); originalSourceFile.fixMissingImports(); logger.debug("元のファイルの不足しているインポートの修正試行を完了。 "); // --- ステップ 10.5: 元ファイルの未使用インポートを整理 --- logger.debug(`元のファイルのインポートを整理試行: ${originalFilePath}`); originalSourceFile.organizeImports(); logger.debug("元のファイルのインポート整理試行を完了。"); } /** * 新しいファイルの内容を生成し、ファイルを作成または既存ファイルに追加する。 */ function generateAndAppendToNewFile( project: Project, declaration: Statement, classifiedDependencies: DependencyClassification[], originalFilePath: string, newFilePath: string, neededExternalImports: NeededExternalImports, ): void { logger.debug( `Generate/Append symbol to file: ${newFilePath} (from ${originalFilePath})`, ); // --- ステップ 1: 必要なインポート情報を計算 (外部 + 内部) --- const requiredImportMap = calculateRequiredImportMap( neededExternalImports, classifiedDependencies, newFilePath, originalFilePath, ); // --- ステップ 2: 追加する宣言の文字列を準備 --- const declarationStrings = prepareDeclarationStrings( declaration, classifiedDependencies, ); // --- ステップ 3: ターゲットファイルを取得または作成し、更新 --- const targetSourceFile = project.getSourceFile(newFilePath); if (targetSourceFile) { // --- 既存ファイルの場合: 新しい updateTargetFile でマージ --- logger.debug(`Target file exists. Updating: ${newFilePath}`); updateTargetFile(targetSourceFile, requiredImportMap, declarationStrings); } else { // --- 新規ファイルの場合: 元の generateNewSourceFileContent を使用 --- logger.debug(`Target file does not exist. Creating: ${newFilePath}`); const newFileContent = generateNewSourceFileContent( declaration, classifiedDependencies, originalFilePath, newFilePath, neededExternalImports, ); logger.debug("Generated new file content."); const newSourceFile = project.createSourceFile(newFilePath, newFileContent); logger.debug(`Created source file: ${newFilePath}`); newSourceFile.organizeImports(); // 新規ファイルでもインポート整理 logger.debug(`Organized imports for new file: ${newFilePath}`); } } /** * 指定されたシンボルを現在のファイルから別ファイル（なければ新規作詞）に移動します。 * ヘルパー関数は成功時に値を返し、失敗時に例外をスローします。 * * @param project ts-morph プロジェクトインスタンス * @param originalFilePath 元のファイルの絶対パス * @param targetFilePath 移動先ファイルの絶対パス * @param symbolToMove 移動するシンボルの名前 * @param declarationKind 移動するシンボルの種類 (オプション) * @returns Promise<void> 処理が完了したら解決される Promise * @throws Error - シンボルが見つからない、デフォルトエクスポート、AST 操作エラーなど */ export async function moveSymbolToFile( project: Project, originalFilePath: string, newFilePath: string, symbolToMove: string, declarationKind?: SyntaxKind, ): Promise<void> { logger.debug( `moveSymbolToFile 開始: Symbol='${symbolToMove}', From='${originalFilePath}', To='${newFilePath}'`, ); const { originalSourceFile, declaration, classifiedDependencies, neededExternalImports, } = await gatherMovePrerequisites( project, originalFilePath, symbolToMove, declarationKind, ); ensureExportsInOriginalFile(classifiedDependencies, originalFilePath); // --- ステップ 6 & 7: 新しいファイルの生成と作成/追加 --- generateAndAppendToNewFile( project, declaration, classifiedDependencies, originalFilePath, newFilePath, neededExternalImports, ); // --- ステップ 8, 9, 10: 参照更新と元のファイル整理 --- await updateReferencesAndOriginalFile( project, originalSourceFile, declaration, classifiedDependencies, originalFilePath, newFilePath, symbolToMove, ); logger.info( `Successfully moved symbol '${symbolToMove}' from '${originalFilePath}' to '${newFilePath}'.`, ); }