From 2823c75bb2dbbc0f42b745d42c1a6ad061d974c5 Mon Sep 17 00:00:00 2001
From: "coderabbitai[bot]"
 <136622811+coderabbitai[bot]@users.noreply.github.com>
Date: Thu, 22 May 2025 10:53:37 +0000
Subject: [PATCH] =?UTF-8?q?=F0=9F=93=9D=20Add=20docstrings=20to=20`fix/sch?=
 =?UTF-8?q?ema-one-of-parser`?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Docstrings generation was requested by @harsha-iiiv.

* https://github.com/harsha-iiiv/openapi-mcp-generator/pull/16#issuecomment-2900651451

The following files were modified:

* `src/parser/extract-tools.ts`
---
 src/parser/extract-tools.ts | 40 +++++++++++++++++++++++++++++++++----
 1 file changed, 36 insertions(+), 4 deletions(-)

diff --git a/src/parser/extract-tools.ts b/src/parser/extract-tools.ts
index 8816095..4c57050 100644
--- a/src/parser/extract-tools.ts
+++ b/src/parser/extract-tools.ts
@@ -153,10 +153,14 @@ export function generateInputSchemaAndDetails(operation: OpenAPIV3.OperationObje
 }
 
 /**
- * Maps an OpenAPI schema to a JSON Schema
+ * Converts an OpenAPI schema or reference object to a JSON Schema representation.
  *
- * @param schema OpenAPI schema object or reference
- * @returns JSON Schema representation
+ * Handles composite schemas (`oneOf`, `anyOf`, `allOf`) by merging subschemas and combining enum values. Removes OpenAPI-specific properties and adjusts types for JSON Schema compatibility, including handling of nullable fields. Recursively processes nested object properties and array items.
+ *
+ * @param schema - The OpenAPI schema object or reference to convert.
+ * @returns The corresponding JSON Schema object or boolean schema.
+ *
+ * @remark If a `$ref` reference cannot be resolved, returns a generic object schema and logs a warning.
  */
 export function mapOpenApiSchemaToJsonSchema(
   schema: OpenAPIV3.SchemaObject | OpenAPIV3.ReferenceObject
@@ -171,7 +175,35 @@ export function mapOpenApiSchemaToJsonSchema(
   if (typeof schema === 'boolean') return schema;
 
   // Create a copy of the schema to modify
-  const jsonSchema: JSONSchema7 = { ...schema } as any;
+  let jsonSchema: JSONSchema7 = { ...schema } as any;
+
+  if (schema.oneOf || schema.anyOf || schema.allOf) {
+    const oneSchema = structuredClone(schema.oneOf || schema.anyOf || schema.allOf);
+
+    if (oneSchema) {
+      const combinedSchema = mapOpenApiSchemaToJsonSchema(oneSchema[0]);
+      
+      for (let i = 1; i < oneSchema.length; i++) {
+        const mappedSubSchema = mapOpenApiSchemaToJsonSchema(oneSchema[i]);
+        if (typeof mappedSubSchema === 'object' && typeof combinedSchema === 'object') {
+          // Handle enum values
+          if (mappedSubSchema.enum) {
+            if (!combinedSchema.enum) {
+              combinedSchema.enum = [];
+            }
+            // Combine enum values from both schemas
+            const uniqueEnums = new Set([
+              ...combinedSchema.enum,
+              ...(mappedSubSchema.enum || [])
+            ]);
+            combinedSchema.enum = Array.from(uniqueEnums);
+          }
+        }
+      }
+
+      jsonSchema = combinedSchema as JSONSchema7;
+    }
+  }
 
   // Convert integer type to number (JSON Schema compatible)
   if (schema.type === 'integer') jsonSchema.type = 'number';