@acrool/rtk-query-codegen-openapi 1.4.0-alpha.0 → 1.4.0-alpha.1

package/lib/index.mjs

@@ -499,6 +499,27 @@ function renameIdentifier(node, oldName, newName) {
     })
   ]).transformed[0];
 }
+function isStringEnumType(node) {
+  if (!ts.isUnionTypeNode(node.type)) return null;
+  const members = [];
+  for (const member of node.type.types) {
+    if (ts.isLiteralTypeNode(member) && ts.isStringLiteral(member.literal)) {
+      members.push(member.literal.text);
+    } else {
+      return null;
+    }
+  }
+  return members.length > 0 ? members : null;
+}
+function generateEnumDeclaration(name, members) {
+  const enumMembers = members.map((value) => {
+    const key = value.charAt(0).toUpperCase() + value.slice(1);
+    return `  ${key} = "${value}"`;
+  });
+  return `export enum ${name} {
+${enumMembers.join(",\n")}
+}`;
+}
 function generateComponentSchemaFile(interfaces, includeOnly) {
   const printer = ts.createPrinter({ newLine: ts.NewLineKind.LineFeed });
   const resultFile = ts.createSourceFile(
@@ -516,6 +537,13 @@ function generateComponentSchemaFile(interfaces, includeOnly) {
     if (includeOnly && !includeOnly.has(originalName)) {
       return;
     }
+    if (ts.isTypeAliasDeclaration(node)) {
+      const enumMembers = isStringEnumType(node);
+      if (enumMembers) {
+        renamedInterfaces.push(generateEnumDeclaration(pascalCaseName, enumMembers));
+        return;
+      }
+    }
     const renamedNode = renameIdentifier(node, originalName, pascalCaseName);
     const printed = printer.printNode(ts.EmitHint.Unspecified, renamedNode, resultFile);
     renamedInterfaces.push(printed);
@@ -693,6 +721,27 @@ function prefixSharedSchemaRefs(node, allSchemaNames, localSchemaNames, declarat
     })
   ]).transformed[0];
 }
+function isStringEnumType2(node) {
+  if (!ts2.isUnionTypeNode(node.type)) return null;
+  const members = [];
+  for (const member of node.type.types) {
+    if (ts2.isLiteralTypeNode(member) && ts2.isStringLiteral(member.literal)) {
+      members.push(member.literal.text);
+    } else {
+      return null;
+    }
+  }
+  return members.length > 0 ? members : null;
+}
+function generateEnumDeclaration2(name, members) {
+  const enumMembers = members.map((value) => {
+    const key = value.charAt(0).toUpperCase() + value.slice(1);
+    return `  ${key} = "${value}"`;
+  });
+  return `export enum ${name} {
+${enumMembers.join(",\n")}
+}`;
+}
 function generateTypesFile(endpointInfos, _options, schemaInterfaces, operationDefinitions, localSchemaOptions) {
   const localSchemaNames = localSchemaOptions?.localSchemaNames ?? /* @__PURE__ */ new Set();
   const localSchemaInterfaces = localSchemaOptions?.localSchemaInterfaces ?? {};
@@ -730,6 +779,13 @@ function generateTypesFile(endpointInfos, _options, schemaInterfaces, operationD
     const allSchemaNameSet = new Set(schemaInterfaces ? Object.keys(schemaInterfaces) : []);
     for (const [originalName, node] of Object.entries(localSchemaInterfaces)) {
       const pascalCaseName = toPascalCase2(originalName);
+      if (ts2.isTypeAliasDeclaration(node)) {
+        const enumMembers = isStringEnumType2(node);
+        if (enumMembers) {
+          localTypeDefs.push(generateEnumDeclaration2(pascalCaseName, enumMembers));
+          continue;
+        }
+      }
       let transformedNode = renameIdentifier2(node, originalName, pascalCaseName);
       transformedNode = prefixSharedSchemaRefs(transformedNode, allSchemaNameSet, localSchemaNames, pascalCaseName);
       const printed = printer.printNode(ts2.EmitHint.Unspecified, transformedNode, resultFile);