import type { FastMCP } from 'fastmcp';
import { z } from 'zod';

import { getController } from '../controller';
import { jsonTextContent } from '../utils/format';

const toolContextSchema = {
  teamName: z.string().min(1),
  claudeDir: z.string().min(1).optional(),
};

export function registerKanbanTools(server: Pick<FastMCP, 'addTool'>) {
  server.addTool({
    name: 'kanban_get',
    description: 'Get current kanban state',
    parameters: z.object({
      ...toolContextSchema,
    }),
    execute: async ({ teamName, claudeDir }) =>
      await Promise.resolve(jsonTextContent(getController(teamName, claudeDir).kanban.getKanbanState())),
  });

  server.addTool({
    name: 'kanban_set_column',
    description: 'Move task to review or approved column',
    parameters: z.object({
      ...toolContextSchema,
      taskId: z.string().min(1),
      column: z.enum(['review', 'approved']),
    }),
    execute: async ({ teamName, claudeDir, taskId, column }) =>
      await Promise.resolve(
        jsonTextContent(getController(teamName, claudeDir).kanban.setKanbanColumn(taskId, column))
      ),
  });

  server.addTool({
    name: 'kanban_clear',
    description: 'Remove task from kanban board',
    parameters: z.object({
      ...toolContextSchema,
      taskId: z.string().min(1),
    }),
    execute: async ({ teamName, claudeDir, taskId }) =>
      await Promise.resolve(jsonTextContent(getController(teamName, claudeDir).kanban.clearKanban(taskId))),
  });

  server.addTool({
    name: 'kanban_list_reviewers',
    description: 'List configured review participants',
    parameters: z.object({
      ...toolContextSchema,
    }),
    execute: async ({ teamName, claudeDir }) =>
      await Promise.resolve(jsonTextContent(getController(teamName, claudeDir).kanban.listReviewers())),
  });

  server.addTool({
    name: 'kanban_add_reviewer',
    description: 'Add a reviewer to kanban configuration',
    parameters: z.object({
      ...toolContextSchema,
      reviewer: z.string().min(1),
    }),
    execute: async ({ teamName, claudeDir, reviewer }) =>
      await Promise.resolve(jsonTextContent(getController(teamName, claudeDir).kanban.addReviewer(reviewer))),
  });

  server.addTool({
    name: 'kanban_remove_reviewer',
    description: 'Remove reviewer from kanban configuration',
    parameters: z.object({
      ...toolContextSchema,
      reviewer: z.string().min(1),
    }),
    execute: async ({ teamName, claudeDir, reviewer }) =>
      await Promise.resolve(
        jsonTextContent(getController(teamName, claudeDir).kanban.removeReviewer(reviewer))
      ),
  });
}