import { Section, Note } from '@/components/ui/section';
import { EmptyState } from '@/components/ui/empty-state';
import { Freshness } from '@/components/ui/freshness';
import { ClaimBadge } from '@/components/ui/badge';
import { SourceBadge, TableProvenance } from '@/components/ui/source-badge';
import { survivalFor } from '@/lib/queries/epidemiology';
import { loadProvenance, toInfo } from '@/lib/queries/provenance';
import { fmtInt, fmtPct, fmtNum, humanize } from '@/lib/format';
import type { CancerBundle } from '../load';

export async function SurvivalTab({ b }: { b: CancerBundle }) {
  const rows = await survivalFor(b.cancer.id);
  if (rows.length === 0) {
    return (
      <Section id="survival" kicker="Survival" title="Population survival">
        <EmptyState knows={[{ label: 'Statistics tab', href: `/cancer/${b.cancer.slug}/statistics` }, { label: 'Methodology: 5-year relative survival', href: '/methodology#five_year_survival' }]}>
          No survival observation is attached. Survival statistics require stage, staging system, diagnosis period, survival type and population — they are only imported with that full context (SEER, once credentials and terms are settled).
        </EmptyState>
      </Section>
    );
  }
  const prov = await loadProvenance(rows.map((r) => r.provenance_id));
  const groups = new Map<string, typeof rows>();
  for (const r of rows) {
    const k = `${r.survival_type}|${r.stage ?? 'all stages'}|${r.geography_name ?? ''}`;
    if (!groups.has(k)) groups.set(k, []);
    groups.get(k)!.push(r);
  }
  return (
    <div className="space-y-8">
      <Note tone="warn">Population survival is not an individual prognosis. Figures apply to the cohort, period, stage and population declared by the source; treatment landscapes change after the diagnosis period shown.</Note>
      {[...groups.entries()].map(([k, rs]) => {
        const f = rs[0]!;
        const firstProv = toInfo(prov.get(f.provenance_id)) ?? { sourceSlug: f.source_slug, sourceName: f.source_name };
        return (
          <Section key={k} id={k.replace(/[^a-z0-9]+/gi, '-')} kicker={f.geography_name ?? 'Population'} title={`${humanize(f.survival_type)} survival${f.stage ? ` — ${f.stage}` : ' — all stages'}`} description={f.staging_system ? `Staging system: ${f.staging_system}` : undefined}>
            <TableProvenance p={firstProv} claim={<ClaimBadge kind="observed" />}>
              {rs.length} observation{rs.length === 1 ? '' : 's'} · diagnosis period, follow-up and method as published; hover a row badge for its dataset and version.
            </TableProvenance>
            <div className="ci-table-wrap">
              <table className="ci-table">
                <thead>
                  <tr>
                    <th scope="col">Diagnosis period</th>
                    <th scope="col">Sex</th>
                    <th scope="col">Age group</th>
                    <th scope="col" className="num">
                      Follow-up (months)
                    </th>
                    <th scope="col" className="num">
                      Survival (%)
                    </th>
                    <th scope="col" className="num">
                      95% CI
                    </th>
                    <th scope="col" className="num">
                      Median (months)
                    </th>
                    <th scope="col" className="num">
                      Cohort (n)
                    </th>
                    <th scope="col">Method</th>
                    <th scope="col">Source</th>
                  </tr>
                </thead>
                <tbody>
                  {rs.map((r) => (
                    <tr key={r.id}>
                      <td>{r.diagnosis_period ?? '—'}</td>
                      <td>{humanize(r.sex)}</td>
                      <td>{r.age_group === 'all' ? 'All ages' : r.age_group}</td>
                      <td className="num">{r.duration_months}</td>
                      <td className="num font-medium">{fmtPct(r.probability)}</td>
                      <td className="num text-ink-3">{r.lower_ci != null && r.upper_ci != null ? `${fmtPct(r.lower_ci)}–${fmtPct(r.upper_ci)}` : '—'}</td>
                      <td className="num">{r.median_months != null ? fmtNum(r.median_months, 1) : '—'}</td>
                      <td className="num">{r.cohort_size != null ? fmtInt(r.cohort_size) : '—'}</td>
                      <td className="text-[12px] text-ink-3">{r.method ?? '—'}</td>
                      <td>
                        <SourceBadge compact p={toInfo(prov.get(r.provenance_id)) ?? { sourceSlug: r.source_slug, sourceName: r.source_name }} />
                      </td>
                    </tr>
                  ))}
                </tbody>
              </table>
            </div>
            <Freshness dataUpdatedAt={rs[0]!.updated_at} sourceVersion={prov.get(rs[0]!.provenance_id)?.dataset_version ?? null} />
          </Section>
        );
      })}
    </div>
  );
}