tauri-workspace/tooling/cli.js/src/helpers/download-binary.ts

import { promisify } from 'util'
import stream from 'stream'
import fs from 'fs'
import { CargoManifest } from '../types/cargo'
import path from 'path'
import { bootstrap } from 'global-agent'
import { fileURLToPath } from 'url'
import { createRequire } from 'module'

// Webpack reads the file at build-time, so this becomes a static var
// @ts-expect-error
import manifest from '../../../cli.rs/Cargo.toml'
const tauriCliManifest = manifest as CargoManifest
const currentDirName = path.dirname(fileURLToPath(import.meta.url))

const require = createRequire(import.meta.url)
/* eslint-disable @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-var-requires */
const got = require('got')
/* eslint-enable @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-var-requires */

// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
const pipeline = promisify(stream.pipeline)

const downloads: { [url: string]: boolean } = {}

async function downloadBinaryRelease(
  tag: string,
  asset: string,
  outPath: string
): Promise<void> {
  const url = `https://github.com/tauri-apps/binary-releases/releases/download/${tag}/${asset}`

  const removeDownloadedCliIfNeeded = (): void => {
    try {
      if (!(url in downloads)) {
        // eslint-disable-next-line security/detect-non-literal-fs-filename
        fs.unlinkSync(outPath)
      }
    } finally {
      process.exit()
    }
  }

  // on exit, we remove the `tauri-cli` file if the download didn't complete
  process.on('exit', removeDownloadedCliIfNeeded)
  process.on('SIGINT', removeDownloadedCliIfNeeded)
  process.on('SIGTERM', removeDownloadedCliIfNeeded)
  process.on('SIGHUP', removeDownloadedCliIfNeeded)
  process.on('SIGBREAK', removeDownloadedCliIfNeeded)

  bootstrap({
    environmentVariableNamespace: ''
  })

  // TODO: Check hash of download
  // eslint-disable-next-line @typescript-eslint/no-unsafe-call, @typescript-eslint/no-unsafe-member-access, security/detect-non-literal-fs-filename
  await pipeline(got.stream(url), fs.createWriteStream(outPath)).catch(
    (e: unknown) => {
      try {
        // eslint-disable-next-line security/detect-non-literal-fs-filename
        fs.unlinkSync(outPath)
      } catch {}
      throw e
    }
  )
  // eslint-disable-next-line security/detect-object-injection
  downloads[url] = true
  // eslint-disable-next-line security/detect-non-literal-fs-filename
  fs.chmodSync(outPath, 0o700)
  console.log('Download Complete')
}

async function downloadCli(): Promise<void> {
  const version = tauriCliManifest.package.version
  let platform: string = process.platform
  if (platform === 'win32') {
    platform = 'windows'
  } else if (platform === 'linux') {
    platform = 'linux'
  } else if (platform === 'darwin') {
    platform = 'macos'
  } else {
    throw Error('Unsupported platform')
  }
  const extension = platform === 'windows' ? '.exe' : ''
  const outPath = path.join(currentDirName, `../../bin/tauri-cli${extension}`)
  console.log('Downloading Rust CLI...')
  await downloadBinaryRelease(
    `tauri-cli-v${version}`,
    `tauri-cli_${platform}${extension}`,
    outPath
  )
}

async function downloadRustup(): Promise<void> {
  const assetName =
    process.platform === 'win32' ? 'rustup-init.exe' : 'rustup-init.sh'
  console.log('Downloading Rustup...')
  return await downloadBinaryRelease(
    'rustup',
    assetName,
    path.join(currentDirName, `../../bin/${assetName}`)
  )
}

export { downloadCli, downloadRustup }