@etherisc/gif-next 0.0.2-a0d749a → 0.0.2-a1053ad-624

package/contracts/registry/Registry.sol

@@ -1,209 +1,423 @@
 // SPDX-License-Identifier: Apache-2.0
-pragma solidity ^0.8.19;
+pragma solidity ^0.8.20;
+
+import {IRegisterable} from "../shared/IRegisterable.sol";
+import {IService} from "../instance/base/IService.sol";
+
+import {ChainNft} from "./ChainNft.sol";
+import {IRegistry} from "./IRegistry.sol";
+import {NftId, toNftId, zeroNftId, NftIdLib} from "../types/NftId.sol";
+import {Version, VersionPart, VersionLib} from "../types/Version.sol";
+import {ObjectType, PROTOCOL, REGISTRY, TOKEN, SERVICE, INSTANCE, STAKE, PRODUCT, DISTRIBUTION, ORACLE, POOL, POLICY, BUNDLE} from "../types/ObjectType.sol";
+import {ITransferInterceptor} from "./ITransferInterceptor.sol";
+
+import {ERC165} from "../shared/ERC165.sol";
+
+
+// IMPORTANT
+// Each NFT minted by registry is accosiated with:
+// 1) NFT owner
+// 2) registred contract OR object stored in registered (parent) contract
+// Four registration flows:
+// 1) non IRegisterable address by registryOwner (TOKEN)
+// 2) IRegisterable address by registryOwner (SERVICE)
+// 3) IRegisterable address by approved service (INSTANCE, COMPONENT)
+// 4) state object by approved service (POLICY, BUNDLE, STAKE)
+
+contract Registry is
+    ERC165,
+    IRegistry
+{
+    // register
+    error NotRegistryService();
+    error ZeroParentAddress();
+    error ContractAlreadyRegistered(address objectAddress);
+    error InvalidServiceVersion(VersionPart majorVersion);
+    error ServiceNameAlreadyRegistered(string name, VersionPart majorVersion);
+            
+    // approve
+    error NotOwner();
+    error NotRegisteredContract(NftId registrarNftId);
+    error NotService(NftId registrarNftId);
+    error InvalidTypesCombination(ObjectType objectType, ObjectType parentType);
+
+    uint256 public constant MAJOR_VERSION_MIN = 3;
+    address public constant NFT_LOCK_ADDRESS = address(0x1);
+    uint256 public constant REGISTRY_SERVICE_TOKEN_SEQUENCE_ID = 3;
+    string public constant EMPTY_URI = "";
+
+    mapping(NftId nftId => ObjectInfo info) internal _info;
+    mapping(address object => NftId nftId) internal _nftIdByAddress;
+
+    mapping(NftId registrator => mapping(
+            ObjectType objectType => bool)) internal _isApproved;
+
+    mapping(ObjectType objectType => mapping(
+            ObjectType parentType => bool)) internal _isValidContractCombination;
+
+    mapping(ObjectType objectType => mapping(
+            ObjectType parentType => bool)) internal _isValidObjectCombination;
+
+    mapping(NftId nftId => string name) internal _string;
+    mapping(bytes32 serviceNameHash => mapping(
+            VersionPart majorVersion => address service)) internal _service;
 
-import {IChainNft} from "./IChainNft.sol";
-import {IRegistry, IRegistryLinked, IRegisterable} from "./IRegistry.sol";
-import {NftId, toNftId} from "../types/NftId.sol";
-import {NftIdLib} from "../types/NftId.sol";
+    NftId internal _registryNftId;
+    NftId internal _serviceNftId; // set in stone upon registry creation
+    ChainNft internal _chainNft;
 
-contract RegistryLinked is IRegistryLinked {
-    IRegistry internal _registry;
 
-    constructor(address registry) {
-        _registry = IRegistry(registry);
+    modifier onlyOwner() {
+        if(msg.sender != getOwner()) {
+            revert NotOwner();
+        }
+        _;
     }
 
-    function getRegistry() external view override returns (IRegistry registry) {
-        return _registry;
+    modifier onlyRegistryService() {
+        
+        if(msg.sender != _info[_serviceNftId].objectAddress) {
+            revert NotRegistryService();
+        }
+        _;
     }
-}
 
-abstract contract Registerable is RegistryLinked, IRegisterable {
-    using NftIdLib for NftId;
+    constructor(address registryOwner, VersionPart majorVersion)
+    {
+        require(registryOwner > address(0), "Registry: registry owner is 0");
+        require(majorVersion.toInt() == MAJOR_VERSION_MIN, "Registry: initial major version of registry service is not MAJOR_VERSION_MIN");
 
-    address private _initialOwner;
+        // deploy NFT 
+        _chainNft = new ChainNft(address(this));// adds 10kb to deployment size
 
-    constructor(address registry) RegistryLinked(registry) {
-        _initialOwner = msg.sender;
-    }
+        // initial registry setup
+        _registerProtocol();
+        _registerRegistry(registryOwner);
+        _registerRegistryService(registryOwner);
 
-    // getType, getData and register need to be implemented by concrete contract
+        // set object parent relations
+        _setupValidObjectParentCombinations();
 
-    function isRegisterable() external pure override returns (bool) {
-        return true;
+        _registerInterface(type(IRegistry).interfaceId);
     }
 
-    function getInitialOwner() public view override returns (address deployer) {
-        return _initialOwner;
-    }
+    /// @dev registry protects only against tampering existing records, registering with invalid types pairs and 0 parent address
+    // TODO service registration means its approval for some type?
+    // TODO registration of precompile addresses
+    function register(ObjectInfo memory info)
+        external
+        onlyRegistryService
+        returns(NftId nftId)
+    {
+        ObjectType objectType = info.objectType;
+        NftId parentNftId = info.parentNftId;
+        ObjectInfo memory parentInfo = _info[parentNftId];
+        ObjectType parentType = parentInfo.objectType; // see function header
+        address parentAddress = parentInfo.objectAddress;
+
+        // parent is contract -> need to check? -> check before minting
+        // special case: global registry nft as parent when not on mainnet -> global registry address is 0
+        // special case: when parentNftId == _chainNft.mint(), check for zero parent address before mint
+        // special case: when parentNftId == _chainNft.mint() && objectAddress == initialOwner
+        if(parentAddress == address(0)) {
+            revert ZeroParentAddress();
+        }
 
-    function isRegistered() public view override returns (bool) {
-        NftId nftId = _registry.getNftId(address(this));
-        return nftId.gtz();
-    }
+        address interceptor = _getInterceptor(info.isInterceptor, info.objectAddress, parentInfo.isInterceptor, parentAddress);
 
-    function getNftId() public view override returns (NftId nftId) {
-        return _registry.getNftId(address(this));
-    }
+        // TODO does external call
+        uint256 mintedTokenId = _chainNft.mint(
+            info.initialOwner,
+            interceptor,
+            EMPTY_URI);
+        nftId = toNftId(mintedTokenId);
+
+        // TODO move nftId out of info struct
+        // getters by nftId -> return struct without nftId
+        // getters by address -> return nftId AND struct
+        info.nftId = nftId;
+        _info[nftId] = info;
+
+        if(info.objectAddress > address(0)) 
+        {
+            // TODO if need to add types latter -> at least call this check from registry service
+            // parent is registered + object-parent types are valid
+            if(_isValidContractCombination[objectType][parentType] == false) {
+                revert InvalidTypesCombination(objectType, parentType);
+            }
+
+            address contractAddress = info.objectAddress;
 
-    function getOwner() public view override returns (address owner) {
-        NftId id = _registry.getNftId(address(this));
-        owner = _registry.getOwner(id);
-        return owner != address(0) ? owner : _initialOwner;
+            if(_nftIdByAddress[contractAddress].gtz()) { 
+                revert ContractAlreadyRegistered(contractAddress);
+            }
+
+            _nftIdByAddress[contractAddress] = nftId;
+
+            // special case
+            if(objectType == SERVICE()) {
+                _registerService(info);
+            }
+        }
+        else
+        {
+            if(_isValidObjectCombination[objectType][parentType] == false) {
+                revert InvalidTypesCombination(objectType, parentType);
+            }
+        }
+
+        emit LogRegistration(info);
     }
-}
 
-contract Registry is IRegistry {
-    using NftIdLib for NftId;
-    string public constant EMPTY_URI = "";
+    // from IRegistry
+    function getObjectCount() external view override returns (uint256) {
+        
+        return _chainNft.totalSupply();
+    }
 
-    mapping(NftId nftId => RegistryInfo info) private _info;
-    mapping(NftId nftId => address owner) private _owner;
-    mapping(address object => NftId nftId) private _nftIdByAddress;
+    function getNftId(address object) external view override returns (NftId id) {
+        return _nftIdByAddress[object];
+    }
 
-    IChainNft private _chainNft;
+    function ownerOf(NftId nftId) public view override returns (address) {
+        return _chainNft.ownerOf(nftId.toInt());
+    }
 
-    function initialize(address chainNft) external {
-        require(
-            address(_chainNft) == address(0),
-            "ERROR:REG-001:ALREADY_INITIALIZED"
-        );
-        _chainNft = IChainNft(chainNft);
+    function ownerOf(address contractAddress) public view returns (address) {
+        
+        return _chainNft.ownerOf(_nftIdByAddress[contractAddress].toInt());
     }
 
-    function TOKEN() public pure override returns (uint256) {
-        return 30;
+    function getObjectInfo(NftId nftId) external view override returns (ObjectInfo memory) {
+        return _info[nftId];
     }
 
-    function INSTANCE() public pure override returns (uint256) {
-        return 40;
+    function getObjectInfo(address object) external view override returns (ObjectInfo memory) {
+        
+        return _info[_nftIdByAddress[object]];
     }
 
-    function PRODUCT() public pure override returns (uint256) {
-        return 50;
+    function isRegistered(NftId nftId) public view override returns (bool) {
+        return _info[nftId].objectType.gtz();
     }
 
-    function ORACLE() public pure override returns (uint256) {
-        return 60;
+    function isRegistered(address object) external view override returns (bool) {
+        return _nftIdByAddress[object].gtz();
     }
 
-    function POOL() public pure override returns (uint256) {
-        return 70;
+    function getServiceName(NftId nftId) external view returns (string memory) {
+        return _string[nftId];
     }
 
-    function POLICY() public pure override returns (uint256) {
-        return 80;
+    // special case to retrive a gif service
+    function getServiceAddress(
+        string memory serviceName, 
+        VersionPart majorVersion
+    ) external view returns (address)
+    {
+        bytes32 serviceNameHash = keccak256(abi.encode(serviceName));
+        return _service[serviceNameHash][majorVersion];
     }
 
-    function BUNDLE() public pure override returns (uint256) {
-        return 90;
+    function getChainNft() external view override returns (ChainNft) {
+        return _chainNft;
     }
 
-    function register(
-        address objectAddress
-    ) external override returns (NftId nftId) {
-        require(
-            _nftIdByAddress[objectAddress].eqz(),
-            "ERROR:REG-002:ALREADY_REGISTERED"
-        );
+    function getOwner() public view returns (address owner) {
+        return ownerOf(address(this));
+    }
 
-        IRegisterable registerable = IRegisterable(objectAddress);
-        require(
-            registerable.isRegisterable(),
-            "ERROR:REG-003:NOT_REGISTERABLE"
-        );
+    // Internals
 
-        // check parent exists (for objects not instances)
-        if (registerable.getType() != INSTANCE()) {
-            RegistryInfo memory parentInfo = _info[
-                registerable.getParentNftId()
-            ];
-            require(parentInfo.nftId.gtz(), "ERROR:REG-004:PARENT_NOT_FOUND");
-            // check validity of parent relation, valid relations are
-            // policy -> product, bundle -> pool, product -> instance, pool -> instance
+    function _registerService(ObjectInfo memory info)
+        internal
+    {
+        (
+            string memory serviceName,
+            VersionPart majorVersion
+        ) = abi.decode(info.data, (string, VersionPart));
+        bytes32 serviceNameHash = keccak256(abi.encode(serviceName));
+
+        // TODO MUST guarantee consistency of service.getVersion() and majorVersion here
+        // TODO update _serviceNftId when registryService with new major version is registered? -> it is fixed in current setup -> can lock up
+        // TODO do not use names -> each object type is registered by corresponding service -> conflicting with approve()
+        if(
+            majorVersion.toInt() < MAJOR_VERSION_MIN ||
+            (majorVersion.toInt() > MAJOR_VERSION_MIN &&
+            _service[serviceNameHash][VersionLib.toVersionPart(majorVersion.toInt() - 1)] == address(0) )
+        ) {
+            revert InvalidServiceVersion(majorVersion);
+        }
+        
+        if(_service[serviceNameHash][majorVersion] != address(0)) {
+            revert ServiceNameAlreadyRegistered(serviceName, majorVersion);
         }
 
-        uint256 mintedTokenId = _chainNft.mint(
-            registerable.getInitialOwner(),
-            EMPTY_URI
-        );
-        nftId = toNftId(mintedTokenId);
+        _string[info.nftId] = serviceName;
+        _service[serviceNameHash][majorVersion] = info.objectAddress;
 
-        RegistryInfo memory info = RegistryInfo(
-            nftId,
-            registerable.getParentNftId(),
-            registerable.getType(),
-            objectAddress,
-            registerable.getInitialOwner()
-        );
+        emit LogServiceNameRegistration(serviceName, majorVersion); 
+    }
 
-        _info[nftId] = info;
-        _nftIdByAddress[objectAddress] = nftId;
+    /// @dev obtain interceptor address for this nft if applicable, address(0) otherwise
+    function _getInterceptor(
+        bool isInterceptor, 
+        address objectAddress,
+        bool parentIsInterceptor,
+        address parentObjectAddress
+    )
+        internal 
+        view 
+        returns (address interceptor) 
+    {
+        if (objectAddress == address(0)) {
+            if (parentIsInterceptor) {
+                return parentObjectAddress;
+            } else {
+                return address(0);
+            }
+        }
+
+        if (isInterceptor) {
+            return objectAddress;
+        }
 
-        // add logging
+        return address(0);
     }
 
-    function registerObjectForInstance(
-        NftId parentNftId,
-        uint256 objectType,
-        address initialOwner
-    )
-        external
-        override
-        returns (
-            // TODO add onlyRegisteredInstance
-            NftId nftId
-        )
+    // Internals called only in constructor
+
+    /// @dev protocol registration used to anchor the dip ecosystem relations
+    function _registerProtocol() 
+        internal
     {
-        // TODO add more validation
-        require(
-            objectType == POLICY() || objectType == BUNDLE(),
-            "ERROR:REG-005:TYPE_INVALID"
+        uint256 protocolId = _chainNft.PROTOCOL_NFT_ID();
+        NftId protocolNftId = toNftId(protocolId);
+
+        _chainNft.mint(NFT_LOCK_ADDRESS, protocolId);
+
+        _info[protocolNftId] = ObjectInfo(
+            protocolNftId,
+            zeroNftId(), // parent
+            PROTOCOL(),
+            false, // isInterceptor
+            address(0), // objectAddress
+            NFT_LOCK_ADDRESS,// initialOwner
+            ""
         );
+    }
 
-        uint256 mintedTokenId = _chainNft.mint(initialOwner, EMPTY_URI);
-        nftId = toNftId(mintedTokenId);
+    /// @dev registry registration
+    /// might also register the global registry when not on mainnet
+    function _registerRegistry(address registryOwner) 
+        internal
+    {
+        uint256 registryId = _chainNft.calculateTokenId(2);
+        NftId registryNftId = toNftId(registryId);
+
+        NftId parentNftId;
+
+        if(registryId != _chainNft.GLOBAL_REGISTRY_ID()) 
+        {// we're not the global registry
+            _registerGlobalRegistry();
+            parentNftId = toNftId(_chainNft.GLOBAL_REGISTRY_ID());
+        }
+        else 
+        {// we are global registry
+            parentNftId = toNftId(_chainNft.PROTOCOL_NFT_ID());
+        }
+
+        _chainNft.mint(registryOwner, registryId);
 
-        RegistryInfo memory info = RegistryInfo(
-            nftId,
+        _info[registryNftId] = ObjectInfo(
+            registryNftId,
             parentNftId,
-            objectType,
-            address(0),
-            initialOwner
+            REGISTRY(),
+            false, // isInterceptor
+            address(this), 
+            registryOwner,
+            "" 
         );
+        _nftIdByAddress[address(this)] = registryNftId;
+        _registryNftId = registryNftId;
+    }
 
-        _info[nftId] = info;
 
-        // add logging
-    }
+    /// @dev global registry registration for non mainnet registries
+    function _registerGlobalRegistry() 
+        internal
+    {
+        uint256 globalRegistryId = _chainNft.GLOBAL_REGISTRY_ID();
 
-    function getObjectCount() external view override returns (uint256) {
-        return _chainNft.totalSupply();
-    }
+        _chainNft.mint(NFT_LOCK_ADDRESS, globalRegistryId);
 
-    function getNftId(
-        address object
-    ) external view override returns (NftId id) {
-        return _nftIdByAddress[object];
-    }
+        NftId globalRegistryNftId = toNftId(globalRegistryId);
 
-    function isRegistered(
-        address object
-    ) external view override returns (bool) {
-        return _nftIdByAddress[object].gtz();
+        _info[globalRegistryNftId] = ObjectInfo(
+            globalRegistryNftId,
+            toNftId(_chainNft.PROTOCOL_NFT_ID()), // parent
+            REGISTRY(),
+            false, // isInterceptor
+            address(0), // objectAddress
+            NFT_LOCK_ADDRESS, // initialOwner
+            "" // data
+        );
     }
 
-    function getInfo(
-        NftId nftId
-    ) external view override returns (RegistryInfo memory info) {
-        return _info[nftId];
-    }
+    function _registerRegistryService(address registryOwner)
+        internal
+    {
+        uint256 serviceId = _chainNft.calculateTokenId(REGISTRY_SERVICE_TOKEN_SEQUENCE_ID);
+        NftId serviceNftId = toNftId(serviceId);        
+
+        _chainNft.mint(registryOwner, serviceId);
+
+        _info[serviceNftId] = ObjectInfo(
+            serviceNftId,
+            _registryNftId,
+            SERVICE(),
+            false, // isInterceptor
+            msg.sender, // service deploys registry
+            registryOwner, // initialOwner,
+            "" 
+        );
 
-    function getOwner(NftId nftId) external view override returns (address) {
-        return _chainNft.ownerOf(nftId.toInt());
+        _nftIdByAddress[msg.sender] = serviceNftId;
+
+        string memory serviceName = "RegistryService";
+        bytes32 serviceNameHash = keccak256(abi.encode(serviceName));
+        _service[serviceNameHash][VersionLib.toVersionPart(MAJOR_VERSION_MIN)] = msg.sender;
+        _string[serviceNftId] = serviceName;
+        _serviceNftId = serviceNftId;
     }
 
-    function getNftAddress() external view override returns (address nft) {
-        return address(_chainNft);
+    /// @dev defines which object - parent types relations are allowed to register
+    // IMPORTANT: 
+    // 1) EACH object type MUST have only one parent type across ALL mappings
+    // 2) DO NOT use object type (e.g. POLCY, BUNDLE, STAKE) as parent type
+    // 3) DO NOT use REGISTRY as object type
+    // 2) DO NOT use PROTOCOL and "zeroObjectType"
+    function _setupValidObjectParentCombinations() 
+        internal 
+    {
+        // registry as parent, ONLY registry owner
+        _isValidContractCombination[TOKEN()][REGISTRY()] = true;
+        _isValidContractCombination[SERVICE()][REGISTRY()] = true;
+
+        // registry as parent, ONLY approved
+        _isValidContractCombination[INSTANCE()][REGISTRY()] = true;
+
+        // instance as parent, ONLY approved
+        _isValidContractCombination[PRODUCT()][INSTANCE()] = true;
+        _isValidContractCombination[DISTRIBUTION()][INSTANCE()] = true;
+        _isValidContractCombination[ORACLE()][INSTANCE()] = true;
+        _isValidContractCombination[POOL()][INSTANCE()] = true;
+
+        // product as parent, ONLY approved
+        _isValidObjectCombination[POLICY()][PRODUCT()] = true;
+
+        // pool as parent, ONLY approved
+        _isValidObjectCombination[BUNDLE()][POOL()] = true;
+        _isValidObjectCombination[STAKE()][POOL()] = true;
     }
 }