sol2uml 2.0.0 → 2.0.1

package/lib/converterClasses2Storage.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.isElementary = exports.calcStorageByteSize = exports.parseStructStorageObject = exports.convertClasses2StorageObjects = exports.StorageType = void 0;
 const umlClass_1 = require("./umlClass");
+const associations_1 = require("./associations");
 var StorageType;
 (function (StorageType) {
     StorageType[StorageType["Contract"] = 0] = "Contract";
@@ -45,9 +46,9 @@ const parseStorage = (umlClass, umlClasses, storages, storageObjects, inheritedC
     inheritedContracts.push(...newInheritedContracts.map((c) => c.targetUmlClassName));
     // Recursively parse each new inherited contract
     newInheritedContracts.forEach((parent) => {
-        const parentClass = umlClasses.find((umlClass) => umlClass.name === parent.targetUmlClassName);
+        const parentClass = (0, associations_1.findAssociatedClass)(parent, umlClass, umlClasses);
         if (!parentClass)
-            throw Error(`Failed to find parent contract ${parent.targetUmlClassName} of ${umlClass.name}`);
+            throw Error(`Failed to find parent contract ${parent.targetUmlClassName} of ${umlClass.absolutePath}`);
         // recursively parse inherited contract
         parseStorage(parentClass, umlClasses, storages, storageObjects, inheritedContracts);
     });

package/lib/sol2uml.js

@@ -101,8 +101,7 @@ program
             ...options,
         };
         const { umlClasses, contractName } = await (0, parserGeneral_1.parserUmlClasses)(fileFolderAddress, combinedOptions);
-        const filteredUmlClasses = (0, filterClasses_1.classesConnectedToBaseContracts)(umlClasses, [combinedOptions.contractName || contractName]);
-        const storageObjects = (0, converterClasses2Storage_1.convertClasses2StorageObjects)(combinedOptions.contractName || contractName, filteredUmlClasses);
+        const storageObjects = (0, converterClasses2Storage_1.convertClasses2StorageObjects)(combinedOptions.contractName || contractName, umlClasses);
         if ((0, regEx_1.isAddress)(fileFolderAddress)) {
             // The first object is the contract
             storageObjects[0].address = fileFolderAddress;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sol2uml",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "Unified Modeling Language (UML) class diagram generator for Solidity contracts",
   "main": "./lib/index.js",
   "types": "./lib/index.d.ts",