Function getMainImplExperimentList

scripts/check_experiments.js:114–179

* Gets list of experiments registered in MainImpl.js.

(mainImplFile, experimentNames)

 * Gets list of experiments registered in MainImpl.js.
 */
function getMainImplExperimentList(mainImplFile, experimentNames) {
  const mainAST = espree.parse(mainImplFile, parseOptions);

  // Find MainImpl Class node
  let mainImplClassNode;
  for (const node of mainAST.body) {
    if (isClassNameDeclaration(node, 'MainImpl')) {
      mainImplClassNode = node;
      break;
    }
  }
  if (!mainImplClassNode) {
    return null;
  }

  // Find function in MainImpl Class
  const initializeExperimentNode = findFunctionInClass(
      mainImplClassNode,
      'initializeExperiments',
  );
  if (!initializeExperimentNode) {
    return null;
  }

  // Get list of experiments
  const experiments = [];
  for (const statement of initializeExperimentNode.value.body.body) {
    let expression = null;
    if (statement.expression && statement.expression.type === 'CallExpression') {
      expression = statement.expression;
    } else if (statement.type === 'VariableDeclaration') {
      expression = statement.declarations[0].init;
    }

    let experimentNameArg = '';
    if (isExperimentRegistrationCall(expression)) {
      // Experiment name is first argument of registration call
      experimentNameArg = expression.arguments[0];
    } else if (isHostExperimentRegistrationCall(expression)) {
      // Experiment name is the value of the `name` property of the first
      // argument of registration call
      experimentNameArg = expression.arguments[0].properties.find(property => property.key.name === 'name').value;
    }
    // Translate the Root.ExperimentNames.ExperimentName to a string
    if (experimentNameArg) {
      const experimentName = isExperimentNameReference(experimentNameArg);
      if (experimentName) {
        const translatedName = experimentNames.get(experimentName);
        if (!translatedName) {
          console.log(
              'Failed to resolve Root.ExperimentNames.${experimentName} to a string',
          );
          process.exit(1);
        }
        experiments.push(translatedName);
      } else {
        console.log(
            'Unexpected argument to Root.Runtime.experiments.register: ',