Handle oneof Const

Author: lucy66hw

tools/proto-convert/src/SchemaModifier.ts

@@ -53,17 +53,34 @@ export class SchemaModifier {
         }
     }
 
-    // Converts `oneOf` schemas with `const` values to boolean types.
-    // Example: oneof: [ {type: 'string', const: 'a'} ] to type: 'boolean' with title: 'schemaName_a'
+    // Converts `oneOf` schemas with `const` values to enum types.
+    // Example: oneof: [ {type: 'string', const: 'a'}, {type: 'string', const: 'b'} ] to enum: ['a', 'b']
+    // For non-string types, uses the type as enum value
     handleOneOfConst(schema: OpenAPIV3.SchemaObject, schemaName: string): void {
         if (schema.oneOf) {
+            const enumValues: string[] = [];
+            let hasConstValues = false;
+            
             for (const item of schema.oneOf) {
-                if (item && !('$ref' in item) && item.type === 'string' && 'const' in item) {
-                    item.type = 'boolean';
-                    item.title = compressMultipleUnderscores(`${schemaName}_${item.const}`);
-                    delete item.const;
+                if (item && !('$ref' in item)) {
+                    if (item.type === 'string' && 'const' in item) {
+                        // String with const value - use const as enum value
+                        enumValues.push(item.const as string);
+                        hasConstValues = true;
+                    } else if (item.type && item.type !== 'string') {
+                        // Non-string type - use type as enum value
+                        enumValues.push(item.type);
+                        hasConstValues = true;
+                    }
                 }
             }
+            
+            // If we found const values or non-string types, convert to enum
+            if (hasConstValues && enumValues.length > 0) {
+                delete schema.oneOf;
+                schema.type = 'string';
+                schema.enum = enumValues;
+            }
         }
     }