import { ZodNativeEnumDef } from "@lowcode-modou/zod";

export type JsonSchema7NativeEnumType = {
  type: "string" | "number" | ["string", "number"];
  enum: (string | number)[];
};

export function parseNativeEnumDef(
  def: ZodNativeEnumDef
): JsonSchema7NativeEnumType {
  const numberValues = Object.values(def.values)
    .filter((value) => typeof value === "number")
    .map(toString);
  const actualValues = Object.values(def.values).filter(
    (_, i) => i >= numberValues.length
  );
  return {
    type:
      numberValues.length === 0
        ? "string"
        : numberValues.length === actualValues.length
        ? "number"
        : ["string", "number"],
    enum: actualValues,
  };
}