embark-area-51/lib/compiler.js

/*jshint esversion: 6, loopfunc: true */
var shelljs = require('shelljs');
var shelljs_global = require('shelljs/global');
var fs = require('fs');
var solc = require('solc');
var merge = require('merge');

var Compiler = function(options) {
  this.plugins = options.plugins;
};

Compiler.prototype.compile_contracts = function(contractFiles) {

  var available_compilers = {
    //".se": this.compile_serpent
    ".sol": this.compile_solidity
  };

  if (this.plugins) {
    var compilerPlugins = this.plugins.getPluginsFor('compilers');
    if (compilerPlugins.length > 0) {
      compilerPlugins.forEach(function(plugin) {
        plugin.compilers.forEach(function(compilerObject) {
          available_compilers[compilerObject.extension] = compilerObject.cb;
        });
      });
    }
  }

  var compiledObject = {};

  // TODO: warn about files it doesn't know how to compile
  for (var extension in available_compilers) {
    var compiler = available_compilers[extension];
    var matchingFiles = contractFiles.filter(function(file) {
      return (file.filename.match(/\.[0-9a-z]+$/)[0] === extension);
    });

    Object.assign(compiledObject, compiler.call(compiler, matchingFiles || []));
  }

  return compiledObject;
};

Compiler.prototype.compile_solidity = function(contractFiles) {
  var input = {};

  // TODO: need to match only the .sol files
  for (var i = 0; i < contractFiles.length; i++){
    // TODO: this depends on the config
    var filename = contractFiles[i].filename.replace('app/contracts/','');
    input[filename] = contractFiles[i].content.toString();
  }

  var output = solc.compile({sources: input}, 1);

  if (output.errors) {
    throw new Error ("Solidity errors: " + output.errors);
  }

  var json = output.contracts;

  compiled_object = {};

  for (var className in json) {
    var contract = json[className];

    compiled_object[className] = {};
    compiled_object[className].code            = contract.bytecode;
    compiled_object[className].runtimeBytecode = contract.runtimeBytecode;
    compiled_object[className].gasEstimates    = contract.gasEstimates;
    compiled_object[className].functionHashes  = contract.functionHashes;
    compiled_object[className].abiDefinition   = JSON.parse(contract.interface);
  }

  return compiled_object;
};

Compiler.prototype.compile_serpent = function(contractFiles) {
  var cmd, result, output, json, compiled_object;

  //TODO: figure out how to compile multiple files and get the correct json
  var contractFile = contractFiles[0];

  cmd = "serpent compile " + contractFile;

  result = exec(cmd, {silent: true});
  code = result.output;

  if (result.code === 1) {
    throw new Error(result.output);
  }

  cmd = "serpent mk_full_signature " + contractFile;
  result = exec(cmd, {silent: true});

  if (result.code === 1) {
    throw new Error(result.output);
  }

  json = JSON.parse(result.output.trim());
  className = contractFile.split('.')[0].split("/").pop();

  for (var i=0; i < json.length; i++) {
    var elem = json[i];

    if (elem.outputs.length > 0) {
      elem.constant = true;
    }
  }

  compiled_object = {};
  compiled_object[className] = {};
  compiled_object[className].code = code.trim();
  compiled_object[className].info = {};
  compiled_object[className].abiDefinition = json;

  return compiled_object;
};

Compiler.prototype.compile = function(contractFiles) {
  var solidity = [], serpent = [];

  for (var i = 0; i < contractFiles.length; i++) {
    var contractParts = contractFiles[i].split('.'),
      extension = contractParts[contractParts.length-1];

    if (extension === 'sol') {
      solidity.push(contractFiles[i]);
    }
    else if (extension === 'se') {
      serpent.push(contractFiles[i]);
    }
    else {
      throw new Error("extension not known, got " + extension);
    }
  }

  var contracts = [];
  if (solidity.length > 0) {
    contracts.concat(this.compile_solidity(solidity));
  }
  if (serpent.length > 0) {
    contracts.concat(this.compile_serpent(serpent));
  }
  return contracts;
};

module.exports = Compiler;