scripts.js

// Copyright 2020 Mozilla Corporation
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

const childProcess = require('child_process');
const path = require('path');
const fs = require('fs');

const exec = (cmd, env) => {
  env = {...process.env, ...env};
  console.log(`> ${cmd}`);
  return childProcess.execSync(cmd, {env, stdio: 'inherit'});
};

const prepare = () => {
  // Copy secrets.sample.json to secrets.json if needed
  const secretsPath = path.join(__dirname, 'secrets.json');
  const secretsSamplePath = path.join(__dirname, 'secrets.sample.json');

  if (!fs.existsSync(secretsPath)) {
    fs.copyFileSync(secretsSamplePath, secretsPath);
  }

  // Install Firefox for Puppeteer
  try {
    process.chdir('node_modules/puppeteer');
  } catch (e) {
    return;
  }
  exec('node install.js', {PUPPETEER_PRODUCT: 'firefox'});
};

if (require.main === module) {
  const {argv} = require('yargs').command(
      '$0 <command>',
      'Run an action',
      (yargs) => {
        yargs
            .positional('command', {
              describe: 'What command to run',
              type: 'string',
              choices: ['unittest', 'prepare']
            });
      }
  );

  switch (argv.command) {
    case 'prepare':
      prepare();
      break;
    case 'unittest':
      exec('nyc mocha --reporter dot --recursive unittest', {NODE_ENV: 'test'});
      break;
    default:
      console.error(`Unknown command ${argv.command}!`);
  }
}