support receiving contract as argument; automatically determine correct deployment order

tasks/deploy.coffee

@@ -23,6 +23,19 @@ module.exports = (grunt) ->
     result  = "web3.setProvider(new web3.providers.HttpProvider('http://#{rpcHost}:#{rpcPort}'));"
     result += "web3.eth.defaultAccount = web3.eth.accounts[0];"
 
+    contractDependencies = {}
+    if contractsConfig?
+      for className, options of contractsConfig
+        continue unless options.args?
+        for arg in options.args
+          if arg[0] is "$"
+            if contractDependencies[className] is undefined
+              contractDependencies[className] = []
+            contractDependencies[className].push(arg.substr(1))
+
+    all_contracts = []
+    contractDB = {}
+
     contractFiles = grunt.file.expand(grunt.config.get("deploy.contracts"))
     for contractFile in contractFiles
       source = grunt.file.read(contractFile)
@@ -31,16 +44,46 @@ module.exports = (grunt) ->
       compiled_contracts = web3.eth.compile.solidity(source)
 
       for className, contract of compiled_contracts
+        all_contracts.push(className)
+        contractDB[className] = contract
+
+    all_contracts.sort (a,b) =>
+      contract_1 = contractDependencies[a]
+      contract_2 = contractDependencies[b]
+
+      if a in contract_1 and b in contract_2
+        grunt.log.writeln("looks like you have a circular dependency between #{a} and #{b}")
+        exit
+      else if b in contract_1
+        1
+      else if a in contract_2
+        -1
+      else
+        0
+
+    deployedContracts = {}
+
+    for className in all_contracts
+      contract = contractDB[className]
       contractGasLimit = contractsConfig?[className]?.gasLimit || gasLimit
       contractGasPrice = contractsConfig?[className]?.gasPrice || gasPrice
 
-        args = contractsConfig?[className]?.args
+      args = contractsConfig?[className]?.args || []
 
       contractObject = web3.eth.contract(contract.info.abiDefinition)
 
-        contractParams = args || []
+      realArgs = []
+
+      for arg in args
+        if arg[0] is "$"
+          realArgs.push(deployedContracts[arg.substr(1)])
+        else
+          realArgs.push(arg)
+
+      contractParams = realArgs
       contractParams.push({from: primaryAddress, data: contract.code, gas: contractGasLimit, gasPrice: contractGasPrice})
       contractAddress = contractObject.new.apply(contractObject, contractParams).address
+      deployedContracts[className] = contractAddress
 
       console.log "address is #{contractAddress}"