import type { PRCheckDetail } from '../../shared/github/check-types'
import type { CheckStatus, IssueInfo, PRInfo } from '../../shared/github/pull-request-types'
import { derivePRCheckStatusFromRollup } from '../../shared/pr-check-status'

// ── REST API check-runs mapping ───────────────────────────────────────
// The REST check-runs endpoint returns separate status + conclusion fields
// (unlike gh pr checks which merges them into a single "state" string).

export function mapCheckRunRESTStatus(status: string): PRCheckDetail['status'] {
  const s = status?.toLowerCase()
  if (s === 'queued') {
    return 'queued'
  }
  if (s === 'in_progress') {
    return 'in_progress'
  }
  return 'completed'
}

const conclusionMap: Record<string, PRCheckDetail['conclusion']> = {
  success: 'success',
  failure: 'failure',
  cancelled: 'cancelled',
  timed_out: 'timed_out',
  skipped: 'skipped',
  neutral: 'neutral',
  action_required: 'action_required',
  stale: 'failure',
  startup_failure: 'failure'
}

export function mapCheckRunRESTConclusion(
  status: string,
  conclusion: string | null
): PRCheckDetail['conclusion'] {
  if (status?.toLowerCase() !== 'completed') {
    return 'pending'
  }
  if (!conclusion) {
    return null
  }
  return conclusionMap[conclusion.toLowerCase()] ?? null
}

// ── REST API commit status mapping ──────────────────────────────────────
// Legacy Jenkins/Prow integrations report commit statuses, not check runs.

export function mapCommitStatusRESTStatus(state: string): PRCheckDetail['status'] {
  const s = state?.toLowerCase()
  return s === 'pending' ? 'queued' : 'completed'
}

export function mapCommitStatusRESTConclusion(state: string): PRCheckDetail['conclusion'] {
  const s = state?.toLowerCase()
  if (s === 'success') {
    return 'success'
  }
  if (s === 'failure' || s === 'error') {
    return 'failure'
  }
  if (s === 'pending') {
    return 'pending'
  }
  return null
}

// ── gh pr checks mapping (single "state" string) ─────────────────────

export function mapCheckStatus(state: string): PRCheckDetail['status'] {
  const s = state?.toUpperCase()
  if (s === 'PENDING' || s === 'QUEUED') {
    return 'queued'
  }
  if (s === 'IN_PROGRESS') {
    return 'in_progress'
  }
  return 'completed'
}

export function mapCheckConclusion(state: string): PRCheckDetail['conclusion'] {
  const s = state?.toUpperCase()
  if (s === 'SUCCESS' || s === 'PASS') {
    return 'success'
  }
  if (s === 'FAILURE' || s === 'FAIL') {
    return 'failure'
  }
  if (s === 'ACTION_REQUIRED') {
    return 'action_required'
  }
  if (s === 'STALE' || s === 'STARTUP_FAILURE') {
    return 'failure'
  }
  if (s === 'CANCELLED') {
    return 'cancelled'
  }
  if (s === 'TIMED_OUT') {
    return 'timed_out'
  }
  if (s === 'SKIPPED') {
    return 'skipped'
  }
  if (s === 'PENDING' || s === 'QUEUED' || s === 'IN_PROGRESS') {
    return 'pending'
  }
  if (s === 'NEUTRAL') {
    return 'neutral'
  }
  return null
}

export function mapPRState(state: string, isDraft?: boolean): PRInfo['state'] {
  const s = state?.toUpperCase()
  if (s === 'MERGED') {
    return 'merged'
  }
  if (s === 'CLOSED') {
    return 'closed'
  }
  if (isDraft) {
    return 'draft'
  }
  return 'open'
}

// ── Issue mapping ────────────────────────────────────────────────────
// REST API returns html_url + lowercase state; gh issue view returns url + mixed-case state.
// This helper normalises both shapes into IssueInfo.

export function mapIssueInfo(data: {
  number: number
  title: string
  state: string
  url?: string
  html_url?: string
  labels?: { name: string }[]
  body?: string | null
}): IssueInfo {
  return {
    number: data.number,
    title: data.title,
    state: data.state?.toLowerCase() === 'open' ? 'open' : 'closed',
    url: data.html_url ?? data.url ?? '',
    labels: (data.labels || []).map((l) => l.name),
    ...(typeof data.body === 'string' ? { description: data.body } : {})
  }
}

export function deriveCheckStatus(rollup: unknown[] | null | undefined): CheckStatus {
  return derivePRCheckStatusFromRollup(rollup)
}