import { z } from 'zod';
import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
import { errorResult, jsonResult, toErrorMessage } from '../shared.js';
import { dockerManager } from './dockerManager.js';

export function registerDockerStopRemoveTools(server: McpServer): void {
  server.registerTool(
    'docker_stop_container',
    {
      title: 'Docker Stop Container',
      description: 'Stop a Docker container started by this server.',
      inputSchema: {
        containerId: z.string().min(1).describe('Container ID returned by docker_run_container.'),
        timeout: z.number().int().positive().optional().default(10).describe('Seconds to wait before killing.'),
      },
    },
    async ({ containerId, timeout }) => {
      try {
        await dockerManager.ensureReachable();
        await dockerManager.getDocker().getContainer(containerId).stop({ t: timeout });
        return jsonResult({ stopped: true, containerId });
      } catch (error) {
        return errorResult(toErrorMessage(error));
      }
    },
  );

  server.registerTool(
    'docker_remove_container',
    {
      title: 'Docker Remove Container',
      description: 'Remove a Docker container started by this server.',
      inputSchema: {
        containerId: z.string().min(1).describe('Container ID returned by docker_run_container.'),
        force: z.boolean().optional().default(false).describe('Force removal even if running.'),
      },
    },
    async ({ containerId, force }) => {
      try {
        await dockerManager.ensureReachable();
        await dockerManager.getDocker().getContainer(containerId).remove({ force });
        dockerManager.untrack(containerId);
        return jsonResult({ removed: true, containerId });
      } catch (error) {
        return errorResult(toErrorMessage(error));
      }
    },
  );
}