@amartus/oas-utils 0.1.0 → 0.3.0

package/README.md

@@ -1,6 +1,6 @@
 # oas-utils
 
-Utilities for working with OpenAPI (OAS) documents. Includes tools to remove unused schemas and to remove entries from oneOf. Use them as a CLI or as Redocly decorators.
+Utilities for working with OpenAPI (OAS) documents. Includes tools to remove unused schemas, remove entries from oneOf, optimize allOf composition, convert allOf + discriminator patterns to oneOf + discriminator, and clean up discriminator mappings. Use them as a CLI or as Redocly decorators.
 
 ## Definition of "unused schema"
 
@@ -26,6 +26,8 @@ This makes the `oas-utils` CLI available system-wide.
 
 ## CLI usage
 
+### remove-unused
+
 ```
 oas-utils remove-unused <input.yaml> -o output.yaml [--keep Name1 Name2] [--aggressive] [--ignore-parents NameX]
 # Read from stdin and write to stdout
@@ -37,6 +39,8 @@ Options:
 - --aggressive: also prune other unused components referenced from paths (responses, headers, requestBodies, parameters, examples, links, callbacks, securitySchemes). Non-referenced entries in these sections are removed.
 - --ignore-parents: schema names that shouldn't promote children via allOf (can be repeated). Useful to avoid allOf upward inclusion when the parent acts as an abstract/base.
 
+### remove-oneof
+
 Remove entries from oneOf and update discriminators:
 
 ```
@@ -47,11 +51,109 @@ oas-utils remove-oneof <input.yaml> --remove Cat
 oas-utils remove-oneof <input.yaml> --parent Pet --remove Cat --guess
 ```
 
-Options (remove-oneof):
+Options:
 - --parent: parent schema name containing oneOf; if omitted, removal is global across the document.
 - --remove: schema name(s) to remove; can be repeated.
 - --guess: expand each name to include variants starting with `<name>_`.
 
+### optimize-allof
+
+Optimize allOf composition by removing redundant references:
+
+```
+oas-utils optimize-allof <input.yaml> -o output.yaml
+```
+
+Options:
+- -o, --output: write result to this file (defaults to stdout).
+
+### allof-to-oneof
+
+Convert allOf + discriminator patterns to oneOf + discriminator. This is useful for transforming inheritance-based polymorphic schemas into composition-based ones.
+
+Specifically, it:
+1. Identifies base schemas with discriminators
+2. Finds concrete schemas that extend the base via allOf
+3. Optionally adds a const property to each concrete schema matching its discriminator value (enabled by default)
+4. Creates a new oneOf wrapper schema containing all concrete types
+5. Replaces references to the base schema with the wrapper schema (in polymorphic contexts)
+
+```
+oas-utils allof-to-oneof <input.yaml> -o output.yaml
+# Optionally remove discriminator from base schema
+oas-utils allof-to-oneof <input.yaml> -o output.yaml --remove-discriminator-from-base
+# Optionally skip adding const to specialization schemas
+oas-utils allof-to-oneof <input.yaml> -o output.yaml --no-add-discriminator-const
+# Optionally skip transformation if only one specialization is found
+oas-utils allof-to-oneof <input.yaml> -o output.yaml --ignore-single-specialization
+```
+
+Options:
+- -o, --output: write result to this file (defaults to stdout).
+- --remove-discriminator-from-base: remove the discriminator from base schemas after conversion.
+- --no-add-discriminator-const: do not add const property with discriminator value to specialization schemas.
+- --ignore-single-specialization: skip oneOf transformation if only one specialization is found (useful for bases with only one concrete implementation).
+
+Example transformation (with addDiscriminatorConst enabled, the default):
+- Base schema `Animal` with discriminator `type` and mapping `{Cat: ..., Dog: ...}`
+- Concrete schemas `Cat` and `Dog` with `allOf: [{$ref: Animal}, {...}]`
+- Creates `AnimalPolymorphic` with `oneOf: [Cat, Dog]` and the same discriminator
+- Adds `type: {const: "Cat"}` to Cat's properties and `type: {const: "Dog"}` to Dog's properties
+- Replaces references to `Animal` with `AnimalPolymorphic` in array items and other polymorphic contexts
+
+### cleanup-discriminators
+
+Clean up discriminator mappings by removing entries that reference non-existent schemas. This is useful when schemas are removed but discriminator mappings are not updated, leaving dangling references.
+
+```
+oas-utils cleanup-discriminators <input.yaml> -o output.yaml
+# Read from stdin and write to stdout
+cat openapi.yaml | oas-utils cleanup-discriminators > cleaned.yaml
+```
+
+Options:
+- -o, --output: write result to this file (defaults to stdout).
+
+Example:
+- Original discriminator mapping: `{cat: '#/components/schemas/Cat', dog: '#/components/schemas/Dog', bird: '#/components/schemas/Bird'}`
+- After removing `Bird` schema: mapping entries `bird` is invalid
+- After cleanup: `{cat: '#/components/schemas/Cat', dog: '#/components/schemas/Dog'}`
+
+### seal-schema
+
+Seal object schemas to prevent additional properties. This ensures every final object shape exposed in the API is sealed (no additional properties allowed), without breaking schemas that are extended via `allOf`.
+
+The algorithm:
+1. Identifies schemas that are bases for extension (referenced in `allOf`)
+2. For each such schema, creates a `Core` variant and a sealed wrapper
+3. Rewrites `allOf` references to point to `Core` variants (allowing extension)
+4. Seals composition roots (`allOf`, `anyOf`, `oneOf`) and direct-use schemas
+
+This ensures:
+- **Objects used directly** in fields or arrays are fully sealed
+- **Objects used as bases** for extension remain unsealed internally but are sealed at the wrapper level
+- **anyOf/oneOf compositions** remain valid alternatives and are sealed at the outer level
+
+```
+oas-utils seal-schema <input.yaml> -o output.yaml
+# Use unevaluatedProperties: false (default, recommended for JSON Schema 2019-09+)
+oas-utils seal-schema <input.yaml> -o output.yaml --use-unevaluated-properties
+# Use additionalProperties: false instead
+oas-utils seal-schema <input.yaml> -o output.yaml --use-additional-properties
+```
+
+Options:
+- -o, --output: write result to this file (defaults to stdout).
+- --use-unevaluated-properties: use `unevaluatedProperties: false` (default, better for JSON Schema 2019-09+).
+- --use-additional-properties: use `additionalProperties: false` instead.
+
+Example transformation:
+- Original `Animal` schema (object-like, referenced in `allOf` by `Cat`)
+- Becomes: `AnimalCore` (unsealed original) + `Animal` wrapper with `allOf: [{$ref: AnimalCore}]` and `unevaluatedProperties: false`
+- `Cat` now uses `allOf: [{$ref: AnimalCore}, {...}]` allowing safe extension
+- Direct references to `Animal` point to the sealed wrapper
+- Inline objects and composition roots are sealed with appropriate keywords
+
 ## As Redocly decorators
 
 1) Add the plugin to `plugins` in your `redocly.yaml` (path relative to the config):
@@ -61,7 +163,7 @@ plugins:
   - ./node_modules/oas-utils/dist/redocly/plugin.js
 ```
 
-2) Enable the decorator:
+2) Enable the decorators:
 
 ```
 decorators:
@@ -75,6 +177,22 @@ decorators:
     parent: Pet             # optional; if omitted, removal is global
     remove: [Cat, Cat_variant1]
     guess: false
+
+  # Optimize allOf composition
+  oas-utils/optimize-allof: {}
+
+  # Convert allOf + discriminator to oneOf + discriminator
+  oas-utils/allof-to-oneof:
+    removeDiscriminatorFromBase: false
+    addDiscriminatorConst: true
+    ignoreSingleSpecialization: false
+
+  # Clean up discriminator mappings
+  oas-utils/cleanup-discriminators: {}
+
+  # Seal object schemas
+  oas-utils/seal-schema:
+    useUnevaluatedProperties: true
 ```
 
 3) Run bundling with Redocly CLI and the decorators will apply the transformations. With `aggressive: true`, unused non-schema components (responses, headers, requestBodies, etc.) are removed as well.
@@ -85,8 +203,20 @@ Notes:
 ## Programmatic usage
 
 ```
-import { removeUnusedSchemas } from 'oas-utils';
-const pruned = removeUnusedSchemas(doc, { keep: ['CommonError'], aggressive: true });
+import { removeUnusedSchemas, allOfToOneOf, sealSchema, cleanupDiscriminatorMappings } from 'oas-utils';
+
+// Remove unused schemas
+removeUnusedSchemas(doc, { keep: ['CommonError'], aggressive: true });
+
+// Convert allOf + discriminator to oneOf + discriminator
+allOfToOneOf(doc, { removeDiscriminatorFromBase: false, addDiscriminatorConst: true });
+
+// Clean up discriminator mappings
+const result = cleanupDiscriminatorMappings(doc);
+console.log(`Removed ${result.mappingsRemoved} invalid mappings from ${result.schemasChecked} schemas`);
+
+// Seal object schemas
+sealSchema(doc, { useUnevaluatedProperties: true });
 ```
 
 ## Notes

package/dist/cli.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import fs from "node:fs/promises";
 import { Command } from "commander";
-import { runRemoveUnused, runRemoveOneOf, optimizeAllOf } from "./lib/cliActions.js";
+import { runRemoveUnused, runRemoveOneOf, optimizeAllOf, runAllOfToOneOf, runSealSchema, runCleanupDiscriminators } from "./lib/cliActions.js";
 import YAML from "yaml";
 const program = new Command();
 program
@@ -49,6 +49,57 @@ program
     .action(async (input, opts) => {
     await optimizeAllOf(opts, format, () => reader(input));
 });
+program
+    .command("allof-to-oneof")
+    .showHelpAfterError()
+    .description("Convert allOf + discriminator patterns to oneOf + discriminator")
+    .argument("[input]", "Path to input OpenAPI file (YAML or JSON). If omitted, reads from stdin")
+    .option("-o, --output <file>", "Write result to this file (defaults to stdout)")
+    .option("--remove-discriminator-from-base", "Remove discriminator from base schemas after conversion", false)
+    .option("--no-add-discriminator-const", "Do not add const property with discriminator value to specialization schemas", true)
+    .option("--ignore-single-specialization", "Skip oneOf transformation if only one specialization is found", false)
+    .action(async (input, opts) => {
+    try {
+        await runAllOfToOneOf(opts, format, () => reader(input));
+    }
+    catch (err) {
+        console.error(`Error: ${err?.message || String(err)}`);
+        process.exitCode = 1;
+    }
+});
+program
+    .command("seal-schema")
+    .showHelpAfterError()
+    .description("Seal object schemas to prevent additional properties")
+    .argument("[input]", "Path to input OpenAPI file (YAML or JSON). If omitted, reads from stdin")
+    .option("-o, --output <file>", "Write result to this file (defaults to stdout)")
+    .option("--use-unevaluated-properties", "Use unevaluatedProperties: false instead of additionalProperties: false (default: true)", true)
+    .option("--use-additional-properties", "Use additionalProperties: false instead of unevaluatedProperties: false", false)
+    .action(async (input, opts) => {
+    try {
+        const useUnevaluated = !opts.useAdditionalProperties;
+        await runSealSchema({ output: opts.output, useUnevaluatedProperties: useUnevaluated }, format, () => reader(input));
+    }
+    catch (err) {
+        console.error(`Error: ${err?.message || String(err)}`);
+        process.exitCode = 1;
+    }
+});
+program
+    .command("cleanup-discriminators")
+    .showHelpAfterError()
+    .description("Clean up discriminator mappings by removing references to non-existent schemas")
+    .argument("[input]", "Path to input OpenAPI file (YAML or JSON). If omitted, reads from stdin")
+    .option("-o, --output <file>", "Write result to this file (defaults to stdout)")
+    .action(async (input, opts) => {
+    try {
+        await runCleanupDiscriminators(opts, format, () => reader(input));
+    }
+    catch (err) {
+        console.error(`Error: ${err?.message || String(err)}`);
+        process.exitCode = 1;
+    }
+});
 if (process.argv.length <= 2) {
     program.help();
 }

package/dist/cli.js.map

@@ -1 +1 @@
-{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACrF,OAAO,IAAI,MAAM,MAAM,CAAC;AAGxB,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAC9B,OAAO;KACJ,OAAO,CAAC,eAAe,CAAC;KACxB,kBAAkB,EAAE;KACpB,WAAW,CAAC,kDAAkD,CAAC;KAC/D,QAAQ,CACP,SAAS,EACT,yEAAyE,CAC1E;KACA,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CACL,mBAAmB,EACnB,4DAA4D,EAC5D,EAAE,CACH;KACA,MAAM,CAAC,cAAc,EAAE,qCAAqC,EAAE,KAAK,CAAC;KACpE,MAAM,CACL,6BAA6B,EAC7B,2DAA2D,EAC3D,EAAE,CACH;KACA,MAAM,CACL,KAAK,EACH,KAAyB,EACzB,IAKC,EACD,EAAE;IACF,IAAI,CAAC;QACH,MAAM,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3D,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;AACH,CAAC,CACF,CAAC;AAEJ,OAAO;KACJ,OAAO,CAAC,cAAc,CAAC;KACvB,kBAAkB,EAAE;KACpB,WAAW,CAAC,yEAAyE,CAAC;KACtF,QAAQ,CAAC,SAAS,EAAE,2CAA2C,CAAC;KAChE,MAAM,CAAC,iBAAiB,EAAE,qCAAqC,CAAC;KAChE,cAAc,CAAC,oBAAoB,EAAE,uDAAuD,CAAC;KAC7F,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CAAC,SAAS,EAAE,6CAA6C,EAAE,KAAK,CAAC;KACvE,MAAM,CACL,KAAK,EACH,KAAa,EACb,IAAuF,EACvF,EAAE;IACF,IAAI,CAAC;QACH,MAAM,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC1D,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;AACH,CAAC,CACF,CAAC;AAEJ,OAAO;KACJ,OAAO,CAAC,gBAAgB,CAAC;KACzB,kBAAkB,EAAE;KACpB,WAAW,CAAC,qDAAqD,CAAC;KAClE,QAAQ,CACP,SAAS,EACT,yEAAyE,CAC1E;KACA,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CAAC,KAAK,EAAE,KAAyB,EAAE,IAAyB,EAAE,EAAE;IACrE,MAAM,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AACzD,CAAC,CAAC,CAAC;AAEL,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;IAC7B,OAAO,CAAC,IAAI,EAAE,CAAC;AACjB,CAAC;KAAM,CAAC;IACN,OAAO,CAAC,KAAK,EAAE,CAAC;AAClB,CAAC;AAED,KAAK,UAAU,SAAS;IACtB,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,OAAO,CAAC,KAAK;QAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACzE,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AAChD,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,KAAa;IAEb,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC7C,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,UAAU,MAAM,CAAC,KAAyB;IAC7C,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IACD,OAAO,SAAS,EAAE,CAAC;AACrB,CAAC;AAGD,SAAS,MAAM,CAAC,GAAQ,EAAE,MAAe;IACvC,MAAM,MAAM,GAAG,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;IACzC,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;AAC5E,CAAC"}
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,aAAa,EAAE,eAAe,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAC/I,OAAO,IAAI,MAAM,MAAM,CAAC;AAGxB,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAC9B,OAAO;KACJ,OAAO,CAAC,eAAe,CAAC;KACxB,kBAAkB,EAAE;KACpB,WAAW,CAAC,kDAAkD,CAAC;KAC/D,QAAQ,CACP,SAAS,EACT,yEAAyE,CAC1E;KACA,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CACL,mBAAmB,EACnB,4DAA4D,EAC5D,EAAE,CACH;KACA,MAAM,CAAC,cAAc,EAAE,qCAAqC,EAAE,KAAK,CAAC;KACpE,MAAM,CACL,6BAA6B,EAC7B,2DAA2D,EAC3D,EAAE,CACH;KACA,MAAM,CACL,KAAK,EACH,KAAyB,EACzB,IAKC,EACD,EAAE;IACF,IAAI,CAAC;QACH,MAAM,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3D,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;AACH,CAAC,CACF,CAAC;AAEJ,OAAO;KACJ,OAAO,CAAC,cAAc,CAAC;KACvB,kBAAkB,EAAE;KACpB,WAAW,CAAC,yEAAyE,CAAC;KACtF,QAAQ,CAAC,SAAS,EAAE,2CAA2C,CAAC;KAChE,MAAM,CAAC,iBAAiB,EAAE,qCAAqC,CAAC;KAChE,cAAc,CAAC,oBAAoB,EAAE,uDAAuD,CAAC;KAC7F,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CAAC,SAAS,EAAE,6CAA6C,EAAE,KAAK,CAAC;KACvE,MAAM,CACL,KAAK,EACH,KAAa,EACb,IAAuF,EACvF,EAAE;IACF,IAAI,CAAC;QACH,MAAM,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC1D,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;AACH,CAAC,CACF,CAAC;AAEJ,OAAO;KACJ,OAAO,CAAC,gBAAgB,CAAC;KACzB,kBAAkB,EAAE;KACpB,WAAW,CAAC,qDAAqD,CAAC;KAClE,QAAQ,CACP,SAAS,EACT,yEAAyE,CAC1E;KACA,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CAAC,KAAK,EAAE,KAAyB,EAAE,IAAyB,EAAE,EAAE;IACrE,MAAM,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AACzD,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,gBAAgB,CAAC;KACzB,kBAAkB,EAAE;KACpB,WAAW,CAAC,iEAAiE,CAAC;KAC9E,QAAQ,CACP,SAAS,EACT,yEAAyE,CAC1E;KACA,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CACL,kCAAkC,EAClC,yDAAyD,EACzD,KAAK,CACN;KACA,MAAM,CACL,8BAA8B,EAC9B,8EAA8E,EAC9E,IAAI,CACL;KACA,MAAM,CACL,gCAAgC,EAChC,+DAA+D,EAC/D,KAAK,CACN;KACA,MAAM,CACL,KAAK,EACH,KAAyB,EACzB,IAAuI,EACvI,EAAE;IACF,IAAI,CAAC;QACH,MAAM,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3D,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;AACH,CAAC,CACF,CAAC;AAEJ,OAAO;KACJ,OAAO,CAAC,aAAa,CAAC;KACtB,kBAAkB,EAAE;KACpB,WAAW,CAAC,sDAAsD,CAAC;KACnE,QAAQ,CACP,SAAS,EACT,yEAAyE,CAC1E;KACA,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CACL,8BAA8B,EAC9B,yFAAyF,EACzF,IAAI,CACL;KACA,MAAM,CACL,6BAA6B,EAC7B,yEAAyE,EACzE,KAAK,CACN;KACA,MAAM,CACL,KAAK,EACH,KAAyB,EACzB,IAAgG,EAChG,EAAE;IACF,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,CAAC,IAAI,CAAC,uBAAuB,CAAC;QACrD,MAAM,aAAa,CACjB,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,wBAAwB,EAAE,cAAc,EAAE,EACjE,MAAM,EACN,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CACpB,CAAC;IACJ,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;AACH,CAAC,CACF,CAAC;AAEJ,OAAO;KACJ,OAAO,CAAC,wBAAwB,CAAC;KACjC,kBAAkB,EAAE;KACpB,WAAW,CAAC,gFAAgF,CAAC;KAC7F,QAAQ,CACP,SAAS,EACT,yEAAyE,CAC1E;KACA,MAAM,CACL,qBAAqB,EACrB,gDAAgD,CACjD;KACA,MAAM,CACL,KAAK,EACH,KAAyB,EACzB,IAAyB,EACzB,EAAE;IACF,IAAI,CAAC;QACH,MAAM,wBAAwB,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACpE,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;AACH,CAAC,CACF,CAAC;AAEJ,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;IAC7B,OAAO,CAAC,IAAI,EAAE,CAAC;AACjB,CAAC;KAAM,CAAC;IACN,OAAO,CAAC,KAAK,EAAE,CAAC;AAClB,CAAC;AAED,KAAK,UAAU,SAAS;IACtB,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,OAAO,CAAC,KAAK;QAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACzE,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AAChD,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,KAAa;IAEb,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC7C,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,UAAU,MAAM,CAAC,KAAyB;IAC7C,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IACD,OAAO,SAAS,EAAE,CAAC;AACrB,CAAC;AAGD,SAAS,MAAM,CAAC,GAAQ,EAAE,MAAe;IACvC,MAAM,MAAM,GAAG,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;IACzC,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;AAC5E,CAAC"}

package/dist/index.d.ts

@@ -1,4 +1,9 @@
 export * from './lib/removeUnusedSchemas.js';
 export * from './lib/removeFromOneOfByName.js';
 export * from './lib/optimizeAllOfComposition.js';
+export * from './lib/allOfToOneOf.js';
+export * from './lib/sealSchema.js';
+export * from './lib/oasUtils.js';
+export * from './lib/cleanupDiscriminatorMappings.js';
+export * from './lib/schemaTransformUtils.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,8BAA8B,CAAC;AAC7C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,mCAAmC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,8BAA8B,CAAC;AAC7C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,mCAAmC,CAAC;AAClD,cAAc,uBAAuB,CAAC;AACtC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uCAAuC,CAAC;AACtD,cAAc,+BAA+B,CAAC"}

package/dist/index.js

@@ -1,4 +1,9 @@
 export * from './lib/removeUnusedSchemas.js';
 export * from './lib/removeFromOneOfByName.js';
 export * from './lib/optimizeAllOfComposition.js';
+export * from './lib/allOfToOneOf.js';
+export * from './lib/sealSchema.js';
+export * from './lib/oasUtils.js';
+export * from './lib/cleanupDiscriminatorMappings.js';
+export * from './lib/schemaTransformUtils.js';
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,8BAA8B,CAAC;AAC7C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,mCAAmC,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,8BAA8B,CAAC;AAC7C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,mCAAmC,CAAC;AAClD,cAAc,uBAAuB,CAAC;AACtC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uCAAuC,CAAC;AACtD,cAAc,+BAA+B,CAAC"}

package/dist/lib/allOfToOneOf.d.ts

@@ -0,0 +1,23 @@
+export interface AllOfToOneOfOptions {
+    /** If true, remove discriminator from base schema and let oneOf wrapper handle it */
+    removeDiscriminatorFromBase?: boolean;
+    /** If true, add const property with discriminator value to specialization schemas (default: true) */
+    addDiscriminatorConst?: boolean;
+    /** If true, skip oneOf transformation if only one specialization is found (default: false) */
+    ignoreSingleSpecialization?: boolean;
+}
+/**
+ * Convert allOf + discriminator patterns to oneOf + discriminator.
+ *
+ * This operation:
+ * 1. Finds base schemas with discriminators
+ * 2. Identifies concrete schemas that extend the base via allOf
+ * 3. For each concrete schema, adds a const property matching the discriminator value
+ * 4. Creates a new oneOf wrapper schema containing all concrete schemas
+ * 5. Replaces all references to the base schema with the wrapper schema (where polymorphism is used)
+ *
+ * @param doc - OpenAPI document to transform
+ * @param opts - Optional configuration
+ */
+export declare function allOfToOneOf(doc: any, opts?: AllOfToOneOfOptions): any;
+//# sourceMappingURL=allOfToOneOf.d.ts.map

package/dist/lib/allOfToOneOf.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"allOfToOneOf.d.ts","sourceRoot":"","sources":["../../src/lib/allOfToOneOf.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,mBAAmB;IAClC,qFAAqF;IACrF,2BAA2B,CAAC,EAAE,OAAO,CAAC;IACtC,qGAAqG;IACrG,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,8FAA8F;IAC9F,0BAA0B,CAAC,EAAE,OAAO,CAAC;CACtC;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,GAAE,mBAAwB,GAAG,GAAG,CAmF1E"}

package/dist/lib/allOfToOneOf.js

@@ -0,0 +1,189 @@
+import { refToName, buildInheritanceGraph } from "./oasUtils.js";
+/**
+ * Convert allOf + discriminator patterns to oneOf + discriminator.
+ *
+ * This operation:
+ * 1. Finds base schemas with discriminators
+ * 2. Identifies concrete schemas that extend the base via allOf
+ * 3. For each concrete schema, adds a const property matching the discriminator value
+ * 4. Creates a new oneOf wrapper schema containing all concrete schemas
+ * 5. Replaces all references to the base schema with the wrapper schema (where polymorphism is used)
+ *
+ * @param doc - OpenAPI document to transform
+ * @param opts - Optional configuration
+ */
+export function allOfToOneOf(doc, opts = {}) {
+    if (!doc || typeof doc !== "object")
+        return doc;
+    const schemas = doc.components?.schemas;
+    if (!schemas || typeof schemas !== "object")
+        return doc;
+    // Step 1: Find all base schemas with discriminators
+    const baseSchemasWithDiscriminator = new Map();
+    for (const [name, schema] of Object.entries(schemas)) {
+        if (schema && typeof schema === "object" && schema.discriminator) {
+            const disc = schema.discriminator;
+            if (disc.propertyName && disc.mapping && typeof disc.mapping === "object") {
+                baseSchemasWithDiscriminator.set(name, {
+                    propertyName: disc.propertyName,
+                    mapping: { ...disc.mapping }
+                });
+            }
+        }
+    }
+    if (baseSchemasWithDiscriminator.size === 0) {
+        return doc; // Nothing to convert
+    }
+    // Step 2: For each base schema with discriminator, find concrete types extending it
+    const inheritanceGraph = buildInheritanceGraph(schemas);
+    const polymorphicWrappers = new Map();
+    for (const [baseName, discInfo] of baseSchemasWithDiscriminator.entries()) {
+        const concreteSchemas = Array.from(inheritanceGraph.get(baseName) || []);
+        if (opts.ignoreSingleSpecialization && concreteSchemas.length === 1) {
+            continue;
+        }
+        if (concreteSchemas.length > 0) {
+            const wrapperName = `${baseName}Polymorphic`;
+            polymorphicWrappers.set(baseName, { name: wrapperName, concreteSchemas });
+            // Step 3: For each concrete schema, optionally add const property for discriminator
+            if (opts.addDiscriminatorConst !== false) {
+                const cS = Object.fromEntries(concreteSchemas.map(name => [name, schemas[name]]));
+                addDiscriminatorConstToConcreteSchemas(cS, discInfo);
+            }
+            // Step 4: Create wrapper schema with oneOf
+            const wrapperSchema = {
+                oneOf: concreteSchemas.map(name => ({ $ref: `#/components/schemas/${name}` })),
+                discriminator: {
+                    propertyName: discInfo.propertyName,
+                    mapping: discInfo.mapping
+                }
+            };
+            // If available, preserve description from base
+            const baseSchema = schemas[baseName];
+            if (baseSchema && baseSchema.description) {
+                wrapperSchema.description = `OneOf polymorphic ${baseName}. Use the "${discInfo.propertyName}" property to identify the concrete schema.`;
+            }
+            schemas[wrapperName] = wrapperSchema;
+        }
+    }
+    // Step 5: Replace references to base schemas with wrapper schemas where polymorphism is used
+    // We need to find where base schemas are referenced and check if they're used polymorphically
+    for (const [baseName, wrapperInfo] of polymorphicWrappers.entries()) {
+        replacePolymorhicReferences(schemas, baseName, wrapperInfo.name);
+    }
+    // Step 6: Optionally remove discriminator from base schemas
+    if (opts.removeDiscriminatorFromBase) {
+        for (const baseName of baseSchemasWithDiscriminator.keys()) {
+            if (schemas[baseName] && polymorphicWrappers.has(baseName)) {
+                delete schemas[baseName].discriminator;
+            }
+        }
+    }
+    return doc;
+}
+/**
+ * Replace references to a polymorphic base schema with references to its wrapper,
+ * but only in contexts where polymorphism would be used (e.g., in array items).
+ * Excludes references within concrete schemas' direct allOf references.
+ */
+function replacePolymorhicReferences(schemas, baseName, wrapperName) {
+    const baseRef = `#/components/schemas/${baseName}`;
+    const wrapperRef = `#/components/schemas/${wrapperName}`;
+    for (const schema of Object.values(schemas)) {
+        if (!schema || typeof schema !== "object")
+            continue;
+        replaceInSchema(schema, baseRef, wrapperRef, true);
+    }
+}
+/**
+ * Recursively replace $ref in a schema, particularly in polymorphic contexts like array items.
+ * When skipAllOfReplacement=true, skip replacing $ref directly within allOf arrays.
+ */
+function replaceInSchema(node, oldRef, newRef, skipDirectAllOfRefs = false) {
+    if (!node || typeof node !== "object")
+        return;
+    if (Array.isArray(node)) {
+        for (const item of node) {
+            if (item && typeof item === "object") {
+                if (item.$ref === oldRef) {
+                    // Don't replace direct $ref in allOf when flag is set
+                    if (skipDirectAllOfRefs) {
+                        continue;
+                    }
+                    item.$ref = newRef;
+                }
+                else {
+                    replaceInSchema(item, oldRef, newRef, skipDirectAllOfRefs);
+                }
+            }
+        }
+        return;
+    }
+    // For object properties, replace all $ref occurrences
+    for (const [key, value] of Object.entries(node)) {
+        if (value && typeof value === "object") {
+            // Special handling for allOf: skip direct $ref replacements
+            if (skipDirectAllOfRefs && key === "allOf" && Array.isArray(value)) {
+                for (const item of value) {
+                    if (item && typeof item === "object" && item.$ref !== oldRef) {
+                        replaceInSchema(item, oldRef, newRef, skipDirectAllOfRefs);
+                    }
+                }
+                continue;
+            }
+            if (value.$ref === oldRef) {
+                value.$ref = newRef;
+            }
+            replaceInSchema(value, oldRef, newRef, skipDirectAllOfRefs);
+        }
+    }
+}
+/**
+ * Add const constraint to each concrete schema matching the discriminator value.
+ * Avoids duplicates by checking if the const constraint already exists.
+ *
+ * @param schemas - Concrete schemas extending a base with discriminator (name -> schema)
+ * @param discInfo - Discriminator info (propertyName, mapping)
+ */
+function addDiscriminatorConstToConcreteSchemas(schemas, discInfo) {
+    // Iterate through concrete schemas by name
+    for (const [concreteName, concreteSchema] of Object.entries(schemas)) {
+        if (!concreteSchema)
+            continue;
+        // Find the discriminator value for this concrete schema
+        let discriminatorValue;
+        for (const [value, ref] of Object.entries(discInfo.mapping)) {
+            if (typeof ref === "string" && refToName(ref) === concreteName) {
+                discriminatorValue = value;
+                break;
+            }
+        }
+        if (discriminatorValue) {
+            // Ensure allOf exists
+            if (!Array.isArray(concreteSchema.allOf)) {
+                concreteSchema.allOf = [];
+            }
+            // Check if const constraint already exists for this schema
+            const constExists = concreteSchema.allOf.some((item) => item &&
+                typeof item === "object" &&
+                item.type === "object" &&
+                item.properties &&
+                item.properties[discInfo.propertyName] &&
+                item.properties[discInfo.propertyName].const === discriminatorValue);
+            // Only add const if it doesn't already exist
+            if (!constExists) {
+                // Add const as a separate allOf constraint (not merged with existing inline objects)
+                const constConstraint = {
+                    type: "object",
+                    properties: {
+                        [discInfo.propertyName]: {
+                            const: discriminatorValue
+                        }
+                    }
+                };
+                concreteSchema.allOf.push(constConstraint);
+            }
+        }
+    }
+}
+//# sourceMappingURL=allOfToOneOf.js.map

package/dist/lib/allOfToOneOf.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"allOfToOneOf.js","sourceRoot":"","sources":["../../src/lib/allOfToOneOf.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAYjE;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,YAAY,CAAC,GAAQ,EAAE,OAA4B,EAAE;IACnE,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,GAAG,CAAC;IAChD,MAAM,OAAO,GAAoC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC;IACzE,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,GAAG,CAAC;IAExD,oDAAoD;IACpD,MAAM,4BAA4B,GAAG,IAAI,GAAG,EAAqE,CAAC;IAElH,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACrD,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACjE,MAAM,IAAI,GAAG,MAAM,CAAC,aAAa,CAAC;YAClC,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,OAAO,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC1E,4BAA4B,CAAC,GAAG,CAAC,IAAI,EAAE;oBACrC,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE;iBAC7B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,4BAA4B,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC5C,OAAO,GAAG,CAAC,CAAC,qBAAqB;IACnC,CAAC;IAED,oFAAoF;IACpF,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;IACxD,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAuD,CAAC;IAG3F,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,4BAA4B,CAAC,OAAO,EAAE,EAAE,CAAC;QAC1E,MAAM,eAAe,GAAa,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;QAEnF,IAAI,IAAI,CAAC,0BAA0B,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpE,SAAS;QACX,CAAC;QAED,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,MAAM,WAAW,GAAG,GAAG,QAAQ,aAAa,CAAC;YAC7C,mBAAmB,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;YAE1E,oFAAoF;YACpF,IAAI,IAAI,CAAC,qBAAqB,KAAK,KAAK,EAAE,CAAC;gBACzC,MAAM,EAAE,GAAG,MAAM,CAAC,WAAW,CAC3B,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CACnD,CAAC;gBACF,sCAAsC,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YACvD,CAAC;YAED,2CAA2C;YAC3C,MAAM,aAAa,GAAQ;gBACzB,KAAK,EAAE,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,wBAAwB,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC9E,aAAa,EAAE;oBACb,YAAY,EAAE,QAAQ,CAAC,YAAY;oBACnC,OAAO,EAAE,QAAQ,CAAC,OAAO;iBAC1B;aACF,CAAC;YAEF,+CAA+C;YAC/C,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YACrC,IAAI,UAAU,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;gBACzC,aAAa,CAAC,WAAW,GAAG,qBAAqB,QAAQ,cAAc,QAAQ,CAAC,YAAY,6CAA6C,CAAC;YAC5I,CAAC;YAED,OAAO,CAAC,WAAW,CAAC,GAAG,aAAa,CAAC;QACvC,CAAC;IACH,CAAC;IAED,6FAA6F;IAC7F,8FAA8F;IAC9F,KAAK,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,IAAI,mBAAmB,CAAC,OAAO,EAAE,EAAE,CAAC;QACpE,2BAA2B,CAAC,OAAO,EAAE,QAAQ,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IACnE,CAAC;IAED,4DAA4D;IAC5D,IAAI,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACrC,KAAK,MAAM,QAAQ,IAAI,4BAA4B,CAAC,IAAI,EAAE,EAAE,CAAC;YAC3D,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC3D,OAAO,OAAO,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC;YACzC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;;;GAIG;AACH,SAAS,2BAA2B,CAClC,OAA4B,EAC5B,QAAgB,EAChB,WAAmB;IAEnB,MAAM,OAAO,GAAG,wBAAwB,QAAQ,EAAE,CAAC;IACnD,MAAM,UAAU,GAAG,wBAAwB,WAAW,EAAE,CAAC;IAEzD,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,SAAS;QACpD,eAAe,CAAC,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;IACrD,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAS,eAAe,CAAC,IAAS,EAAE,MAAc,EAAE,MAAc,EAAE,sBAA+B,KAAK;IACtG,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO;IAE9C,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YACxB,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACrC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACzB,sDAAsD;oBACtD,IAAI,mBAAmB,EAAE,CAAC;wBACxB,SAAS;oBACX,CAAC;oBACD,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC;gBACrB,CAAC;qBAAM,CAAC;oBACN,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC;gBAC7D,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO;IACT,CAAC;IAED,sDAAsD;IACtD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAChD,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,4DAA4D;YAC5D,IAAI,mBAAmB,IAAI,GAAG,KAAK,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACnE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBAC7D,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC;oBAC7D,CAAC;gBACH,CAAC;gBACD,SAAS;YACX,CAAC;YACD,IAAK,KAAa,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAClC,KAAa,CAAC,IAAI,GAAG,MAAM,CAAC;YAC/B,CAAC;YACD,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,SAAS,sCAAsC,CAC7C,OAA4B,EAC5B,QAAmE;IAEnE,2CAA2C;IAC3C,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACrE,IAAI,CAAC,cAAc;YAAE,SAAS;QAE9B,wDAAwD;QACxD,IAAI,kBAAsC,CAAC;QAC3C,KAAK,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5D,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,SAAS,CAAC,GAAG,CAAC,KAAK,YAAY,EAAE,CAAC;gBAC/D,kBAAkB,GAAG,KAAK,CAAC;gBAC3B,MAAM;YACR,CAAC;QACH,CAAC;QAED,IAAI,kBAAkB,EAAE,CAAC;YACvB,sBAAsB;YACtB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzC,cAAc,CAAC,KAAK,GAAG,EAAE,CAAC;YAC5B,CAAC;YAED,2DAA2D;YAC3D,MAAM,WAAW,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAC3C,CAAC,IAAS,EAAE,EAAE,CACZ,IAAI;gBACJ,OAAO,IAAI,KAAK,QAAQ;gBACxB,IAAI,CAAC,IAAI,KAAK,QAAQ;gBACtB,IAAI,CAAC,UAAU;gBACf,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC;gBACtC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,KAAK,KAAK,kBAAkB,CACtE,CAAC;YAEF,6CAA6C;YAC7C,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,qFAAqF;gBACrF,MAAM,eAAe,GAAQ;oBAC3B,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;4BACvB,KAAK,EAAE,kBAAkB;yBAC1B;qBACF;iBACF,CAAC;gBACF,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/lib/cleanupDiscriminatorMappings.d.ts

@@ -0,0 +1,21 @@
+/**
+ * Cleans up discriminator mappings by removing entries that point to non-existent schemas.
+ *
+ * This function:
+ * 1. Iterates through all schemas looking for discriminator definitions
+ * 2. For each discriminator mapping entry, checks if the referenced schema exists
+ * 3. Removes mapping entries that reference undefined schemas
+ * 4. Returns count of cleaned mappings for reporting
+ *
+ * @param doc - OpenAPI document to clean
+ * @returns Object with cleanup statistics
+ */
+export declare function cleanupDiscriminatorMappings(doc: any): {
+    schemasChecked: number;
+    mappingsRemoved: number;
+    details: Array<{
+        schema: string;
+        removed: string[];
+    }>;
+};
+//# sourceMappingURL=cleanupDiscriminatorMappings.d.ts.map

package/dist/lib/cleanupDiscriminatorMappings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cleanupDiscriminatorMappings.d.ts","sourceRoot":"","sources":["../../src/lib/cleanupDiscriminatorMappings.ts"],"names":[],"mappings":"AAGA;;;;;;;;;;;GAWG;AACH,wBAAgB,4BAA4B,CAAC,GAAG,EAAE,GAAG,GAAG;IACtD,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;IACxB,OAAO,EAAE,KAAK,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC,CAAC;CACvD,CAyDA"}

package/dist/lib/cleanupDiscriminatorMappings.js

@@ -0,0 +1,64 @@
+import { refToName } from "./oasUtils.js";
+import { updateDiscriminatorMappings } from "./schemaTransformUtils.js";
+/**
+ * Cleans up discriminator mappings by removing entries that point to non-existent schemas.
+ *
+ * This function:
+ * 1. Iterates through all schemas looking for discriminator definitions
+ * 2. For each discriminator mapping entry, checks if the referenced schema exists
+ * 3. Removes mapping entries that reference undefined schemas
+ * 4. Returns count of cleaned mappings for reporting
+ *
+ * @param doc - OpenAPI document to clean
+ * @returns Object with cleanup statistics
+ */
+export function cleanupDiscriminatorMappings(doc) {
+    if (!doc || typeof doc !== "object") {
+        return { schemasChecked: 0, mappingsRemoved: 0, details: [] };
+    }
+    const schemas = doc.components?.schemas;
+    if (!schemas || typeof schemas !== "object") {
+        return { schemasChecked: 0, mappingsRemoved: 0, details: [] };
+    }
+    // Collect all existing schema names for validation
+    const existingSchemas = new Set(Object.keys(schemas));
+    let totalMappingsRemoved = 0;
+    let schemasChecked = 0;
+    const details = [];
+    // Iterate through all schemas
+    for (const [schemaName, schema] of Object.entries(schemas)) {
+        if (!schema || typeof schema !== "object")
+            continue;
+        // Check if schema has discriminator with mapping
+        if (schema.discriminator && typeof schema.discriminator === "object") {
+            const discriminator = schema.discriminator;
+            if (discriminator.mapping && typeof discriminator.mapping === "object") {
+                schemasChecked++;
+                const removedMappings = [];
+                // Update discriminator mappings, removing invalid references
+                const removed = updateDiscriminatorMappings(schema, (key, ref) => {
+                    const referencedSchemaName = refToName(ref);
+                    // Keep this mapping only if the referenced schema exists
+                    const shouldKeep = !referencedSchemaName || existingSchemas.has(referencedSchemaName);
+                    if (!shouldKeep) {
+                        removedMappings.push(key);
+                    }
+                    return shouldKeep;
+                });
+                totalMappingsRemoved += removed;
+                if (removedMappings.length > 0) {
+                    details.push({
+                        schema: schemaName,
+                        removed: removedMappings,
+                    });
+                }
+            }
+        }
+    }
+    return {
+        schemasChecked,
+        mappingsRemoved: totalMappingsRemoved,
+        details,
+    };
+}
+//# sourceMappingURL=cleanupDiscriminatorMappings.js.map

package/dist/lib/cleanupDiscriminatorMappings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cleanupDiscriminatorMappings.js","sourceRoot":"","sources":["../../src/lib/cleanupDiscriminatorMappings.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC1C,OAAO,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AAExE;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,4BAA4B,CAAC,GAAQ;IAKnD,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QACpC,OAAO,EAAE,cAAc,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;IAChE,CAAC;IAED,MAAM,OAAO,GAAoC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC;IACzE,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAC5C,OAAO,EAAE,cAAc,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;IAChE,CAAC;IAED,mDAAmD;IACnD,MAAM,eAAe,GAAG,IAAI,GAAG,CAAS,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAE9D,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAC7B,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,MAAM,OAAO,GAAiD,EAAE,CAAC;IAEjE,8BAA8B;IAC9B,KAAK,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3D,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,SAAS;QAEpD,iDAAiD;QACjD,IAAI,MAAM,CAAC,aAAa,IAAI,OAAO,MAAM,CAAC,aAAa,KAAK,QAAQ,EAAE,CAAC;YACrE,MAAM,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;YAE3C,IAAI,aAAa,CAAC,OAAO,IAAI,OAAO,aAAa,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACvE,cAAc,EAAE,CAAC;gBACjB,MAAM,eAAe,GAAa,EAAE,CAAC;gBAErC,6DAA6D;gBAC7D,MAAM,OAAO,GAAG,2BAA2B,CAAC,MAAM,EAAE,CAAC,GAAW,EAAE,GAAW,EAAE,EAAE;oBAC/E,MAAM,oBAAoB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;oBAC5C,yDAAyD;oBACzD,MAAM,UAAU,GAAG,CAAC,oBAAoB,IAAI,eAAe,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;oBACtF,IAAI,CAAC,UAAU,EAAE,CAAC;wBAChB,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAC5B,CAAC;oBACD,OAAO,UAAU,CAAC;gBACpB,CAAC,CAAC,CAAC;gBAEH,oBAAoB,IAAI,OAAO,CAAC;gBAEhC,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC/B,OAAO,CAAC,IAAI,CAAC;wBACX,MAAM,EAAE,UAAU;wBAClB,OAAO,EAAE,eAAe;qBACzB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,cAAc;QACd,eAAe,EAAE,oBAAoB;QACrC,OAAO;KACR,CAAC;AACJ,CAAC"}