@axboot-mcp/mcp-server 1.0.0

package/src/handlers/generate-from-requirements.ts

@@ -0,0 +1,250 @@
+import { promises as fs } from 'fs';
+import * as path from 'path';
+import { spawn } from 'child_process';
+import { analyzeInterface } from './analyze.js';
+import { getTemplate } from '../templates/index.js';
+import { buildTemplateVariables } from './generate.js';
+import {
+  extractFeaturesFromRequirements,
+  matchTypes,
+  findBestMatch,
+} from './analyze-requirements.js';
+import { generateMultiStores, detectMultiListPattern } from './generate-multi-stores.js';
+
+/**
+ * 요구사항 기반 Store 생성 (단일/다중 자동 판별)
+ */
+export async function generateStoreFromRequirements(args: {
+  interfacePath: string;
+  outputPath: string;
+  requirements: string;
+}): Promise<{
+  content: Array<{ type: string; text: string }>;
+}> {
+  const { interfacePath, outputPath, requirements } = args;
+
+  // outputPath에서 store 이름 추출 (예: /path/to/useOrderStore.ts → useOrderStore)
+  const storeName = path.basename(outputPath, '.ts');
+
+  try {
+    // 1. 다중 리스트 패턴 우선 감지
+    const multiPattern = detectMultiListPattern(requirements);
+    if (multiPattern && multiPattern.isMultiList) {
+      // 다중 리스트인 경우 generateMultiStores 호출 (사용자가 제시한 storeName 사용)
+      return await generateMultiStores({
+        interfacePath,
+        outputPath,
+        requirements,
+        baseStoreName: storeName,
+      });
+    }
+
+    // 2. 단일 리스트인 경우 기존 로직
+    const analyzeResult = await analyzeInterface({ interfacePath });
+    const analyzeData = JSON.parse(analyzeResult.content[0].text);
+
+    if (!analyzeData.success) {
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({
+              success: false,
+              error: analyzeData.error || '인터페이스 분석 실패',
+            }),
+          },
+        ],
+      };
+    }
+
+    // 2.5. 유효성 검사: 리스트 메서드가 없는 경우 스토어 생성 건너뜀
+    const listMethods = analyzeData.methods?.filter((m: any) => m.isListMethod) || [];
+    if (listMethods.length === 0) {
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({
+              success: false,
+              error: `리스트 메서드를 찾을 수 없습니다. Repository에 목록 조회 메서드가 필요합니다. (${analyzeData.className})`,
+              skip: true,
+            }),
+          },
+        ],
+      };
+    }
+
+    // 3. 요구사항 분석
+    const extractedFeatures = extractFeaturesFromRequirements(requirements);
+    const matchedTypes = matchTypes(extractedFeatures);
+    const recommendedType = findBestMatch(matchedTypes);
+
+    if (!recommendedType) {
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({
+              success: false,
+              error: '요구사항에 맞는 Store 타입을 찾을 수 없습니다',
+              extractedFeatures,
+              matchedTypes,
+            }),
+          },
+        ],
+      };
+    }
+
+    // 4. 템플릿 로드
+    const template = getTemplate(recommendedType.type);
+    if (!template) {
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({
+              success: false,
+              error: `지원하지 않는 Store 타입입니다: ${recommendedType.type}`,
+            }),
+          },
+        ],
+      };
+    }
+
+    // 5. 템플릿 변수 빌드 (사용자가 제시한 storeName 사용)
+    const variables = buildTemplateVariables(analyzeData, recommendedType.type, storeName) as unknown as Record<string, string>;
+
+    // 6. 템플릿 적용
+    const storeCode = applyTemplate(template, variables);
+
+    // 7. 파일 생성
+    const outputDir = path.dirname(outputPath);
+    await fs.mkdir(outputDir, { recursive: true });
+    await fs.writeFile(outputPath, storeCode, 'utf-8');
+
+    // 8. Lint 및 Prettier 실행
+    runLintAndPrettier(outputPath).catch((err) => {
+      console.warn(`Lint/Prettier 실행 중 경고: ${err}`);
+    });
+
+    return {
+      content: [
+        {
+          type: 'text',
+          text: JSON.stringify(
+            {
+              success: true,
+              message: 'Store 파일 생성 완료',
+              outputPath,
+              requirements,
+              extractedFeatures,
+              recommendedType: {
+                type: recommendedType.type,
+                name: recommendedType.name,
+                description: recommendedType.description,
+                matchScore: recommendedType.matchScore,
+                matchedFeatures: recommendedType.matchedFeatures,
+                missingFeatures: recommendedType.missingFeatures,
+              },
+              analysis: {
+                className: analyzeData.className,
+                serviceName: analyzeData.serviceName,
+                methodCount: analyzeData.methods.length,
+              },
+            },
+            null,
+            2
+          ),
+        },
+      ],
+    };
+  } catch (error) {
+    return {
+      content: [
+        {
+          type: 'text',
+          text: JSON.stringify({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+          }),
+        },
+      ],
+    };
+  }
+}
+
+/**
+ * 템플릿 적용 - Mustache.js 사용
+ */
+function applyTemplate(template: string, variables: Record<string, any>): string {
+  const Mustache = require('mustache');
+
+  // TypeScript 코드 생성이므로 HTML 이스케이프 비활성화
+  // 기본 escape 함수를 덮어써서 원본 그대로 반환
+  const originalEscape = Mustache.escape;
+  Mustache.escape = (text: any) => String(text);
+
+  try {
+    // Mustache.js를 사용하여 템플릿 렌더링
+    // {{#KEY}}...{{/KEY}}: 조건부 블록 (truthy일 때 내용 포함)
+    // {{^KEY}}...{{/KEY}}: 역조건부 블록 (falsy일 때 내용 포함)
+    // {{KEY}}: 변수 치환
+    let result = Mustache.render(template, variables);
+
+    // 후처리: 빈 줄 제거만 수행 (정규식 제거로 인한 함수 시그니처 손실 방지)
+    result = result
+      // 빈 줄 제거 (연속된 2개 이상의 줄바꿈을 하나로 줄임)
+      .replace(/\n\s*\n\s*\n/g, '\n\n')
+      // 빈 줄 정리
+      .replace(/\n\s*\n/g, '\n');
+
+    return result;
+  } finally {
+    // 원래 escape 함수 복원
+    Mustache.escape = originalEscape;
+  }
+}
+
+/**
+ * Lint 및 Prettier 실행
+ */
+async function runLintAndPrettier(filePath: string): Promise<void> {
+  const fileDir = path.dirname(filePath);
+
+  // Prettier 실행
+  await runCommand('npx', ['prettier', '--write', filePath], fileDir);
+
+  // ESLint 실행
+  await runCommand('npx', ['eslint', '--fix', filePath], fileDir);
+}
+
+/**
+ * 명령 실행 헬퍼
+ */
+function runCommand(command: string, args: string[], cwd: string): Promise<void> {
+  return new Promise((resolve, reject) => {
+    const proc = spawn(command, args, {
+      cwd,
+      stdio: 'pipe',
+      shell: true,
+    });
+
+    let stderr = '';
+
+    proc.stderr?.on('data', (data) => {
+      stderr += data.toString();
+    });
+
+    proc.on('close', (code) => {
+      if (code !== 0) {
+        reject(new Error(`Command failed with code ${code}: ${stderr}`));
+      } else {
+        resolve();
+      }
+    });
+
+    proc.on('error', (err) => {
+      reject(err);
+    });
+  });
+}