main.ts

import * as core from '@actions/core'
import {mkdirp} from './src/downloader'
import {restoreCache, saveCache} from '@actions/cache'
import process from 'process'
import {spawnSync} from 'child_process'
import {
  getArtifactMetadata,
  getViaGit,
  gitForWindowsUsrBinPath
} from './src/git'
import {getViaCIArtifacts} from './src/ci_artifacts'
import * as fs from 'fs'

const flavor = core.getInput('flavor')
const architecture = core.getInput('architecture')

/**
 * Some Azure VM types have a temporary disk which is local to the VM and therefore provides
 * _much_ faster disk IO than the OS Disk (or any other attached disk).
 *
 * Hosted GitHub Actions runners also leverage this disk and do their work in D:/a/_work, so let's
 * use it too if we can. It leads to a ~25% speed increase when doing heavy IO operations.
 *
 * https://learn.microsoft.com/en-us/azure/virtual-machines/managed-disks-overview#temporary-disk
 */
function getDriveLetterPrefix(): string {
  if (fs.existsSync('D:/')) {
    core.info('Found a fast, temporary disk on this VM (D:/). Will use that.')
    return 'D:/'
  }

  return 'C:/'
}

async function run(): Promise<void> {
  try {
    if (process.platform !== 'win32') {
      core.warning(
        `Skipping this Action because it only works on Windows, not on ${process.platform}`
      )
      return
    }

    const githubToken = core.getInput('github-token')
    const verbose = core.getInput('verbose')
    const msysMode = core.getInput('msys') === 'true'

    const {artifactName, download, id} =
      flavor === 'minimal'
        ? await getViaCIArtifacts(architecture, githubToken)
        : await getViaGit(flavor, architecture, githubToken)
    const outputDirectory =
      core.getInput('path') || `${getDriveLetterPrefix()}${artifactName}`
    core.setOutput('result', outputDirectory)

    let useCache: boolean
    switch (core.getInput('cache')) {
      case 'true':
        useCache = true
        break
      case 'auto':
        useCache = !['full', 'minimal'].includes(flavor)
        break
      default:
        useCache = false
    }

    let needToDownload = true
    try {
      if (useCache && (await restoreCache([outputDirectory], id))) {
        core.info(`Cached ${id} was successfully restored`)
        needToDownload = false
      }
    } catch (e) {
      core.warning(`Cannot use @actions/cache (${e})`)
      useCache = false
    }

    if (needToDownload) {
      core.info(`Downloading ${artifactName}`)
      await download(
        outputDirectory,
        verbose.match(/^\d+$/) ? parseInt(verbose) : verbose === 'true'
      )

      try {
        if (useCache && !(await saveCache([outputDirectory], id))) {
          core.warning(`Failed to cache ${id}`)
        }
      } catch (e) {
        core.warning(
          `Failed to cache ${id}: ${e instanceof Error ? e.message : e}`
        )
      }
    }

    const mingw = {
      i686: 'MINGW32',
      x86_64: 'MINGW64',
      aarch64: 'CLANGARM64'
    }[architecture]

    if (mingw === undefined) {
      core.setFailed(`Invalid architecture ${architecture} specified`)
      return
    }

    const msystem = msysMode ? 'MSYS' : mingw

    const binPaths = [
      // Set up PATH so that Git for Windows' SDK's `bash.exe`, `prove` and `gcc` are found
      '/usr/bin/core_perl',
      '/usr/bin',
      `/${mingw.toLocaleLowerCase()}/bin`
    ]

    for (const binPath of msysMode ? binPaths.reverse() : binPaths) {
      core.addPath(`${outputDirectory}${binPath}`)
    }

    core.exportVariable('MSYSTEM', msystem)
    if (
      !('LANG' in process.env) &&
      !('LC_ALL' in process.env) &&
      !('LC_CTYPE' in process.env)
    ) {
      core.exportVariable('LC_CTYPE', 'C.UTF-8')
    }

    // ensure that /dev/fd/*, /dev/mqueue and friends exist
    for (const path of ['/dev/mqueue', '/dev/shm']) {
      mkdirp(`${outputDirectory}${path}`)
    }

    const ln = (linkPath: string, target: string): void => {
      const child = spawnSync(
        flavor === 'minimal' ? 'ln.exe' : 'usr\\bin\\ln.exe',
        ['-s', target, linkPath],
        {
          cwd: outputDirectory,
          env: {
            MSYS: 'winsymlinks:sys'
          }
        }
      )
      if (child.error) throw child.error
    }
    for (const [linkPath, target] of Object.entries({
      fd: 'fd',
      stdin: 'fd/0',
      stdout: 'fd/1',
      stderr: 'fd/2'
    })) {
      ln(`/dev/${linkPath}`, `/proc/self/${target}`)
    }
  } catch (error) {
    core.setFailed(error instanceof Error ? error.message : `${error}`)
  }
}

function cleanup(): void {
  if (core.getInput('cleanup') !== 'true') {
    core.info(
      `Won't clean up SDK files as the 'cleanup' input was not provided or doesn't equal 'true'.`
    )
    return
  }

  const outputDirectory =
    core.getInput('path') ||
    `${getDriveLetterPrefix()}${
      getArtifactMetadata(flavor, architecture).artifactName
    }`

  /**
   * Shelling out to `rm -rf` is more than twice as fast as Node's `fs.rmSync` method.
   * Let's use it if it's available, and otherwise fall back to `fs.rmSync`.
   */
  const cleanupMethod = fs.existsSync(`${gitForWindowsUsrBinPath}/bash.exe`)
    ? 'rm -rf'
    : 'node'

  core.info(
    `Cleaning up ${outputDirectory} using the "${cleanupMethod}" method...`
  )

  if (cleanupMethod === 'rm -rf') {
    const child = spawnSync(
      `${gitForWindowsUsrBinPath}/bash.exe`,
      ['-c', `rm -rf "${outputDirectory}"`],
      {
        encoding: 'utf-8',
        env: {PATH: '/usr/bin'}
      }
    )

    if (child.error) throw child.error
    if (child.stderr) core.error(child.stderr)
  } else {
    fs.rmSync(outputDirectory, {recursive: true, force: true})
  }

  core.info(`Finished cleaning up ${outputDirectory}.`)
}

/**
 * Indicates whether the POST action is running
 */
export const isPost = !!core.getState('isPost')

if (!isPost) {
  run()
  /*
   * Publish a variable so that when the POST action runs, it can determine it should run the cleanup logic.
   * This is necessary since we don't have a separate entry point.
   * Inspired by https://github.com/actions/checkout/blob/v3.1.0/src/state-helper.ts#L56-L60
   */
  core.saveState('isPost', 'true')
} else {
  // If the POST action is running, we cleanup our artifacts
  cleanup()
}