@contractspec/lib.source-extractors 2.7.6 → 2.7.10

package/dist/extractors/index.js

@@ -318,101 +318,54 @@ class BaseExtractor {
     ctx.ir.schemas.push(schema);
   }
 }
-// src/extractors/nestjs/extractor.ts
+// src/extractors/elysia/extractor.ts
 var PATTERNS = {
-  controller: /@Controller\s*\(\s*['"`]([^'"`]*)['"`]\s*\)/g,
-  route: /@(Get|Post|Put|Patch|Delete|Head|Options)\s*\(\s*(?:['"`]([^'"`]*)['"`])?\s*\)/g,
-  body: /@Body\s*\(\s*\)/g,
-  param: /@Param\s*\(\s*(?:['"`]([^'"`]*)['"`])?\s*\)/g,
-  query: /@Query\s*\(\s*(?:['"`]([^'"`]*)['"`])?\s*\)/g,
-  dto: /class\s+(\w+(?:Dto|DTO|Request|Response|Input|Output))\s*\{/g,
-  classValidator: /@(IsString|IsNumber|IsBoolean|IsArray|IsOptional|IsNotEmpty|Min|Max|Length|Matches)/g
+  route: /\.(get|post|put|patch|delete)\s*\(\s*['"`]([^'"`]+)['"`]/gi,
+  tSchema: /body:\s*t\.\w+/g
 };
 
-class NestJsExtractor extends BaseExtractor {
-  id = "nestjs";
-  name = "NestJS Extractor";
-  frameworks = ["nestjs"];
-  priority = 20;
+class ElysiaExtractor extends BaseExtractor {
+  id = "elysia";
+  name = "Elysia Extractor";
+  frameworks = ["elysia"];
+  priority = 15;
   async doExtract(ctx) {
     const { project, options, fs } = ctx;
     const pattern = options.scope?.length ? options.scope.map((s) => `${s}/**/*.ts`).join(",") : "**/*.ts";
     const files = await fs.glob(pattern, { cwd: project.rootPath });
     ctx.ir.stats.filesScanned = files.length;
     for (const file of files) {
-      if (file.includes("node_modules") || file.includes(".spec.") || file.includes(".test.")) {
+      if (file.includes("node_modules") || file.includes(".test."))
         continue;
-      }
       const fullPath = `${project.rootPath}/${file}`;
       const content = await fs.readFile(fullPath);
-      await this.extractControllers(ctx, file, content);
-      await this.extractDtos(ctx, file, content);
-    }
-  }
-  async extractControllers(ctx, file, content) {
-    const controllerMatches = [...content.matchAll(PATTERNS.controller)];
-    for (const controllerMatch of controllerMatches) {
-      const basePath = controllerMatch[1] || "";
-      const controllerIndex = controllerMatch.index ?? 0;
-      const afterDecorator = content.slice(controllerIndex);
-      const classMatch = afterDecorator.match(/class\s+(\w+)/);
-      const controllerName = classMatch?.[1] ?? "UnknownController";
-      const nextController = content.indexOf("@Controller", controllerIndex + 1);
-      const controllerBlock = nextController > 0 ? content.slice(controllerIndex, nextController) : content.slice(controllerIndex);
-      const routeMatches = [...controllerBlock.matchAll(PATTERNS.route)];
-      for (const routeMatch of routeMatches) {
-        const method = routeMatch[1]?.toUpperCase();
-        const routePath = routeMatch[2] || "";
-        const fullPath = this.normalizePath(`/${basePath}/${routePath}`);
-        const afterRoute = controllerBlock.slice(routeMatch.index ?? 0);
-        const methodMatch = afterRoute.match(/(?:async\s+)?(\w+)\s*\([^)]*\)\s*(?::\s*\w+(?:<[^>]+>)?)?\s*\{/);
-        const handlerName = methodMatch?.[1] ?? "unknownHandler";
-        const absoluteIndex = controllerIndex + (routeMatch.index ?? 0);
-        const lineNumber = content.slice(0, absoluteIndex).split(`
-`).length;
-        const hasBody = PATTERNS.body.test(afterRoute.slice(0, 200));
-        const hasParams = PATTERNS.param.test(afterRoute.slice(0, 200));
-        const hasQuery = PATTERNS.query.test(afterRoute.slice(0, 200));
-        const endpoint = {
-          id: this.generateEndpointId(method, fullPath, handlerName),
-          method,
-          path: fullPath,
-          kind: this.methodToOpKind(method),
-          handlerName,
-          controllerName,
-          source: this.createLocation(file, lineNumber, lineNumber + 10),
-          confidence: this.createConfidence("medium", "decorator-hints"),
-          frameworkMeta: {
-            hasBody,
-            hasParams,
-            hasQuery
-          }
-        };
-        this.addEndpoint(ctx, endpoint);
-      }
+      if (!content.includes("elysia"))
+        continue;
+      await this.extractRoutes(ctx, file, content);
     }
   }
-  async extractDtos(ctx, file, content) {
-    const dtoMatches = [...content.matchAll(PATTERNS.dto)];
-    for (const match of dtoMatches) {
-      const name = match[1] ?? "UnknownDto";
+  async extractRoutes(ctx, file, content) {
+    const matches = [...content.matchAll(PATTERNS.route)];
+    for (const match of matches) {
+      const method = match[1]?.toUpperCase() ?? "GET";
+      const path = match[2] ?? "/";
       const index = match.index ?? 0;
       const lineNumber = content.slice(0, index).split(`
 `).length;
-      const hasClassValidator = content.includes("class-validator") || content.includes("@IsString") || content.includes("@IsNumber");
-      const schema = {
-        id: this.generateSchemaId(name, file),
-        name,
-        schemaType: hasClassValidator ? "class-validator" : "typescript",
-        source: this.createLocation(file, lineNumber, lineNumber + 20),
-        confidence: this.createConfidence(hasClassValidator ? "high" : "medium", hasClassValidator ? "explicit-schema" : "inferred-types")
+      const afterMatch = content.slice(index, index + 500);
+      const hasTSchema = PATTERNS.tSchema.test(afterMatch);
+      const endpoint = {
+        id: this.generateEndpointId(method, path),
+        method,
+        path,
+        kind: this.methodToOpKind(method),
+        handlerName: "handler",
+        source: this.createLocation(file, lineNumber, lineNumber + 5),
+        confidence: this.createConfidence(hasTSchema ? "high" : "medium", hasTSchema ? "explicit-schema" : "decorator-hints")
       };
-      this.addSchema(ctx, schema);
+      this.addEndpoint(ctx, endpoint);
     }
   }
-  normalizePath(path) {
-    return "/" + path.replace(/\/+/g, "/").replace(/^\/+|\/+$/g, "");
-  }
 }
 // src/extractors/express/extractor.ts
 var PATTERNS2 = {
@@ -561,119 +514,104 @@ class HonoExtractor extends BaseExtractor {
     }
   }
 }
-// src/extractors/elysia/extractor.ts
+// src/extractors/nestjs/extractor.ts
 var PATTERNS5 = {
-  route: /\.(get|post|put|patch|delete)\s*\(\s*['"`]([^'"`]+)['"`]/gi,
-  tSchema: /body:\s*t\.\w+/g
+  controller: /@Controller\s*\(\s*['"`]([^'"`]*)['"`]\s*\)/g,
+  route: /@(Get|Post|Put|Patch|Delete|Head|Options)\s*\(\s*(?:['"`]([^'"`]*)['"`])?\s*\)/g,
+  body: /@Body\s*\(\s*\)/g,
+  param: /@Param\s*\(\s*(?:['"`]([^'"`]*)['"`])?\s*\)/g,
+  query: /@Query\s*\(\s*(?:['"`]([^'"`]*)['"`])?\s*\)/g,
+  dto: /class\s+(\w+(?:Dto|DTO|Request|Response|Input|Output))\s*\{/g,
+  classValidator: /@(IsString|IsNumber|IsBoolean|IsArray|IsOptional|IsNotEmpty|Min|Max|Length|Matches)/g
 };
 
-class ElysiaExtractor extends BaseExtractor {
-  id = "elysia";
-  name = "Elysia Extractor";
-  frameworks = ["elysia"];
-  priority = 15;
+class NestJsExtractor extends BaseExtractor {
+  id = "nestjs";
+  name = "NestJS Extractor";
+  frameworks = ["nestjs"];
+  priority = 20;
   async doExtract(ctx) {
     const { project, options, fs } = ctx;
     const pattern = options.scope?.length ? options.scope.map((s) => `${s}/**/*.ts`).join(",") : "**/*.ts";
     const files = await fs.glob(pattern, { cwd: project.rootPath });
     ctx.ir.stats.filesScanned = files.length;
     for (const file of files) {
-      if (file.includes("node_modules") || file.includes(".test."))
+      if (file.includes("node_modules") || file.includes(".spec.") || file.includes(".test.")) {
         continue;
+      }
       const fullPath = `${project.rootPath}/${file}`;
       const content = await fs.readFile(fullPath);
-      if (!content.includes("elysia"))
-        continue;
-      await this.extractRoutes(ctx, file, content);
+      await this.extractControllers(ctx, file, content);
+      await this.extractDtos(ctx, file, content);
     }
   }
-  async extractRoutes(ctx, file, content) {
-    const matches = [...content.matchAll(PATTERNS5.route)];
-    for (const match of matches) {
-      const method = match[1]?.toUpperCase() ?? "GET";
-      const path = match[2] ?? "/";
-      const index = match.index ?? 0;
-      const lineNumber = content.slice(0, index).split(`
+  async extractControllers(ctx, file, content) {
+    const controllerMatches = [...content.matchAll(PATTERNS5.controller)];
+    for (const controllerMatch of controllerMatches) {
+      const basePath = controllerMatch[1] || "";
+      const controllerIndex = controllerMatch.index ?? 0;
+      const afterDecorator = content.slice(controllerIndex);
+      const classMatch = afterDecorator.match(/class\s+(\w+)/);
+      const controllerName = classMatch?.[1] ?? "UnknownController";
+      const nextController = content.indexOf("@Controller", controllerIndex + 1);
+      const controllerBlock = nextController > 0 ? content.slice(controllerIndex, nextController) : content.slice(controllerIndex);
+      const routeMatches = [...controllerBlock.matchAll(PATTERNS5.route)];
+      for (const routeMatch of routeMatches) {
+        const method = routeMatch[1]?.toUpperCase();
+        const routePath = routeMatch[2] || "";
+        const fullPath = this.normalizePath(`/${basePath}/${routePath}`);
+        const afterRoute = controllerBlock.slice(routeMatch.index ?? 0);
+        const methodMatch = afterRoute.match(/(?:async\s+)?(\w+)\s*\([^)]*\)\s*(?::\s*\w+(?:<[^>]+>)?)?\s*\{/);
+        const handlerName = methodMatch?.[1] ?? "unknownHandler";
+        const absoluteIndex = controllerIndex + (routeMatch.index ?? 0);
+        const lineNumber = content.slice(0, absoluteIndex).split(`
 `).length;
-      const afterMatch = content.slice(index, index + 500);
-      const hasTSchema = PATTERNS5.tSchema.test(afterMatch);
-      const endpoint = {
-        id: this.generateEndpointId(method, path),
-        method,
-        path,
-        kind: this.methodToOpKind(method),
-        handlerName: "handler",
-        source: this.createLocation(file, lineNumber, lineNumber + 5),
-        confidence: this.createConfidence(hasTSchema ? "high" : "medium", hasTSchema ? "explicit-schema" : "decorator-hints")
-      };
-      this.addEndpoint(ctx, endpoint);
-    }
-  }
-}
-// src/extractors/trpc/extractor.ts
-var PATTERNS6 = {
-  procedure: /\.(query|mutation)\s*\(\s*(?:\{[^}]*\}|[^)]+)\)/gi,
-  procedureName: /(\w+)\s*:\s*(?:publicProcedure|protectedProcedure|procedure)/g,
-  zodInput: /\.input\s*\(\s*(\w+)/g,
-  zodOutput: /\.output\s*\(\s*(\w+)/g
-};
-
-class TrpcExtractor extends BaseExtractor {
-  id = "trpc";
-  name = "tRPC Extractor";
-  frameworks = ["trpc"];
-  priority = 15;
-  async doExtract(ctx) {
-    const { project, options, fs } = ctx;
-    const pattern = options.scope?.length ? options.scope.map((s) => `${s}/**/*.ts`).join(",") : "**/*.ts";
-    const files = await fs.glob(pattern, { cwd: project.rootPath });
-    ctx.ir.stats.filesScanned = files.length;
-    for (const file of files) {
-      if (file.includes("node_modules") || file.includes(".test."))
-        continue;
-      const fullPath = `${project.rootPath}/${file}`;
-      const content = await fs.readFile(fullPath);
-      if (!content.includes("trpc") && !content.includes("Procedure"))
-        continue;
-      await this.extractProcedures(ctx, file, content);
+        const hasBody = PATTERNS5.body.test(afterRoute.slice(0, 200));
+        const hasParams = PATTERNS5.param.test(afterRoute.slice(0, 200));
+        const hasQuery = PATTERNS5.query.test(afterRoute.slice(0, 200));
+        const endpoint = {
+          id: this.generateEndpointId(method, fullPath, handlerName),
+          method,
+          path: fullPath,
+          kind: this.methodToOpKind(method),
+          handlerName,
+          controllerName,
+          source: this.createLocation(file, lineNumber, lineNumber + 10),
+          confidence: this.createConfidence("medium", "decorator-hints"),
+          frameworkMeta: {
+            hasBody,
+            hasParams,
+            hasQuery
+          }
+        };
+        this.addEndpoint(ctx, endpoint);
+      }
     }
   }
-  async extractProcedures(ctx, file, content) {
-    const nameMatches = [...content.matchAll(PATTERNS6.procedureName)];
-    for (const match of nameMatches) {
-      const procedureName = match[1] ?? "unknownProcedure";
+  async extractDtos(ctx, file, content) {
+    const dtoMatches = [...content.matchAll(PATTERNS5.dto)];
+    for (const match of dtoMatches) {
+      const name = match[1] ?? "UnknownDto";
       const index = match.index ?? 0;
       const lineNumber = content.slice(0, index).split(`
 `).length;
-      const afterMatch = content.slice(index, index + 500);
-      const isQuery = afterMatch.includes(".query(");
-      const isMutation = afterMatch.includes(".mutation(");
-      if (!isQuery && !isMutation)
-        continue;
-      const hasZodInput = PATTERNS6.zodInput.test(afterMatch);
-      const hasZodOutput = PATTERNS6.zodOutput.test(afterMatch);
-      const hasSchema = hasZodInput || hasZodOutput;
-      const method = isMutation ? "POST" : "GET";
-      const endpoint = {
-        id: `trpc.${procedureName}`,
-        method,
-        path: `/trpc/${procedureName}`,
-        kind: isMutation ? "command" : "query",
-        handlerName: procedureName,
-        source: this.createLocation(file, lineNumber, lineNumber + 10),
-        confidence: this.createConfidence(hasSchema ? "high" : "medium", hasSchema ? "explicit-schema" : "inferred-types"),
-        frameworkMeta: {
-          procedureType: isMutation ? "mutation" : "query",
-          hasInput: hasZodInput,
-          hasOutput: hasZodOutput
-        }
+      const hasClassValidator = content.includes("class-validator") || content.includes("@IsString") || content.includes("@IsNumber");
+      const schema = {
+        id: this.generateSchemaId(name, file),
+        name,
+        schemaType: hasClassValidator ? "class-validator" : "typescript",
+        source: this.createLocation(file, lineNumber, lineNumber + 20),
+        confidence: this.createConfidence(hasClassValidator ? "high" : "medium", hasClassValidator ? "explicit-schema" : "inferred-types")
       };
-      this.addEndpoint(ctx, endpoint);
+      this.addSchema(ctx, schema);
     }
   }
+  normalizePath(path) {
+    return "/" + path.replace(/\/+/g, "/").replace(/^\/+|\/+$/g, "");
+  }
 }
 // src/extractors/next-api/extractor.ts
-var PATTERNS7 = {
+var PATTERNS6 = {
   appRouterExport: /export\s+(?:async\s+)?function\s+(GET|POST|PUT|PATCH|DELETE|HEAD|OPTIONS)/gi,
   pagesHandler: /export\s+default\s+(?:async\s+)?function/g,
   zodSchema: /z\.\w+\(/g
@@ -707,13 +645,13 @@ class NextApiExtractor extends BaseExtractor {
   async extractAppRoutes(ctx, file, content) {
     const pathMatch = file.match(/app\/api\/(.+)\/route\.ts$/);
     const routePath = pathMatch ? `/api/${pathMatch[1]}` : "/api";
-    const matches = [...content.matchAll(PATTERNS7.appRouterExport)];
+    const matches = [...content.matchAll(PATTERNS6.appRouterExport)];
     for (const match of matches) {
       const method = match[1]?.toUpperCase() ?? "GET";
       const index = match.index ?? 0;
       const lineNumber = content.slice(0, index).split(`
 `).length;
-      const hasZod = PATTERNS7.zodSchema.test(content);
+      const hasZod = PATTERNS6.zodSchema.test(content);
       const endpoint = {
         id: this.generateEndpointId(method, routePath),
         method,
@@ -730,10 +668,10 @@ class NextApiExtractor extends BaseExtractor {
   async extractPagesRoutes(ctx, file, content) {
     const pathMatch = file.match(/pages\/api\/(.+)\.ts$/);
     const routePath = pathMatch ? `/api/${pathMatch[1]}` : "/api";
-    if (!PATTERNS7.pagesHandler.test(content))
+    if (!PATTERNS6.pagesHandler.test(content))
       return;
     const lineNumber = 1;
-    const _hasZod = PATTERNS7.zodSchema.test(content);
+    const _hasZod = PATTERNS6.zodSchema.test(content);
     const methods = ["GET", "POST"];
     for (const method of methods) {
       const endpoint = {
@@ -750,6 +688,68 @@ class NextApiExtractor extends BaseExtractor {
     }
   }
 }
+// src/extractors/trpc/extractor.ts
+var PATTERNS7 = {
+  procedure: /\.(query|mutation)\s*\(\s*(?:\{[^}]*\}|[^)]+)\)/gi,
+  procedureName: /(\w+)\s*:\s*(?:publicProcedure|protectedProcedure|procedure)/g,
+  zodInput: /\.input\s*\(\s*(\w+)/g,
+  zodOutput: /\.output\s*\(\s*(\w+)/g
+};
+
+class TrpcExtractor extends BaseExtractor {
+  id = "trpc";
+  name = "tRPC Extractor";
+  frameworks = ["trpc"];
+  priority = 15;
+  async doExtract(ctx) {
+    const { project, options, fs } = ctx;
+    const pattern = options.scope?.length ? options.scope.map((s) => `${s}/**/*.ts`).join(",") : "**/*.ts";
+    const files = await fs.glob(pattern, { cwd: project.rootPath });
+    ctx.ir.stats.filesScanned = files.length;
+    for (const file of files) {
+      if (file.includes("node_modules") || file.includes(".test."))
+        continue;
+      const fullPath = `${project.rootPath}/${file}`;
+      const content = await fs.readFile(fullPath);
+      if (!content.includes("trpc") && !content.includes("Procedure"))
+        continue;
+      await this.extractProcedures(ctx, file, content);
+    }
+  }
+  async extractProcedures(ctx, file, content) {
+    const nameMatches = [...content.matchAll(PATTERNS7.procedureName)];
+    for (const match of nameMatches) {
+      const procedureName = match[1] ?? "unknownProcedure";
+      const index = match.index ?? 0;
+      const lineNumber = content.slice(0, index).split(`
+`).length;
+      const afterMatch = content.slice(index, index + 500);
+      const isQuery = afterMatch.includes(".query(");
+      const isMutation = afterMatch.includes(".mutation(");
+      if (!isQuery && !isMutation)
+        continue;
+      const hasZodInput = PATTERNS7.zodInput.test(afterMatch);
+      const hasZodOutput = PATTERNS7.zodOutput.test(afterMatch);
+      const hasSchema = hasZodInput || hasZodOutput;
+      const method = isMutation ? "POST" : "GET";
+      const endpoint = {
+        id: `trpc.${procedureName}`,
+        method,
+        path: `/trpc/${procedureName}`,
+        kind: isMutation ? "command" : "query",
+        handlerName: procedureName,
+        source: this.createLocation(file, lineNumber, lineNumber + 10),
+        confidence: this.createConfidence(hasSchema ? "high" : "medium", hasSchema ? "explicit-schema" : "inferred-types"),
+        frameworkMeta: {
+          procedureType: isMutation ? "mutation" : "query",
+          hasInput: hasZodInput,
+          hasOutput: hasZodOutput
+        }
+      };
+      this.addEndpoint(ctx, endpoint);
+    }
+  }
+}
 // src/extractors/zod/extractor.ts
 var PATTERNS8 = {
   zodSchema: /(?:export\s+)?const\s+(\w+)\s*=\s*z\.(?:object|string|number|boolean|array|enum|union|intersection|literal|tuple|record)/g,

package/dist/index.d.ts

@@ -21,9 +21,9 @@
  * const ir = await extractFromProject('./my-project', { framework });
  * ```
  */
-export * from './types';
-export * from './registry';
+export * as codegen from './codegen/index';
 export * from './detect';
 export * from './extract';
 export * as extractors from './extractors/index';
-export * as codegen from './codegen/index';
+export * from './registry';
+export * from './types';