studio-desktop/scripts/downloadUV.js

import axios from 'axios';
import extractZip from 'extract-zip';
import fs from 'fs-extra';
import os from 'node:os';
import path from 'node:path';
import * as tar from 'tar';

import packageJson from './getPackage.js';

const uvVer = packageJson.config.uvVersion;

/** @typedef {{ [key]: { zipFile: string, uvOutputFolderName: string, zip: boolean } }} UvDownloadOptions */
const options = {
  win32: {
    zipFile: 'uv-x86_64-pc-windows-msvc.zip',
    uvOutputFolderName: 'win',
    zip: true,
  },
  darwin: {
    zipFile: 'uv-aarch64-apple-darwin.tar.gz',
    uvOutputFolderName: 'macos',
    zip: false,
  },
  linux: {
    zipFile: 'uv-x86_64-unknown-linux-gnu.tar.gz',
    uvOutputFolderName: 'linux',
    zip: false,
  },
};

async function downloadUV() {
  const allFlag = process.argv[2];
  const baseDownloadURL = `https://github.com/astral-sh/uv/releases/download/${uvVer}/`;
  if (allFlag) {
    if (allFlag === 'all') {
      await downloadAndExtract(baseDownloadURL, options.win32);
      await downloadAndExtract(baseDownloadURL, options.darwin);
      await downloadAndExtract(baseDownloadURL, options.linux);
      return;
    }
    if (allFlag === 'none') {
      return;
    }
  }

  const uvDownloaded = fs.existsSync(path.join('./assets', 'uv'));
  if (!uvDownloaded) {
    // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
    await downloadAndExtract(baseDownloadURL, options[os.platform()]);
    return;
  }
  console.log('< UV Folder Exists, Skipping >');
}

/** @param {UvDownloadOptions[any]} options */
async function downloadAndExtract(baseURL, options) {
  const { zipFile, uvOutputFolderName, zip } = options;
  const zipFilePath = path.join('./assets', zipFile);
  const outputUVFolder = path.join('./assets', 'uv', uvOutputFolderName);
  await fs.mkdir(outputUVFolder, {
    recursive: true,
  });
  const downloadedFile = await axios({
    method: 'GET',
    url: baseURL + zipFile,
    responseType: 'arraybuffer',
  });
  // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
  fs.writeFileSync(zipFilePath, downloadedFile.data);
  if (zip) {
    await extractZip(zipFilePath, { dir: path.resolve(outputUVFolder) });
  } else {
    tar.extract({
      sync: true,
      file: zipFilePath,
      C: outputUVFolder,
      'strip-components': 1,
    });
  }
  await fs.unlink(zipFilePath);
  console.log(`FINISHED DOWNLOAD AND EXTRACT UV ${uvOutputFolderName}`);
}

//** Download and Extract UV. Default uses OS.Platfrom. Add 'all' will download all. Add 'none' will skip */
await downloadUV();