/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

import { execFileSync, spawn } from 'node:child_process';
import { createHash } from 'node:crypto';
import { createServer } from 'node:http';
import { createRequire } from 'node:module';
import {
  closeSync,
  existsSync,
  mkdirSync,
  mkdtempSync,
  openSync,
  readFileSync,
  rmSync,
  statSync,
  writeFileSync,
  writeSync,
} from 'node:fs';
import { tmpdir } from 'node:os';
import { basename, join, resolve } from 'node:path';
import { pathToFileURL } from 'node:url';
import { validateCliReleaseArtifactMetrics } from './release-cli-artifact-policy.mjs';
import { findReleaseTarball } from './release-cli-eval-support.mjs';
import {
  collectRuntimeHostFailureDiagnostic,
  renderRuntimeHostFailureDiagnostic,
  retireCollectedRuntimeHostStartupDiagnostic,
} from './release-cli-runtime-host-diagnostics.mjs';
import { npmSpawnOptions } from './npm-spawn.mjs';

const MAX_OUTPUT_BYTES = 16 * 1024 * 1024;
const PROCESS_TIMEOUT_MS = 90_000;
const RUNTIME_HOST_SHUTDOWN_TIMEOUT_MS = 45_000;
const MODEL_ID = 'maka-release-smoke-model';
const CONNECTION_SLUG = 'maka-release-smoke';
const API_KEY = 'maka-release-smoke-key';
const FILE_SENTINEL = 'MAKA_RELEASE_FILESYSTEM_WORKER_OK';
const RESPONSE_SENTINEL = 'MAKA_RELEASE_SMOKE_OK';
const INSTALLED_ROOT_ENV = 'MAKA_CLI_RELEASE_INSTALLED_ROOT';
const require = createRequire(import.meta.url);

const repoRoot = resolve(import.meta.dirname, '..');
const tarballPath = resolve(
  process.argv[2] ?? findReleaseTarball(join(repoRoot, 'packages/cli/release')),
);

const installedRoot = process.env[INSTALLED_ROOT_ENV];
if (installedRoot) await runInstalledVerifier(resolve(installedRoot));
else await main();

async function main() {
  validateReleaseArtifact(tarballPath);
  const root = mkdtempSync(join(tmpdir(), 'maka-cli-tarball-smoke-'));
  let primaryError;
  let cleanupError;
  try {
    const prefix = join(root, 'prefix');
    const cache = join(root, 'empty-npm-cache');
    logStep('installing the immutable tarball with an empty offline npm cache');
    execFileSync(
      'npm',
      [
        'install',
        '--global',
        '--prefix',
        prefix,
        '--cache',
        cache,
        '--offline',
        '--no-audit',
        '--no-fund',
        tarballPath,
      ],
      npmSpawnOptions({
        cwd: root,
        env: { ...process.env, npm_config_registry: 'http://127.0.0.1:9/' },
        stdio: 'inherit',
      }),
    );
    logStep('validating the installed product in an isolated verifier process');
    execFileSync(process.execPath, [process.argv[1], tarballPath], {
      cwd: repoRoot,
      env: { ...process.env, [INSTALLED_ROOT_ENV]: root },
      stdio: 'inherit',
    });
  } catch (error) {
    primaryError = error;
  } finally {
    logStep('removing the isolated installation');
    try {
      rmSync(root, { recursive: true, force: true, maxRetries: 10, retryDelay: 100 });
    } catch (error) {
      cleanupError = error;
    }
  }
  if (primaryError && cleanupError) {
    throw new AggregateError(
      [primaryError, cleanupError],
      'Installed CLI validation and isolated-install cleanup both failed',
    );
  }
  if (primaryError) throw primaryError;
  if (cleanupError) throw cleanupError;
}

async function runInstalledVerifier(root) {
  try {
    await validateInstalledProduct(root);
    // Native PTY libraries can retain an event-loop handle after their product
    // processes settle. This child owns those libraries, so exiting here both
    // unloads them and gives the parent a reliable cleanup boundary.
    process.exit(0);
  } catch (error) {
    writeSync(2, `${formatError(error)}\n`);
    process.exit(1);
  }
}

async function validateInstalledProduct(root) {
  const prefix = join(root, 'prefix');
  const packageRoot =
    process.platform === 'win32'
      ? join(prefix, 'node_modules/maka-agent')
      : join(prefix, 'lib/node_modules/maka-agent');
  const baseEnvironment = isolatedEnvironment(join(root, 'home'));
  const maka = process.platform === 'win32' ? join(prefix, 'maka.cmd') : join(prefix, 'bin/maka');
  const cliEntrypoint = join(packageRoot, 'dist/cli.js');
  const manifest = JSON.parse(readFileSync(join(packageRoot, 'package.json'), 'utf8'));
  const crossSpawnModule = await importInstalled(packageRoot, 'node_modules/cross-spawn/index.js');
  const crossSpawn = crossSpawnModule.default ?? crossSpawnModule;
  if (typeof crossSpawn.sync !== 'function') {
    throw new Error('Installed cross-spawn sync API is unavailable');
  }

  logStep('checking bins, Eval assets, and patched runtime files');
  const version = runSync(crossSpawn.sync, maka, ['--version'], baseEnvironment, root).trim();
  if (version !== manifest.version) {
    throw new Error(`Installed CLI reports ${version}; package manifest is ${manifest.version}`);
  }
  assertOutput(runSync(crossSpawn.sync, maka, ['--help'], baseEnvironment, root), 'Usage: maka');
  assertOutput(
    runSync(crossSpawn.sync, maka, ['eval', '--help'], baseEnvironment, root),
    'usage: maka eval run',
  );
  validateInstalledRuntimeFiles(packageRoot);

  logStep('checking installed Eval spec decoding and framework preflight');
  smokeEvalPreflight({ crossSpawn: crossSpawn.sync, environment: baseEnvironment, maka, root });

  logStep('checking installed native PTY and file-lock modules');
  const nodePty = await importInstalled(packageRoot, 'node_modules/node-pty/lib/index.js');
  const ptySpawn = nodePty.spawn ?? nodePty.default?.spawn;
  if (typeof ptySpawn !== 'function') throw new Error('Installed node-pty has no spawn function');
  await smokePty(ptySpawn, baseEnvironment, root);
  await smokeNativeFileLock(packageRoot, root);
  await smokeRuntimeHostPeerProtocol({ packageRoot, cliEntrypoint, root });

  logStep('checking the interactive TUI setup path');
  await smokeInteractiveTui({
    packageRoot,
    cliEntrypoint,
    ptySpawn,
    root: join(root, 'first-run'),
  });

  logStep('checking the managed Runtime Host lifecycle');
  await smokeRuntimeHostService({
    packageRoot,
    cliEntrypoint,
    ptySpawn,
    root: join(root, 'runtime-host-service'),
  });

  logStep('checking a filesystem-backed controlled model turn');
  await smokeControlledRun({
    packageRoot,
    cliEntrypoint,
    root: join(root, 'controlled-run'),
  });

  console.log(
    `[release-cli-validation] OK — installed ${basename(tarballPath)} offline as ${version}`,
  );
}

async function smokeRuntimeHostPeerProtocol({ packageRoot, cliEntrypoint, root }) {
  const peerArtifact = await importInstalled(packageRoot, 'dist/runtime-host-peer-artifact.js');
  const server = await importInstalled(
    packageRoot,
    'node_modules/@maka/runtime-host/dist/server/index.js',
  );
  const client = await importInstalled(
    packageRoot,
    'node_modules/@maka/runtime-host/dist/client/index.js',
  );
  const mesh = await importInstalled(
    packageRoot,
    'node_modules/@maka/runtime-host/dist/peer-mesh/index.js',
  );
  const access = await importInstalled(packageRoot, 'dist/runtime-host-access-command.js');
  const clientDataRoot = join(root, 'peer-client');
  const hostRoot = join(root, 'peer-host');
  const hostKeyPath = join(root, 'peer-host.key');
  mkdirSync(clientDataRoot, { recursive: true });
  mkdirSync(hostRoot, { recursive: true });

  const previousNativePath = process.env.MAKA_RUNTIME_HOST_PEER_NATIVE_PATH;
  const previousKeyPath = process.env.MAKA_RUNTIME_HOST_PEER_KEY_PATH;
  let host;
  let connection;
  let meshAuthorityOwner;
  let meshMemberOwner;
  try {
    delete process.env.MAKA_RUNTIME_HOST_PEER_NATIVE_PATH;
    delete process.env.MAKA_RUNTIME_HOST_PEER_KEY_PATH;
    const configured = await peerArtifact.configureRuntimeHostPeerClient({
      cliPath: cliEntrypoint,
      clientDataRoot,
      environment: process.env,
    });
    if (!configured) throw new Error('Installed CLI could not resolve its direct-peer artifact');
    const nativePath = process.env.MAKA_RUNTIME_HOST_PEER_NATIVE_PATH;
    if (!nativePath) throw new Error('Installed CLI did not configure its direct-peer artifact');
    const addon = require(nativePath);
    const peerId = await addon.ensurePeerIdentity(hostKeyPath);
    const unrelatedPeerId = await addon.ensurePeerIdentity(join(root, 'unrelated-peer.key'));
    try {
      addon.startPeerEndpoint({ keyPath: hostKeyPath, expectedPeerId: unrelatedPeerId });
      throw new Error('Installed direct-peer addon accepted the wrong persisted identity');
    } catch (error) {
      if (!String(error).includes('peer_identity_mismatch')) throw error;
    }
    meshAuthorityOwner = await mesh.openRuntimeHostPeerMeshOwner({
      nativePath,
      keyPath: hostKeyPath,
      expectedPeerId: peerId,
      dataRoot: join(root, 'mesh-authority'),
      listenAddresses: ['/ip4/127.0.0.1/udp/0/quic-v1'],
    });
    host = await server.startExecutionRuntimeHostService({
      rootPath: hostRoot,
      peer: { client: meshAuthorityOwner.client },
    });
    const listener = host.peerListeners[0];
    if (!listener || listener.peerId !== peerId || listener.listenAddresses.length === 0) {
      throw new Error('Installed Runtime Host direct-peer listener did not become ready');
    }
    const issued = await access.issueRuntimeHostAccessCredential({
      rootPath: hostRoot,
      expectedRootId: host.rootId,
      principalKind: 'remote_owner',
      principalId: 'release-smoke-peer-client',
      operationGrants: [],
      canPublishClientCapabilities: false,
      canUseHostPaths: false,
      preset: 'terminal-client',
    });
    const meshMemberKeyPath = join(root, 'mesh-member.key');
    const meshMemberDataRoot = join(root, 'mesh-member');
    meshMemberOwner = await mesh.openRuntimeHostPeerMeshOwner({
      nativePath,
      keyPath: meshMemberKeyPath,
      dataRoot: meshMemberDataRoot,
      listenAddresses: ['/ip4/127.0.0.1/udp/0/quic-v1'],
    });
    const meshAuthority = meshAuthorityOwner.mesh;
    let meshMember = meshMemberOwner.mesh;
    const created = await meshAuthority.create();
    const joined = await meshMember.join(await meshAuthority.invite(created.roster.roster.meshId));
    if (joined.roster.roster.members.length !== 2) {
      throw new Error('Installed Runtime Host peer Mesh did not admit the invited peer');
    }
    connection = await client.connectRemoteRuntimeHostProfile({
      profile: {
        id: 'release-smoke-peer',
        name: 'Release smoke peer',
        kind: 'remote',
        rootId: host.rootId,
        transport: {
          kind: 'libp2p-direct',
          peerId,
          routeHints: ['/ip4/127.0.0.1/udp/1/quic-v1'],
          coordinationRelays: [],
        },
      },
      credential: issued.credential,
      clientInstanceId: 'release-smoke-peer-client',
      peerClient: meshMemberOwner.client,
      connectTimeoutMs: 10_000,
      handshakeTimeoutMs: 10_000,
      readyTimeoutMs: 10_000,
    });
    const status = await connection.status(10_000);
    if (status.state !== 'ready') {
      throw new Error(`Installed Runtime Host direct-peer status is ${status.state}`);
    }
    const removed = await meshAuthority.remove(
      created.roster.roster.meshId,
      meshMemberOwner.client.identity().peerId,
    );
    if (removed.roster.roster.members.length !== 1) {
      throw new Error('Installed Runtime Host peer Mesh did not remove the invited peer');
    }
    await meshMemberOwner.close();
    meshMemberOwner = await mesh.openRuntimeHostPeerMeshOwner({
      nativePath,
      keyPath: meshMemberKeyPath,
      dataRoot: meshMemberDataRoot,
      listenAddresses: ['/ip4/127.0.0.1/udp/0/quic-v1'],
    });
    meshMember = meshMemberOwner.mesh;
    const stale = meshMember.status()[0];
    if (stale?.roster.roster.revision !== joined.roster.roster.revision) {
      throw new Error('Installed Runtime Host peer Mesh did not recover the last-known roster');
    }
    const rejoined = await meshMember.join(
      await meshAuthority.invite(created.roster.roster.meshId),
    );
    if (
      rejoined.roster.roster.members.length !== 2 ||
      rejoined.roster.roster.revision <= stale.roster.roster.revision
    ) {
      throw new Error('Installed Runtime Host peer Mesh did not re-admit the removed peer');
    }
    await meshAuthority.remove(
      created.roster.roster.meshId,
      meshMemberOwner.client.identity().peerId,
    );
    await meshMember.reconcile();
    if (meshMember.status().length !== 0) {
      throw new Error('Installed Runtime Host peer Mesh did not propagate member removal');
    }
  } finally {
    await connection?.close().catch(() => undefined);
    await host?.close().catch(() => undefined);
    await meshMemberOwner?.close().catch(() => undefined);
    await meshAuthorityOwner?.close().catch(() => undefined);
    restoreEnvironment('MAKA_RUNTIME_HOST_PEER_NATIVE_PATH', previousNativePath);
    restoreEnvironment('MAKA_RUNTIME_HOST_PEER_KEY_PATH', previousKeyPath);
  }
}

function restoreEnvironment(name, value) {
  if (value === undefined) delete process.env[name];
  else process.env[name] = value;
}

function validateReleaseArtifact(path) {
  if (!existsSync(path)) throw new Error(`Release tarball does not exist: ${path}`);
  const checksumPath = `${path}.sha256`;
  const inventoryPath = `${path}.files.json`;
  if (!existsSync(checksumPath) || !existsSync(inventoryPath)) {
    throw new Error('Release tarball checksum or file inventory is missing');
  }
  const checksum = readFileSync(checksumPath, 'utf8').trim().split(/\s+/u);
  if (checksum.length !== 2 || checksum[1] !== basename(path)) {
    throw new Error('Release tarball checksum sidecar is malformed');
  }
  const actual = createHash('sha256').update(readFileSync(path)).digest('hex');
  if (checksum[0] !== actual) throw new Error('Release tarball checksum does not match');

  const files = JSON.parse(readFileSync(inventoryPath, 'utf8'));
  if (!Array.isArray(files)) throw new Error('Release tarball file inventory must be an array');
  let unpackedBytes = 0;
  for (const file of files) {
    if (!Number.isSafeInteger(file?.size) || file.size < 0 || typeof file.path !== 'string') {
      throw new Error('Release tarball file inventory contains an invalid entry');
    }
    unpackedBytes += file.size;
  }
  validateCliReleaseArtifactMetrics({
    compressedBytes: statSync(path).size,
    unpackedBytes,
    entryCount: files.length,
  });
}

function validateInstalledRuntimeFiles(packageRoot) {
  for (const path of [
    // Incubator policy: the installed package carries the incubating
    // disclaimer next to LICENSE/NOTICE, like every other Maka release.
    'DISCLAIMER-WIP',
    'node_modules/@maka/runtime/dist/workers/filesystem-worker.js',
    'node_modules/@maka/runtime-host/dist/execution-candidate-main.js',
    'node_modules/@maka/eval/dist/index.js',
    'node_modules/@maka/eval/harbor/relay_agent.py',
    'node_modules/@maka/eval/harbor/egress-proxy/network-policy',
  ]) {
    if (!existsSync(join(packageRoot, path))) {
      throw new Error(`Installed runtime file is missing: ${path}`);
    }
  }
  assertOutput(
    readFileSync(join(packageRoot, 'node_modules/node-pty/lib/unixTerminal.js'), 'utf8'),
    'CustomWriteStream.prototype._ownsFileDescriptor',
  );
  assertOutput(
    readFileSync(join(packageRoot, 'node_modules/@ai-sdk/provider-utils/dist/index.js'), 'utf8'),
    'function absentIfBlank',
  );
}

function smokeEvalPreflight({ crossSpawn, environment, maka, root }) {
  const evalRoot = join(root, 'eval-preflight');
  mkdirSync(evalRoot, { recursive: true });
  const machineEnvironment = {
    ...environment,
    MAKA_RELEASE_EVAL_PYTHON: process.execPath,
    MAKA_RELEASE_EVAL_TASKS: evalRoot,
    MAKA_RELEASE_EVAL_TRIALS: join(evalRoot, 'trials'),
  };
  for (const framework of ['harbor', 'pier']) {
    const specPath = join(evalRoot, `${framework}.json`);
    writeFileSync(specPath, `${JSON.stringify(evalPreflightSpec(framework))}\n`, 'utf8');
    const result = crossSpawn(maka, ['eval', 'run', specPath, '--out', join(evalRoot, framework)], {
      cwd: evalRoot,
      env: machineEnvironment,
      encoding: 'utf8',
      timeout: 30_000,
      maxBuffer: MAX_OUTPUT_BYTES,
    });
    if (result.error) throw result.error;
    if (result.status === 0) {
      throw new Error(`${framework} Eval preflight unexpectedly accepted the Node executable`);
    }
    assertOutput(
      `${result.stdout ?? ''}\n${result.stderr ?? ''}`,
      `${framework} Python environment MAKA_RELEASE_EVAL_PYTHON is unavailable or does not provide`,
    );
  }
}

function evalPreflightSpec(framework) {
  return {
    schemaVersion: 'maka.eval.v1',
    id: `release-${framework}-preflight`,
    benchmark: {
      id: 'release-preflight',
      version: '0000000000000000000000000000000000000000',
      config: { repository: 'https://invalid.invalid/release-preflight.git' },
    },
    executor: {
      kind: framework,
      config: {
        frameworkVersion: framework === 'harbor' ? '0.20.0' : '0.3.0',
        pythonPathEnv: 'MAKA_RELEASE_EVAL_PYTHON',
        trialsRootEnv: 'MAKA_RELEASE_EVAL_TRIALS',
        ...(framework === 'pier' ? { tasksRootEnv: 'MAKA_RELEASE_EVAL_TASKS' } : {}),
        environment: { type: 'docker', delete: true },
        preparationEnvironment: [],
        mounts: [],
      },
    },
    subjects: [
      {
        id: 'subject',
        kind: 'external',
        credentials: [],
        config: { command: process.execPath, args: ['--version'], result: 'exit-code' },
      },
    ],
    tasks: [
      {
        id: 'task',
        input: 'Do not execute this preflight-only task.',
        config: framework === 'harbor' ? { harbor: { path: 'task' } } : { pier: { path: 'task' } },
      },
    ],
    repetitions: 1,
    budget: { timeoutMultiplier: 1 },
    verifier: { reward: 'reward' },
  };
}

async function smokePty(ptySpawn, environment, cwd) {
  const result = await runPtyScenario({
    ptySpawn,
    command: process.execPath,
    args: ['-e', 'process.stdout.write("maka-pty-ok")'],
    cwd,
    environment,
    marker: 'maka-pty-ok',
    timeoutMs: 30_000,
  });
  if (result.exitCode !== 0) throw new Error(`Installed PTY exited with ${result.exitCode}`);
}

async function smokeNativeFileLock(packageRoot, root) {
  const imported = await importInstalled(packageRoot, 'node_modules/fs-native-extensions/index.js');
  const native = imported.default ?? imported;
  if (typeof native.tryLock !== 'function' || typeof native.unlock !== 'function') {
    throw new Error('Installed fs-native-extensions lock API is unavailable');
  }
  const lockPath = join(root, 'native-lock');
  const handle = openSync(lockPath, 'a+');
  try {
    if (native.tryLock(handle) !== true) throw new Error('Native file lock was not granted');
    native.unlock(handle);
  } finally {
    closeSync(handle);
  }
}

async function smokeInteractiveTui({ packageRoot, cliEntrypoint, ptySpawn, root }) {
  mkdirSync(root, { recursive: true });
  const home = join(root, 'home');
  const workspace = join(root, 'workspace');
  mkdirSync(workspace, { recursive: true });
  const environment = isolatedEnvironment(home);
  const dataRoots = await resolveInstalledDataRoots(packageRoot, environment, home);
  await withCleanup(
    async () => {
      const result = await runPtyScenario({
        ptySpawn,
        command: process.execPath,
        args: [cliEntrypoint],
        cwd: workspace,
        environment,
        marker: 'Set Up Provider',
        onOutput: (terminal, output) => {
          if (!output.includes('/setup')) return false;
          terminal.write('/setup\r');
          return true;
        },
        onMarker: (terminal) => {
          terminal.write('\x03');
          setTimeout(() => terminal.write('\x04'), 250);
        },
        timeoutMs: PROCESS_TIMEOUT_MS,
      });
      if (result.exitCode !== 0) throw new Error(`Interactive TUI exited with ${result.exitCode}`);
    },
    (completed) => settleRuntimeHost(packageRoot, dataRoots.workspaceRoot, completed),
  );
}

async function smokeRuntimeHostService({ packageRoot, cliEntrypoint, ptySpawn, root }) {
  mkdirSync(root, { recursive: true });
  const environment = isolatedEnvironment(join(root, 'home'));
  const clientDataRoot = join(root, 'client');
  const stateRoot = join(root, 'state');
  mkdirSync(clientDataRoot, { recursive: true });
  mkdirSync(stateRoot, { recursive: true });
  const configPath = join(clientDataRoot, 'runtime-host-service.json');
  writeFileSync(
    configPath,
    `${JSON.stringify({
      schemaVersion: 2,
      rootPath: stateRoot,
      projectDirectoryRoots: [{ label: '~', path: root }],
      websocket: {
        host: '127.0.0.1',
        port: await allocateLoopbackPort(),
        path: '/runtime-host',
      },
      launch: { nodePath: process.execPath, cliPath: cliEntrypoint },
    })}\n`,
    { mode: 0o600 },
  );
  let ready;
  await withCleanup(
    async () => {
      const result = await runPtyScenario({
        ptySpawn,
        command: process.execPath,
        args: [
          cliEntrypoint,
          'runtime-host',
          'serve',
          '--managed-service-config',
          configPath,
          '--json',
        ],
        cwd: root,
        environment,
        marker: '"event":"runtime_host_ready"',
        onMarker: (terminal, output) => {
          const line = output
            .split(/\r?\n/u)
            .map((candidate) => candidate.trim())
            .find((candidate) => candidate.includes('"event":"runtime_host_ready"'));
          ready = line ? parseTerminalJsonObject(line) : undefined;
          terminal.write('\x03');
        },
        timeoutMs: PROCESS_TIMEOUT_MS,
      });
      if (result.exitCode !== 0) {
        throw new Error(`Runtime Host service exited with ${result.exitCode}`);
      }
      if (
        ready?.schemaVersion !== 1 ||
        ready.protocol?.version === undefined ||
        !ready.hostEpoch ||
        !ready.rootId ||
        !ready.listeners?.some((listener) => listener.kind === 'local_ipc')
      ) {
        throw new Error('Runtime Host ready event is incomplete');
      }
    },
    (completed) => settleRuntimeHost(packageRoot, stateRoot, completed),
  );
}

async function allocateLoopbackPort() {
  const server = createServer();
  await new Promise((resolve, reject) => {
    server.once('error', reject);
    server.listen(0, '127.0.0.1', resolve);
  });
  const address = server.address();
  await new Promise((resolve, reject) =>
    server.close((error) => (error ? reject(error) : resolve())),
  );
  if (!address || typeof address === 'string') throw new Error('Unable to allocate a TCP port');
  return address.port;
}

async function smokeControlledRun({ packageRoot, cliEntrypoint, root }) {
  mkdirSync(root, { recursive: true });
  const home = join(root, 'home');
  const workspace = join(root, 'workspace');
  mkdirSync(workspace, { recursive: true });
  writeFileSync(join(workspace, 'smoke.txt'), `${FILE_SENTINEL}\n`, 'utf8');
  const environment = isolatedEnvironment(home);
  const dataRoots = await resolveInstalledDataRoots(packageRoot, environment, home);
  const provider = await startProvider();
  await withCleanup(
    async () => {
      await configureRuntimePolicy(packageRoot, dataRoots.workspaceRoot, provider.baseUrl);
      const result = await runProcess(
        process.execPath,
        [
          cliEntrypoint,
          'run',
          'Read smoke.txt, then report the exact release-smoke result.',
          '--cwd',
          workspace,
          '--connection',
          CONNECTION_SLUG,
          '--model',
          MODEL_ID,
          '--timeout',
          '45',
          '--max-steps',
          '4',
          '--yolo',
        ],
        {
          cwd: workspace,
          environment,
          timeoutMs: PROCESS_TIMEOUT_MS,
        },
      );
      if (result.exitCode !== 0) {
        throw new Error(`Controlled maka run exited with ${result.exitCode}: ${result.stderr}`);
      }
      if (provider.error) throw provider.error;
      assertOutput(result.stdout, RESPONSE_SENTINEL);
      if (!provider.sawReadTool || !provider.sawFileSentinel) {
        throw new Error('Controlled maka run did not execute the installed filesystem worker');
      }
    },
    (completed) =>
      runCleanupSteps([
        () => provider.close(),
        () => settleRuntimeHost(packageRoot, dataRoots.workspaceRoot, completed),
      ]),
  );
}

async function configureRuntimePolicy(packageRoot, rootPath, baseUrl) {
  const authority = await importInstalled(
    packageRoot,
    'node_modules/@maka/storage/dist/root-authority.js',
  );
  const policyModule = await importInstalled(
    packageRoot,
    'node_modules/@maka/storage/dist/runtime-policy-stores.js',
  );
  const capability = await authority.resolveStorageRoot({ path: rootPath, kind: 'interactive' });
  const owner = await authority.tryAcquireInteractiveRootOwner(capability);
  if (!owner) throw new Error('Unable to acquire the controlled-run Runtime Host root');
  try {
    const policy = await policyModule.openInteractiveRuntimePolicyStoresForWrite(owner.lease);
    const initial = await policy.connectionCatalog.getSnapshot();
    const created = await policy.connectionCatalog.create({
      expectedCatalogRevision: initial.revision,
      connection: {
        slug: CONNECTION_SLUG,
        name: 'Maka release smoke',
        providerType: 'moonshot',
        baseUrl,
        enabled: true,
        enabledModelIds: [MODEL_ID],
      },
    });
    if (created.kind !== 'committed') throw new Error(`Connection setup failed: ${created.kind}`);
    const connection = created.snapshot.connections.find(
      (candidate) => candidate.slug === CONNECTION_SLUG,
    );
    if (!connection) throw new Error('Connection setup omitted the controlled connection');
    const credential = await policy.credentialVault.set({
      locator: {
        scope: 'connection',
        connectionId: connection.connectionId,
        kind: 'api_key',
      },
      expected: null,
      secret: API_KEY,
    });
    if (credential.kind !== 'committed') {
      throw new Error(`Credential setup failed: ${credential.kind}`);
    }
    const fetch = await policy.operations.beginModelFetch(connection.connectionId);
    if (fetch.kind !== 'ready') throw new Error(`Model setup failed: ${fetch.kind}`);
    const discovered = await policy.operations.completeModelFetch(fetch.ticket, {
      models: [
        {
          id: MODEL_ID,
          capabilities: { chat: true, functionCalling: true },
          contextWindow: 8_192,
          maxOutputTokens: 256,
        },
      ],
      source: 'fetched',
      fetchedAt: Date.now(),
    });
    if (discovered.kind !== 'committed') {
      throw new Error(`Model setup commit failed: ${discovered.kind}`);
    }
    const defaulted = await policy.connectionCatalog.setDefaultTarget({
      expectedCatalogRevision: discovered.snapshot.revision,
      target: { connectionId: connection.connectionId, modelId: MODEL_ID },
    });
    if (defaulted.kind !== 'committed') {
      throw new Error(`Default model setup failed: ${defaulted.kind}`);
    }
  } finally {
    await owner.close();
  }
}

async function startProvider() {
  let streamRequestCount = 0;
  let sawReadTool = false;
  let sawFileSentinel = false;
  let providerError;
  const server = createServer((request, response) => {
    void handleRequest(request, response).catch((error) => {
      providerError ??= error instanceof Error ? error : new Error(String(error));
      response.writeHead(500, { 'content-type': 'text/plain' });
      response.end('release smoke provider failure');
    });
  });
  await new Promise((resolve, reject) => {
    server.once('error', reject);
    server.listen(0, '127.0.0.1', () => {
      server.off('error', reject);
      resolve();
    });
  });
  const address = server.address();
  if (!address || typeof address === 'string') throw new Error('Provider did not bind TCP');

  async function handleRequest(request, response) {
    if (request.method === 'GET' && request.url?.endsWith('/models')) {
      response.writeHead(200, { 'content-type': 'application/json' });
      response.end(JSON.stringify({ object: 'list', data: [{ id: MODEL_ID, object: 'model' }] }));
      return;
    }
    if (request.method !== 'POST') throw new Error(`Unexpected provider method: ${request.method}`);
    if (request.headers.authorization !== `Bearer ${API_KEY}`) {
      throw new Error('Controlled provider received the wrong authorization header');
    }
    const body = JSON.parse(await readRequestBody(request));
    if (body.stream !== true) {
      respondNonStreaming(response);
      return;
    }
    const toolNames = (body.tools ?? []).map((tool) => tool.function?.name ?? tool.name);
    if (toolNames.includes('Read')) {
      streamRequestCount += 1;
      if (streamRequestCount === 1) {
        sawReadTool = true;
        respondToolCall(response, 'Read', { path: 'smoke.txt' });
        return;
      }
      sawFileSentinel ||= JSON.stringify(body).includes(FILE_SENTINEL);
      if (!sawFileSentinel) throw new Error('Read tool result did not contain the smoke file');
    }
    respondText(response, RESPONSE_SENTINEL);
  }

  return {
    baseUrl: `http://127.0.0.1:${address.port}/v1`,
    get sawReadTool() {
      return sawReadTool;
    },
    get sawFileSentinel() {
      return sawFileSentinel;
    },
    get error() {
      return providerError;
    },
    close: () =>
      new Promise((resolve, reject) => {
        server.close((error) => (error ? reject(error) : resolve()));
      }),
  };
}

function respondToolCall(response, name, args) {
  respondSse(response, [
    {
      id: 'chatcmpl-release-smoke-tool',
      object: 'chat.completion.chunk',
      created: 1,
      model: MODEL_ID,
      choices: [
        {
          index: 0,
          delta: {
            role: 'assistant',
            tool_calls: [
              {
                index: 0,
                id: 'release-smoke-read',
                type: 'function',
                function: { name, arguments: JSON.stringify(args) },
              },
            ],
          },
          finish_reason: null,
        },
      ],
    },
    {
      id: 'chatcmpl-release-smoke-tool',
      object: 'chat.completion.chunk',
      created: 1,
      model: MODEL_ID,
      choices: [{ index: 0, delta: {}, finish_reason: 'tool_calls' }],
      usage: { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 },
    },
  ]);
}

function respondText(response, text) {
  respondSse(response, [
    {
      id: 'chatcmpl-release-smoke-text',
      object: 'chat.completion.chunk',
      created: 2,
      model: MODEL_ID,
      choices: [{ index: 0, delta: { role: 'assistant', content: text }, finish_reason: null }],
    },
    {
      id: 'chatcmpl-release-smoke-text',
      object: 'chat.completion.chunk',
      created: 2,
      model: MODEL_ID,
      choices: [{ index: 0, delta: {}, finish_reason: 'stop' }],
      usage: { prompt_tokens: 2, completion_tokens: 1, total_tokens: 3 },
    },
  ]);
}

function respondSse(response, events) {
  response.writeHead(200, { 'content-type': 'text/event-stream' });
  for (const event of events) response.write(`data: ${JSON.stringify(event)}\n\n`);
  response.end('data: [DONE]\n\n');
}

function respondNonStreaming(response) {
  response.writeHead(200, { 'content-type': 'application/json' });
  response.end(
    JSON.stringify({
      id: 'chatcmpl-release-smoke-summary',
      object: 'chat.completion',
      created: 3,
      model: MODEL_ID,
      choices: [
        {
          index: 0,
          message: { role: 'assistant', content: RESPONSE_SENTINEL },
          finish_reason: 'stop',
        },
      ],
      usage: { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 },
    }),
  );
}

async function readRequestBody(request) {
  let body = '';
  request.setEncoding('utf8');
  for await (const chunk of request) {
    body += chunk;
    if (Buffer.byteLength(body) > MAX_OUTPUT_BYTES)
      throw new Error('Provider request is too large');
  }
  return body;
}

async function resolveInstalledDataRoots(packageRoot, environment, home) {
  const storage = await importInstalled(
    packageRoot,
    'node_modules/@maka/storage/dist/workspace-root.js',
  );
  return storage.resolveMakaDataRoots({ env: environment, homeDir: home, profileName: 'Maka' });
}

async function waitForRuntimeHostShutdown(packageRoot, rootPath) {
  const authority = await importInstalled(
    packageRoot,
    'node_modules/@maka/storage/dist/root-authority.js',
  );
  const capability = await authority.resolveStorageRoot({ path: rootPath, kind: 'interactive' });
  const { controlDirectory } = await authority.prepareStorageRootControlDirectory(capability);
  const registrationPath = join(controlDirectory, 'registration.json');
  const deadline = Date.now() + RUNTIME_HOST_SHUTDOWN_TIMEOUT_MS;
  while (Date.now() < deadline) {
    if (!existsSync(registrationPath)) {
      const owner = await authority.tryAcquireInteractiveRootOwner(capability);
      if (owner) {
        await owner.close();
        return;
      }
    }
    await delay(100);
  }
  throw new Error(`Runtime Host did not release its root: ${rootPath}`);
}

async function withCleanup(action, cleanup) {
  let primaryError;
  let cleanupError;
  try {
    await action();
  } catch (error) {
    primaryError = error;
  }
  try {
    await cleanup(primaryError === undefined);
  } catch (error) {
    cleanupError = error;
  }
  if (primaryError && cleanupError) {
    throw new AggregateError([primaryError, cleanupError], 'Validation and cleanup both failed');
  }
  if (primaryError) throw primaryError;
  if (cleanupError) throw cleanupError;
}

async function runCleanupSteps(steps) {
  const errors = [];
  for (const step of steps) {
    try {
      await step();
    } catch (error) {
      errors.push(error);
    }
  }
  if (errors.length === 1) throw errors[0];
  if (errors.length > 1) throw new AggregateError(errors, 'Multiple cleanup steps failed');
}

async function settleRuntimeHost(packageRoot, rootPath, requireNaturalShutdown) {
  if (!requireNaturalShutdown) {
    await reportRuntimeHostFailureDiagnostics(packageRoot, rootPath).catch(() => undefined);
  }
  let naturalShutdownError;
  try {
    await waitForRuntimeHostShutdown(packageRoot, rootPath);
    return;
  } catch (error) {
    naturalShutdownError = error;
  }
  let forcedCleanupError;
  try {
    const authority = await importInstalled(
      packageRoot,
      'node_modules/@maka/storage/dist/root-authority.js',
    );
    const capability = await authority.resolveStorageRoot({ path: rootPath, kind: 'interactive' });
    const { controlDirectory } = await authority.prepareStorageRootControlDirectory(capability);
    const registrationPath = join(controlDirectory, 'registration.json');
    if (existsSync(registrationPath)) {
      const registration = JSON.parse(readFileSync(registrationPath, 'utf8'));
      if (
        registration.rootId !== capability.rootId ||
        !Number.isSafeInteger(registration.pid) ||
        registration.pid <= 0
      ) {
        throw new Error('Refusing to clean up a Runtime Host with an unexpected registration');
      }
      await terminateRegisteredProcess(registration.pid);
    }
  } catch (error) {
    forcedCleanupError = error;
  }
  if (naturalShutdownError && forcedCleanupError) {
    throw new AggregateError(
      [naturalShutdownError, forcedCleanupError],
      'Runtime Host did not stop naturally and forced cleanup failed',
    );
  }
  if (forcedCleanupError) throw forcedCleanupError;
  if (requireNaturalShutdown) throw naturalShutdownError;
}

async function reportRuntimeHostFailureDiagnostics(packageRoot, rootPath) {
  if (process.platform !== 'win32') return;
  const diagnostic = await collectRuntimeHostFailureDiagnostic(packageRoot, rootPath);
  const rendered = renderRuntimeHostFailureDiagnostic(diagnostic);
  writeSync(2, `[release-cli-validation] Runtime Host failure diagnostics: ${rendered}\n`);
  await retireCollectedRuntimeHostStartupDiagnostic(packageRoot, diagnostic);
}

async function terminateRegisteredProcess(pid) {
  try {
    process.kill(pid, 'SIGTERM');
  } catch (error) {
    if (error?.code !== 'ESRCH') throw error;
  }
  let deadline = Date.now() + 5_000;
  while (Date.now() < deadline && processExists(pid)) await delay(100);
  if (processExists(pid)) {
    try {
      process.kill(pid, 'SIGKILL');
    } catch (error) {
      if (error?.code !== 'ESRCH') throw error;
    }
    deadline = Date.now() + 5_000;
    while (Date.now() < deadline && processExists(pid)) await delay(100);
    if (processExists(pid)) throw new Error(`Runtime Host process ${pid} did not exit`);
  }
}

function runPtyScenario({
  ptySpawn,
  command,
  args,
  cwd,
  environment,
  marker,
  onOutput,
  onMarker,
  timeoutMs,
}) {
  return new Promise((resolvePromise, reject) => {
    let output = '';
    let markerSeen = false;
    let outputActionApplied = false;
    let settled = false;
    const terminal = ptySpawn(command, args, {
      cwd,
      env: environment,
      name: 'xterm-256color',
      cols: 100,
      rows: 30,
    });
    const timer = setTimeout(() => {
      if (settled) return;
      settled = true;
      terminal.kill();
      reject(
        new Error(
          `PTY command timed out waiting for ${JSON.stringify(marker)}: ${output.slice(-4_000)}`,
        ),
      );
    }, timeoutMs);
    terminal.onData((chunk) => {
      if (settled) return;
      try {
        output = appendBounded(output, chunk);
      } catch (error) {
        settled = true;
        clearTimeout(timer);
        terminal.kill();
        reject(error);
        return;
      }
      if (!outputActionApplied && onOutput) {
        try {
          outputActionApplied = onOutput(terminal, output) === true;
        } catch (error) {
          settled = true;
          clearTimeout(timer);
          terminal.kill();
          reject(error);
          return;
        }
      }
      if (!markerSeen && output.includes(marker)) {
        markerSeen = true;
        try {
          onMarker?.(terminal, output);
        } catch (error) {
          settled = true;
          clearTimeout(timer);
          terminal.kill();
          reject(error);
        }
      }
    });
    terminal.onExit(({ exitCode, signal }) => {
      if (settled) return;
      settled = true;
      clearTimeout(timer);
      if (!markerSeen) {
        reject(new Error(`PTY command exited before ${JSON.stringify(marker)}: ${output}`));
        return;
      }
      resolvePromise({ exitCode, signal, output });
    });
  });
}

function runProcess(command, args, { cwd, environment, timeoutMs }) {
  return new Promise((resolvePromise, reject) => {
    const child = spawn(command, args, {
      cwd,
      env: environment,
      stdio: ['ignore', 'pipe', 'pipe'],
      windowsHide: true,
    });
    let stdout = '';
    let stderr = '';
    let timedOut = false;
    let settled = false;
    const fail = (error) => {
      if (settled) return;
      settled = true;
      clearTimeout(timer);
      child.kill('SIGKILL');
      reject(error);
    };
    const timer = setTimeout(() => {
      timedOut = true;
      child.kill('SIGKILL');
    }, timeoutMs);
    child.stdout.setEncoding('utf8');
    child.stderr.setEncoding('utf8');
    child.stdout.on('data', (chunk) => {
      try {
        stdout = appendBounded(stdout, chunk);
      } catch (error) {
        fail(error);
      }
    });
    child.stderr.on('data', (chunk) => {
      try {
        stderr = appendBounded(stderr, chunk);
      } catch (error) {
        fail(error);
      }
    });
    child.once('error', (error) => {
      fail(error);
    });
    child.once('exit', (exitCode, signal) => {
      if (settled) return;
      settled = true;
      clearTimeout(timer);
      if (timedOut) {
        reject(new Error(`Command timed out after ${timeoutMs}ms: ${command}`));
        return;
      }
      resolvePromise({ exitCode, signal, stdout, stderr });
    });
  });
}

function isolatedEnvironment(home) {
  const appData = join(home, 'AppData', 'Roaming');
  const localAppData = join(home, 'AppData', 'Local');
  const temporaryDirectory = join(home, 'tmp');
  for (const path of [home, appData, localAppData, join(home, '.config'), temporaryDirectory]) {
    mkdirSync(path, { recursive: true });
  }
  return {
    ...process.env,
    HOME: home,
    USERPROFILE: home,
    XDG_CONFIG_HOME: join(home, '.config'),
    APPDATA: appData,
    LOCALAPPDATA: localAppData,
    TMPDIR: temporaryDirectory,
    TEMP: temporaryDirectory,
    TMP: temporaryDirectory,
    NODE_PATH: '',
    TERM: 'xterm-256color',
    NO_PROXY: '127.0.0.1,localhost',
    no_proxy: '127.0.0.1,localhost',
  };
}

function importInstalled(packageRoot, relativePath) {
  return import(pathToFileURL(join(packageRoot, relativePath)).href);
}

function runSync(spawnSync, command, args, environment, cwd) {
  const result = spawnSync(command, args, {
    cwd,
    env: environment,
    encoding: 'utf8',
    timeout: 30_000,
    maxBuffer: MAX_OUTPUT_BYTES,
  });
  if (result.error) throw result.error;
  if (result.status !== 0) {
    throw new Error(
      `Installed command exited with ${result.status ?? result.signal}: ${result.stderr}`,
    );
  }
  return result.stdout;
}

function appendBounded(previous, chunk) {
  const next = previous + chunk;
  if (Buffer.byteLength(next) > MAX_OUTPUT_BYTES) {
    throw new Error(`Command output exceeded ${MAX_OUTPUT_BYTES} bytes`);
  }
  return next;
}

function parseTerminalJsonObject(line) {
  const start = line.indexOf('{');
  const end = line.lastIndexOf('}');
  if (start < 0 || end < start) throw new Error('Terminal output did not contain a JSON object');
  return JSON.parse(line.slice(start, end + 1));
}

function assertOutput(output, marker) {
  if (!output.includes(marker)) {
    throw new Error(`Expected output to contain ${JSON.stringify(marker)}`);
  }
}

function logStep(message) {
  console.log(`[release-cli-validation] ${message}`);
}

function formatError(error) {
  const rendered = error instanceof Error ? (error.stack ?? error.message) : String(error);
  if (!(error instanceof AggregateError)) return rendered;
  return [rendered, ...error.errors.map((cause) => `Caused by:\n${formatError(cause)}`)].join('\n');
}

function delay(ms) {
  return new Promise((resolvePromise) => setTimeout(resolvePromise, ms));
}

function processExists(pid) {
  try {
    process.kill(pid, 0);
    return true;
  } catch (error) {
    if (error?.code === 'ESRCH') return false;
    if (error?.code === 'EPERM') return true;
    throw error;
  }
}