import type {
  RootlyIncidentActionResponse,
  RootlyMitigateIncidentParams,
} from '@/tools/rootly/types'
import type { ToolConfig } from '@/tools/types'

export const rootlyMitigateIncidentTool: ToolConfig<
  RootlyMitigateIncidentParams,
  RootlyIncidentActionResponse
> = {
  id: 'rootly_mitigate_incident',
  name: 'Rootly Mitigate Incident',
  description: 'Transition a Rootly incident to the mitigated state.',
  version: '1.0.0',

  params: {
    apiKey: {
      type: 'string',
      required: true,
      visibility: 'user-only',
      description: 'Rootly API key',
    },
    incidentId: {
      type: 'string',
      required: true,
      visibility: 'user-or-llm',
      description: 'The ID of the incident to mitigate',
    },
    mitigationMessage: {
      type: 'string',
      required: false,
      visibility: 'user-or-llm',
      description: 'How was the incident mitigated?',
    },
  },

  request: {
    url: (params) => `https://api.rootly.com/v1/incidents/${params.incidentId.trim()}/mitigate`,
    method: 'PUT',
    headers: (params) => ({
      'Content-Type': 'application/vnd.api+json',
      Authorization: `Bearer ${params.apiKey}`,
    }),
    body: (params) => {
      const attributes: Record<string, unknown> = {}
      if (params.mitigationMessage) attributes.mitigation_message = params.mitigationMessage
      return { data: { type: 'incidents', attributes } }
    },
  },

  transformResponse: async (response: Response) => {
    if (!response.ok) {
      const errorData = await response.json().catch(() => ({}))
      return {
        success: false,
        output: { incident: {} as RootlyIncidentActionResponse['output']['incident'] },
        error: errorData.errors?.[0]?.detail || `HTTP ${response.status}: ${response.statusText}`,
      }
    }

    const data = await response.json()
    const attrs = data.data?.attributes || {}
    return {
      success: true,
      output: {
        incident: {
          id: data.data?.id ?? null,
          sequentialId: attrs.sequential_id ?? null,
          title: attrs.title ?? '',
          slug: attrs.slug ?? null,
          kind: attrs.kind ?? null,
          summary: attrs.summary ?? null,
          status: attrs.status ?? null,
          private: attrs.private ?? false,
          url: attrs.url ?? null,
          shortUrl: attrs.short_url ?? null,
          severityName: attrs.severity?.data?.attributes?.name ?? null,
          severityId: attrs.severity?.data?.id ?? null,
          createdAt: attrs.created_at ?? '',
          updatedAt: attrs.updated_at ?? '',
          startedAt: attrs.started_at ?? null,
          mitigatedAt: attrs.mitigated_at ?? null,
          resolvedAt: attrs.resolved_at ?? null,
          closedAt: attrs.closed_at ?? null,
        },
      },
    }
  },

  outputs: {
    incident: {
      type: 'object',
      description: 'The mitigated incident',
      properties: {
        id: { type: 'string', description: 'Unique incident ID' },
        sequentialId: { type: 'number', description: 'Sequential incident number' },
        title: { type: 'string', description: 'Incident title' },
        slug: { type: 'string', description: 'Incident slug' },
        kind: { type: 'string', description: 'Incident kind' },
        summary: { type: 'string', description: 'Incident summary' },
        status: { type: 'string', description: 'Incident status' },
        private: { type: 'boolean', description: 'Whether the incident is private' },
        url: { type: 'string', description: 'URL to the incident' },
        shortUrl: { type: 'string', description: 'Short URL to the incident' },
        severityName: { type: 'string', description: 'Severity name' },
        severityId: { type: 'string', description: 'Severity ID' },
        createdAt: { type: 'string', description: 'Creation date' },
        updatedAt: { type: 'string', description: 'Last update date' },
        startedAt: { type: 'string', description: 'Start date' },
        mitigatedAt: { type: 'string', description: 'Mitigation date' },
        resolvedAt: { type: 'string', description: 'Resolution date' },
        closedAt: { type: 'string', description: 'Closed date' },
      },
    },
  },
}
