import type { z } from 'zod'

export interface MarketMcpToolResult {
  [key: string]: unknown
  content: Array<{ type: 'text'; text: string } | { type: 'image'; data: string; mimeType: string }>
  structuredContent?: Record<string, unknown>
}

interface MarketMcpToolAnnotations {
  readOnlyHint?: boolean
  destructiveHint?: boolean
  idempotentHint?: boolean
  openWorldHint?: boolean
}

interface MarketMcpToolConfig<Input extends z.ZodType | undefined> {
  title?: string
  description?: string
  inputSchema?: Input
  annotations?: MarketMcpToolAnnotations
}

type MarketMcpToolHandler<Input extends z.ZodType | undefined> = Input extends z.ZodType
  ? (input: z.output<Input>) => Promise<MarketMcpToolResult>
  : () => Promise<MarketMcpToolResult>

export interface MarketMcpServer {
  registerTool<Input extends z.ZodType | undefined = undefined>(
    name: string,
    config: MarketMcpToolConfig<Input>,
    handler: MarketMcpToolHandler<Input>,
  ): unknown
}

export interface MarketMcpCapabilities {
  generate?: boolean
  agent?: boolean
}

export interface MarketMcpOptions {
  fetch?: typeof globalThis.fetch
  capabilities?: MarketMcpCapabilities
}