embark-area-51/lib/contracts/compiler.js

124 lines
4.0 KiB
JavaScript
Raw Normal View History

2017-01-29 06:28:01 +00:00
/*jshint esversion: 6, loopfunc: true */
2017-03-29 17:18:00 +00:00
let async = require('../utils/async_extend.js');
let SolcW = require('./solcW.js');
2017-02-17 12:14:44 +00:00
2017-03-30 11:12:39 +00:00
class Compiler {
constructor(options) {
this.plugins = options.plugins;
this.logger = options.logger;
this.solcVersion = options.solcVersion;
2017-03-30 11:12:39 +00:00
}
2017-01-29 02:31:09 +00:00
2017-03-30 11:12:39 +00:00
compile_contracts(contractFiles, cb) {
2017-01-29 02:31:09 +00:00
2017-03-30 11:12:39 +00:00
let available_compilers = {
//".se": this.compile_serpent
".sol": this.compile_solidity.bind(this)
};
2017-01-29 02:31:09 +00:00
2017-03-30 11:12:39 +00:00
if (this.plugins) {
let compilerPlugins = this.plugins.getPluginsFor('compilers');
if (compilerPlugins.length > 0) {
compilerPlugins.forEach(function (plugin) {
plugin.compilers.forEach(function (compilerObject) {
available_compilers[compilerObject.extension] = compilerObject.cb;
});
2017-01-29 06:28:01 +00:00
});
2017-03-30 11:12:39 +00:00
}
2017-01-29 06:28:01 +00:00
}
2017-01-29 02:31:09 +00:00
2017-03-30 11:12:39 +00:00
let compiledObject = {};
2017-01-29 02:31:09 +00:00
2017-03-30 11:12:39 +00:00
async.eachObject(available_compilers,
function (extension, compiler, callback) {
// TODO: warn about files it doesn't know how to compile
let matchingFiles = contractFiles.filter(function (file) {
2017-06-29 11:39:12 +00:00
let fileMatch = file.filename.match(/\.[0-9a-z]+$/);
return (fileMatch && (fileMatch[0] === extension));
2017-03-30 11:12:39 +00:00
});
2016-08-14 12:04:34 +00:00
2017-03-30 11:12:39 +00:00
compiler.call(compiler, matchingFiles || [], function (err, compileResult) {
Object.assign(compiledObject, compileResult);
callback(err, compileResult);
});
},
function (err) {
cb(err, compiledObject);
}
2017-03-30 11:12:39 +00:00
);
}
2017-03-02 02:08:28 +00:00
2017-03-30 11:12:39 +00:00
compile_solidity(contractFiles, cb) {
let self = this;
let input = {};
let solcW;
async.waterfall([
function prepareInput(callback) {
async.each(contractFiles,
function(file, fileCb) {
let filename = file.filename.replace('app/contracts/', '');
file.content(function(fileContent) {
input[filename] = fileContent;
fileCb();
});
},
function (err) {
callback(err);
}
);
2017-03-30 11:12:39 +00:00
},
function loadCompiler(callback) {
// TODO: there ino need to load this twice
solcW = new SolcW({logger: self.logger, solcVersion: self.solcVersion});
2017-03-30 11:12:39 +00:00
if (solcW.isCompilerLoaded()) {
return callback();
}
2017-02-25 03:49:34 +00:00
2017-03-30 11:12:39 +00:00
self.logger.info("loading solc compiler..");
solcW.load_compiler(function () {
callback();
});
},
function compileContracts(callback) {
self.logger.info("compiling contracts...");
solcW.compile({sources: input}, 1, function (output) {
if (output.errors) {
return callback(new Error("Solidity errors: " + output.errors).message);
}
callback(null, output);
});
},
function createCompiledObject(output, callback) {
let json = output.contracts;
let compiled_object = {};
2017-02-25 03:49:34 +00:00
for (let contractName in json) {
let contract = json[contractName];
// Pull out filename:classname
// [0] filename:classname
// [1] filename
// [2] classname
const regex = /(.*):(.*)/;
const className = contractName.match(regex)[2];
2017-02-25 03:49:34 +00:00
2017-03-30 11:12:39 +00:00
compiled_object[className] = {};
compiled_object[className].code = contract.bytecode;
compiled_object[className].runtimeBytecode = contract.runtimeBytecode;
compiled_object[className].realRuntimeBytecode = contract.runtimeBytecode.slice(0, -68);
compiled_object[className].swarmHash = contract.runtimeBytecode.slice(-68).slice(0, 64);
compiled_object[className].gasEstimates = contract.gasEstimates;
compiled_object[className].functionHashes = contract.functionHashes;
compiled_object[className].abiDefinition = JSON.parse(contract.interface);
}
callback(null, compiled_object);
2017-02-25 03:49:34 +00:00
}
2017-03-30 11:12:39 +00:00
], function (err, result) {
cb(err, result);
});
2017-03-30 11:38:14 +00:00
}
2017-03-30 11:12:39 +00:00
}
2016-08-14 12:04:34 +00:00
module.exports = Compiler;