napi-rs/cli/src/artifacts.ts

import { join, parse } from 'path'

import { Command, Option } from 'clipanion'
import * as chalk from 'colorette'
import { fdir } from 'fdir'

import { getNapiConfig } from './consts'
import { debugFactory } from './debug'
import { UniArchsByPlatform } from './parse-triple'
import { readFileAsync, writeFileAsync } from './utils'

const debug = debugFactory('artifacts')

export class ArtifactsCommand extends Command {
  static usage = Command.Usage({
    description: 'Copy artifacts from Github Actions into specified dir',
  })

  static paths = [['artifacts']]

  sourceDir = Option.String('-d,--dir', 'artifacts')

  distDir = Option.String('--dist', 'npm')

  configFileName?: string = Option.String('-c,--config')

  async execute() {
    const { platforms, binaryName, packageJsonPath } = getNapiConfig(
      this.configFileName,
    )

    const packageJsonDir = parse(packageJsonPath).dir

    const sourceApi = new fdir()
      .withFullPaths()
      .crawl(join(process.cwd(), this.sourceDir))

    const distDirs = platforms.map((platform) =>
      join(process.cwd(), this.distDir, platform.platformArchABI),
    )

    const universalSourceBins = new Set(
      platforms
        .filter((platform) => platform.arch === 'universal')
        .flatMap((p) =>
          UniArchsByPlatform[p.platform].map((a) => `${p.platform}-${a}`),
        ),
    )

    await sourceApi.withPromise().then((output) =>
      Promise.all(
        (output as string[]).map(async (filePath) => {
          debug(`Read [${chalk.yellowBright(filePath)}]`)
          const sourceContent = await readFileAsync(filePath)
          const parsedName = parse(filePath)
          const [_binaryName, platformArchABI] = parsedName.name.split('.')
          if (_binaryName !== binaryName) {
            debug(
              `[${chalk.yellowBright(
                _binaryName,
              )}] is not matched with [${chalk.greenBright(binaryName)}], skip`,
            )
            return
          }
          const dir = distDirs.find((dir) => dir.includes(platformArchABI))
          if (!dir && universalSourceBins.has(platformArchABI)) {
            debug(
              `[${chalk.yellowBright(
                platformArchABI,
              )}] has no dist dir but it is source bin for universal arch, skip`,
            )
            return
          }
          if (!dir) {
            throw new TypeError(`No dist dir found for ${filePath}`)
          }
          const distFilePath = join(dir, parsedName.base)
          debug(`Write file content to [${chalk.yellowBright(distFilePath)}]`)
          await writeFileAsync(distFilePath, sourceContent)
          const distFilePathLocal = join(packageJsonDir, parsedName.base)
          debug(
            `Write file content to [${chalk.yellowBright(distFilePathLocal)}]`,
          )
          await writeFileAsync(distFilePathLocal, sourceContent)
        }),
      ),
    )
  }
}