import { z } from "zod"

export type RegistryItemCategory = "agents" | "tools" | "prompts"

export const registryItemTypeSchema = z.enum([
	"registry:agent",
	"registry:tool",
	"registry:prompt",
	"registry:lib",
])

// File with required content (for fetched items)
export const registryItemFileSchema = z.object({
	path: z.string(),
	content: z.string(),
	type: registryItemTypeSchema,
	target: z.string().optional(),
})

// Base fields shared between index entries and full items
const registryItemBaseSchema = z.object({
	$schema: z.string().optional(),
	name: z.string(),
	type: registryItemTypeSchema,
	title: z.string().optional(),
	author: z.string().min(2).optional(),
	description: z.string().optional(),
	dependencies: z.array(z.string()).optional(),
	devDependencies: z.array(z.string()).optional(),
	registryDependencies: z.array(z.string()).optional(),
	meta: z.record(z.string(), z.any()).optional(),
	docs: z.string().optional(),
	categories: z.array(z.string()).optional(),
})

// Full registry item - has files with content
export const registryItemSchema = registryItemBaseSchema.extend({
	files: z.array(registryItemFileSchema),
})

export type RegistryItem = z.infer<typeof registryItemSchema>

// Registry index - items don't have files
export const registrySchema = z.object({
	name: z.string(),
	homepage: z.string(),
	items: z.array(registryItemBaseSchema),
})

export type Registry = z.infer<typeof registrySchema>

export const registryIndexSchema = z.array(registryItemBaseSchema)

export const registryResolvedItemsTreeSchema = registryItemSchema.pick({
	dependencies: true,
	devDependencies: true,
	files: true,
	docs: true,
})

export const registryConfigItemSchema = z.union([
	z.string().refine((s) => s.includes("{name}") && s.includes("{type}"), {
		message: "Registry URL must include {name} and {type} placeholders",
	}),
	z.object({
		url: z
			.string()
			.refine((s) => s.includes("{name}") && s.includes("{type}"), {
				message: "Registry URL must include {name} and {type} placeholders",
			}),
		params: z.record(z.string(), z.string()).optional(),
		headers: z.record(z.string(), z.string()).optional(),
	}),
])

export const registryConfigSchema = z.record(
	z.string().refine((key) => key.startsWith("@"), {
		message: "Registry names must start with @ (e.g., @awesome-ai)",
	}),
	registryConfigItemSchema,
)