Fix compilation issue

OpenZeppelin · Dec 24, 2024 · 2112bbf · 2112bbf
1 parent 4d5a3d5
commit 2112bbf
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/packages/core-cairo/src/generate/sources.ts b/packages/core-cairo/src/generate/sources.ts
@@ -99,10 +99,24 @@ function generateContractSubset(subset: Subset, kind?: Kind): GeneratedContract[
     return contracts;
   } else {
     const getParents = (c: GeneratedContract) => c.contract.components.map(p => p.path);
+    function filterByUpgradeableSetTo(isUpgradeable: boolean) {
+      return (c: GeneratedContract) => {
+        switch (c.options.kind) {
+          case 'Vesting':
+            return isUpgradeable === false;
+          case 'Account':
+          case 'ERC20':
+          case 'ERC721':
+          case 'ERC1155':
+          case 'Governor':
+          case 'Custom':
+            return c.options.upgradeable === isUpgradeable;
+        }
+      }
+    }
     return [
-      ...findCover(contracts.filter(c => c.options.upgradeable === undefined), getParents),
-      ...findCover(contracts.filter(c => c.options.upgradeable === true), getParents),
-      ...findCover(contracts.filter(c => c.options.upgradeable === false), getParents),
+      ...findCover(contracts.filter(filterByUpgradeableSetTo(true)), getParents),
+      ...findCover(contracts.filter(filterByUpgradeableSetTo(false)), getParents),
     ];
   }
 }