import type { CsoStack } from '../../lib/cso/preparation';

type RuntimeStack = Exclude<CsoStack, 'java'>;
import { canonical, sha256 } from '../../lib/cso/contracts';
import { ISOLATION_POLICY_HASH } from '../../lib/cso/docker';
import {
  CSO_HELPER_ABI,
  type QualifiedRuntime,
  type RuntimeCatalog,
  type RuntimePlatform,
} from '../../lib/cso/runtime-catalog';

const STACKS = ['node', 'bun', 'python', 'rails', 'postgresql'] as const;
const PLATFORMS = ['linux/amd64', 'linux/arm64'] as const;
const SOURCE_COMMIT = 'b'.repeat(40);
const WORKFLOW = 'https://github.com/garrytan/gstack/actions/runs/1';
const DIGEST = `sha256:${'a'.repeat(64)}`;

const VERSIONS: Record<RuntimeStack | 'postgresql', Record<string, string>> = {
  node: { node: '24.1.0', npm: '11.3.0', 'cso-preparation': '1.0.0' },
  bun: { bun: '1.3.10', 'cso-preparation': '1.0.0' },
  python: { python: '3.12.9', uv: '0.8.0', 'cso-preparation': '1.0.0' },
  rails: { ruby: '3.3.6', bundler: '2.6.9', 'cso-preparation': '1.0.0' },
  postgresql: { postgresql: '17.2' },
};

function runtimeId(stack: RuntimeStack | 'postgresql', platform: RuntimePlatform): string {
  return `${stack}-qualified-test-${platform === 'linux/amd64' ? 'amd64' : 'arm64'}`;
}

export function qualifiedRuntimeFixture(
  stack: RuntimeStack | 'postgresql',
  platform: RuntimePlatform = 'linux/amd64',
): QualifiedRuntime {
  const arch = platform === 'linux/amd64' ? 'amd64' : 'arm64';
  const qualification = stack === 'postgresql'
    ? {
        kind: 'postgresql' as const,
        sourceCommit: SOURCE_COMMIT,
        workflow: WORKFLOW,
        sbomDigest: DIGEST,
        provenanceDigest: DIGEST,
        verifiedProvenance: true as const,
        containmentPassed: true as const,
        coldStartPassed: true as const,
        multiDatabasePassed: true as const,
        readinessPassed: true as const,
      }
    : {
        kind: 'application' as const,
        sourceCommit: SOURCE_COMMIT,
        workflow: WORKFLOW,
        sbomDigest: DIGEST,
        provenanceDigest: DIGEST,
        verifiedProvenance: true as const,
        containmentPassed: true as const,
        coldStartPassed: true as const,
        positiveNegativeAssertionsPassed: true as const,
        heldOutRepairPassed: true as const,
      };

  return {
    id: runtimeId(stack, platform),
    stack,
    platform,
    state: 'qualified',
    image: `ghcr.io/garrytan/gstack/cso-staging/${stack}-${arch}@${DIGEST}`,
    entrypoint: '/opt/cso/entrypoint',
    helperAbi: CSO_HELPER_ABI,
    versions: { ...VERSIONS[stack] },
    policyVersion: 'cso-isolation-v1',
    qualifiedAt: '2026-09-09T00:00:00Z',
    qualification,
  };
}

/** A fresh catalog satisfying the complete reviewed and promoted runtime matrices. */
export function completeRuntimeCatalogFixture(
  revision = 'runtime-test-v1',
  previousRevision: string | null = null,
): RuntimeCatalog {
  const runtimes = STACKS.flatMap(stack => PLATFORMS.map(platform => qualifiedRuntimeFixture(stack, platform)));
  return {
    schemaVersion: 1,
    revision,
    previousRevision,
    helperAbi: CSO_HELPER_ABI,
    buildRevision: 'runtime-test-build-v1',
    profiles: runtimes.map(runtime => ({
      id: runtime.id,
      stack: runtime.stack,
      platform: runtime.platform,
      state: 'build_reviewed' as const,
      versions: { ...runtime.versions },
      reviewedAt: '2026-09-08T00:00:00Z',
    })),
    promotion: {
      sourceCommit: SOURCE_COMMIT,
      workflow: WORKFLOW,
      evidenceDigest: `sha256:${sha256(canonical(runtimes))}`,
      qualificationEvidenceDigest: DIGEST,
      requalification: {
        helperAbi: CSO_HELPER_ABI,
        isolationPolicyHash: ISOLATION_POLICY_HASH,
        preparationSha256: 'c'.repeat(64),
        buildInputsRevision: 'runtime-test-build-v1',
      },
    },
    runtimes,
  };
}

/**
 * Put a staged runtime into its stack/platform slot under the reviewed profile
 * id, as promotion does, and re-bind the promotion evidence digest the
 * validator recomputes from the runtime matrix.
 */
export function installStagedRuntime(catalog: RuntimeCatalog, runtime: QualifiedRuntime): QualifiedRuntime {
  const index = catalog.runtimes.findIndex(item => item.stack === runtime.stack && item.platform === runtime.platform);
  const profile = catalog.profiles.find(item => item.stack === runtime.stack && item.platform === runtime.platform);
  if (index < 0 || !profile || !catalog.promotion) throw new Error(`Runtime catalog fixture lacks ${runtime.stack} on ${runtime.platform}`);
  const installed = { ...runtime, id: profile.id, versions: { ...runtime.versions } };
  catalog.runtimes[index] = installed;
  profile.versions = { ...installed.versions };
  catalog.promotion.evidenceDigest = `sha256:${sha256(canonical(catalog.runtimes))}`;
  return installed;
}