swallowkit 1.0.0-beta.9 → 1.0.1

package/src/core/scaffold/model-parser.ts

@@ -3,8 +3,11 @@
  */
 
 import * as fs from "fs";
+import { execFileSync } from "child_process";
 import * as path from "path";
-import { pathToFileURL } from "url";
+import * as ts from "typescript";
+
+import { ModelConnectorConfig, ModelAuthPolicy } from "../../types";
 
 export interface ModelInfo {
   name: string; // モデル名（例: "Todo"）
@@ -16,12 +19,16 @@ export interface ModelInfo {
   hasCreatedAt: boolean; // createdAt フィールドがあるか
   hasUpdatedAt: boolean; // updatedAt フィールドがあるか
   nestedSchemaRefs: NestedSchemaRef[]; // ネストしたスキーマ参照
+  connectorConfig?: ModelConnectorConfig; // コネクタメタデータ（外部データソース用）
+  authPolicy?: ModelAuthPolicy; // 認可ポリシー（ロールベースアクセス制御用）
+  partitionKey: string; // Cosmos DB パーティションキーパス（デフォルト: "/id"）
 }
 
 export interface FieldInfo {
   name: string;
   type: string; // "string" | "number" | "boolean" | "date" | "object" | "array"
   isOptional: boolean;
+  isNullable?: boolean;
   isArray: boolean;
   enumValues?: string[]; // enum の場合の選択肢
   isForeignKey?: boolean; // 外部キーかどうか
@@ -60,11 +67,15 @@ export async function parseModelFile(modelPath: string): Promise<ModelInfo> {
   const modelName = toPascalCase(fileName);
   
   // スキーマ変数名を抽出
+  // コメント内のコードパターンを誤認識しないよう、先にコメントを除去してから正規表現を適用する
   // パターン1: export const todoSchema = z.object({ ... })  (camelCase + Schema接尾辞)
   // パターン2: export const Todo = z.object({ ... })         (Zod公式パターン)
-  let schemaMatch = content.match(/export\s+const\s+(\w+Schema)\s*=/);
+  const contentWithoutComments = content
+    .replace(/\/\*[\s\S]*?\*\//g, '')  // ブロックコメントを除去
+    .replace(/\/\/.*/g, '');           // 行コメントを除去
+  let schemaMatch = contentWithoutComments.match(/export\s+const\s+(\w+Schema)\s*=/);
   if (!schemaMatch) {
-    schemaMatch = content.match(/export\s+const\s+(\w+)\s*=\s*z\.object\s*\(/);
+    schemaMatch = contentWithoutComments.match(/export\s+const\s+(\w+)\s*=\s*z\.object\s*\(/);
   }
   if (!schemaMatch) {
     throw new Error(`Could not find exported schema in ${modelPath}. Expected patterns:\n  - export const xxxSchema = z.object({ ... })\n  - export const Xxx = z.object({ ... })`);
@@ -75,6 +86,15 @@ export async function parseModelFile(modelPath: string): Promise<ModelInfo> {
   // displayName を抽出（例: export const displayName = 'Task'）
   const displayNameMatch = content.match(/export\s+const\s+displayName\s*=\s*['"]([^'"]+)['"]/);  
   const displayName = displayNameMatch ? displayNameMatch[1] : modelName;
+
+  // connectorConfig を抽出（外部データソース用メタデータ）
+  const connectorConfig = parseConnectorConfig(content);
+
+  // authPolicy を抽出（ロールベースアクセス制御用メタデータ）
+  const authPolicy = parseAuthPolicy(content);
+
+  // partitionKey を抽出（Cosmos DB パーティションキー）
+  const partitionKey = parsePartitionKey(content);
   
   // ネストしたスキーマ参照を検出
   const nestedSchemaRefs = detectNestedSchemaRefs(modelPath, content, schemaName);
@@ -89,6 +109,17 @@ export async function parseModelFile(modelPath: string): Promise<ModelInfo> {
   const hasId = fields.some(f => f.name === "id");
   const hasCreatedAt = fields.some(f => f.name === "createdAt");
   const hasUpdatedAt = fields.some(f => f.name === "updatedAt");
+
+  // パーティションキーのバリデーション
+  if (partitionKey !== '/id') {
+    if (!partitionKey.startsWith('/')) {
+      console.warn(`⚠️  [${modelName}] partitionKey should start with '/': got '${partitionKey}'`);
+    }
+    const pkField = partitionKey.startsWith('/') ? partitionKey.slice(1) : partitionKey;
+    if (fields.length > 0 && !fields.some(f => f.name === pkField)) {
+      console.warn(`⚠️  [${modelName}] partitionKey field '${pkField}' not found in schema fields`);
+    }
+  }
   
   return {
     name: modelName,
@@ -100,6 +131,9 @@ export async function parseModelFile(modelPath: string): Promise<ModelInfo> {
     hasCreatedAt,
     hasUpdatedAt,
     nestedSchemaRefs,
+    partitionKey,
+    ...(connectorConfig ? { connectorConfig } : {}),
+    ...(authPolicy ? { authPolicy } : {}),
   };
 }
 
@@ -291,79 +325,46 @@ function mergeNestedSchemaInfo(fields: FieldInfo[], nestedRefs: NestedSchemaRef[
   }
 }
 
+function compileModelModule(modelPath: string, directory: string, modules = new Map<string, string>()): string {
+  const sourcePath = path.resolve(modelPath);
+  const existing = modules.get(sourcePath);
+  if (existing) return existing;
+  const outputPath = path.join(directory, `model-${modules.size}.mjs`);
+  modules.set(sourcePath, outputPath);
+  const output = ts.transpileModule(fs.readFileSync(sourcePath, "utf8"), {
+    fileName: sourcePath,
+    compilerOptions: { target: ts.ScriptTarget.ES2022, module: ts.ModuleKind.ESNext },
+    transformers: {
+      after: [context => source => {
+        const visit: ts.Visitor = node => {
+          if (ts.isStringLiteral(node) &&
+              (ts.isImportDeclaration(node.parent) || ts.isExportDeclaration(node.parent)) &&
+              node.text.startsWith(".")) {
+            const base = path.resolve(path.dirname(sourcePath), node.text);
+            const candidates = [base.replace(/\.js$/, ".ts"), `${base}.ts`, path.join(base, "index.ts"), base];
+            const dependency = candidates.find(candidate => fs.existsSync(candidate) && fs.statSync(candidate).isFile());
+            if (!dependency) throw new Error(`Cannot resolve ${node.text} from ${sourcePath}`);
+            const compiled = compileModelModule(dependency, directory, modules);
+            return context.factory.createStringLiteral(`./${path.basename(compiled)}`);
+          }
+          return ts.visitEachChild(node, visit, context);
+        };
+        return ts.visitNode(source, visit) as ts.SourceFile;
+      }],
+    },
+  });
+  fs.writeFileSync(outputPath, output.outputText);
+  return outputPath;
+}
+
 /**
  * Zodスキーマから動的にフィールド情報を抽出
  */
 async function extractFieldsFromSchema(modelPath: string, schemaName: string): Promise<FieldInfo[]> {
   const fields: FieldInfo[] = [];
+  let tempDirectory: string | undefined;
   
   try {
-    // child_processでtsxを実行
-    const { execSync } = require('child_process');
-    const { tmpdir } = require('os');
-    const { join } = require('path');
-    const { writeFileSync, unlinkSync, readFileSync } = require('fs');
-    
-    // モデルファイルの内容を読み込む
-    let modelContent = readFileSync(modelPath, 'utf8');
-    const modelDir = path.dirname(path.resolve(modelPath));
-    
-    // ローカル相対インポートを絶対パスの .mjs import 文に変換して保持
-    // パターン: import { categorySchema } from './category'
-    const localImports: string[] = [];
-    modelContent = modelContent.replace(
-      /import\s*\{([^}]+)\}\s*from\s*['"](\.[^'"]+)['"]\s*;?/g,
-      (_match: string, imports: string, importPath: string) => {
-        // インポートされた変数名を取得
-        const importedNames = imports.split(',').map(s => {
-          const trimmed = s.trim();
-          // 'as' エイリアスに対応
-          return trimmed.split(/\s+as\s+/).pop()!.trim();
-        }).filter(s => s.length > 0);
-        
-        // 相対パスを絶対パスに解決
-        let resolvedPath = path.resolve(modelDir, importPath);
-        if (!resolvedPath.endsWith('.ts') && !resolvedPath.endsWith('.js')) {
-          resolvedPath += '.ts';
-        }
-        if (fs.existsSync(resolvedPath)) {
-          // インポート先の内容を読み込み、インライン化する
-          let refContent = readFileSync(resolvedPath, 'utf8');
-          // インポート先の import 文を除去
-          refContent = refContent.replace(/import\s+.*?\s+from\s+['"](?!\.).*?['"];?\s*/g, '');
-          refContent = refContent.replace(/import\s+.*?\s+from\s+['"]\..*?['"];?\s*/g, '');
-          refContent = refContent.replace(/export\s+(const|type|interface|class|function)\s+/g, '$1 ');
-          refContent = refContent.replace(/^type\s+\w+\s*=\s*[^;]+;/gm, '');
-          refContent = refContent.replace(/^interface\s+\w+\s*\{[\s\S]*?\}/gm, '');
-          refContent = refContent.replace(/\/\*[\s\S]*?\*\//g, '');
-          refContent = refContent.replace(/\/\/.*/g, '');
-          
-          // インポートされていない const 宣言を除去（displayName 等の重複を防ぐ）
-          const importedNameSet = new Set(importedNames);
-          refContent = refContent.replace(/^const\s+(\w+)\s*=\s*[^;]+;/gm, (constMatch: string, varName: string) => {
-            return importedNameSet.has(varName) ? constMatch : '';
-          });
-          
-          localImports.push(refContent.trim());
-        }
-        return ''; // 元のインポート文は除去
-      }
-    );
-    
-    // zod 以外のパッケージインポートを除去
-    modelContent = modelContent.replace(/import\s+.*?\s+from\s+['"].*?['"];?\s*/g, '');
-    modelContent = modelContent.replace(/export\s+(const|type|interface|class|function)\s+/g, '$1 ');
-    // type宣言を削除（ランタイムでは不要）
-    modelContent = modelContent.replace(/^type\s+\w+\s*=\s*[^;]+;/gm, '');
-    // interface宣言を削除（ランタイムでは不要、複数行対応）
-    modelContent = modelContent.replace(/^interface\s+\w+\s*\{[\s\S]*?\}/gm, '');
-    // コメントを削除
-    modelContent = modelContent.replace(/\/\*[\s\S]*?\*\//g, '');
-    modelContent = modelContent.replace(/\/\/.*/g, '');
-    
-    // インライン化したローカルインポートを先頭に追加
-    const inlinedDeps = localImports.length > 0 ? localImports.join('\n\n') + '\n\n' : '';
-    
     // プロジェクトルートを探す（package.jsonがある場所）
     let projectRoot = path.dirname(modelPath);
     while (projectRoot !== path.dirname(projectRoot)) {
@@ -374,16 +375,11 @@ async function extractFieldsFromSchema(modelPath: string, schemaName: string): P
     }
     
     // プロジェクトルート内に一時スクリプトファイルを作成
-    const tempScript = path.join(projectRoot, `.swallowkit-parser-${Date.now()}.mjs`);
+    tempDirectory = fs.mkdtempSync(path.join(projectRoot, '.swallowkit-parser-'));
+    const compiledModel = compileModelModule(modelPath, tempDirectory);
+    const tempScript = path.join(tempDirectory, 'inspect.mjs');
     const scriptCode = `
-import { z } from 'zod/v4';
-
-// インライン化した依存スキーマ
-${inlinedDeps}
-// モデルファイルの内容を評価
-${modelContent}
-
-const schema = ${schemaName};
+  import { ${schemaName} as schema } from './${path.basename(compiledModel)}';
 
 // Zod v3とv4の両方に対応
 const isObject = (schema && schema._def && 
@@ -395,10 +391,11 @@ if (isObject) {
     const field = shape[key];
     let type = 'string';
     let isOptional = false;
+    let isNullable = false;
     let isArray = false;
     let enumValues = undefined;
     
-    // ZodOptional, ZodDefault, ZodEffects を unwrap
+    // ZodOptional, ZodNullable, ZodDefault, ZodEffects を unwrap
     let fieldDef = field;
     const getTypeName = (def) => def?._def?.typeName || def?.constructor?.name || '';
     
@@ -412,6 +409,10 @@ if (isObject) {
         isOptional = true;
         fieldDef = fieldDef._def.innerType;
         unwrapped = true;
+      } else if (typeName === 'ZodNullable') {
+        isNullable = true;
+        fieldDef = fieldDef._def.innerType;
+        unwrapped = true;
       } else if (typeName === 'ZodDefault') {
         // .default() は optional と同様に扱う
         isOptional = true;
@@ -427,7 +428,7 @@ if (isObject) {
     // ZodArray をチェック
     if (getTypeName(fieldDef) === 'ZodArray') {
       isArray = true;
-      fieldDef = fieldDef._def.type || fieldDef._def.element;
+      fieldDef = fieldDef._def.element || fieldDef._def.type;
     }
     
     // 基本型を判定
@@ -468,22 +469,17 @@ if (isObject) {
       isForeignKey = true;
     }
     
-    return { name: key, type, isOptional, isArray, enumValues, isForeignKey, referencedModel };
+    return { name: key, type, isOptional, isNullable, isArray, enumValues, isForeignKey, referencedModel };
   });
   
   console.log(JSON.stringify(fields));
 }
 `;
     
-    writeFileSync(tempScript, scriptCode, 'utf8');
+    fs.writeFileSync(tempScript, scriptCode, 'utf8');
     
     try {
-      // プロジェクトルートでtsxを実行 (npm/pnpm 両対応)
-      // Detect package manager from project lockfile
-      const { detectFromProject, getCommands } = require('../../utils/package-manager');
-      const pm = detectFromProject(projectRoot);
-      const pmCmd = getCommands(pm);
-      const result = execSync(`${pmCmd.exec} tsx "${tempScript}"`, {
+      const result = execFileSync(process.execPath, [tempScript], {
         encoding: 'utf8',
         cwd: projectRoot,
       });
@@ -497,16 +493,17 @@ if (isObject) {
     } finally {
       // 一時ファイルを削除
       try {
-        unlinkSync(tempScript);
+        fs.unlinkSync(tempScript);
       } catch (e) {
         // ファイル削除失敗は無視
       }
     }
   } catch (error) {
-    // tsxが使えない場合は正規表現フォールバック
     console.warn('Failed to use dynamic import, falling back to regex parsing');
     console.warn('Error:', error);
     return extractFieldsWithRegex(modelPath, schemaName);
+  } finally {
+    if (tempDirectory) fs.rmSync(tempDirectory, { recursive: true, force: true });
   }
   
   return fields;
@@ -545,6 +542,7 @@ function extractFieldsWithRegex(modelPath: string, schemaName: string): FieldInf
       name: fieldName,
       type: mapZodTypeToTs(zodType),
       isOptional: fieldDef.includes(".optional()"),
+      isNullable: fieldDef.includes(".nullable()"),
       isArray: fieldDef.includes(".array()"),
     });
   }
@@ -586,6 +584,162 @@ export function toCamelCase(str: string): string {
   return pascal.charAt(0).toLowerCase() + pascal.slice(1);
 }
 
+/**
+ * モデルファイルから connectorConfig エクスポートを静的解析で抽出する
+ */
+export function parseConnectorConfig(content: string): ModelConnectorConfig | undefined {
+  // export const connectorConfig = { ... } を検出
+  const connectorMatch = content.match(/export\s+const\s+connectorConfig\s*=\s*\{/);
+  if (!connectorMatch) {
+    return undefined;
+  }
+
+  // connectorConfig オブジェクトの内容を抽出
+  const startIdx = content.indexOf('{', connectorMatch.index!);
+  let braceCount = 1;
+  let endIdx = startIdx + 1;
+  while (braceCount > 0 && endIdx < content.length) {
+    if (content[endIdx] === '{') braceCount++;
+    if (content[endIdx] === '}') braceCount--;
+    endIdx++;
+  }
+
+  const objectStr = content.substring(startIdx, endIdx);
+
+  // connector 名を抽出
+  const connectorNameMatch = objectStr.match(/connector\s*:\s*['"]([^'"]+)['"]/);
+  if (!connectorNameMatch) {
+    return undefined;
+  }
+  const connector = connectorNameMatch[1];
+
+  // operations を抽出
+  const opsMatch = objectStr.match(/operations\s*:\s*\[([^\]]*)\]/);
+  const operations: string[] = [];
+  if (opsMatch) {
+    const opsStr = opsMatch[1];
+    const opEntries = opsStr.match(/['"]([^'"]+)['"]/g);
+    if (opEntries) {
+      for (const entry of opEntries) {
+        operations.push(entry.replace(/['"]/g, ''));
+      }
+    }
+  }
+
+  // table を抽出（RDB 固有）
+  const tableMatch = objectStr.match(/table\s*:\s*['"]([^'"]+)['"]/);
+
+  // idColumn を抽出（RDB 固有）
+  const idColumnMatch = objectStr.match(/idColumn\s*:\s*['"]([^'"]+)['"]/);
+
+  // endpoints を抽出（API 固有）— endpoint値に {id} 等のプレースホルダが含まれるためbrace-countingで抽出
+  let endpoints: Record<string, string> | undefined;
+  const endpointsStart = objectStr.match(/endpoints\s*:\s*\{/);
+  if (endpointsStart) {
+    const epStartIdx = objectStr.indexOf('{', endpointsStart.index!);
+    let epBraceCount = 1;
+    let epEndIdx = epStartIdx + 1;
+    while (epBraceCount > 0 && epEndIdx < objectStr.length) {
+      const ch = objectStr[epEndIdx];
+      // 文字列内の波括弧をスキップ
+      if (ch === "'" || ch === '"') {
+        epEndIdx++;
+        while (epEndIdx < objectStr.length && objectStr[epEndIdx] !== ch) {
+          epEndIdx++;
+        }
+      } else if (ch === '{') {
+        epBraceCount++;
+      } else if (ch === '}') {
+        epBraceCount--;
+      }
+      epEndIdx++;
+    }
+    const epStr = objectStr.substring(epStartIdx + 1, epEndIdx - 1);
+    endpoints = {};
+    const endpointEntries = epStr.matchAll(/(\w+)\s*:\s*['"]([^'"]+)['"]/g);
+    for (const entry of endpointEntries) {
+      endpoints[entry[1]] = entry[2];
+    }
+  }
+
+  if (tableMatch) {
+    // RDB コネクタ
+    return {
+      connector,
+      operations: operations as ModelConnectorConfig['operations'],
+      table: tableMatch[1],
+      ...(idColumnMatch ? { idColumn: idColumnMatch[1] } : {}),
+    };
+  }
+
+  // API コネクタ
+  return {
+    connector,
+    operations: operations as ModelConnectorConfig['operations'],
+    ...(endpoints ? { endpoints } : {}),
+  };
+}
+
+/**
+ * authPolicy をモデルファイルから抽出
+ * パターン: export const authPolicy = { roles: [...], read: [...], write: [...] }
+ */
+export function parseAuthPolicy(content: string): ModelAuthPolicy | undefined {
+  const policyMatch = content.match(/export\s+const\s+authPolicy\s*=\s*\{/);
+  if (!policyMatch) {
+    return undefined;
+  }
+
+  const startIdx = content.indexOf('{', policyMatch.index!);
+  let braceCount = 1;
+  let endIdx = startIdx + 1;
+  while (braceCount > 0 && endIdx < content.length) {
+    if (content[endIdx] === '{') braceCount++;
+    if (content[endIdx] === '}') braceCount--;
+    endIdx++;
+  }
+
+  const objectStr = content.substring(startIdx, endIdx);
+
+  const extractRoles = (key: string): string[] | undefined => {
+    const match = objectStr.match(new RegExp(`${key}\\s*:\\s*\\[([^\\]]*)\\]`));
+    if (!match) return undefined;
+    const entries = match[1].match(/['"]([^'"]+)['"]/g);
+    if (!entries) return [];
+    return entries.map(e => e.replace(/['"]/g, ''));
+  };
+
+  const roles = extractRoles('roles');
+  const read = extractRoles('read');
+  const write = extractRoles('write');
+
+  const extractName = (key: string): string | undefined => objectStr.match(new RegExp(`${key}\\s*:\\s*['"]([^'"]+)['"]`))?.[1];
+  const namedPolicy = extractName('policy');
+  const namedRead = extractName('read');
+  const namedWrite = extractName('write');
+
+  if (!roles && !read && !write && !namedPolicy && !namedRead && !namedWrite) {
+    return undefined;
+  }
+
+  return {
+    ...(roles ? { roles } : {}),
+    ...(namedRead ? { read: namedRead } : read ? { read } : {}),
+    ...(namedWrite ? { write: namedWrite } : write ? { write } : {}),
+    ...(namedPolicy ? { policy: namedPolicy } : {}),
+  };
+}
+
+/**
+ * パーティションキーを抽出する
+ * export const partitionKey = '/tenantId' のようなエクスポートを検出
+ * 未指定の場合はデフォルト '/id' を返す
+ */
+export function parsePartitionKey(content: string): string {
+  const match = content.match(/export\s+const\s+partitionKey\s*=\s*['"]([^'"]+)['"]/);
+  return match ? match[1] : '/id';
+}
+
 /**
  * 文字列を kebab-case に変換
  */