From 8541e5d05c453a70237aaaa46864d5db93af3db7 Mon Sep 17 00:00:00 2001
From: rapiddenis <41779817+rapidddenis@users.noreply.github.com>
Date: Fri, 6 Sep 2024 19:01:53 +0000
Subject: [PATCH] reduce policy and pool services size

---
 contracts/pool/IPoolService.sol     |  3 ---
 contracts/pool/PoolService.sol      |  6 +++---
 contracts/product/PolicyService.sol | 14 ++++++++------
 contracts/staking/Staking.sol       |  2 +-
 4 files changed, 12 insertions(+), 13 deletions(-)

diff --git a/contracts/pool/IPoolService.sol b/contracts/pool/IPoolService.sol
index 4fcd4fb26..fd0fa372b 100644
--- a/contracts/pool/IPoolService.sol
+++ b/contracts/pool/IPoolService.sol
@@ -18,7 +18,6 @@ interface IPoolService is IService {
     event LogPoolServiceWalletFunded(NftId poolNftId, address poolOwner, Amount amount);
     event LogPoolServiceWalletDefunded(NftId poolNftId, address poolOwner, Amount amount);
 
-    event LogPoolServiceBundleCreated(NftId instanceNftId, NftId poolNftId, NftId bundleNftId);
     event LogPoolServiceBundleClosed(NftId instanceNftId, NftId poolNftId, NftId bundleNftId);
 
     event LogPoolServiceBundleStaked(NftId instanceNftId, NftId poolNftId, NftId bundleNftId, Amount amount, Amount netAmount);
@@ -36,8 +35,6 @@ interface IPoolService is IService {
 
     error ErrorPoolServicePoolNotExternallyManaged(NftId poolNftId);
     error ErrorPoolServicePolicyPoolMismatch(NftId policyNftId, NftId productNftId, NftId expectedProductNftId);
-    error ErrorPoolServiceBundleOwnerRoleAlreadySet(NftId poolNftId);
-    error ErrorPoolServiceInvalidTransferAmount(Amount expectedAmount, Amount actualAmount);
     error ErrorPoolServiceBundlePoolMismatch(NftId bundleNftId, NftId poolNftId);
     error ErrorPoolServiceMaxBalanceAmountExceeded(NftId poolNftId, Amount maxBalanceAmount, Amount currentBalanceAmount, Amount transferAmount);
     error ErrorPoolServiceFeesWithdrawAmountExceedsLimit(Amount amount, Amount limit);
diff --git a/contracts/pool/PoolService.sol b/contracts/pool/PoolService.sol
index c87f218b4..8cee72312 100644
--- a/contracts/pool/PoolService.sol
+++ b/contracts/pool/PoolService.sol
@@ -131,13 +131,13 @@ contract PoolService is
         (NftId poolNftId, IInstance instance) = _getAndVerifyActivePool();
         InstanceReader instanceReader = instance.getInstanceReader();
         NftId productNftId = getRegistry().getParentNftId(poolNftId);
+        NftId policyProductNftId = getRegistry().getParentNftId(policyNftId);
 
         // check policy matches with calling pool
-        IPolicy.PolicyInfo memory policyInfo = instanceReader.getPolicyInfo(policyNftId);
-        if(policyInfo.productNftId != productNftId) {
+        if(policyProductNftId != productNftId) {
             revert ErrorPoolServicePolicyPoolMismatch(
                 policyNftId, 
-                policyInfo.productNftId, 
+                policyProductNftId, 
                 productNftId);
         }
 
diff --git a/contracts/product/PolicyService.sol b/contracts/product/PolicyService.sol
index 80d924ff3..28edc1471 100644
--- a/contracts/product/PolicyService.sol
+++ b/contracts/product/PolicyService.sol
@@ -83,9 +83,10 @@ contract PolicyService is
         (IInstance instance,,) = _getAndVerifyCallerForPolicy(applicationNftId);
 
         // check policy is in state applied
-        if (instance.getInstanceReader().getPolicyState(applicationNftId) != APPLIED()) {
-            revert ErrorPolicyServicePolicyStateNotApplied(applicationNftId);
-        }
+        // commented to reduce contract size, the only "APPLIED -> DECLINED" transition exists
+        //if (instance.getInstanceReader().getPolicyState(applicationNftId) != APPLIED()) {
+        //    revert ErrorPolicyServicePolicyStateNotApplied(applicationNftId);
+        //}
 
         // effects
         // store updated policy info
@@ -117,10 +118,11 @@ contract PolicyService is
         ) = _getAndVerifyCallerForPolicy(applicationNftId);
 
         // check policy is in state applied
+        // commented to reduce contract size, the only "APPLIED -> COLLATERALIZED" transition exists
         InstanceReader instanceReader = instance.getInstanceReader();
-        if (instanceReader.getPolicyState(applicationNftId) != APPLIED()) {
-            revert ErrorPolicyServicePolicyStateNotApplied(applicationNftId);
-        }
+        //if (instanceReader.getPolicyState(applicationNftId) != APPLIED()) {
+        //    revert ErrorPolicyServicePolicyStateNotApplied(applicationNftId);
+        //}
 
         PolicyServiceStorage storage $ = _getPolicyServiceStorage();
 
diff --git a/contracts/staking/Staking.sol b/contracts/staking/Staking.sol
index a28a6a930..c1b30e70a 100644
--- a/contracts/staking/Staking.sol
+++ b/contracts/staking/Staking.sol
@@ -73,7 +73,7 @@ contract Staking is
         }
 
         StakingStorage storage $ = _getStakingStorage();
-        address dipToken = _getStakingStorage()._tokenRegistry.getDipTokenAddress();
+        address dipToken = $._tokenRegistry.getDipTokenAddress();
         $._tokenHandler = TokenHandlerDeployerLib.deployTokenHandler(
             address(getRegistry()),
             address(this),