From 403f2daee2e18ff4d6eb946da76fb54936e71696 Mon Sep 17 00:00:00 2001
From: Richard Ramos <info@richardramos.me>
Date: Thu, 22 Mar 2018 17:11:11 -0400
Subject: [PATCH] Added relationships between contracts

---
 lib/cmds/graph.js | 61 ++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 47 insertions(+), 14 deletions(-)

diff --git a/lib/cmds/graph.js b/lib/cmds/graph.js
index 8bc670865..b9dfcb4c7 100644
--- a/lib/cmds/graph.js
+++ b/lib/cmds/graph.js
@@ -9,44 +9,77 @@ class GraphGenerator {
     generate() {
         let id = 0;
         let contractString = "";
+        let relationshipString = "";
+        let idMapping = {};
+        let contractInheritance = {};
 
+        
         for (let contract in this.engine.contractsManager.contracts) {
             id++;
+
+            idMapping[contract] = id;
+
             let contractLabel = "";
 
             contractLabel += `${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 += "|";
+
+            for(let i = 0; i < this.engine.contractsManager.contracts[contract].abiDefinition.length; i++){
+                switch(this.engine.contractsManager.contracts[contract].abiDefinition[i].type){
+                    case 'fallback': 
+                        contractLabel += "«fallback»()" + '\\l';
+                        break;
+                    case 'constructor':
+                        contractLabel += "«constructor»(";
+                        for(let j = 0; j < this.engine.contractsManager.contracts[contract].abiDefinition[i].inputs.length; j++){
+                            contractLabel += (j == 0 ? "" : ", ") + this.engine.contractsManager.contracts[contract].abiDefinition[i].inputs[j].type;
+                        }
+                        contractLabel += ")\\l";
+                        break;
+                    default: break;
+                }       
+            }
+
             let fHashes = this.engine.contractsManager.contracts[contract].functionHashes;
             if(fHashes != {} && fHashes != undefined){
-                contractLabel += "|";
                 for(let method in this.engine.contractsManager.contracts[contract].functionHashes){
                     contractLabel += method + '\\l';
                 }
             }
-
-            for(let i = 0; i < this.engine.contractsManager.contracts[contract].abiDefinition.length; i++){
-                if(this.engine.contractsManager.contracts[contract].abiDefinition[i].type == 'fallback')
-                    contractLabel += "«fallback»()" + '\\l';
-            }
-
-        /*if(c.methods.length > 0){
-            contractLabel += "|";
-            c.methods.forEach(function(a){
-                contractLabel += a + '\\l';
-            })
-            
-        } */
+          
 
             contractString += `${id}[label = "{${contractLabel}}"]`
             contractString += "\n";
         }
 
+        for (let c in this.engine.contractsManager.contractDependencies){
+            let contractDependencies = Array.from(new Set(this.engine.contractsManager.contractDependencies[c]));
+            contractDependencies.forEach(function(d){
+                if(idMapping[c] !== undefined && idMapping[d] !== undefined){
+                    relationshipString += `${idMapping[d]}->${idMapping[c]}[constraint=true, arrowtail=diamond]`
+                    relationshipString += "\n";
+                }
+            });
+        }
+
+        for (let c in contractInheritance){
+            relationshipString += `${idMapping[contractInheritance[c]]}->${idMapping[c]}`
+            relationshipString += "\n";
+        }
+
 
         let dot = `
         digraph hierarchy {
         node[shape=record,style=filled,fillcolor=gray95]
         edge[dir=back, arrowtail=empty]
             ${contractString}
+            ${relationshipString}
         }`;
 
         let svg = Viz(dot);