routstrd/tests/install-script.test.ts

import { afterEach, describe, expect, test } from "bun:test";
import { createHash } from "crypto";
import { chmodSync, mkdtempSync, readFileSync, rmSync, statSync, writeFileSync } from "fs";
import { tmpdir } from "os";
import { join } from "path";
import { releaseArchiveName } from "../src/utils/standalone-update";

const INSTALL_SCRIPT = join(import.meta.dir, "..", "install.sh");
const REPO = "Routstr/routstrd";
const VERSION = "9.9.9";
const PLATFORMS = ["linux", "darwin"] as const;
const ARCHS = ["x64", "arm64"] as const;

const tempDirs: string[] = [];
const servers: ReturnType<typeof Bun.serve>[] = [];

function tempDir(prefix: string): string {
  const dir = mkdtempSync(join(tmpdir(), prefix));
  tempDirs.push(dir);
  return dir;
}

afterEach(() => {
  for (const server of servers.splice(0)) server.stop(true);
  for (const dir of tempDirs.splice(0)) rmSync(dir, { recursive: true, force: true });
});

function sha256Hex(bytes: Uint8Array): string {
  return createHash("sha256").update(bytes).digest("hex");
}

/** Builds a tar.gz that mimics a real release archive: a single `routstrd` entry. */
function buildArchive(dir: string, version: string): Uint8Array {
  const stage = join(dir, "stage");
  const archivePath = join(dir, "archive.tar.gz");
  writeFileSync(
    join(dir, "routstrd"),
    `#!/bin/sh\necho ${version}\n`,
  );
  Bun.spawnSync([
    "sh",
    "-c",
    `mkdir -p ${JSON.stringify(stage)} && cp ${JSON.stringify(join(dir, "routstrd"))} ${JSON.stringify(join(stage, "routstrd"))} && tar -czf ${JSON.stringify(archivePath)} -C ${JSON.stringify(stage)} routstrd`,
  ]);
  return new Uint8Array(readFileSync(archivePath));
}

type FakeReleaseOptions = {
  version?: string;
  asset?: string;
  archive?: Uint8Array;
  checksums?: string;
  omitAsset?: boolean;
};

/**
 * Serves the subset of the GitHub API and release download URLs that install.sh
 * consumes, so the installer can be exercised end to end without network access.
 */
function serveFakeRelease(options: FakeReleaseOptions = {}): string {
  const version = options.version ?? VERSION;
  const asset = options.asset ?? releaseArchiveName(version, "linux", "x64");
  const server = Bun.serve({
    port: 0,
    fetch(request) {
      const { pathname } = new URL(request.url);
      if (pathname === `/repos/${REPO}/releases/latest`) {
        return Response.json({
          tag_name: `v${version}`,
          assets: [{ name: asset }, { name: "SHA256SUMS" }],
        });
      }
      if (pathname === `/dl/v${version}/SHA256SUMS`) {
        return new Response(options.checksums ?? "");
      }
      if (pathname === `/dl/v${version}/${asset}`) {
        if (options.omitAsset) return new Response("not found", { status: 404 });
        return new Response(options.archive ?? new Uint8Array());
      }
      return new Response("not found", { status: 404 });
    },
  });
  servers.push(server);
  const origin = `http://127.0.0.1:${server.port}`;
  return origin;
}

function runInstaller(args: string[], env: Record<string, string> = {}) {
  return Bun.spawnSync(["sh", INSTALL_SCRIPT, ...args], {
    env: { ...process.env, ...env },
    stdout: "pipe",
    stderr: "pipe",
  });
}

describe("install.sh", () => {
  test("is valid POSIX shell", () => {
    const result = Bun.spawnSync(["sh", "-n", INSTALL_SCRIPT], { stderr: "pipe" });
    expect(result.stderr.toString()).toBe("");
    expect(result.exitCode).toBe(0);
  });

  test("resolves the same asset names as releaseArchiveName", () => {
    for (const platform of PLATFORMS) {
      for (const arch of ARCHS) {
        const result = runInstaller([
          "--print-asset",
          "--version",
          VERSION,
          "--platform",
          platform,
          "--arch",
          arch,
        ]);
        expect(result.exitCode).toBe(0);
        expect(result.stdout.toString().trim()).toBe(
          releaseArchiveName(VERSION, platform, arch),
        );
      }
    }
  });

  test("accepts a v-prefixed version", () => {
    const result = runInstaller([
      "--print-asset",
      "--version",
      `v${VERSION}`,
      "--platform",
      "linux",
      "--arch",
      "x64",
    ]);
    expect(result.stdout.toString().trim()).toBe(releaseArchiveName(VERSION, "linux", "x64"));
  });

  test("rejects unsupported platforms and architectures", () => {
    for (const args of [
      ["--print-asset", "--version", VERSION, "--platform", "windows", "--arch", "x64"],
      ["--print-asset", "--version", VERSION, "--platform", "linux", "--arch", "riscv64"],
      ["--print-asset", "--version", "../../etc/passwd", "--platform", "linux", "--arch", "x64"],
    ]) {
      const result = runInstaller(args);
      expect(result.exitCode).not.toBe(0);
    }
  });

  test("installs the executable from a release archive", () => {
    const dir = tempDir("routstrd-install-e2e-");
    const installDir = join(dir, "bin");
    const asset = releaseArchiveName(VERSION, process.platform, process.arch);
    const archive = buildArchive(dir, VERSION);
    const origin = serveFakeRelease({
      asset,
      archive,
      checksums: `${sha256Hex(archive)}  ${asset}\n`,
    });

    const result = runInstaller([
      "--dir",
      installDir,
      "--api-base-url",
      origin,
      "--download-base-url",
      `${origin}/dl`,
    ]);

    expect(result.stderr.toString()).toContain(`Installed routstrd v${VERSION}`);
    expect(result.exitCode).toBe(0);

    const target = join(installDir, "routstrd");
    expect(statSync(target).mode & 0o111).not.toBe(0);
    const version = Bun.spawnSync([target, "--version"]);
    expect(version.stdout.toString().trim()).toBe(VERSION);
  });

  test("installs a specific version without querying the API", () => {
    const dir = tempDir("routstrd-install-pinned-");
    const installDir = join(dir, "bin");
    const asset = releaseArchiveName(VERSION, process.platform, process.arch);
    const archive = buildArchive(dir, VERSION);
    const origin = serveFakeRelease({
      asset,
      archive,
      checksums: `${sha256Hex(archive)}  ${asset}\n`,
    });

    const result = runInstaller([
      "--version",
      VERSION,
      "--dir",
      installDir,
      "--api-base-url",
      `${origin}/broken`,
      "--download-base-url",
      `${origin}/dl`,
    ]);

    expect(result.exitCode).toBe(0);
    expect(statSync(join(installDir, "routstrd")).isFile()).toBe(true);
  });

  test("refuses to install when the checksum does not match", () => {
    const dir = tempDir("routstrd-install-tampered-");
    const installDir = join(dir, "bin");
    const asset = releaseArchiveName(VERSION, process.platform, process.arch);
    const archive = buildArchive(dir, VERSION);
    const origin = serveFakeRelease({
      asset,
      archive,
      checksums: `${sha256Hex(new TextEncoder().encode("tampered"))}  ${asset}\n`,
    });

    const result = runInstaller([
      "--dir",
      installDir,
      "--api-base-url",
      origin,
      "--download-base-url",
      `${origin}/dl`,
    ]);

    expect(result.exitCode).not.toBe(0);
    expect(result.stderr.toString()).toContain("checksum mismatch");
    expect(() => statSync(join(installDir, "routstrd"))).toThrow();
  });

  test("reports a missing asset for the current platform", () => {
    const dir = tempDir("routstrd-install-missing-");
    const origin = serveFakeRelease({
      asset: releaseArchiveName(VERSION, process.platform, process.arch),
      omitAsset: true,
    });

    const result = runInstaller([
      "--dir",
      join(dir, "bin"),
      "--api-base-url",
      origin,
      "--download-base-url",
      `${origin}/dl`,
    ]);

    expect(result.exitCode).not.toBe(0);
    expect(result.stderr.toString()).toContain("may not include a build");
  });

  test("reports a release without SHA256SUMS", () => {
    const dir = tempDir("routstrd-install-nosums-");
    const asset = releaseArchiveName(VERSION, process.platform, process.arch);
    const archive = buildArchive(dir, VERSION);
    const origin = serveFakeRelease({ asset, archive, checksums: "" });

    const result = runInstaller([
      "--dir",
      join(dir, "bin"),
      "--api-base-url",
      origin,
      "--download-base-url",
      `${origin}/dl`,
    ]);

    expect(result.exitCode).not.toBe(0);
    expect(result.stderr.toString()).toContain("does not contain");
  });

  test("honours ROUTSTRD_INSTALL_DIR when --dir is absent", () => {
    const dir = tempDir("routstrd-install-env-");
    const installDir = join(dir, "bin");
    const asset = releaseArchiveName(VERSION, process.platform, process.arch);
    const archive = buildArchive(dir, VERSION);
    const origin = serveFakeRelease({
      asset,
      archive,
      checksums: `${sha256Hex(archive)}  ${asset}\n`,
    });

    const result = runInstaller(
      [
        "--api-base-url",
        origin,
        "--download-base-url",
        `${origin}/dl`,
      ],
      { ROUTSTRD_INSTALL_DIR: installDir },
    );

    expect(result.exitCode).toBe(0);
    const installed = join(installDir, "routstrd");
    chmodSync(installed, 0o755);
    expect(statSync(installed).isFile()).toBe(true);
  });
});