41 lines
2.0 KiB
Solidity
41 lines
2.0 KiB
Solidity
|
// SPDX-License-Identifier: MIT
|
||
|
|
pragma solidity ^0.8.20;
|
||
|
|
|
||
|
|
import {Script, console} from "forge-std/Script.sol";
|
||
|
|
import {Chain138MainnetCheckpoint} from "../../contracts/mainnet-checkpoint/Chain138MainnetCheckpoint.sol";
|
||
|
|
import {ExtensionIds} from "../../contracts/mainnet-checkpoint/libraries/ExtensionIds.sol";
|
||
|
|
import {TokenTransferFilterExtension} from "../../contracts/mainnet-checkpoint/extensions/TokenTransferFilterExtension.sol";
|
||
|
|
import {ICheckpointExtension} from "../../contracts/mainnet-checkpoint/interfaces/ICheckpointExtension.sol";
|
||
|
|
|
||
|
|
/// @notice Deploy fixed TokenTransferFilterExtension and swap on hub (revoke + register).
|
||
|
|
contract ReplaceTokenTransferFilterExtension is Script {
|
||
|
|
function run() external {
|
||
|
|
uint256 pk = vm.envUint("PRIVATE_KEY");
|
||
|
|
address hubAddr = vm.envAddress("CHAIN138_MAINNET_CHECKPOINT_PROXY");
|
||
|
|
Chain138MainnetCheckpoint hub = Chain138MainnetCheckpoint(hubAddr);
|
||
|
|
|
||
|
|
vm.startBroadcast(pk);
|
||
|
|
TokenTransferFilterExtension filter = new TokenTransferFilterExtension();
|
||
|
|
console.log("New TokenTransferFilter:", address(filter));
|
||
|
|
|
||
|
|
bool allowNative = vm.envOr("CHECKPOINT_TOKEN_FILTER_ALLOW_NATIVE", true);
|
||
|
|
uint256 minNative = vm.envOr("CHECKPOINT_TOKEN_FILTER_MIN_NATIVE_WEI", uint256(0));
|
||
|
|
filter.setAllowNative(allowNative, minNative);
|
||
|
|
|
||
|
|
bytes32 id = ExtensionIds.TOKEN_FILTER;
|
||
|
|
(, , bool wasActive) = hub.getExtension(id);
|
||
|
|
if (!wasActive) {
|
||
|
|
hub.setExtensionActive(id, true);
|
||
|
|
}
|
||
|
|
hub.revokeExtension(id);
|
||
|
|
uint32 hooks = ICheckpointExtension(address(filter)).HOOK_BEFORE_SUBMIT()
|
||
|
|
| ICheckpointExtension(address(filter)).HOOK_AFTER_SUBMIT()
|
||
|
|
| ICheckpointExtension(address(filter)).HOOK_ON_CCIP()
|
||
|
|
| ICheckpointExtension(address(filter)).HOOK_VERIFY_LEAF();
|
||
|
|
hub.registerExtension(id, address(filter), hooks);
|
||
|
|
hub.setExtensionActive(id, true);
|
||
|
|
console.log("TOKEN_FILTER registered and active");
|
||
|
|
vm.stopBroadcast();
|
||
|
|
}
|
||
|
|
}
|