feat(angular-mcp-server): add save location for contracts

Karnaukhov-kh · Karnaukhov-kh · commit 80d3611aa2b6 · 2025-09-14T15:10:40.000+03:00
diff --git a/packages/angular-mcp-server/src/lib/tools/ds/component-contract/builder/build-component-contract.tool.ts b/packages/angular-mcp-server/src/lib/tools/ds/component-contract/builder/build-component-contract.tool.ts
@@ -5,43 +5,36 @@ import {
 import { buildComponentContractSchema } from './models/schema.js';
 import { buildComponentContract } from './utils/build-contract.js';
 import {
-  saveContract,
   generateContractSummary,
 } from '../shared/utils/contract-file-ops.js';
 import { ContractResult } from './models/types.js';
 import { resolveCrossPlatformPath } from '../../shared/utils/cross-platform-path.js';
 
 interface BuildComponentContractOptions extends BaseHandlerOptions {
-  directory: string;
+  saveLocation: string;
   templateFile: string;
   styleFile: string;
   typescriptFile: string;
-  dsComponentName: string;
+  dsComponentName?: string;
 }
 
 export const buildComponentContractHandler = createHandler<
   BuildComponentContractOptions,
   ContractResult
 >(
   buildComponentContractSchema.name,
-  async (params, { cwd, workspaceRoot }) => {
+  async (params, { cwd, workspaceRoot: _workspaceRoot }) => {
     const {
-      directory,
+      saveLocation,
       templateFile,
       styleFile,
       typescriptFile,
-      dsComponentName,
+      dsComponentName = '',
     } = params;
 
-    const effectiveTemplatePath = resolveCrossPlatformPath(
-      directory,
-      templateFile,
-    );
-    const effectiveScssPath = resolveCrossPlatformPath(directory, styleFile);
-    const effectiveTypescriptPath = resolveCrossPlatformPath(
-      directory,
-      typescriptFile,
-    );
+    const effectiveTemplatePath = resolveCrossPlatformPath(cwd, templateFile);
+    const effectiveScssPath = resolveCrossPlatformPath(cwd, styleFile);
+    const effectiveTypescriptPath = resolveCrossPlatformPath(cwd, typescriptFile);
 
     const contract = await buildComponentContract(
       effectiveTemplatePath,
@@ -50,18 +43,40 @@ export const buildComponentContractHandler = createHandler<
       effectiveTypescriptPath,
     );
 
-    const { contractFilePath, hash } = await saveContract(
+    // Custom save logic using the provided saveLocation
+    const contractString = JSON.stringify(contract, null, 2);
+    const hash = require('node:crypto').createHash('sha256').update(contractString).digest('hex');
+    
+    const effectiveSaveLocation = resolveCrossPlatformPath(cwd, saveLocation);
+    
+    // Ensure directory exists
+    const { mkdir, writeFile } = await import('node:fs/promises');
+    const { dirname } = await import('node:path');
+    await mkdir(dirname(effectiveSaveLocation), { recursive: true });
+    
+    const contractData = {
       contract,
-      workspaceRoot,
-      effectiveTemplatePath,
-      effectiveScssPath,
-      cwd,
-      dsComponentName,
+      hash: `sha256-${hash}`,
+      metadata: {
+        templatePath: effectiveTemplatePath,
+        scssPath: effectiveScssPath,
+        typescriptPath: effectiveTypescriptPath,
+        timestamp: new Date().toISOString(),
+        dsComponentName,
+      },
+    };
+
+    await writeFile(
+      effectiveSaveLocation,
+      JSON.stringify(contractData, null, 2),
+      'utf-8',
     );
 
+    const contractFilePath = effectiveSaveLocation;
+
     return {
       contract,
-      hash,
+      hash: `sha256-${hash}`,
       contractFilePath,
     };
   },
diff --git a/packages/angular-mcp-server/src/lib/tools/ds/component-contract/builder/models/schema.ts b/packages/angular-mcp-server/src/lib/tools/ds/component-contract/builder/models/schema.ts
@@ -1,7 +1,6 @@
 import { ToolSchemaOptions } from '@push-based/models';
 import {
   COMMON_ANNOTATIONS,
-  createProjectAnalysisSchema,
 } from '../../../shared';
 
 /**
@@ -12,32 +11,38 @@ export const buildComponentContractSchema: ToolSchemaOptions = {
   description:
     "Generate a static surface contract for a component's template and SCSS.",
   inputSchema: {
-    ...createProjectAnalysisSchema({
+    type: 'object',
+    properties: {
+      saveLocation: {
+        type: 'string',
+        description:
+          'Path where to save the contract file. Supports both absolute and relative paths.',
+      },
       templateFile: {
         type: 'string',
         description:
-          'File name of the component template file (.html) or TypeScript component file (.ts) for inline templates',
+          'Path to the component template file (.html) or TypeScript component file (.ts) for inline templates. Supports both absolute and relative paths.',
       },
       styleFile: {
         type: 'string',
         description:
-          'File name of the component style file (.scss, .sass, .less, .css)',
+          'Path to the component style file (.scss, .sass, .less, .css). Supports both absolute and relative paths.',
       },
       typescriptFile: {
         type: 'string',
-        description: 'File name of the TypeScript component file (.ts)',
+        description: 'Path to the TypeScript component file (.ts). Supports both absolute and relative paths.',
       },
       dsComponentName: {
         type: 'string',
         description: 'The name of the design system component being used',
+        default: '',
       },
-    }),
+    },
     required: [
-      'directory',
+      'saveLocation',
       'templateFile',
       'styleFile',
       'typescriptFile',
-      'dsComponentName',
     ],
   },
   annotations: {
diff --git a/packages/angular-mcp-server/src/lib/tools/ds/component-contract/diff/diff-component-contract.tool.ts b/packages/angular-mcp-server/src/lib/tools/ds/component-contract/diff/diff-component-contract.tool.ts
@@ -9,8 +9,6 @@ import {
 import { diffComponentContractSchema } from './models/schema.js';
 import type { DomPathDictionary } from '../shared/models/types.js';
 import { loadContract } from '../shared/utils/contract-file-ops.js';
-import { componentNameToKebabCase } from '../../shared/utils/component-validation.js';
-import { basename } from 'node:path';
 import {
   consolidateAndPruneRemoveOperationsWithDeduplication,
   groupChangesByDomainAndType,
@@ -20,10 +18,10 @@ import { writeFile, mkdir } from 'node:fs/promises';
 import diff from 'microdiff';
 
 interface DiffComponentContractOptions extends BaseHandlerOptions {
-  directory: string;
+  saveLocation: string;
   contractBeforePath: string;
   contractAfterPath: string;
-  dsComponentName: string;
+  dsComponentName?: string;
 }
 
 export const diffComponentContractHandler = createHandler<
@@ -34,15 +32,11 @@ export const diffComponentContractHandler = createHandler<
   }
 >(
   diffComponentContractSchema.name,
-  async (params, { workspaceRoot }) => {
-    const effectiveBeforePath = resolveCrossPlatformPath(
-      params.directory,
-      params.contractBeforePath,
-    );
-    const effectiveAfterPath = resolveCrossPlatformPath(
-      params.directory,
-      params.contractAfterPath,
-    );
+  async (params, { cwd, workspaceRoot }) => {
+    const { saveLocation, contractBeforePath, contractAfterPath, dsComponentName = '' } = params;
+    
+    const effectiveBeforePath = resolveCrossPlatformPath(cwd, contractBeforePath);
+    const effectiveAfterPath = resolveCrossPlatformPath(cwd, contractAfterPath);
 
     const contractBefore = await loadContract(effectiveBeforePath);
     const contractAfter = await loadContract(effectiveAfterPath);
@@ -57,7 +51,7 @@ export const diffComponentContractHandler = createHandler<
     const diffData = {
       before: effectiveBeforePath,
       after: effectiveAfterPath,
-      dsComponentName: params.dsComponentName,
+      dsComponentName,
       timestamp: new Date().toISOString(),
       domPathDictionary: domPathDict.paths,
       changes: groupedChanges,
@@ -67,25 +61,14 @@ export const diffComponentContractHandler = createHandler<
     // Normalize absolute paths to relative paths for portability
     const normalizedDiffData = normalizePathsInObject(diffData, workspaceRoot);
 
-    // Create component-specific diffs directory
-    const componentKebab = componentNameToKebabCase(params.dsComponentName);
-    const diffDir = resolveCrossPlatformPath(
-      workspaceRoot,
-      `.cursor/tmp/contracts/${componentKebab}/diffs`,
-    );
-    await mkdir(diffDir, { recursive: true });
-
-    // Generate simplified diff filename: diff-{componentName}-{timestamp}.json
-    const componentBaseName = basename(
-      effectiveBeforePath,
-      '.contract.json',
-    ).split('-')[0]; // Extract component name before timestamp
-    const timestamp = new Date()
-      .toISOString()
-      .replace(/[-:]/g, '')
-      .replace(/\.\d+Z$/, 'Z');
-    const diffFileName = `diff-${componentBaseName}-${timestamp}.json`;
-    const diffFilePath = resolveCrossPlatformPath(diffDir, diffFileName);
+    // Use the provided saveLocation
+    const effectiveSaveLocation = resolveCrossPlatformPath(cwd, saveLocation);
+    
+    // Ensure directory exists
+    const { dirname } = await import('node:path');
+    await mkdir(dirname(effectiveSaveLocation), { recursive: true });
+    
+    const diffFilePath = effectiveSaveLocation;
 
     const formattedJson = JSON.stringify(normalizedDiffData, null, 2);
     await writeFile(diffFilePath, formattedJson, 'utf-8');
diff --git a/packages/angular-mcp-server/src/lib/tools/ds/component-contract/diff/models/schema.ts b/packages/angular-mcp-server/src/lib/tools/ds/component-contract/diff/models/schema.ts
@@ -1,6 +1,5 @@
 import { ToolSchemaOptions } from '@push-based/models';
 import {
-  createProjectAnalysisSchema,
   COMMON_ANNOTATIONS,
 } from '../../../shared';
 
@@ -12,25 +11,31 @@ export const diffComponentContractSchema: ToolSchemaOptions = {
   description:
     'Compare before/after contracts for parity and surface breaking changes.',
   inputSchema: {
-    ...createProjectAnalysisSchema({
+    type: 'object',
+    properties: {
+      saveLocation: {
+        type: 'string',
+        description:
+          'Path where to save the diff result file. Supports both absolute and relative paths.',
+      },
       contractBeforePath: {
         type: 'string',
-        description: 'Path to the contract file before refactoring',
+        description: 'Path to the contract file before refactoring. Supports both absolute and relative paths.',
       },
       contractAfterPath: {
         type: 'string',
-        description: 'Path to the contract file after refactoring',
+        description: 'Path to the contract file after refactoring. Supports both absolute and relative paths.',
       },
       dsComponentName: {
         type: 'string',
         description: 'The name of the design system component being used',
+        default: '',
       },
-    }),
+    },
     required: [
-      'directory',
+      'saveLocation',
       'contractBeforePath',
       'contractAfterPath',
-      'dsComponentName',
     ],
   },
   annotations: {
