chore(evm): refactor zkgm to fit max contract size

Author: hussein-aitlahcen

evm/contracts/apps/ucs/03-zkgm/Zkgm.sol

@@ -361,22 +361,14 @@ contract UCS03Zkgm is
         return address(ibcHandler);
     }
 
-    function transferAndCall(
+    function internalSendOverChannel(
         uint32 channelId,
         bytes calldata receiver,
         address baseToken,
         uint256 baseAmount,
         bytes calldata quoteToken,
-        uint256 quoteAmount,
-        bytes calldata contractAddress,
-        bytes calldata contractCalldata,
-        uint64 timeoutHeight,
-        uint64 timeoutTimestamp,
-        bytes32 salt
-    ) public {
-        if (baseAmount == 0) {
-            revert ZkgmLib.ErrInvalidAmount();
-        }
+        uint256 quoteAmount
+    ) internal returns (Instruction memory) {
         uint256 origin = tokenOrigin[baseToken];
         // Verify the unwrap
         (address wrappedToken,) =
@@ -397,13 +389,14 @@ contract UCS03Zkgm is
             );
             channelBalance[channelId][address(baseToken)] += baseAmount;
         }
+
         // TODO: make this non-failable as it's not guaranteed to exist
-        Instruction[] memory instructions = new Instruction[](2);
         IERC20Metadata sentTokenMeta = IERC20Metadata(baseToken);
         string memory symbol = sentTokenMeta.symbol();
         string memory name = sentTokenMeta.name();
         uint8 decimals = sentTokenMeta.decimals();
-        instructions[0] = Instruction({
+
+        return Instruction({
             version: ZkgmLib.INSTR_VERSION_1,
             opcode: ZkgmLib.OP_FUNGIBLE_ASSET_ORDER,
             operand: ZkgmLib.encodeFungibleAssetOrder(
@@ -421,6 +414,28 @@ contract UCS03Zkgm is
                 })
             )
         });
+    }
+
+    function transferAndCall(
+        uint32 channelId,
+        bytes calldata receiver,
+        address baseToken,
+        uint256 baseAmount,
+        bytes calldata quoteToken,
+        uint256 quoteAmount,
+        bytes calldata contractAddress,
+        bytes calldata contractCalldata,
+        uint64 timeoutHeight,
+        uint64 timeoutTimestamp,
+        bytes32 salt
+    ) public {
+        if (baseAmount == 0) {
+            revert ZkgmLib.ErrInvalidAmount();
+        }
+        Instruction[] memory instructions = new Instruction[](2);
+        instructions[0] = internalSendOverChannel(
+            channelId, receiver, baseToken, baseAmount, quoteToken, quoteAmount
+        );
         instructions[1] = Instruction({
             version: ZkgmLib.INSTR_VERSION_0,
             opcode: ZkgmLib.OP_MULTIPLEX,
@@ -453,39 +468,6 @@ contract UCS03Zkgm is
         );
     }
 
-    function call(
-        uint32 channelId,
-        bytes calldata contractAddress,
-        bytes calldata contractCalldata,
-        uint64 timeoutHeight,
-        uint64 timeoutTimestamp,
-        bytes32 salt
-    ) public {
-        ibcHandler.sendPacket(
-            channelId,
-            timeoutHeight,
-            timeoutTimestamp,
-            ZkgmLib.encode(
-                ZkgmPacket({
-                    salt: keccak256(abi.encodePacked(msg.sender, salt)),
-                    path: 0,
-                    instruction: Instruction({
-                        version: ZkgmLib.INSTR_VERSION_0,
-                        opcode: ZkgmLib.OP_MULTIPLEX,
-                        operand: ZkgmLib.encodeMultiplex(
-                            Multiplex({
-                                sender: abi.encodePacked(msg.sender),
-                                eureka: true,
-                                contractAddress: contractAddress,
-                                contractCalldata: contractCalldata
-                            })
-                        )
-                    })
-                })
-            )
-        );
-    }
-
     function transferV2(
         uint32 channelId,
         bytes calldata receiver,
@@ -501,52 +483,11 @@ contract UCS03Zkgm is
         if (baseAmount == 0) {
             revert ZkgmLib.ErrInvalidAmount();
         }
-        uint256 origin = tokenOrigin[baseToken];
-        // Verify the unwrap
-        (address wrappedToken,) =
-            internalPredictWrappedTokenMemory(0, channelId, quoteToken);
-        // Only allow unwrapping if the quote asset is the unwrapped asset.
-        if (
-            ZkgmLib.lastChannelFromPath(origin) == channelId
-                && abi.encodePacked(baseToken).eq(abi.encodePacked(wrappedToken))
-        ) {
-            IZkgmERC20(baseToken).burn(msg.sender, baseAmount);
-        } else {
-            // We reset the origin, the asset will not be unescrowed on the destination
-            origin = 0;
-            // TODO: extract this as a step before verifying to allow for ERC777
-            // send hook
-            SafeERC20.safeTransferFrom(
-                IERC20(baseToken), msg.sender, address(this), baseAmount
-            );
-            channelBalance[channelId][address(baseToken)] += baseAmount;
-        }
-
-        // TODO: make calls to sentTokenMeta non-failable as it's not guaranteed to exist
-        IERC20Metadata sentTokenMeta = IERC20Metadata(baseToken);
-        uint256 nbOfInstructions = 1;
-        if (msg.value > 0) {
-            nbOfInstructions = 2;
-        }
-        Instruction[] memory instructions = new Instruction[](nbOfInstructions);
-        instructions[0] = Instruction({
-            version: ZkgmLib.INSTR_VERSION_1,
-            opcode: ZkgmLib.OP_FUNGIBLE_ASSET_ORDER,
-            operand: ZkgmLib.encodeFungibleAssetOrder(
-                FungibleAssetOrder({
-                    sender: abi.encodePacked(msg.sender),
-                    receiver: receiver,
-                    baseToken: abi.encodePacked(baseToken),
-                    baseTokenPath: origin,
-                    baseTokenSymbol: sentTokenMeta.symbol(),
-                    baseTokenName: sentTokenMeta.name(),
-                    baseTokenDecimals: sentTokenMeta.decimals(),
-                    baseAmount: baseAmount,
-                    quoteToken: quoteToken,
-                    quoteAmount: quoteAmount
-                })
-            )
-        });
+        Instruction[] memory instructions =
+            new Instruction[](msg.value > 0 ? 2 : 1);
+        instructions[0] = internalSendOverChannel(
+            channelId, receiver, baseToken, baseAmount, quoteToken, quoteAmount
+        );
         if (msg.value > 0) {
             weth.deposit{value: msg.value}();
             address wethBaseToken = address(weth);
@@ -1398,9 +1339,7 @@ contract UCS03Zkgm is
             delete inFlightPacket[packetHash];
         } else {
             ZkgmPacket calldata zkgmPacket = ZkgmLib.decode(ibcPacket.data);
-            timeoutInternal(
-                ibcPacket, relayer, zkgmPacket.instruction
-            );
+            timeoutInternal(ibcPacket, relayer, zkgmPacket.instruction);
         }
     }
 
@@ -1439,27 +1378,21 @@ contract UCS03Zkgm is
                 revert ZkgmLib.ErrUnsupportedVersion();
             }
             timeoutBatch(
-                ibcPacket,
-                relayer,
-                ZkgmLib.decodeBatch(instruction.operand)
+                ibcPacket, relayer, ZkgmLib.decodeBatch(instruction.operand)
             );
         } else if (instruction.opcode == ZkgmLib.OP_FORWARD) {
             if (instruction.version > ZkgmLib.INSTR_VERSION_0) {
                 revert ZkgmLib.ErrUnsupportedVersion();
             }
             timeoutForward(
-                ibcPacket,
-                relayer,
-                ZkgmLib.decodeForward(instruction.operand)
+                ibcPacket, relayer, ZkgmLib.decodeForward(instruction.operand)
             );
         } else if (instruction.opcode == ZkgmLib.OP_MULTIPLEX) {
             if (instruction.version > ZkgmLib.INSTR_VERSION_0) {
                 revert ZkgmLib.ErrUnsupportedVersion();
             }
             timeoutMultiplex(
-                ibcPacket,
-                relayer,
-                ZkgmLib.decodeMultiplex(instruction.operand)
+                ibcPacket, relayer, ZkgmLib.decodeMultiplex(instruction.operand)
             );
         } else {
             revert ZkgmLib.ErrUnknownOpcode();