Function main

link-package/build_deps.ts:58–112

* Build bazel dependencies for the packages specified by the repeated argument * tfjs_package. If the package is a Bazel package, also build the package * itself. * * @example 'yarn build-deps-for tfjs-react-native' builds all bazel * dependencies for @tensorflow/tfjs-react-native. * * @examp

()

 * @example 'yarn build-deps-for --all' builds all bazel packages.
 */
async function main() {
  const args = parser.parse_args();
  let packageNames: string[] = args.tfjs_package;

  let targets: string[];
  if (args.all) {
    targets = [...BAZEL_PACKAGES].map(dirToTarget);
  } else {
    const bazelDeps = findTransitiveBazelDeps(packageNames);
    targets = [...bazelDeps].map(dirToTarget);
  }

  if (targets.length > 0) {
    if (process.platform === 'win32') {
      const child = exec('yarn bazel build --color=yes '
        + `${args.bazel_options} ${targets.join(' ')}`);
      await new Promise((resolve, reject) => {
        child.stdout.pipe(process.stdout);
        child.stderr.pipe(process.stderr);
        child.on('exit', code => {
          if (code !== 0) {
            reject(code);
          }
          resolve(code);
        });
      });
    } else {
      // Use spawnSync intead of exec for prettier printing.
      const bazelArgs = ['bazel', 'build'];
      if (args.bazel_options) {
        bazelArgs.push(args.bazel_options);
      }
      bazelArgs.push(...targets);
      spawnSync('yarn', bazelArgs, {stdio:'inherit'});
    }
  }

  const tfjsDir = `${__dirname}/node_modules/@tensorflow`;
  rimraf.sync(tfjsDir);
  fs.mkdirSync(tfjsDir, {recursive: true});

  // Copy all built packages to node_modules. Note that this does not install
  // their dependencies, but that's okay since the node resolution algorithm
  // will find dependencies in the root node_modules folder of the repository.
  for (const pkg of BAZEL_PACKAGES) {
    const pkgPath = path.normalize(
      `${__dirname}/../dist/bin/${pkg}/${pkg}_pkg`);

    if (fs.existsSync(pkgPath)) {
      const newPath = path.join(tfjsDir, pkg);
      copyRecursive(pkgPath, newPath);
    }
  }
  chmodRecursive(tfjsDir);
}

/**
 * Get all dependencies and devDependencies of a tfjs package.