test-pack-package.mts

import path from 'path'
import fs from 'fs-extra'
import os from 'os'
import execa from 'execa'
import { randomBytes } from 'crypto'
import { fileURLToPath } from 'url'

const main = async () => {
  const __dirname = fileURLToPath(new URL('.', import.meta.url))
  const repoRoot = path.dirname(__dirname)
  const pkgsDir = path.join(repoRoot, 'packages')
  const currentPkgDirname = process.argv[2]

  const getPackedPkgPath = (pkgDirname: string) =>
    path.join(pkgsDir, pkgDirname, `packed-${pkgDirname}.tgz`)
  const getPackageJsonPath = (pkgDirname: string) =>
    path.join(pkgsDir, pkgDirname, `package.json`)

  const allPkgDirnames = (await fs.readdir(pkgsDir)).filter(
    (item) => !item.startsWith('.')
  )

  if (!allPkgDirnames.includes(currentPkgDirname)) {
    throw new Error(`Unknown package '${currentPkgDirname}'`)
  }

  const currentPkgDir = path.join(pkgsDir, currentPkgDirname)

  const tmpPkgPath = path.join(
    os.tmpdir(),
    `${currentPkgDirname}-${randomBytes(32).toString('hex')}`
  )
  console.log(`Packing '${currentPkgDirname}' in '${tmpPkgPath}'.`)

  const packageJsonPath = getPackageJsonPath(currentPkgDirname)
  const packageJson = await fs.readJson(packageJsonPath)
  const dependencies = packageJson.dependencies

  if (packageJson?.scripts?.prepublishOnly) {
    // There's a bug in `pnpm pack` where it will run
    // the prepublishOnly script and that will fail.
    // See https://github.com/pnpm/pnpm/issues/2941
    delete packageJson.scripts.prepublishOnly
  }

  // @next/swc is devDependency in next, but we want to include it anyway
  if (currentPkgDirname === 'next') {
    dependencies['@next/swc'] = '*'
  }

  // Modify dependencies to point to packed packages
  if (dependencies) {
    await Promise.all(
      allPkgDirnames.map(async (depPkgDirname) => {
        const { name: depPkgName } = await fs.readJson(
          getPackageJsonPath(depPkgDirname)
        )
        if (depPkgName in dependencies) {
          dependencies[depPkgName] = getPackedPkgPath(depPkgDirname)
        }
      })
    )
  }

  // Ensure that we bundle binaries with swc
  if (currentPkgDirname === 'next-swc') {
    packageJson.files = packageJson.files ?? []
    packageJson.files.push('native/*')

    console.log('using swc binaries:')
    await execa('ls', [
      path.join(path.dirname(packageJsonPath), 'native'),
    ]).stdout?.pipe(process.stdout)
  }

  // Allow overriding nateve swc version in next
  if (currentPkgDirname === 'next' && process.env.NEXT_SWC_VERSION) {
    dependencies['@next/swc-linux-x64-gnu'] = process.env.NEXT_SWC_VERSION
  }

  try {
    await fs.copy(currentPkgDir, tmpPkgPath, {
      filter: (item) =>
        !item.includes('node_modules') &&
        !item.includes('.DS_Store') &&
        // Exclude Rust compilation files
        (currentPkgDirname !== 'next' ||
          !/build[\\/]swc[\\/]target/.test(item)) &&
        (currentPkgDirname !== 'next-swc' || !/target/.test(item)),
    })
    await fs.writeJson(path.join(tmpPkgPath, 'package.json'), packageJson)
    // Copied from pnpm source: https://github.com/pnpm/pnpm/blob/5a5512f14c47f4778b8d2b6d957fb12c7ef40127/releasing/plugin-commands-publishing/src/pack.ts#L96
    const tmpTarball = path.join(
      tmpPkgPath,
      `${packageJson.name.replace('@', '').replace('/', '-')}-${
        packageJson.version
      }.tgz`
    )
    await execa('pnpm', ['pack'], {
      cwd: tmpPkgPath,
    })
    await fs.copyFile(tmpTarball, getPackedPkgPath(currentPkgDirname))
  } finally {
    await fs.remove(tmpPkgPath).catch()
  }
}

main()