From b43bda19d5caa5836c1abbc31befc99d8db8c06e Mon Sep 17 00:00:00 2001 From: Iuri Matias Date: Fri, 20 Jul 2018 20:03:55 +0300 Subject: [PATCH] move graph into a module --- lib/cmds/graph.js | 118 ---------------------------------------------- 1 file changed, 118 deletions(-) delete mode 100644 lib/cmds/graph.js diff --git a/lib/cmds/graph.js b/lib/cmds/graph.js deleted file mode 100644 index ba45743a8..000000000 --- a/lib/cmds/graph.js +++ /dev/null @@ -1,118 +0,0 @@ -const Viz = require('viz.js'); -const fs = require('fs'); - -class GraphGenerator { - constructor(engine) { - this.engine = engine; - } - - generate(options) { - let id = 0; - let contractString = ""; - let relationshipString = ""; - let idMapping = {}; - let contractInheritance = {}; - - for (let contract in this.engine.contractsManager.contracts) { - if(options.skipUndeployed && !this.engine.contractsManager.contracts[contract].deploy) continue; - - id++; - - idMapping[contract] = id; - - let contractLabel = ""; - - contractLabel += `${contract}`; - let tooltip = contract; - - if(this.engine.contractsManager.contracts[contract].instanceOf !== undefined && - this.engine.contractsManager.contracts[this.engine.contractsManager.contracts[contract].instanceOf] !== undefined){ - contractInheritance[contract] = this.engine.contractsManager.contracts[contract].instanceOf; - contractLabel += ": " + this.engine.contractsManager.contracts[contract].instanceOf; - tooltip += " instance of " + this.engine.contractsManager.contracts[contract].instanceOf; - } else { - if(!(options.skipFunctions === true && options.skipEvents === true)) contractLabel += "|"; - - for(let i = 0; i < this.engine.contractsManager.contracts[contract].abiDefinition.length; i++){ - let abiDef = this.engine.contractsManager.contracts[contract].abiDefinition[i]; - if(abiDef.type == 'event' && options.skipEvents) continue; - if(['constructor', 'fallback'].indexOf(abiDef.type) > -1 && options.skipFunctions) continue; - - switch(abiDef.type){ - case 'fallback': - contractLabel += "«fallback»()\\l"; - break; - case 'constructor': - contractLabel += "«constructor»("; - abiDef.inputs.forEach(function(elem, index){ - contractLabel += (index == 0 ? "" : ", ") + elem.type; - }); - contractLabel += ")\\l"; - break; - case 'event': - contractLabel += "«event»" + abiDef.name + "("; - abiDef.inputs.forEach(function(elem, index){ - contractLabel += (index == 0 ? "" : ", ") + elem.type; - }); - contractLabel += ")\\l"; - break; - default: break; - } - } - - let fHashes = this.engine.contractsManager.contracts[contract].functionHashes; - if(fHashes != {} && fHashes != undefined && !options.skipFunctions){ - for(let method in this.engine.contractsManager.contracts[contract].functionHashes){ - contractLabel += method + '\\l'; - } - } - } - - let others = ''; - if(!this.engine.contractsManager.contracts[contract].deploy){ - others = 'fontcolor="#c3c3c3", color="#a0a0a0"'; - tooltip += " (not deployed)"; - } - - contractString += `${id}[label = "{${contractLabel}}", tooltip="${tooltip}", fillcolor=gray95, ${others}]\n`; - - } - - for (let c in this.engine.contractsManager.contractDependencies){ - let contractDependencies = Array.from(new Set(this.engine.contractsManager.contractDependencies[c])); - contractDependencies.forEach((d) => { - if(idMapping[c] !== undefined && idMapping[d] !== undefined){ - if((options.skipUndeployed && this.engine.contractsManager.contracts[c].deploy && this.engine.contractsManager.contracts[d].deploy) || !options.skipUndeployed){ - relationshipString += `${idMapping[d]}->${idMapping[c]}[constraint=true, arrowtail=diamond, tooltip="${c} uses ${d}"]\n`; - } - } - }); - } - - for (let c in contractInheritance){ - if(options.skipUndeployed && !this.engine.contractsManager.contracts[contractInheritance[c]].deploy) continue; - - relationshipString += `${idMapping[contractInheritance[c]]}->${idMapping[c]}[tooltip="${c} instance of ${contractInheritance[c]}"]\n`; - } - - let dot = ` - digraph Contracts { - node[shape=record,style=filled] - edge[dir=back, arrowtail=empty] - ${contractString} - ${relationshipString} - }`; - - let svg = Viz(dot); - - let filename = "diagram.svg"; - - fs.writeFileSync(filename, svg, (err) => { - if (err) throw err; - }); - - - } -} - -module.exports = GraphGenerator;