@effect/openapi-generator 4.0.0-beta.6 → 4.0.0-beta.62

package/src/JsonSchemaGenerator.ts

@@ -3,6 +3,19 @@ import * as JsonSchema from "effect/JsonSchema"
 import * as Rec from "effect/Record"
 import * as SchemaRepresentation from "effect/SchemaRepresentation"
 
+type Source = "openapi-3.0" | "openapi-3.1"
+
+interface GenerateOptions {
+  readonly onEnter?: ((js: JsonSchema.JsonSchema) => JsonSchema.JsonSchema) | undefined
+}
+
+interface GenerateHttpApiOptions extends GenerateOptions {
+  readonly multipartSchemaRefs?: {
+    readonly singleFile: string
+    readonly files: string
+  } | undefined
+}
+
 export function make() {
   const store: Record<string, JsonSchema.JsonSchema> = {}
 
@@ -15,80 +28,260 @@ export function make() {
   }
 
   function generate(
-    source: "openapi-3.0" | "openapi-3.1",
+    source: Source,
     components: JsonSchema.Definitions,
     typeOnly: boolean,
-    options?: {
-      readonly onEnter?: ((js: JsonSchema.JsonSchema) => JsonSchema.JsonSchema) | undefined
+    options?: GenerateOptions
+  ) {
+    const generated = makeCodeDocument(source, components, options)
+    if (generated === undefined) {
+      return ""
     }
+
+    const nonRecursiveReferences = generated.codeDocument.references.nonRecursives
+    const recursiveReferences = Object.entries(generated.codeDocument.references.recursives)
+
+    const nonRecursives = nonRecursiveReferences.map(({ $ref, code }) =>
+      renderSchemaTypeAndRuntime($ref, code, typeOnly)
+    )
+
+    const recursiveDeclarations: Array<string> = []
+    const recursives: Array<string> = []
+
+    if (typeOnly) {
+      for (const [$ref, code] of recursiveReferences) {
+        recursives.push(renderSchemaTypeAndRuntime($ref, code, true))
+      }
+    } else {
+      const recursivelyForwardReferenced = collectForwardReferencedRecursives(
+        nonRecursiveReferences,
+        recursiveReferences
+      )
+      const recursiveInternalNames = makeRecursiveInternalNameMap(
+        recursivelyForwardReferenced,
+        [
+          ...nonRecursiveReferences.map(({ $ref }) => $ref),
+          ...recursiveReferences.map(([$ref]) => $ref),
+          ...generated.nameMap
+        ]
+      )
+
+      for (const [$ref, code] of recursiveReferences) {
+        if (recursivelyForwardReferenced.has($ref)) {
+          const internalName = recursiveInternalNames.get($ref)!
+          recursiveDeclarations.push(renderRecursiveReferenceDeclaration($ref, code, internalName))
+          recursives.push(`const ${internalName} = ${code.runtime}`)
+          continue
+        }
+
+        recursives.push(renderSchemaTypeAndRuntime($ref, code, false))
+      }
+    }
+
+    const codes = generated.codeDocument.codes.map((code, i) =>
+      renderSchemaTypeAndRuntime(generated.nameMap[i], code, typeOnly)
+    )
+
+    return render("recursive declarations", recursiveDeclarations) +
+      render("non-recursive definitions", nonRecursives) +
+      render("recursive definitions", recursives) +
+      render("schemas", codes)
+  }
+
+  function generateHttpApi(
+    source: Source,
+    components: JsonSchema.Definitions,
+    options?: GenerateHttpApiOptions
   ) {
+    const generated = makeCodeDocument(source, components, options)
+    if (generated === undefined) {
+      return ""
+    }
+
+    const nonRecursiveReferences = generated.codeDocument.references.nonRecursives
+    const recursiveReferences = Object.entries(generated.codeDocument.references.recursives)
+
+    const nonRecursives = nonRecursiveReferences.map(({ $ref, code }) =>
+      renderSchemaTypeAndRuntime($ref, code, false, options?.multipartSchemaRefs)
+    )
+
+    const recursivelyForwardReferenced = collectForwardReferencedRecursives(nonRecursiveReferences, recursiveReferences)
+    const recursiveInternalNames = makeRecursiveInternalNameMap(
+      recursivelyForwardReferenced,
+      [
+        ...nonRecursiveReferences.map(({ $ref }) => $ref),
+        ...recursiveReferences.map(([$ref]) => $ref),
+        ...generated.nameMap
+      ]
+    )
+
+    const recursiveDeclarations: Array<string> = []
+    const recursives: Array<string> = []
+
+    for (const [$ref, code] of recursiveReferences) {
+      if (recursivelyForwardReferenced.has($ref)) {
+        const internalName = recursiveInternalNames.get($ref)!
+        recursiveDeclarations.push(renderRecursiveReferenceDeclaration($ref, code, internalName))
+        recursives.push(`const ${internalName} = ${code.runtime}`)
+        continue
+      }
+
+      recursives.push(renderSchemaTypeAndRuntime($ref, code, false, options?.multipartSchemaRefs))
+    }
+
+    const codes = generated.codeDocument.codes.map((code, i) =>
+      renderSchemaTypeAndRuntime(generated.nameMap[i], code, false, options?.multipartSchemaRefs)
+    )
+
+    return render("recursive declarations", recursiveDeclarations) +
+      render("non-recursive definitions", nonRecursives) +
+      render("recursive definitions", recursives) +
+      render("schemas", codes)
+  }
+
+  function makeCodeDocument(
+    source: Source,
+    components: JsonSchema.Definitions,
+    options?: GenerateOptions
+  ): {
+    readonly nameMap: Array<string>
+    readonly codeDocument: SchemaRepresentation.CodeDocument
+  } | undefined {
     const nameMap: Array<string> = []
     const schemas: Array<JsonSchema.JsonSchema> = []
 
     const definitions: JsonSchema.Definitions = Rec.map(
       components,
-      (js) => fromSchemaOpenApi(js).schema
+      (js) => fromSchemaOpenApi(source, js).schema
     )
 
     for (const [name, js] of Object.entries(store)) {
       nameMap.push(name)
-      schemas.push(fromSchemaOpenApi(js).schema)
+      schemas.push(fromSchemaOpenApi(source, js).schema)
+    }
+
+    if (!Arr.isArrayNonEmpty(schemas)) {
+      return
     }
 
-    if (Arr.isArrayNonEmpty(schemas)) {
-      const multiDocument: SchemaRepresentation.MultiDocument = SchemaRepresentation.fromJsonSchemaMultiDocument({
-        dialect: "draft-2020-12",
-        schemas,
-        definitions
-      }, {
-        onEnter(js) {
-          const out = { ...js }
-          if (out.type === "object" && out.additionalProperties === undefined) {
-            out.additionalProperties = false
-          }
-          return options?.onEnter?.(out) ?? out
+    const multiDocument: SchemaRepresentation.MultiDocument = SchemaRepresentation.fromJsonSchemaMultiDocument({
+      dialect: "draft-2020-12",
+      schemas,
+      definitions
+    }, {
+      onEnter(js) {
+        const out = { ...js }
+        if (out.type === "object" && out.additionalProperties === undefined) {
+          out.additionalProperties = false
         }
-      })
+        return options?.onEnter?.(out) ?? out
+      }
+    })
 
-      const codeDocument = SchemaRepresentation.toCodeDocument(multiDocument)
+    return {
+      nameMap,
+      codeDocument: SchemaRepresentation.toCodeDocument(multiDocument)
+    }
+  }
 
-      const nonRecursives = codeDocument.references.nonRecursives.map(({ $ref, code }) => renderSchema($ref, code))
-      const recursives = Object.entries(codeDocument.references.recursives).map(([$ref, code]) =>
-        renderSchema($ref, code)
-      )
-      const codes = codeDocument.codes.map((code, i) => renderSchema(nameMap[i], code))
+  return { addSchema, generate, generateHttpApi } as const
+}
 
-      const s = render("non-recursive definitions", nonRecursives) +
-        render("recursive definitions", recursives) +
-        render("schemas", codes)
+function fromSchemaOpenApi(source: Source, jsonSchema: JsonSchema.JsonSchema) {
+  switch (source) {
+    case "openapi-3.1":
+      return JsonSchema.fromSchemaOpenApi3_1(jsonSchema)
+    case "openapi-3.0":
+      return JsonSchema.fromSchemaOpenApi3_0(jsonSchema)
+  }
+}
 
-      return s
-    } else {
-      return ""
+function renderSchemaTypeAndRuntime(
+  $ref: string,
+  code: SchemaRepresentation.Code,
+  typeOnly: boolean,
+  multipartSchemaRefs?: {
+    readonly singleFile: string
+    readonly files: string
+  }
+) {
+  if (!typeOnly && multipartSchemaRefs !== undefined) {
+    if ($ref === multipartSchemaRefs.singleFile) {
+      return [
+        `export type ${$ref} = Multipart.PersistedFile`,
+        `export const ${$ref} = Multipart.SingleFileSchema`
+      ].join("\n")
     }
-
-    function fromSchemaOpenApi(jsonSchema: JsonSchema.JsonSchema) {
-      switch (source) {
-        case "openapi-3.1":
-          return JsonSchema.fromSchemaOpenApi3_1(jsonSchema)
-        case "openapi-3.0":
-          return JsonSchema.fromSchemaOpenApi3_0(jsonSchema)
-      }
+    if ($ref === multipartSchemaRefs.files) {
+      return [
+        `export type ${$ref} = ReadonlyArray<Multipart.PersistedFile>`,
+        `export const ${$ref} = Multipart.FilesSchema`
+      ].join("\n")
     }
+  }
+
+  const strings = [`export type ${$ref} = ${code.Type}`]
+  if (!typeOnly) {
+    strings.push(`export const ${$ref} = ${code.runtime}`)
+  }
+  return strings.join("\n")
+}
+
+function renderRecursiveReferenceDeclaration(
+  $ref: string,
+  code: SchemaRepresentation.Code,
+  internalName: string
+): string {
+  return [
+    `export type ${$ref} = ${code.Type}`,
+    `export const ${$ref} = Schema.suspend((): Schema.Codec<${$ref}> => ${internalName})`
+  ].join("\n")
+}
+
+function render(title: string, as: ReadonlyArray<string>) {
+  if (as.length === 0) return ""
+  return "// " + title + "\n" + as.join("\n") + "\n"
+}
+
+const tokenPattern = /[A-Za-z_$][A-Za-z0-9_$]*/g
+
+function collectForwardReferencedRecursives(
+  nonRecursives: ReadonlyArray<{
+    readonly $ref: string
+    readonly code: SchemaRepresentation.Code
+  }>,
+  recursives: ReadonlyArray<readonly [string, SchemaRepresentation.Code]>
+): Set<string> {
+  const recursiveNames = new Set(recursives.map(([name]) => name))
+  const referenced = new Set<string>()
 
-    function renderSchema($ref: string, code: SchemaRepresentation.Code) {
-      const strings = [`export type ${$ref} = ${code.Type}`]
-      if (!typeOnly) {
-        strings.push(`export const ${$ref} = ${code.runtime}`)
+  for (const { code } of nonRecursives) {
+    for (const token of code.runtime.matchAll(tokenPattern)) {
+      const identifier = token[0]
+      if (recursiveNames.has(identifier)) {
+        referenced.add(identifier)
       }
-      return strings.join("\n")
     }
+  }
+
+  return referenced
+}
+
+function makeRecursiveInternalNameMap(
+  recursiveNames: ReadonlySet<string>,
+  existingNames: ReadonlyArray<string>
+): Map<string, string> {
+  const usedNames = new Set(existingNames)
+  const internalNames = new Map<string, string>()
 
-    function render(title: string, as: ReadonlyArray<string>) {
-      if (as.length === 0) return ""
-      return "// " + title + "\n" + as.join("\n") + "\n"
+  for (const name of recursiveNames) {
+    let candidate = `__recursive_${name}`
+    while (usedNames.has(candidate)) {
+      candidate = `_${candidate}`
     }
+    usedNames.add(candidate)
+    internalNames.set(name, candidate)
   }
 
-  return { addSchema, generate } as const
+  return internalNames
 }