@amritk/generate-examples 0.1.1 → 0.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@amritk/generate-examples",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "description": "Generate fast-check arbitraries and example values from JSON Schemas.",
   "module": "./dist/index.js",
   "type": "module",
@@ -33,23 +33,23 @@
     "access": "public"
   },
   "scripts": {
-    "build": "bun run build:code && bun run build:types",
-    "build:code": "bun build ./src/index.ts --outdir=dist --target=node",
-    "build:types": "tsc -p ."
+    "build": "tsgo -p tsconfig.build.json && tsc-alias -p tsconfig.build.json -f",
+    "types:check": "tsgo -p . --noEmit",
+    "test": "NODE_ENV=production vitest run --root ../.. generate-examples"
   },
   "imports": {
     "#generators/*": "./src/generators/*.ts"
   },
   "exports": {
     ".": {
-      "bun": "./src/index.ts",
+      "development": "./src/index.ts",
       "default": "./dist/index.js",
       "types": "./dist/index.d.ts"
     }
   },
   "dependencies": {
     "json-schema-typed": "^8.0.1",
-    "@amritk/helpers": "0.6.2"
+    "@amritk/helpers": "0.7.0"
   },
   "peerDependencies": {
     "fast-check": ">=3"

package/src/generators/build-schema.ts

@@ -1,10 +1,5 @@
-import { buildDynamicRefMap } from '@amritk/helpers/build-dynamic-ref-map'
-import { extractRefs } from '@amritk/helpers/extract-refs'
-import { refToFilename } from '@amritk/helpers/ref-to-filename'
-import { refToName } from '@amritk/helpers/ref-to-name'
-import { resolveDynamicRefs } from '@amritk/helpers/resolve-dynamic-refs'
-import { resolveRef } from '@amritk/helpers/resolve-ref'
-import { upgradeDraft07Schema } from '@amritk/helpers/upgrade-draft07-schema'
+import { generateIndexBarrel } from '@amritk/helpers/generate-index-barrel'
+import { walkRefGraph } from '@amritk/helpers/walk-ref-graph'
 import type { JSONSchema } from 'json-schema-typed/draft-2020-12'
 
 import { generateExampleFile } from './generate-files'
@@ -19,7 +14,8 @@ export type GeneratedFile = {
 
 /**
  * Builds all TypeScript example files from a JSON Schema by traversing all
- * $ref references recursively, mirroring the generate-parsers pipeline.
+ * `$ref` / `$dynamicRef` references recursively (via the shared
+ * `@amritk/helpers/walk-ref-graph` walker).
  *
  * Each generated file exports:
  * - A TypeScript type definition
@@ -45,83 +41,22 @@ export const buildExampleSchema = async (
   rootTypeName: string,
   typeSuffix = '',
 ): Promise<GeneratedFile[]> => {
-  rootSchema = upgradeDraft07Schema(rootSchema as Record<string, unknown>) as JSONSchema
-
   const files: GeneratedFile[] = []
-  const processedRefs = new Set<string>()
-  const processedFilenames = new Set<string>()
-  const refsToProcess: string[] = []
-
-  const dynamicRefMap = buildDynamicRefMap(rootSchema)
-
-  // Root schema
-  const processedRootSchema = resolveDynamicRefs(rootSchema, dynamicRefMap)
-  const rootContent = generateExampleFile(processedRootSchema, rootTypeName, {
-    rootSchema: rootSchema as Record<string, unknown>,
-    typeSuffix,
-  })
-  const rootFilename = rootTypeName.toLowerCase()
 
-  if (rootFilename !== 'index') {
-    processedFilenames.add(rootFilename)
-    files.push({ filename: `${rootFilename}.ts`, content: rootContent })
-  }
+  walkRefGraph(rootSchema, rootTypeName, { typeSuffix }, (node) => {
+    // `index` is reserved for the barrel below, so never let a definition of
+    // that name overwrite it.
+    if (node.filename === 'index') return
 
-  const rootRefs = extractRefs(rootSchema)
-  refsToProcess.push(...rootRefs)
-
-  while (refsToProcess.length > 0) {
-    const ref = refsToProcess.shift()
-    if (!ref || processedRefs.has(ref)) continue
-    processedRefs.add(ref)
-
-    const resolvedSchema = resolveRef(ref, rootSchema as Record<string, unknown>)
-    if (!resolvedSchema) {
-      console.warn(`Warning: Could not resolve ref: ${ref}`)
-      continue
-    }
-
-    const typeName = refToName(ref, typeSuffix)
-    const filename = refToFilename(ref)
-    const processedSchema = resolveDynamicRefs(resolvedSchema as JSONSchema, dynamicRefMap)
-    const content = generateExampleFile(processedSchema, typeName, {
-      selfRef: ref,
-      rootSchema: rootSchema as Record<string, unknown>,
+    const content = generateExampleFile(node.schema, node.typeName, {
+      rootSchema: node.rootSchema,
       typeSuffix,
+      ...(node.ref !== undefined ? { selfRef: node.ref } : {}),
     })
+    files.push({ filename: `${node.filename}.ts`, content })
+  })
 
-    if (filename !== 'index' && !processedFilenames.has(filename)) {
-      processedFilenames.add(filename)
-      files.push({ filename: `${filename}.ts`, content })
-    }
-
-    for (const nestedRef of extractRefs(resolvedSchema as JSONSchema)) {
-      if (!processedRefs.has(nestedRef)) refsToProcess.push(nestedRef)
-    }
-  }
-
-  // Generate index.ts barrel
-  const TYPE_EXPORT_RE = /^export type (\w+)/gm
-  const CONST_EXPORT_RE = /^export const (\w+)/gm
-
-  const sortedFiles = [...files].sort((a, b) => a.filename.localeCompare(b.filename))
-  let indexContent = ''
-
-  for (const file of sortedFiles) {
-    const moduleName = file.filename.replace(/\.ts$/, '')
-    const typeNames: string[] = []
-    const constNames: string[] = []
-
-    for (const match of file.content.matchAll(TYPE_EXPORT_RE)) typeNames.push(match[1] as string)
-    for (const match of file.content.matchAll(CONST_EXPORT_RE)) constNames.push(match[1] as string)
-
-    if (typeNames.length === 0 && constNames.length === 0) continue
-
-    const typeExports = typeNames.map((n) => `type ${n}`)
-    indexContent += `export { ${[...typeExports, ...constNames].join(', ')} } from './${moduleName}';\n`
-  }
-
-  files.push({ filename: 'index.ts', content: indexContent })
+  files.push({ filename: 'index.ts', content: generateIndexBarrel(files) })
 
   return files
 }