import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { eq } from "../drizzle.js";
import { z } from "zod";
import { db } from "../db.js";
import { objects } from "../schema.js";
import { toolCatch, toolErr, toolOk } from "./tool-result.js";

const updateObjectInputSchema = z.object({
  id: z.string().uuid(),
  title: z.string().min(1).max(500).optional(),
  description: z.string().nullable().optional(),
  status: z.string().nullable().optional(),
  icon: z.string().nullable().optional(),
  coverImage: z.string().nullable().optional(),
});

export function registerUpdateObjectTool(mcp: McpServer): void {
  mcp.registerTool(
    "update_object",
    {
      description: "Update an existing object (only provided fields are changed).",
      inputSchema: updateObjectInputSchema,
    },
    async (args) => {
      try {
        const input = updateObjectInputSchema.parse(args);
        const [updated] = await db
          .update(objects)
          .set({
            ...(input.title !== undefined ? { title: input.title } : {}),
            ...(input.description !== undefined ? { description: input.description } : {}),
            ...(input.status !== undefined ? { status: input.status } : {}),
            ...(input.icon !== undefined ? { icon: input.icon } : {}),
            ...(input.coverImage !== undefined ? { coverImage: input.coverImage } : {}),
            updatedAt: new Date(),
          })
          .where(eq(objects.id, input.id))
          .returning();

        if (!updated) {
          return toolErr(`Object not found: ${input.id}`);
        }
        return toolOk(updated);
      } catch (e) {
        return toolCatch(e);
      }
    },
  );
}