// packages/opencode/src/ui-intent/index.ts
import { z } from "zod"
import { BusEvent } from "../bus/bus-event"

export const FormField = z.discriminatedUnion("type", [
  z.object({
    type: z.literal("select"),
    id: z.string(),
    label: z.string(),
    description: z.string().optional(),
    options: z.array(
      z.object({
        value: z.string(),
        label: z.string(),
        description: z.string().optional(),
      }),
    ),
    default: z.string().optional(),
  }),
  z.object({
    type: z.literal("multiselect"),
    id: z.string(),
    label: z.string(),
    description: z.string().optional(),
    options: z.array(
      z.object({
        value: z.string(),
        label: z.string(),
        description: z.string().optional(),
      }),
    ),
    default: z.array(z.string()).optional(),
  }),
  z.object({
    type: z.literal("confirm"),
    id: z.string(),
    label: z.string(),
    description: z.string().optional(),
    default: z.boolean().optional(),
  }),
  z.object({
    type: z.literal("text"),
    id: z.string(),
    label: z.string(),
    description: z.string().optional(),
    placeholder: z.string().optional(),
    default: z.string().optional(),
  }),
])

export const UIIntent = z.object({
  id: z.string(),
  sessionID: z.string(),
  messageID: z.string(),
  callID: z.string().optional(),
  source: z.enum(["core", "plugin"]),
  plugin: z.string().optional(),

  intent: z.discriminatedUnion("type", [
    z.object({
      type: z.literal("form"),
      title: z.string(),
      description: z.string().optional(),
      fields: z.array(FormField).min(1).max(10),
      submitLabel: z.string().default("Submit"),
      cancelLabel: z.string().default("Cancel"),
    }),
    z.object({
      type: z.literal("confirm"),
      title: z.string(),
      message: z.string(),
      confirmLabel: z.string().default("Confirm"),
      cancelLabel: z.string().default("Cancel"),
      variant: z.enum(["info", "warning", "danger"]).default("info"),
    }),
    z.object({
      type: z.literal("progress"),
      title: z.string(),
      message: z.string().optional(),
      value: z.number().min(0).max(100).optional(),
      indeterminate: z.boolean().default(false),
      cancellable: z.boolean().default(false),
    }),
    z.object({
      type: z.literal("toast"),
      message: z.string(),
      variant: z.enum(["info", "success", "warning", "error"]),
      duration: z.number().default(5000),
    }),
  ]),
})

export const UIIntentResponse = z.object({
  intentID: z.string(),
  sessionID: z.string(),
  response: z.union([
    z.object({ type: z.literal("submit"), data: z.record(z.string(), z.any()) }),
    z.object({ type: z.literal("cancel") }),
  ]),
})

export const Event = {
  Request: BusEvent.define("ui.intent.request", UIIntent),
  Response: BusEvent.define("ui.intent.response", UIIntentResponse),
}

// packages/opencode/src/server/server.ts
.post("/session/:sessionID/ui-intent/:intentID", async (c) => {
  const { sessionID, intentID } = c.req.param()
  const body = UIIntentResponse.parse(await c.req.json())
  await UIIntent.respond({ sessionID, intentID, response: body.response })
  return c.json({ success: true })
})

// packages/plugin/src/index.ts
export type PluginInput = {
  // ... existing fields
  ui: {
    form: (input: FormIntent) => Promise<Record<string, any>>
    confirm: (input: ConfirmIntent) => Promise<boolean>
    toast: (input: ToastIntent) => void
  }
}