chore: normalise contract filename for storage diagrams to help find …

…the correct Solidity file
naddison36 · Aug 14, 2022 · dac62a9 · dac62a9
1 parent 055e09f
commit dac62a9
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/lib/converterClasses2Storage.js b/lib/converterClasses2Storage.js
diff --git a/src/ts/converterAST2Classes.ts b/src/ts/converterAST2Classes.ts
@@ -145,9 +145,14 @@ export function convertAST2UmlClasses(
                                   }
                               })
                             : [],
-                    })
+                    }
+                    debug(
+                        `Added Etherscan import ${newImport.absolutePath} with class names: ${newImport.classNames}`
+                    )
+                    imports.push(newImport)
                 }
             }
+            // TODO add file level constants
         })
     } else {
         throw new Error(`AST node not of type SourceUnit`)

diff --git a/src/ts/converterClasses2Storage.ts b/src/ts/converterClasses2Storage.ts
@@ -76,7 +76,7 @@ export const convertClasses2Storages = (
         }
         return (
             name === contractName &&
-            (relativePath == contractFilename ||
+            (relativePath == path.normalize(contractFilename) ||
                 path.basename(relativePath) ===
                     path.normalize(contractFilename))
         )