let async = require('async'); let fs = require('../core/fs.js'); const utils = require('../utils/utils.js'); require('ejs'); const Templates = { utils: require('./code_templates/utils.js.ejs'), vanilla_contract: require('./code_templates/vanilla-contract.js.ejs'), embarkjs_contract: require('./code_templates/embarkjs-contract.js.ejs'), exec_when_ready: require('./code_templates/exec-when-ready.js.ejs'), load_manager: require('./code_templates/load-manager.js.ejs'), define_when_env_loaded: require('./code_templates/define-when-env-loaded.js.ejs'), main_context: require('./code_templates/main-context.js.ejs'), define_web3_simple: require('./code_templates/define-web3-simple.js.ejs'), web3_connector: require('./code_templates/web3-connector.js.ejs'), do_when_loaded: require('./code_templates/do-when-loaded.js.ejs'), exec_when_env_loaded: require('./code_templates/exec-when-env-loaded.js.ejs'), embark_building_placeholder: require('./code_templates/embark-building-placeholder.html.ejs') }; class CodeGenerator { constructor(options) { this.blockchainConfig = options.blockchainConfig || {}; this.rpcHost = this.blockchainConfig.rpcHost || ''; this.rpcPort = this.blockchainConfig.rpcPort || ''; this.contractsConfig = options.contractsConfig || {}; this.storageConfig = options.storageConfig || {}; this.communicationConfig = options.communicationConfig || {}; this.namesystemConfig = options.namesystemConfig || {}; this.env = options.env || 'development'; this.plugins = options.plugins; this.events = options.events; } listenToCommands() { let self = this; this.events.setCommandHandler('provider-code', function(cb) { let providerCode = self.generateProvider(false); cb(providerCode); }); // new events this.events.setCommandHandler('code-vanila', function(cb) { self.events.request("contracts:list", (_err, contractsList) => { let vanillaABI = self.generateABI(contractsList, {useEmbarkJS: false}); let contractsJSON = self.generateContractsJSON(contractsList); cb(vanillaABI, contractsJSON); }); }); this.events.setCommandHandler('code', function(cb) { self.events.request("contracts:list", (_err, contractsList) => { let embarkJSABI = self.generateABI(contractsList, {useEmbarkJS: true}); let contractsJSON = self.generateContractsJSON(contractsList); cb(embarkJSABI, contractsJSON); }); }); this.events.setCommandHandler('code-contracts-vanila', function(cb) { self.events.request("contracts:list", (_err, contractsList) => { let vanillaContractsABI = self.generateContracts(contractsList, false, true, false); let contractsJSON = self.generateContractsJSON(contractsList); cb(vanillaContractsABI, contractsJSON); }); }); this.events.setCommandHandler('code-vanila-deployment', function(cb) { self.events.request("contracts:list", (_err, contractsList) => { let vanillaABI = self.generateABI(contractsList, {useEmbarkJS: false, deployment: true}); let contractsJSON = self.generateContractsJSON(contractsList); cb(vanillaABI, contractsJSON); }); }); this.events.setCommandHandler('code-generator:web3js', function(cb) { self.buildWeb3JS(cb); }); self.events.setCommandHandler('code-generator:contract', (contractName, cb) => { self.events.request('contracts:contract', contractName, (contract) => { self.buildContractJS(contractName, self.generateContractJSON(contract, contract), cb); }); }); self.events.setCommandHandler('code-generator:contract:vanilla', (contract, gasLimit, cb) => { cb(self.generateContractCode(contract, gasLimit)); }); this.events.setCommandHandler('embark-building-placeholder', (cb) => { self.buildPlaceholderPage(cb); }); } generateContext() { let result = ""; result += Templates.main_context(); result += Templates.load_manager(); return result; } generateProvider(isDeployment) { let self = this; let result = ""; let providerPlugins; // TODO: check contractsConfig for enabled if (self.blockchainConfig === {} || self.blockchainConfig.enabled === false) { return ""; } result += Templates.utils(); result += Templates.main_context(); result += Templates.load_manager(); result += Templates.define_when_env_loaded(); if (this.plugins) { providerPlugins = this.plugins.getPluginsFor('clientWeb3Provider'); } if (this.plugins && providerPlugins.length > 0) { providerPlugins.forEach(function(plugin) { result += plugin.generateProvider(self) + "\n"; }); } else { let web3Load; if (isDeployment) { let connection = "http://" + this.contractsConfig.deployment.host + ":" + this.contractsConfig.deployment.port; web3Load = Templates.define_web3_simple({url: connection, done: 'done();'}); } else { let connectionList = "[" + this.contractsConfig.dappConnection.map((x) => '"' + x + '"').join(',') + "]"; if (self.env === 'development') { web3Load = Templates.web3_connector({connectionList: connectionList, done: 'done();', warnAboutMetamask: true}); } else { web3Load = Templates.web3_connector({connectionList: connectionList, done: 'done();'}); } } result += Templates.do_when_loaded({block: web3Load}); } return result; } generateContracts(contractsList, useEmbarkJS, isDeployment, useLoader) { let self = this; let result = "\n"; let contractsPlugins; if (useLoader === false) { for (let contract of contractsList) { let abi = JSON.stringify(contract.abiDefinition); result += Templates.vanilla_contract({className: contract.className, abi: abi, contract: contract, gasLimit: 6000000}); } return result; } if (self.blockchainConfig === {} || self.blockchainConfig.enabled === false) { return ""; } if (this.plugins) { contractsPlugins = this.plugins.getPluginsFor('contractGeneration'); } if (this.plugins && contractsPlugins.length > 0) { contractsPlugins.forEach(function (plugin) { result += plugin.generateContracts({contracts: contractsList}); }); } else { for (let contract of contractsList) { let abi = JSON.stringify(contract.abiDefinition); let gasEstimates = JSON.stringify(contract.gasEstimates); let block = ""; if (useEmbarkJS) { let contractAddress = contract.deployedAddress ? ("'" + contract.deployedAddress + "'") : "undefined"; block += Templates.embarkjs_contract({className: contract.className, abi: abi, contract: contract, contractAddress: contractAddress, gasEstimates: gasEstimates}); } else { block += Templates.vanilla_contract({className: contract.className, abi: abi, contract: contract, gasLimit: (isDeployment ? 6000000 : false)}); } result += Templates.exec_when_ready({block: block}); } } return result; } generateContractCode(contract, gasLimit) { let abi = JSON.stringify(contract.abiDefinition); let block = ""; block += Templates.vanilla_contract({className: contract.className, abi: abi, contract: contract, gasLimit: gasLimit}); return block; } generateNamesInitialization(useEmbarkJS) { if (!useEmbarkJS || this.namesystemConfig === {}) return ""; let result = "\n"; result += Templates.define_when_env_loaded(); result += this._getInitCode('names', this.namesystemConfig); return result; } generateStorageInitialization(useEmbarkJS) { if (!useEmbarkJS || this.storageConfig === {}) return ""; let result = "\n"; result += Templates.define_when_env_loaded(); result += this._getInitCode('storage', this.storageConfig); return result; } generateCommunicationInitialization(useEmbarkJS) { if (!useEmbarkJS || this.communicationConfig === {}) return ""; let result = "\n"; result += Templates.define_when_env_loaded(); result += this._getInitCode('communication', this.communicationConfig); return result; } _getInitCode(codeType, config) { let result = ""; let pluginsWithCode = this.plugins.getPluginsFor('initCode'); for (let plugin of pluginsWithCode) { let initCodes = plugin.embarkjs_init_code[codeType] || []; for (let initCode of initCodes) { let [block, shouldInit] = initCode; if (shouldInit.call(plugin, config)) { result += Templates.exec_when_env_loaded({block: block}); } } } return result; } generateABI(contractsList, options) { let result = ""; result += this.generateProvider(options.deployment); result += this.generateContracts(contractsList, options.useEmbarkJS, options.deployment, true); result += this.generateStorageInitialization(options.useEmbarkJS); result += this.generateCommunicationInitialization(options.useEmbarkJS); result += this.generateNamesInitialization(options.useEmbarkJS); return result; } generateContractJSON(className, contract) { let contractJSON = {}; contractJSON.contract_name = className; contractJSON.address = contract.deployedAddress; contractJSON.code = contract.code; contractJSON.runtime_bytecode = contract.runtimeBytecode; contractJSON.real_runtime_bytecode = contract.realRuntimeBytecode; contractJSON.swarm_hash = contract.swarmHash; contractJSON.gas_estimates = contract.gasEstimates; contractJSON.function_hashes = contract.functionHashes; contractJSON.abi = contract.abiDefinition; return contractJSON; } generateContractsJSON(contractsList) { let contracts = {}; for (let contract of contractsList) { contracts[contract.className] = this.generateContractJSON(contract.className, contract); } return contracts; } buildEmbarkJS(cb) { const self = this; let embarkjsCode = fs.readFileSync(fs.embarkPath('js/embark.js')).toString(); let code = ""; async.waterfall([ function getWeb3Location(next) { self.events.request("version:get:web3", function(web3Version) { if (web3Version === "1.0.0-beta") { return next(null, fs.embarkPath("js/web3-1.0.min.js")); } else { self.events.request("version:getPackageLocation", "web3", web3Version, function(err, location) { return next(null, fs.dappPath(location)); }); } }); }, function getImports(web3Location, next) { web3Location = web3Location.replace(/\\/g, '/'); // Import paths must always have forward slashes code += "\nimport Web3 from '" + web3Location + "';\n"; code += "\nimport web3 from 'Embark/web3';\n"; next(); }, function getJSCode(next) { code += "\n" + embarkjsCode + "\n"; let pluginsWithCode = self.plugins.getPluginsFor('embarkjsCode'); for (let plugin of pluginsWithCode) { code += plugin.embarkjs_code.join('\n'); } code += self.generateCommunicationInitialization(true); code += self.generateStorageInitialization(true); code += self.generateNamesInitialization(true); next(); }, function writeFile(next) { fs.mkdirpSync(fs.dappPath(".embark")); fs.writeFileSync(fs.dappPath(".embark", 'embark.js'), code); next(); } ], function(_err, _result) { cb(); }); } buildContractJS(contractName, contractJSON, cb) { let contractCode = ""; contractCode += "import web3 from 'Embark/web3';\n"; contractCode += "import EmbarkJS from 'Embark/EmbarkJS';\n"; contractCode += "let " + contractName + "JSONConfig = " + JSON.stringify(contractJSON) + ";\n"; contractCode += "let " + contractName + " = new EmbarkJS.Contract(" + contractName + "JSONConfig);\n"; contractCode += "\n__embarkContext.execWhenReady(function() {\n"; contractCode += "\n" + contractName + ".setProvider(web3.currentProvider);\n"; contractCode += "\n" + contractName + ".options.from = web3.eth.defaultAccount;\n"; contractCode += "\n});\n"; contractCode += "export default " + contractName + ";\n"; cb(contractCode); } buildWeb3JS(cb) { const self = this; let code = ""; async.waterfall([ function getWeb3Location(next) { self.events.request("version:get:web3", function(web3Version) { if (web3Version === "1.0.0-beta") { return next(null, utils.joinPath(fs.embarkPath("js/web3-1.0.min.js"))); } else { self.events.request("version:getPackageLocation", "web3", web3Version, function(err, location) { return next(null, fs.dappPath(location)); }); } }); }, function getImports(web3Location, next) { web3Location = web3Location.replace(/\\/g, '/'); // Import paths must always have forward slashes code += "\nimport Web3 from '" + web3Location + "';\n"; code += "\n if (typeof web3 !== 'undefined') {"; code += "\n } else {"; code += "\n var web3 = new Web3();\n"; code += "\n }"; let providerCode = self.generateProvider(false); code += providerCode; code += "\nglobal.__embarkContext = __mainContext.__loadManagerInstance;\n"; code += "\nwindow.web3 = web3;\n"; code += "\nexport default web3;\n"; next(null, code); } ], cb); } buildPlaceholderPage(cb) { let html = Templates.embark_building_placeholder({buildingMsg: __('Embark is building, please wait...')}); cb(html); } } module.exports = CodeGenerator;