embark/lib/modules/profiler/index.js

const asciiTable = require('ascii-table');

class Profiler {
  constructor(embark) {
    this.embark = embark;
    this.logger = embark.logger;
    this.events = embark.events;
    this.plugins = embark.plugins;

    this.registerConsoleCommand();
  }

  profile(contractName, contract) {
    const self = this;
    let table = new asciiTable(contractName);
    table.setHeading('Function', 'Payable', 'Mutability', 'Inputs', 'Outputs');
    contract.abiDefinition.forEach((abiMethod) => {
      switch(abiMethod.type) {
        case "constructor": 
          table.addRow("constructor", abiMethod.payable, abiMethod.stateMutability, this.formatParams(abiMethod.inputs), this.formatParams(abiMethod.outputs));
          break;
        default:
          table.addRow(abiMethod.name, abiMethod.payable, abiMethod.stateMutability, this.formatParams(abiMethod.inputs), this.formatParams(abiMethod.outputs));
      }
    });
    self.logger.info(table.toString());
  }

  formatParams(params) {
    if (undefined !== params && params.length) {
      let paramString = "(";
      let mappedParams = params.map(param => param.type);
      paramString += mappedParams.join(',')
      paramString += ")";
      return paramString;
    }
    return "()";
  }

  registerConsoleCommand() {
    const self = this;
    self.embark.registerConsoleCommand((cmd, _options) => {
      let cmdName = cmd.split(' ')[0];
      let contractName = cmd.split(' ')[1];
      if (cmdName === 'profile') {
          self.events.request('contracts:contract', contractName, (contract) => {
               self.logger.info("--  profile for " + contractName);
               this.profile(contractName, contract);
          });
          return "profiled...";
      }
    });
  }
}

module.exports = Profiler;