mirror of https://github.com/embarklabs/embark.git
117 lines
3.7 KiB
JavaScript
117 lines
3.7 KiB
JavaScript
/*jshint esversion: 6, loopfunc: true */
|
|
let async = require('../utils/async_extend.js');
|
|
let SolcW = require('./solcW.js');
|
|
|
|
class Compiler {
|
|
constructor(options) {
|
|
this.plugins = options.plugins;
|
|
this.logger = options.logger;
|
|
}
|
|
|
|
compile_contracts(contractFiles, cb) {
|
|
|
|
let available_compilers = {
|
|
//".se": this.compile_serpent
|
|
".sol": this.compile_solidity.bind(this)
|
|
};
|
|
|
|
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;
|
|
});
|
|
});
|
|
}
|
|
}
|
|
|
|
let compiledObject = {};
|
|
|
|
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) {
|
|
let fileMatch = file.filename.match(/\.[0-9a-z]+$/);
|
|
return (fileMatch && (fileMatch[0] === extension));
|
|
});
|
|
|
|
compiler.call(compiler, matchingFiles || [], function (err, compileResult) {
|
|
Object.assign(compiledObject, compileResult);
|
|
callback(err, compileResult);
|
|
});
|
|
},
|
|
function (err) {
|
|
cb(err, compiledObject);
|
|
}
|
|
);
|
|
}
|
|
|
|
compile_solidity(contractFiles, cb) {
|
|
let self = this;
|
|
let input = {};
|
|
let solcW;
|
|
async.waterfall([
|
|
function prepareInput(callback) {
|
|
for (let i = 0; i < contractFiles.length; i++) {
|
|
// TODO: this depends on the config
|
|
let filename = contractFiles[i].filename.replace('app/contracts/', '');
|
|
input[filename] = contractFiles[i].content.toString();
|
|
}
|
|
callback();
|
|
},
|
|
function loadCompiler(callback) {
|
|
// TODO: there ino need to load this twice
|
|
solcW = new SolcW();
|
|
if (solcW.isCompilerLoaded()) {
|
|
return callback();
|
|
}
|
|
|
|
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 = {};
|
|
|
|
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];
|
|
|
|
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);
|
|
}
|
|
], function (err, result) {
|
|
cb(err, result);
|
|
});
|
|
}
|
|
}
|
|
|
|
module.exports = Compiler;
|