/*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;