pantry/scripts/check-core-sizes.ts

#!/usr/bin/env -S tea -E

/*---
args:
  - deno
  - run
  - --allow-read
  - --allow-env
  - --allow-write
  - --import-map={{ srcroot }}/import-map.json
---*/

import { usePantry } from "hooks"
import * as ARGV from "./utils/args.ts"

const pantry = usePantry()

const pkgs = await ARGV.toArray(ARGV.pkgs())

let coreSize = 2

for (const pkg of pkgs) {
  const yml = await pantry.getYAML(pkg).parse()
  if (yml?.build?.cores) {
    coreSize = Math.max(yml.build.cores, coreSize)
  }
}

const output = `GHA_LINUX_BUILD_SIZE=${imageName(coreSize)}\n`

Deno.stdout.write(new TextEncoder().encode(output))

if (Deno.env.get("GITHUB_ENV")) {
  const envFile = Deno.env.get("GITHUB_ENV")!
  await Deno.writeTextFile(envFile, output, { append: true})
}

function imageName(size: number) {
  switch (size) {
    case 0:
    case 1:
    case 2:
      return "ubuntu-latest"
    case 4:
    case 8:
    case 16:
      return `ubuntu-latest-${size}-cores`
    default:
      throw new Error("Invalid core size")
  }
}