embarklabs
/
embark
mirror of https://github.com/embarklabs/embark.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

add find contract

This commit is contained in:
Iuri Matias 2018-07-30 15:12:49 -04:00
parent 77dd5b4944
commit 358c9b2531
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
lib/modules/graph/index.js
View File

@ -6,6 +6,7 @@ class GraphGenerator {
constructor(embark, options) { constructor(embark, options) {
const self = this; const self = this;
this.events = embark.events; this.events = embark.events;
this.contracts = [];
this.events.setCommandHandler("graph:create", function(options, cb) { this.events.setCommandHandler("graph:create", function(options, cb) {
self.generate(options); self.generate(options);
@ -20,13 +21,12 @@ class GraphGenerator {
let relationshipString = ""; let relationshipString = "";
let idMapping = {}; let idMapping = {};
let contractInheritance = {}; let contractInheritance = {};
let contracts = {};
let contractsDependencies = {}; let contractsDependencies = {};
async.waterfall([ async.waterfall([
function getContractList(next) { function getContractList(next) {
self.events.request('contracts:list', (err, _contracts) => { self.events.request('contracts:list', (err, contracts) => {
contracts = _contracts; self.contracts = contracts;
next(); next();
}); });
}, },
@ -37,7 +37,7 @@ class GraphGenerator {
}); });
}, },
function (next) { function (next) {
for (let contract of contracts) { for (let contract of self.contracts) {
if (options.skipUndeployed && !contract.deploy) continue; if (options.skipUndeployed && !contract.deploy) continue;
id++; id++;
@ -49,7 +49,7 @@ class GraphGenerator {
contractLabel += `${contract.className}`; contractLabel += `${contract.className}`;
let tooltip = contract.className; let tooltip = contract.className;
if (contract.instanceOf !== undefined && contracts[contract.instanceOf] !== undefined) { if (contract.instanceOf !== undefined && self.getContract(contract.instanceOf) !== undefined) {
contractInheritance[contract.className] = contract.instanceOf; contractInheritance[contract.className] = contract.instanceOf;
contractLabel += ": " + contract.instanceOf; contractLabel += ": " + contract.instanceOf;
tooltip += " instance of " + contract.instanceOf; tooltip += " instance of " + contract.instanceOf;
@ -106,7 +106,7 @@ class GraphGenerator {
let contractDependencies = Array.from(new Set(contractsDependencies[c])); let contractDependencies = Array.from(new Set(contractsDependencies[c]));
contractDependencies.forEach((d) => { contractDependencies.forEach((d) => {
if (idMapping[c] !== undefined && idMapping[d] !== undefined) { if (idMapping[c] !== undefined && idMapping[d] !== undefined) {
if ((options.skipUndeployed && contracts[c].deploy && contracts[d].deploy) || !options.skipUndeployed) { if ((options.skipUndeployed && self.getContract(c).deploy && self.getContract(d).deploy) || !options.skipUndeployed) {
relationshipString += `${idMapping[d]}->${idMapping[c]}[constraint=true, arrowtail=diamond, tooltip="${c} uses ${d}"]\n`; relationshipString += `${idMapping[d]}->${idMapping[c]}[constraint=true, arrowtail=diamond, tooltip="${c} uses ${d}"]\n`;
} }
} }
@ -116,13 +116,10 @@ class GraphGenerator {
}, },
function (next) { function (next) {
for (let c in contractInheritance){ for (let c in contractInheritance){
if(options.skipUndeployed && !contracts[contractInheritance[c]].deploy) continue; if(options.skipUndeployed && !self.getContract(contractInheritance[c]).deploy) continue;
relationshipString += `${idMapping[contractInheritance[c]]}->${idMapping[c]}[tooltip="${c} instance of ${contractInheritance[c]}"]\n`; relationshipString += `${idMapping[contractInheritance[c]]}->${idMapping[c]}[tooltip="${c} instance of ${contractInheritance[c]}"]\n`;
} }
console.log("-----");
console.log(relationshipString);
console.log("-----");
next(); next();
}, },
function (next) { function (next) {
@ -134,7 +131,6 @@ class GraphGenerator {
${relationshipString} ${relationshipString}
}`; }`;
console.log(dot);
let svg = Viz(dot); let svg = Viz(dot);
let filename = "diagram.svg"; let filename = "diagram.svg";
@ -147,6 +143,10 @@ class GraphGenerator {
], function(_err, _result) { ], function(_err, _result) {
}); });
} }
getContract(contractName) {
return this.contracts.find((contract) => { return contract.className === contractName });
}
} }
module.exports = GraphGenerator; module.exports = GraphGenerator;