From ad796bc8f89a294c6f3369498b59370578990f9c Mon Sep 17 00:00:00 2001
From: Jonathan Rainville <rainville.jonathan@gmail.com>
Date: Thu, 4 Jul 2019 16:36:09 -0400
Subject: [PATCH] fix(@embark/site): update contractGeneration plugin api

---
 site/source/docs/plugin_reference.md | 39 ++++++++++++++++------------
 1 file changed, 22 insertions(+), 17 deletions(-)

diff --git a/site/source/docs/plugin_reference.md b/site/source/docs/plugin_reference.md
index 87687e769..9c806a59e 100644
--- a/site/source/docs/plugin_reference.md
+++ b/site/source/docs/plugin_reference.md
@@ -164,31 +164,36 @@ example:
 
 ## .registerContractsGeneration(callback(options))
 
-By default Embark will use EmbarkJS to declare contracts in the dapp. You can override and use your own client side library.
+By default Embark will use EmbarkJS to declare contracts in the Dapp. You can override that and use your own client side library.
 
 Available options:
   * contracts - Hash of objects containing all the deployed contracts. (key: contractName, value: contract object)
-  * abiDefinition
-  * code
-  * deployedAddress
-  * gasEstimates
-  * gas
-  * gasPrice
-  * runtimeByteCode
+    * abiDefinition
+    * code
+    * deployedAddress
+    * gasEstimates
+    * gas
+    * gasPrice
+    * runtimeByteCode
 
 Returns `string`
 
-```
-embark.registerContractsGeneration(function (options) {
-  const contractGenerations = [];
-  Object.keys(options.contracts).map(className => {
-    const contract = options.contracts[className];
-    const abi = JSON.stringify(contract.abiDefinition);
+ The generated string will be used to create the contract objects in the Embark console and will be generated in `embarkArtifacts` so that the Dapp can use them.
 
-    contractGenerations.push(`${className} = web3.eth.contract('${abi}').at('${contract.deployedAddress}')`);
+```
+module.exports = (embark) => {
+  embark.registerContractsGeneration((options) => {
+    const contractGenerations = [];
+    Object.keys(options.contracts).map(className => {
+      const contract = options.contracts[className];
+      const abi = JSON.stringify(contract.abiDefinition);
+
+      contractGenerations.push(`${className} = new web3.eth.Contract(${abi}, '${contract.deployedAddress}');
+      module.exports = ${className};`);
+    });
+    return contractGenerations.join('\n');
   });
-  return contractGenerations.join('\n');
-});
+};
 ```
 
 ## .registerConsoleCommand(options)