atom-nuclide/pkg/nuclide-rpc/bin/regenerate-baselines.js

A unified developer experience for web and mobile development, built as a suite of features on top of Atom to provide hackability and the support of an active community.

#!/usr/bin/env node
'use strict';
/* @noflow */

/*
 * Copyright (c) 2015-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the license found in the LICENSE file in
 * the root directory of this source tree.
 */

/* NON-TRANSPILED FILE */
/* eslint-disable babel/func-params-comma-dangle, prefer-object-spread/prefer-object-spread */

// Regenerates the .proxy baseline files in the spec/fixtures directory.

require('../../nuclide-node-transpiler');

const parseServiceDefinition = require('../lib/service-parser').parseServiceDefinition;
const generateProxy = require('../lib/proxy-generator').generateProxy;
const stripLocationsFileName = require('../lib/location').stripLocationsFileName;

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

const dir = path.join(__dirname, '../spec/fixtures');
for (const file of fs.readdirSync(dir)) {
  if (file.endsWith('.def')) {
    const serviceName = path.basename(file, '.def');
    const preserveFunctionNames = false;
    const definitionPath = path.join(dir, file);

    const definitionSource = fs.readFileSync(definitionPath, 'utf8');
    const definitions = parseServiceDefinition(
      definitionPath,
      definitionSource
    );

    stripLocationsFileName(definitions);

    const json = mapDefinitions(definitions);
    fs.writeFileSync(
      definitionPath.replace('.def', '.def.json'),
      JSON.stringify(json, null, 4),
      'utf8'
    );

    const code = generateProxy(serviceName, preserveFunctionNames, definitions);
    fs.writeFileSync(definitionPath.replace('.def', '.proxy'), code, 'utf8');
  }
}

function mapDefinitions(map) {
  const obj = {};
  for (const it of map.values()) {
    let value;
    switch (it.kind) {
      case 'interface':
        value = {
          constructorArgs: it.constructorArgs,
          instanceMethods: mapToJSON(it.instanceMethods),
          staticMethods: mapToJSON(it.staticMethods),
        };
        break;
      default:
        value = it;
        break;
    }
    obj[it.name] = value;
  }
  return obj;
}

function mapToJSON(map) {
  const result = {};
  for (const it of map.entries()) {
    result[it[0]] = it[1];
  }
  return result;
}