Merge pull request #745 from embark-framework/deployIf

deployIf field + minor refactor

lib/core/engine.js

@@ -167,7 +167,7 @@ class Engine {
     this.registerModule('solidity', {ipc: self.ipc, useDashboard: this.useDashboard});
     this.registerModule('vyper');
     this.registerModule('profiler');
-    this.registerModule('deploytracker');
+    this.registerModule('deploytracker', {trackContracts: options.trackContracts});
     this.registerModule('specialconfigs');
     this.registerModule('console_listener', {ipc: self.ipc});
     this.registerModule('contracts_manager');

lib/core/plugins.js

@@ -138,15 +138,18 @@ Plugins.prototype.runActionsForEvent = function(eventName, args, cb) {
   let actionPlugins = this.getPluginsProperty('eventActions', 'eventActions', eventName);
 
   if (actionPlugins.length === 0) {
-    return cb();
+    return cb(args);
   }
 
-  async.eachLimit(actionPlugins, 1, function(plugin, nextEach) {
+  async.reduce(actionPlugins, args, function(current_args, plugin, nextEach) {
     if (typeof (args) === 'function') {
-      plugin.call(plugin, nextEach);
+      plugin.call(plugin, (params) => {
+        nextEach(null, (params || current_args));
+      });
     } else {
-      //plugin.call(plugin, ...args, nextEach);
-      plugin.call(plugin, args, nextEach);
+      plugin.call(plugin, args, (params) => {
+        nextEach(null, (params || current_args));
+      });
     }
   }, cb);
 };

lib/modules/contracts_manager/index.js

@@ -7,7 +7,7 @@ let utils = require('../../utils/utils.js');
 // TODO: create a contract object
 
 class ContractsManager {
-  constructor(embark) {
+  constructor(embark, options) {
     const self = this;
     this.logger = embark.logger;
     this.events = embark.events;
@@ -16,6 +16,7 @@ class ContractsManager {
     this.contractDependencies = {};
     this.deployOnlyOnConfig = false;
     this.compileError = false;
+    this.compileOnceOnly = options.compileOnceOnly;
 
     self.events.setCommandHandler('contracts:list', (cb) => {
       cb(self.compileError, self.listContracts());
@@ -77,7 +78,7 @@ class ContractsManager {
       },
       function compileContracts(callback) {
         self.events.emit("status", __("Compiling..."));
-        if (process.env.isTest && self.compiledContracts && Object.keys(self.compiledContracts).length) {
+        if (self.compileOnceOnly && self.compiledContracts && Object.keys(self.compiledContracts).length) {
           // Only compile once for tests
           return callback();
         }

lib/modules/deployment/contract_deployer.js

@@ -140,13 +140,12 @@ class ContractDeployer {
           return next();
         }
 
-        // TODO find a better way to do that
-        if (process.env.isTest) {
-          return self.deployContract(contract, next);
-        }
-        // TODO: this should be a plugin API instead, if not existing, it should by default deploy the contract
-        self.events.request("deploy:contract:shouldDeploy", contract, function(trackedContract) {
-          if (!trackedContract) {
+        self.plugins.emitAndRunActionsForEvent('deploy:contract:shouldDeploy', {contract: contract, shouldDeploy: true}, function(_err, params) {
+          let trackedContract = params.contract;
+          if (!params.shouldDeploy) {
+            return next();
+          }
+          if (!trackedContract.address) {
             return self.deployContract(contract, next);
           }
 
@@ -213,7 +212,9 @@ class ContractDeployer {
         });
       },
       function applyBeforeDeploy(next) {
-        self.plugins.emitAndRunActionsForEvent('deploy:contract:beforeDeploy', {contract: contract}, next);
+        self.plugins.emitAndRunActionsForEvent('deploy:contract:beforeDeploy', {contract: contract}, (_params) => {
+          next();
+        });
       },
       function getGasPriceForNetwork(next) {
         self.events.request("blockchain:gasPrice", (gasPrice) => {

lib/modules/deploytracker/index.js

@@ -6,6 +6,8 @@ class DeployTracker {
   constructor(embark, options) {
     this.logger = embark.logger;
     this.events = embark.events;
+    this.embark = embark;
+    this.trackContracts = (options.trackContracts !== false);
 
     // TODO: unclear where it comes from
     this.env = options.env;
@@ -24,9 +26,20 @@ class DeployTracker {
       self.save();
     });
 
-    this.events.setCommandHandler("deploy:contract:shouldDeploy", (contract, cb) => {
+    self.embark.registerActionForEvent("deploy:contract:shouldDeploy", (params, cb) => {
+      if (!self.trackContracts) {
+        return cb(params);
+      }
+
+      let contract = params.contract;
       let trackedContract = self.getContract(contract.className, contract.realRuntimeBytecode, contract.realArgs);
-      cb(trackedContract);
+      if (trackedContract) {
+        params.contract.address = trackedContract.address;
+      }
+      if (params.shouldDeploy && trackedContract) {
+         params.shouldDeploy = true;
+      }
+      cb(params);
     });
   }
 

lib/modules/specialconfigs/index.js

@@ -12,6 +12,7 @@ class SpecialConfigs {
 
     this.registerAfterDeployAction();
     this.registerOnDeployAction();
+    this.registerDeployIfAction();
   }
 
   replaceWithAddresses(cmd, cb) {
@@ -104,6 +105,30 @@ class SpecialConfigs {
     });
   }
 
+  registerDeployIfAction() {
+    const self = this;
+
+    self.embark.registerActionForEvent("deploy:contract:shouldDeploy", (params, cb) => {
+      let cmd = params.contract.deployIf;
+      if (!cmd) {
+        return cb(params);
+      }
+
+      self.events.request('runcode:eval', cmd, (err, result) => {
+        if (err) {
+          self.logger.error(params.contract.className + ' deployIf directive has an error; contract will not deploy');
+          self.logger.error(err);
+          params.shouldDeploy = false;
+        } else if (!result) {
+          self.logger.info(params.contract.className + ' deployIf directive returned false; contract will not deploy');
+          params.shouldDeploy = false;
+        }
+
+        cb(params);
+      });
+    });
+  }
+
 }
 
 module.exports = SpecialConfigs;

lib/tests/run_tests.js

@@ -24,7 +24,6 @@ function getFilesFromDir(filePath, cb) {
 
 module.exports = {
   run: function (options) {
-    process.env.isTest = true;
     let failures = 0;
     let filePath = options.file;
     const loglevel = options.loglevel || 'warn';

lib/tests/test.js

@@ -108,7 +108,8 @@ class Test {
       web3: this.web3
     });
     this.engine.startService("deployment", {
-      trackContracts: false
+      trackContracts: false,
+      compileOnceOnly: true
     });
     this.events.request('deploy:setGasLimit', 6000000);
     this.engine.startService("codeCoverage");

test_apps/test_app/config/contracts.js

@@ -53,6 +53,7 @@ module.exports = {
         }
       },
       SomeContract: {
+        deployIf: 'MyToken.methods.isAvailable().call()',
         args: [
           ["$MyToken2", "$SimpleStorage"],
           100

test_apps/test_app/contracts/token.sol

@@ -63,4 +63,7 @@ contract Token {
   function safeToAdd(uint a, uint b) internal pure returns (bool) {
     return (a + b >= a);
   }
+  function isAvailable() public constant returns (bool) {
+    return false;
+  }
 }

test_apps/test_app/test/token_spec.js

@@ -5,6 +5,7 @@ const MyToken = require('Embark/contracts/MyToken');
 const MyToken2 = require('Embark/contracts/MyToken2');
 const AlreadyDeployedToken = require('Embark/contracts/AlreadyDeployedToken');
 const Test = require('Embark/contracts/Test');
+const SomeContract = require('Embark/contracts/SomeContract');
 
 config({
   contracts: {
@@ -40,6 +41,7 @@ config({
       }
     },
     SomeContract: {
+      deployIf: "MyToken.methods.isAvailable().call()",
       args: [
         ["$MyToken2", "$SimpleStorage"],
         100
@@ -79,4 +81,8 @@ describe("Token", function () {
     let result = await Test.methods.addr().call();
     assert.strictEqual(result, MyToken.options.address);
   });
+
+  it("should not deploy if deployIf returns false", async function() {
+    assert.ok(!SomeContract.options.address);
+  });
 });