stash to remove auto scaling

axelarnetwork · Jan 9, 2025 · fb0464f · fb0464f
1 parent 98365e4
commit fb0464f
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/contracts/InterchainTokenFactory.sol b/contracts/InterchainTokenFactory.sol
@@ -27,6 +27,7 @@ contract InterchainTokenFactory is IInterchainTokenFactory, ITokenManagerType, M
     bytes32 internal constant PREFIX_CANONICAL_TOKEN_SALT = keccak256('canonical-token-salt');
     bytes32 internal constant PREFIX_INTERCHAIN_TOKEN_SALT = keccak256('interchain-token-salt');
     bytes32 internal constant PREFIX_DEPLOY_APPROVAL = keccak256('deploy-approval');
+    bytes32 internal constant PREFIX_CUSTOM_TOKEN_SALT = keccak256('custom-token-salt');
     address private constant TOKEN_FACTORY_DEPLOYER = address(0);
 
     IInterchainTokenService public immutable interchainTokenService;
@@ -491,6 +492,53 @@ contract InterchainTokenFactory is IInterchainTokenFactory, ITokenManagerType, M
         tokenId = deployRemoteCanonicalInterchainToken(originalTokenAddress, destinationChain, gasValue);
     }
 
+        function linkedTokenDeploySalt(address deployer, bytes32 salt) public view returns (bytes32 deploySalt) {
+        deploySalt = keccak256(abi.encode(PREFIX_CUSTOM_TOKEN_SALT, chainNameHash, deployer, salt));
+    }
+
+    function linkedTokenId(address deployer, bytes32 salt) external view returns (bytes32 tokenId) {
+        bytes32 deploySalt = linkedTokenDeploySalt(deployer, salt);
+        tokenId = _interchainTokenId(deploySalt);
+    }
+
+    function registerCustomToken(
+        bytes32 salt,
+        address tokenAddress,
+        TokenManagerType tokenManagerType,
+        address operator,
+        uint256 gasValue
+    ) external payable returns (bytes32 tokenId) {
+        bytes32 deploySalt = linkedTokenDeploySalt(msg.sender, salt);
+        bytes memory operatorBytes = '';
+        string memory currentChain = '';
+        if (operator != address(0)) {
+            operatorBytes = operator.toBytes();
+        }
+
+        tokenId = interchainTokenService.linkToken(deploySalt, currentChain, tokenAddress.toBytes(), tokenManagerType, operatorBytes, 0);
+
+        interchainTokenService.registerTokenMetadata{ value: gasValue }(tokenAddress, gasValue);
+    }
+
+    function linkToken(
+        bytes32 salt,
+        string calldata destinationChain,
+        bytes calldata destinationTokenAddress,
+        TokenManagerType tokenManagerType,
+        bytes calldata linkParams,
+        uint256 gasValue
+    ) external payable returns (bytes32 tokenId) {
+        bytes32 deploySalt = linkedTokenDeploySalt(msg.sender, salt);
+        tokenId = interchainTokenService.linkToken(
+            deploySalt,
+            destinationChain,
+            destinationTokenAddress,
+            tokenManagerType,
+            linkParams,
+            gasValue
+        );
+    }
+
     /********************\
     |* Pure Key Getters *|
     \********************/