import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
import type { AnyNodeId } from '@pascal-app/core/schema'
import { ZoneNode } from '@pascal-app/core/schema'
import { z } from 'zod'
import type { SceneOperations } from '../operations'
import { ErrorCode, throwMcpError } from './errors'
import { publishLiveSceneSnapshot } from './live-sync'
import { NodeIdSchema, Vec2Schema } from './schemas'

export const setZoneInput = {
  levelId: NodeIdSchema,
  polygon: z.array(Vec2Schema).min(3),
  label: z.string().min(1),
  properties: z.record(z.string(), z.unknown()).optional(),
}

export const setZoneOutput = {
  zoneId: z.string(),
}

export function registerSetZone(server: McpServer, bridge: SceneOperations): void {
  server.registerTool(
    'set_zone',
    {
      title: 'Set zone',
      description:
        'Create a polygonal zone on the given level. label is stored as the zone name and properties are merged into metadata.',
      inputSchema: setZoneInput,
      outputSchema: setZoneOutput,
    },
    async ({ levelId, polygon, label, properties }) => {
      const parent = bridge.getNode(levelId as AnyNodeId)
      if (!parent) {
        throwMcpError(ErrorCode.InvalidParams, `Level not found: ${levelId}`)
      }
      if (parent.type !== 'level') {
        throwMcpError(
          ErrorCode.InvalidParams,
          `Node ${levelId} is a ${parent.type}, expected level`,
        )
      }
      if (
        typeof parent.metadata === 'object' &&
        parent.metadata !== null &&
        'role' in parent.metadata &&
        parent.metadata.role === 'roof'
      ) {
        throwMcpError(
          ErrorCode.InvalidParams,
          `Roof support level ${levelId} is not an occupied story; create zones on an occupied level instead`,
        )
      }

      const zone = ZoneNode.parse({
        name: label,
        polygon: polygon as Array<[number, number]>,
        metadata: properties ?? {},
      })
      const id = bridge.createNode(zone, levelId as AnyNodeId)
      await publishLiveSceneSnapshot(bridge, 'set_zone')

      const payload = { zoneId: id as string }
      return {
        content: [{ type: 'text' as const, text: JSON.stringify(payload) }],
        structuredContent: payload,
      }
    },
  )
}