import { z } from "@hono/zod-openapi";

const DefaultOptionsSchema = z.object({
  internal: z.boolean().optional(),
  external: z.boolean().optional(),
  setting: z.string().optional(),
  label: z.string().optional(),
  description: z.string().optional(),
  accepts: z.array(z.string()).optional(),
  hidden: z.boolean().optional(),
});

const NodeInputFloatSchema = z.object({
  ...DefaultOptionsSchema.shape,
  type: z.literal("float"),
  element: z.literal("slider").optional(),
  value: z.number().optional(),
  min: z.number().optional(),
  max: z.number().optional(),
  step: z.number().optional(),
});

const NodeInputIntegerSchema = z.object({
  ...DefaultOptionsSchema.shape,
  type: z.literal("integer"),
  element: z.literal("slider").optional(),
  value: z.number().optional(),
  min: z.number().optional(),
  max: z.number().optional(),
});

const NodeInputBooleanSchema = z.object({
  ...DefaultOptionsSchema.shape,
  type: z.literal("boolean"),
  value: z.boolean().optional(),
});

const NodeInputSelectSchema = z.object({
  ...DefaultOptionsSchema.shape,
  type: z.literal("select"),
  options: z.array(z.string()).optional(),
  value: z.number().optional(),
});

const NodeInputSeedSchema = z.object({
  ...DefaultOptionsSchema.shape,
  type: z.literal("seed"),
  value: z.number().optional(),
});

const NodeInputVec3Schema = z.object({
  ...DefaultOptionsSchema.shape,
  type: z.literal("vec3"),
  value: z.array(z.number()).optional(),
});

const NodeInputGeometrySchema = z.object({
  ...DefaultOptionsSchema.shape,
  type: z.literal("geometry"),
});

const NodeInputPathSchema = z.object({
  ...DefaultOptionsSchema.shape,
  type: z.literal("path"),
});

export const NodeInputSchema = z
  .union([
    NodeInputSeedSchema,
    NodeInputBooleanSchema,
    NodeInputFloatSchema,
    NodeInputIntegerSchema,
    NodeInputSelectSchema,
    NodeInputSeedSchema,
    NodeInputVec3Schema,
    NodeInputGeometrySchema,
    NodeInputPathSchema,
  ])
  .openapi("NodeInput");

export type NodeInput = z.infer<typeof NodeInputSchema>;