@etherisc/gif-next 0.0.2-e6a90b9 → 0.0.2-e6ad61d-739

package/contracts/instance/treasury/ITreasury.sol

@@ -0,0 +1,91 @@
+// SPDX-License-Identifier: Apache-2.0
+pragma solidity ^0.8.19;
+
+import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
+
+import {IRegistryLinked} from "../../registry/IRegistry.sol";
+
+import {NftId} from "../../types/NftId.sol";
+import {UFixed} from "../../types/UFixed.sol";
+import {Fee} from "../../types/Fee.sol";
+
+import {TokenHandler} from "./TokenHandler.sol";
+
+interface ITreasury {
+    // TODO add events
+    // TODO add errors
+}
+
+interface ITreasuryModule is IRegistryLinked, ITreasury {
+    struct ProductSetup {
+        NftId productNftId;
+        NftId distributorNftId;
+        NftId poolNftId;
+        IERC20 token;
+        TokenHandler tokenHandler;
+        address wallet;
+        Fee policyFee;
+        Fee processingFee;
+    }
+
+    struct DistributorSetup {
+        NftId distributorNftId;
+        address wallet;
+        Fee commissionFees;
+    }
+
+    struct PoolSetup {
+        NftId poolNftId;
+        address wallet;
+        Fee stakingFee;
+        Fee performanceFee;
+    }
+
+    function registerProduct(
+        NftId productNftId,
+        NftId distributorNftId,
+        NftId poolNftId,
+        IERC20 token,
+        address wallet,
+        Fee memory policyFee,
+        Fee memory processingFee
+    ) external;
+
+    function setProductFees(
+        NftId productNftId,
+        Fee memory policyFee,
+        Fee memory processingFee
+    ) external;
+
+    function registerPool(
+        NftId poolNftId,
+        address wallet,
+        Fee memory stakingFee,
+        Fee memory performanceFee
+    ) external;
+
+    function setPoolFees(
+        NftId poolNftId,
+        Fee memory stakingFee,
+        Fee memory performanceFee
+    ) external;
+
+    function processPremium(NftId policyNftId, NftId productNftId) external;
+
+    function getTokenHandler(
+        NftId productNftId
+    ) external view returns (TokenHandler tokenHandler);
+
+    function getProductSetup(
+        NftId productNftId
+    ) external view returns (ProductSetup memory setup);
+
+    function getPoolSetup(
+        NftId poolNftId
+    ) external view returns (PoolSetup memory setup);
+
+    function calculateFeeAmount(
+        uint256 amount,
+        Fee memory fee
+    ) external pure returns (uint256 feeAmount, uint256 netAmount);
+}

package/contracts/instance/treasury/TokenHandler.sol

@@ -0,0 +1,24 @@
+// SPDX-License-Identifier: Apache-2.0
+pragma solidity ^0.8.19;
+
+import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
+
+contract TokenHandler {
+    // TODO use oz safeTransferFrom
+
+    IERC20 _token;
+
+    constructor(address token) {
+        _token = IERC20(token);
+    }
+
+    // TODO add logging
+    function transfer(
+        address from,
+        address to,
+        uint256 amount
+    ) external // TODO add authz (only treasury)
+    {
+        _token.transferFrom(from, to, amount);
+    }
+}

package/contracts/instance/treasury/TreasuryModule.sol

@@ -0,0 +1,168 @@
+// SPDX-License-Identifier: Apache-2.0
+pragma solidity ^0.8.19;
+
+import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
+
+import {NftId} from "../../types/NftId.sol";
+import {Fee, feeIsZero} from "../../types/Fee.sol";
+import {UFixed, UFixedMathLib} from "../../types/UFixed.sol";
+import {IProductComponent} from "../../components/IProduct.sol";
+import {IPolicy, IPolicyModule} from "../policy/IPolicy.sol";
+import {TokenHandler} from "./TokenHandler.sol";
+import {ITreasuryModule} from "./ITreasury.sol";
+import {TokenHandler} from "./TokenHandler.sol";
+
+abstract contract TreasuryModule is ITreasuryModule {
+    mapping(NftId productNftId => ProductSetup setup) private _productSetup;
+    mapping(NftId distributorNftId => DistributorSetup setup)
+        private _distributorSetup;
+    mapping(NftId poolNftId => PoolSetup setup) private _poolSetup;
+
+    IPolicyModule private _policyModule;
+
+    constructor() {
+        _policyModule = IPolicyModule(address(this));
+    }
+
+    function registerProduct(
+        NftId productNftId,
+        NftId distributorNftId,
+        NftId poolNftId,
+        IERC20 token,
+        address wallet,
+        Fee memory policyFee,
+        Fee memory processingFee
+    ) external override // TODO add authz (only component module)
+    {
+        // TODO add validation
+
+        // deploy product specific handler contract
+        TokenHandler tokenHandler = new TokenHandler(address(token));
+
+        _productSetup[productNftId] = ProductSetup(
+            productNftId,
+            distributorNftId,
+            poolNftId,
+            token,
+            tokenHandler,
+            wallet,
+            policyFee,
+            processingFee
+        );
+
+        // TODO add logging
+    }
+
+    function setProductFees(
+        NftId productNftId,
+        Fee memory policyFee,
+        Fee memory processingFee
+    ) external override // TODO add authz (only component owner service)
+    {
+        // TODO add validation
+
+        ProductSetup storage setup = _productSetup[productNftId];
+        setup.policyFee = policyFee;
+        setup.processingFee = processingFee;
+
+        // TODO add logging
+    }
+
+    function registerPool(
+        NftId poolNftId,
+        address wallet,
+        Fee memory stakingFee,
+        Fee memory performanceFee
+    ) external override // TODO add authz (only component module)
+    {
+        // TODO add validation
+
+        _poolSetup[poolNftId] = PoolSetup(
+            poolNftId,
+            wallet,
+            stakingFee,
+            performanceFee
+        );
+
+        // TODO add logging
+    }
+
+    function setPoolFees(
+        NftId poolNftId,
+        Fee memory stakingFee,
+        Fee memory performanceFee
+    ) external override // TODO add authz (only component owner service)
+    {
+        // TODO add validation
+
+        PoolSetup storage setup = _poolSetup[poolNftId];
+        setup.stakingFee = stakingFee;
+        setup.performanceFee = performanceFee;
+
+        // TODO add logging
+    }
+
+    function getTokenHandler(
+        NftId productNftId
+    ) external view override returns (TokenHandler tokenHandler) {
+        return _productSetup[productNftId].tokenHandler;
+    }
+
+    function getProductSetup(
+        NftId productNftId
+    ) external view override returns (ProductSetup memory setup) {
+        return _productSetup[productNftId];
+    }
+
+    function getPoolSetup(
+        NftId poolNftId
+    ) external view override returns (PoolSetup memory setup) {
+        return _poolSetup[poolNftId];
+    }
+
+    function processPremium(
+        NftId policyNftId,
+        NftId productNftId
+    ) external override // TODO add authz (only product service)
+    {
+        IPolicy.PolicyInfo memory policyInfo = _policyModule.getPolicyInfo(
+            policyNftId
+        );
+        require(
+            policyInfo.nftId == policyNftId,
+            "ERROR:TRS-020:POLICY_UNKNOWN"
+        );
+
+        ProductSetup memory product = _productSetup[productNftId];
+        TokenHandler tokenHandler = product.tokenHandler;
+        address policyOwner = this.getRegistry().getOwner(policyNftId);
+        address poolWallet = _poolSetup[product.poolNftId].wallet;
+        // TODO add validation
+
+        if (feeIsZero(product.policyFee)) {
+            tokenHandler.transfer(
+                policyOwner,
+                poolWallet,
+                policyInfo.premiumAmount
+            );
+        } else {
+            (uint256 feeAmount, uint256 netAmount) = calculateFeeAmount(
+                policyInfo.premiumAmount,
+                product.policyFee
+            );
+
+            tokenHandler.transfer(policyOwner, product.wallet, feeAmount);
+            tokenHandler.transfer(policyOwner, poolWallet, netAmount);
+        }
+    }
+
+    function calculateFeeAmount(
+        uint256 amount,
+        Fee memory fee
+    ) public pure override returns (uint256 feeAmount, uint256 netAmount) {
+        UFixed fractionalAmount = UFixedMathLib.itof(amount) *
+            fee.fractionalFee;
+        feeAmount = UFixedMathLib.ftoi(fractionalAmount) + fee.fixedFee;
+        netAmount = amount - feeAmount;
+    }
+}

package/contracts/registry/ChainNft.sol

@@ -4,10 +4,7 @@ pragma solidity ^0.8.19;
 import {ERC721, ERC721Enumerable} from "@openzeppelin/contracts/token/ERC721/extensions/ERC721Enumerable.sol";
 import {IChainNft} from "./IChainNft.sol";
 
-contract ChainNft is
-    ERC721Enumerable,
-    IChainNft
-{
+contract ChainNft is ERC721Enumerable, IChainNft {
     string public constant NAME = "Dezentralized Insurance Protocol Registry";
     string public constant SYMBOL = "DIPR";
 
@@ -18,22 +15,18 @@ contract ChainNft is
     address private _registry;
 
     // only used for _getNextTokenId
-    uint256 internal _chainIdInt; 
+    uint256 internal _chainIdInt;
     uint256 internal _chainIdDigits;
     uint256 internal _chainIdMultiplier;
     uint256 internal _idNext;
     uint256 internal _totalMinted;
 
-
     modifier onlyRegistry() {
         require(msg.sender == _registry, "ERROR:NFT-001:CALLER_NOT_REGISTRY");
         _;
     }
 
-
-    constructor(address registry)
-        ERC721(NAME, SYMBOL)
-    {
+    constructor(address registry) ERC721(NAME, SYMBOL) {
         require(registry != address(0), "ERROR:NFT-010:REGISTRY_ZERO");
 
         _registry = registry;
@@ -43,88 +36,59 @@ contract ChainNft is
         _chainIdMultiplier = 10 ** _chainIdDigits;
 
         // on mainnet/goerli start /1 (reserved for protocol nft) on other chains with 2
-        if(block.chainid == 1 || block.chainid == 5) {
+        if (block.chainid == 1 || block.chainid == 5) {
             _idNext = 1;
         } else {
             _idNext = 2;
         }
     }
 
-
     function mint(
         address to,
         string memory uri
-    )
-        external
-        override
-        onlyRegistry
-        returns(uint256 tokenId)
-    {
+    ) external override onlyRegistry returns (uint256 tokenId) {
         tokenId = _getNextTokenId();
         _totalMinted++;
 
         _safeMint(to, tokenId);
 
-        if(bytes(uri).length > 0) {
+        if (bytes(uri).length > 0) {
             _uri[tokenId] = uri;
         }
     }
 
-
-    function burn(uint256 tokenId)
-        external
-        override
-        onlyRegistry
-    {
+    function burn(uint256 tokenId) external override onlyRegistry {
         _requireMinted(tokenId);
         _burn(tokenId);
         delete _uri[tokenId];
     }
 
-
-    function setURI(uint256 tokenId, string memory uri)
-        external
-        override
-        onlyRegistry
-    {
+    function setURI(
+        uint256 tokenId,
+        string memory uri
+    ) external override onlyRegistry {
         require(bytes(uri).length > 0, "ERROR:CRG-011:URI_EMPTY");
 
         _requireMinted(tokenId);
         _uri[tokenId] = uri;
     }
 
-
-    function exists(uint256 tokenId)
-        external
-        view
-        override
-        returns(bool)
-    {
+    function exists(uint256 tokenId) external view override returns (bool) {
         return _exists(tokenId);
     }
 
-
-    function tokenURI(uint256 tokenId)
-        public
-        view
-        override 
-        returns(string memory)
-    {
+    function tokenURI(
+        uint256 tokenId
+    ) public view override returns (string memory) {
         _requireMinted(tokenId);
         return _uri[tokenId];
     }
 
-
-    function getRegistryAddress()
-        external
-        view
-        override
-        returns(address)
-    {
+    function getRegistryAddress() external view override returns (address) {
         return _registry;
     }
 
-    function totalMinted() external override view returns(uint256) {
+    function totalMinted() external view override returns (uint256) {
         return _totalMinted;
     }
 
@@ -153,17 +117,15 @@ contract ChainNft is
     // (42 * 10 ** 10 + 9876543210) * 100 + 10
     // (index * 10 ** digits + chainid) * 100 + digits (1 < digits < 100)
 
-    function _getNextTokenId() private returns(uint256 id) {
-        id = (_idNext * _chainIdMultiplier + _chainIdInt) * 100 + _chainIdDigits;
+    function _getNextTokenId() private returns (uint256 id) {
+        id =
+            (_idNext * _chainIdMultiplier + _chainIdInt) *
+            100 +
+            _chainIdDigits;
         _idNext++;
     }
 
-
-    function _countDigits(uint256 num)
-        private 
-        pure 
-        returns (uint256 count)
-    {
+    function _countDigits(uint256 num) private pure returns (uint256 count) {
         count = 0;
         while (num != 0) {
             count++;

package/contracts/registry/IChainNft.sol

@@ -3,16 +3,19 @@ pragma solidity ^0.8.19;
 
 import {IERC721Enumerable} from "@openzeppelin/contracts/token/ERC721/extensions/IERC721Enumerable.sol";
 
-interface IChainNft is 
-    IERC721Enumerable 
-{
+interface IChainNft is IERC721Enumerable {
+    function mint(
+        address to,
+        string memory uri
+    ) external returns (uint256 tokenId);
 
-    function mint(address to, string memory uri) external returns(uint256 tokenId);
     function burn(uint256 tokenId) external;
+
     function setURI(uint256 tokenId, string memory uri) external;
 
-    function exists(uint256 tokenId) external view returns(bool);
-    function totalMinted() external view returns(uint256);
+    function exists(uint256 tokenId) external view returns (bool);
+
+    function totalMinted() external view returns (uint256);
 
-    function getRegistryAddress() external view returns(address registry);
+    function getRegistryAddress() external view returns (address registry);
 }

package/contracts/registry/IRegistry.sol

@@ -1,68 +1,67 @@
 // SPDX-License-Identifier: Apache-2.0
 pragma solidity ^0.8.19;
 
+import {NftId} from "../types/NftId.sol";
+import {ObjectType} from "../types/ObjectType.sol";
+
 interface IOwnable {
-    function getOwner() external view returns(address owner);
+    function getOwner() external view returns (address owner);
 }
 
 interface IRegistryLinked {
-
     event LogDebug(uint256 idx, address module, string comment);
 
-    function getRegistry() external view returns(IRegistry registry);
+    function getRegistry() external view returns (IRegistry registry);
 }
 
-interface IRegisterable is 
-    IOwnable,
-    IRegistryLinked
-{
-
-    function register() external returns(uint256 nftId);
-    
-    function getNftId() external view returns(uint256 nftId);
-    function getParentNftId() external view returns(uint256 parentNftId);
-    function getType() external view returns(uint256 objectType);
-    function getData() external view returns(bytes memory data);
-    function isRegisterable() external pure returns(bool);
-    function getInitialOwner() external view returns(address initialOwner);
-
-    function isRegistered() external view returns(bool);
-}
+interface IRegisterable is IOwnable, IRegistryLinked {
+    function register() external returns (NftId nftId);
 
+    function getNftId() external view returns (NftId nftId);
 
-interface IRegistry {
+    function getParentNftId() external view returns (NftId parentNftId);
+
+    function getType() external view returns (ObjectType objectType);
+
+    function getData() external view returns (bytes memory data);
+
+    function isRegisterable() external pure returns (bool);
 
+    function getInitialOwner() external view returns (address initialOwner);
+
+    function isRegistered() external view returns (bool);
+}
+
+interface IRegistry {
     struct RegistryInfo {
-        uint256 nftId;
-        uint256 parentNftId;
-        uint256 objectType;
+        NftId nftId;
+        NftId parentNftId;
+        ObjectType objectType;
         address objectAddress;
         address initialOwner;
     }
 
-    function TOKEN() external pure returns(uint256);
-    function INSTANCE() external pure returns(uint256);
-    function PRODUCT() external pure returns(uint256);
-    function ORACLE() external pure returns(uint256);
-    function POOL() external pure returns(uint256);
-    function POLICY() external pure returns(uint256);
-    function BUNDLE() external pure returns(uint256);
+    function register(address objectAddress) external returns (NftId nftId);
 
-    function register(address objectAddress) external returns(uint256 nftId);
     function registerObjectForInstance(
-        uint256 parentNftid,
-        uint256 objectType,
+        NftId parentNftid,
+        ObjectType objectType,
         address initialOwner
-    )
-        external returns(uint256 nftId);
+    ) external returns (NftId nftId);
+
+    function getObjectCount() external view returns (uint256);
+
+    function getNftId(
+        address objectAddress
+    ) external view returns (NftId nftId);
 
-    function getObjectCount() external view returns(uint256);
+    function getInfo(
+        NftId nftId
+    ) external view returns (RegistryInfo memory info);
 
-    function getNftId(address objectAddress) external view returns(uint256 nftId);
-    function getInfo(uint256 nftId) external view returns(RegistryInfo memory info);
-    function getOwner(uint256 nftId) external view returns(address ownerAddress);
+    function getOwner(NftId nftId) external view returns (address ownerAddress);
 
-    function isRegistered(address objectAddress) external view returns(bool);
+    function isRegistered(address objectAddress) external view returns (bool);
 
-    function getNftAddress() external view returns(address nft);
+    function getNftAddress() external view returns (address nft);
 }