@pikku/inspector 0.12.94 → 0.12.96

package/dist/utils/schema-generator.js

@@ -2,7 +2,7 @@ import * as ts from 'typescript';
 import { createHash } from 'crypto';
 import { existsSync, mkdirSync, readFileSync, statSync, writeFileSync, } from 'fs';
 import { dirname, join, resolve } from 'path';
-import { createGenerator, RootlessError } from 'ts-json-schema-generator';
+import { createFormatter, createParser, DEFAULT_CONFIG, RootlessError, SchemaGenerator, } from 'ts-json-schema-generator';
 import { register } from 'tsx/esm/api';
 import * as z from 'zod';
 import { findZodTransform } from './find-zod-transform.js';
@@ -184,6 +184,34 @@ export function writeDiskTSSchemas(logger, cacheDir, key, schemas, deps) {
         logger.debug(`Failed to persist TS-schema cache: ${e.message}`);
     }
 }
+// The base class walks every source file again for each type it looks up by
+// name, which was nearly all of schema generation (8.7s for 677 types on e2e).
+class IndexedSchemaGenerator extends SchemaGenerator {
+    projectTypes;
+    allTypes;
+    findNamedNode(fullName) {
+        const typeChecker = this.program.getTypeChecker();
+        if (!this.projectTypes) {
+            this.projectTypes = new Map();
+            this.appendTypes(this.partitionFiles().projectFiles, typeChecker, this.projectTypes);
+        }
+        const projectType = this.projectTypes.get(fullName);
+        if (projectType)
+            return projectType;
+        if (!this.allTypes) {
+            this.allTypes = new Map(this.projectTypes);
+            this.appendTypes(this.partitionFiles().externalFiles, typeChecker, this.allTypes);
+        }
+        const type = this.allTypes.get(fullName);
+        if (type)
+            return type;
+        throw new RootlessError(fullName);
+    }
+}
+const createIndexedGenerator = (program, config) => {
+    const completedConfig = { ...DEFAULT_CONFIG, ...config };
+    return new IndexedSchemaGenerator(program, createParser(program, completedConfig), createFormatter(completedConfig), completedConfig);
+};
 function createProgramWithVirtualFile(logger, tsconfig, virtualFilePath, virtualFileContent) {
     const configPath = resolve(tsconfig);
     // Cache the parsed tsconfig — it doesn't change between runs
@@ -271,8 +299,7 @@ function generateTSSchemas(logger, tsconfig, customTypesContent, typesMap, funct
     logger.debug(`generateTSSchemas needed for ${uncoveredSchemas.length} types: ${uncoveredSchemas.slice(0, 3).join(', ')}${uncoveredSchemas.length > 3 ? '...' : ''}`);
     const virtualFilePath = join(dirname(resolve(tsconfig)), '__pikku_virtual_types__.ts');
     const program = createProgramWithVirtualFile(logger, tsconfig, virtualFilePath, customTypesContent);
-    const generator = createGenerator({
-        tsProgram: program,
+    const generator = createIndexedGenerator(program, {
         skipTypeCheck: true,
         topRef: false,
         discriminatorType: 'open-api',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pikku/inspector",
-  "version": "0.12.94",
+  "version": "0.12.96",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/pikkujs/pikku.git",
@@ -54,10 +54,10 @@
     "zod": "^4.4.3"
   },
   "peerDependencies": {
-    "@pikku/core": "^0.12.128"
+    "@pikku/core": "^0.12.130"
   },
   "devDependencies": {
-    "@pikku/core": "^0.12.128",
+    "@pikku/core": "^0.12.130",
     "@types/node": "^24.13.3"
   },
   "engines": {