onnxruntime/js/node/script/prepack.ts

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

import * as fs from 'fs-extra';
import klawSync from 'klaw-sync';
import * as path from 'path';
import {pack} from 'tar-stream';
import * as zlib from 'zlib';

function updatePackageJson() {
  const commonPackageJsonPath = path.join(__dirname, '..', '..', 'common', 'package.json');
  const selfPackageJsonPath = path.join(__dirname, '..', 'package.json');
  console.log(`=== start to update package.json: ${selfPackageJsonPath}`);
  const packageCommon = fs.readJSONSync(commonPackageJsonPath);
  const packageSelf = fs.readJSONSync(selfPackageJsonPath);
  const version = packageCommon.version;
  packageSelf.dependencies['onnxruntime-common'] = `~${version}`;
  fs.writeJSONSync(selfPackageJsonPath, packageSelf, {spaces: 2});
  console.log('=== finished updating package.json.');
}

function packPrebuild() {
  // build path
  const ROOT_FOLDER = path.join(__dirname, '..');
  const BIN_FOLDER = path.join(ROOT_FOLDER, 'bin');
  const PREBUILDS_FOLDER = path.join(ROOT_FOLDER, 'prebuilds');

  // start pack
  const tarName = `${process.env.npm_package_name}-v${process.env.npm_package_version}-napi-v3-${process.platform}-${
      process.arch}.tar.gz`;
  const tarPath = path.join(PREBUILDS_FOLDER, tarName);

  const tarStream = pack();
  fs.ensureDirSync(PREBUILDS_FOLDER);
  const ws = fs.createWriteStream(tarPath);

  tarStream.pipe(zlib.createGzip({level: 9})).pipe(ws);

  // enumerate all files under BIN folder
  const entries =
      klawSync(BIN_FOLDER, {nodir: true}).map(i => ({
                                                path: i.path,
                                                name: path.relative(ROOT_FOLDER, i.path).replace(/\\/g, '/'),
                                                size: i.stats.size,
                                                // eslint-disable-next-line no-bitwise
                                                mode: i.stats.mode | parseInt('444', 8) | parseInt('222', 8),
                                                gid: i.stats.gid,
                                                uid: i.stats.uid
                                              }));

  console.log(`=== start to pack prebuild: ${tarName}`);

  function packNextFile(): void {
    const nextEntry = entries.shift();
    if (nextEntry) {
      console.log(`  adding file: ${nextEntry.name}`);
      const stream = tarStream.entry(nextEntry);
      fs.createReadStream(nextEntry.path).pipe(stream).on('finish', packNextFile);
    } else {
      console.log('=== finished packing prebuild.');
      tarStream.finalize();
    }
  }

  packNextFile();
}

// update version of dependency "onnxruntime-common" before packing
updatePackageJson();
// pack prebuilds
packPrebuild();