Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feat/pump staking integration #78

Merged
merged 3 commits into from
Aug 30, 2024
Merged

Feat/pump staking integration #78

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
c578a61
Create decoder
crispymangoes Aug 16, 2024
a8b4226
Create PumpStaking Decoder and Sanitizer and test it
crispymangoes Aug 16, 2024
8ce6e62
Merge branch 'dev/august-2024' of https://github.com/Se7en-Seas/borin…
crispymangoes Aug 30, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 28 additions & 0 deletions src/base/DecodersAndSanitizers/Protocols/PumpStakingDecoderAndSanitizer.sol
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
// SPDX-License-Identifier: UNLICENSED
pragma solidity 0.8.21;

import {BaseDecoderAndSanitizer} from "src/base/DecodersAndSanitizers/BaseDecoderAndSanitizer.sol";

abstract contract PumpStakingDecoderAndSanitizer is BaseDecoderAndSanitizer {
//============================== PUMP STAKING ===============================

function stake(uint256 /*amount*/ ) external pure virtual returns (bytes memory addressesFound) {
return addressesFound;
}

function unstakeRequest(uint256 /*amount*/ ) external pure virtual returns (bytes memory addressesFound) {
return addressesFound;
}

function claimSlot(uint8 /*slot*/ ) external pure virtual returns (bytes memory addressesFound) {
return addressesFound;
}

function claimAll() external pure virtual returns (bytes memory addressesFound) {
return addressesFound;
}

function unstakeInstant(uint256 /*amount*/ ) external pure virtual returns (bytes memory addressesFound) {
return addressesFound;
}
}
12 changes: 12 additions & 0 deletions src/base/DecodersAndSanitizers/PumpBtcDecoderAndSanitizer.sol
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
// SPDX-License-Identifier: UNLICENSED
pragma solidity 0.8.21;

import {BaseDecoderAndSanitizer} from "src/base/DecodersAndSanitizers/BaseDecoderAndSanitizer.sol";
import {PumpStakingDecoderAndSanitizer} from
"src/base/DecodersAndSanitizers/Protocols/PumpStakingDecoderAndSanitizer.sol";

contract PumpBtcDecoderAndSanitizer is BaseDecoderAndSanitizer, PumpStakingDecoderAndSanitizer {
constructor(address _boringVault) BaseDecoderAndSanitizer(_boringVault) {}

//============================== HANDLE FUNCTION COLLISIONS ===============================
}
305 changes: 305 additions & 0 deletions test/integrations/PumpStakingIntegration.t.sol
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,305 @@
// SPDX-License-Identifier: Apache-2.0
pragma solidity 0.8.21;

import {MainnetAddresses} from "test/resources/MainnetAddresses.sol";
import {BoringVault} from "src/base/BoringVault.sol";
import {ManagerWithMerkleVerification} from "src/base/Roles/ManagerWithMerkleVerification.sol";
import {SafeTransferLib} from "@solmate/utils/SafeTransferLib.sol";
import {FixedPointMathLib} from "@solmate/utils/FixedPointMathLib.sol";
import {ERC20} from "@solmate/tokens/ERC20.sol";
import {ERC4626} from "@solmate/tokens/ERC4626.sol";
import {
PumpBtcDecoderAndSanitizer,
PumpStakingDecoderAndSanitizer
} from "src/base/DecodersAndSanitizers/PumpBtcDecoderAndSanitizer.sol";
import {DecoderCustomTypes} from "src/interfaces/DecoderCustomTypes.sol";
import {RolesAuthority, Authority} from "@solmate/auth/authorities/RolesAuthority.sol";
import {MerkleTreeHelper} from "test/resources/MerkleTreeHelper/MerkleTreeHelper.sol";

import {Test, stdStorage, StdStorage, stdError, console} from "@forge-std/Test.sol";

contract PumpStakingIntegrationTest is Test, MerkleTreeHelper {
using SafeTransferLib for ERC20;
using FixedPointMathLib for uint256;
using stdStorage for StdStorage;

ManagerWithMerkleVerification public manager;
BoringVault public boringVault;
address public rawDataDecoderAndSanitizer;
RolesAuthority public rolesAuthority;

uint8 public constant MANAGER_ROLE = 1;
uint8 public constant STRATEGIST_ROLE = 2;
uint8 public constant MANGER_INTERNAL_ROLE = 3;
uint8 public constant ADMIN_ROLE = 4;
uint8 public constant BORING_VAULT_ROLE = 5;
uint8 public constant BALANCER_VAULT_ROLE = 6;

function setUp() external {
setSourceChainName("mainnet");
// Setup forked environment.
string memory rpcKey = "MAINNET_RPC_URL";
uint256 blockNumber = 20543218;

_startFork(rpcKey, blockNumber);

boringVault = new BoringVault(address(this), "Boring Vault", "BV", 18);

manager =
new ManagerWithMerkleVerification(address(this), address(boringVault), getAddress(sourceChain, "vault"));

rawDataDecoderAndSanitizer = address(new PumpBtcDecoderAndSanitizer(address(boringVault)));

setAddress(false, sourceChain, "boringVault", address(boringVault));
setAddress(false, sourceChain, "rawDataDecoderAndSanitizer", rawDataDecoderAndSanitizer);
setAddress(false, sourceChain, "manager", address(manager));
setAddress(false, sourceChain, "managerAddress", address(manager));
setAddress(false, sourceChain, "accountantAddress", address(1));

rolesAuthority = new RolesAuthority(address(this), Authority(address(0)));
boringVault.setAuthority(rolesAuthority);
manager.setAuthority(rolesAuthority);

// Setup roles authority.
rolesAuthority.setRoleCapability(
MANAGER_ROLE,
address(boringVault),
bytes4(keccak256(abi.encodePacked("manage(address,bytes,uint256)"))),
true
);
rolesAuthority.setRoleCapability(
MANAGER_ROLE,
address(boringVault),
bytes4(keccak256(abi.encodePacked("manage(address[],bytes[],uint256[])"))),
true
);

rolesAuthority.setRoleCapability(
STRATEGIST_ROLE,
address(manager),
ManagerWithMerkleVerification.manageVaultWithMerkleVerification.selector,
true
);
rolesAuthority.setRoleCapability(
MANGER_INTERNAL_ROLE,
address(manager),
ManagerWithMerkleVerification.manageVaultWithMerkleVerification.selector,
true
);
rolesAuthority.setRoleCapability(
ADMIN_ROLE, address(manager), ManagerWithMerkleVerification.setManageRoot.selector, true
);
rolesAuthority.setRoleCapability(
BORING_VAULT_ROLE, address(manager), ManagerWithMerkleVerification.flashLoan.selector, true
);
rolesAuthority.setRoleCapability(
BALANCER_VAULT_ROLE, address(manager), ManagerWithMerkleVerification.receiveFlashLoan.selector, true
);

// Grant roles
rolesAuthority.setUserRole(address(this), STRATEGIST_ROLE, true);
rolesAuthority.setUserRole(address(manager), MANGER_INTERNAL_ROLE, true);
rolesAuthority.setUserRole(address(this), ADMIN_ROLE, true);
rolesAuthority.setUserRole(address(manager), MANAGER_ROLE, true);
rolesAuthority.setUserRole(address(boringVault), BORING_VAULT_ROLE, true);
rolesAuthority.setUserRole(getAddress(sourceChain, "vault"), BALANCER_VAULT_ROLE, true);
}

function testPumpStakingIntegrationStake() external {
deal(getAddress(sourceChain, "WBTC"), address(boringVault), 10e8);

// approve
// Call deposit
// withdraw
// complete withdraw
ManageLeaf[] memory leafs = new ManageLeaf[](8);
_addLeafsForPumpStaking(leafs, getAddress(sourceChain, "pumpStaking"), getERC20(sourceChain, "WBTC"));

bytes32[][] memory manageTree = _generateMerkleTree(leafs);

manager.setManageRoot(address(this), manageTree[manageTree.length - 1][0]);

ManageLeaf[] memory manageLeafs = new ManageLeaf[](2);
manageLeafs[0] = leafs[0];
manageLeafs[1] = leafs[1];

bytes32[][] memory manageProofs = _getProofsUsingTree(manageLeafs, manageTree);

address[] memory targets = new address[](2);
targets[0] = getAddress(sourceChain, "WBTC");
targets[1] = getAddress(sourceChain, "pumpStaking");

bytes[] memory targetData = new bytes[](2);
targetData[0] = abi.encodeWithSignature(
"approve(address,uint256)", getAddress(sourceChain, "pumpStaking"), type(uint256).max
);
targetData[1] = abi.encodeWithSignature("stake(uint256)", 10e8);

address[] memory decodersAndSanitizers = new address[](2);
decodersAndSanitizers[0] = rawDataDecoderAndSanitizer;
decodersAndSanitizers[1] = rawDataDecoderAndSanitizer;

uint256[] memory values = new uint256[](2);

manager.manageVaultWithMerkleVerification(manageProofs, decodersAndSanitizers, targets, targetData, values);

assertEq(
getERC20(sourceChain, "pumpBTC").balanceOf(address(boringVault)),
10e8,
"BoringVault should have received 10 pumpBTC"
);
}

function testPumpStakingIntegrationUnstake() external {
// Spoof PumpStaking owner to allow for usntaking.
PumpStaking ps = PumpStaking(getAddress(sourceChain, "pumpStaking"));
address owner = ps.owner();
vm.startPrank(owner);
ps.setOnlyAllowStake(false);
ps.setOperator(address(this));
vm.stopPrank();

deal(getAddress(sourceChain, "pumpBTC"), address(boringVault), 10e8);
deal(getAddress(sourceChain, "WBTC"), address(this), 10e8);

// approve
// Call deposit
// withdraw
// complete withdraw
ManageLeaf[] memory leafs = new ManageLeaf[](8);
_addLeafsForPumpStaking(leafs, getAddress(sourceChain, "pumpStaking"), getERC20(sourceChain, "WBTC"));

bytes32[][] memory manageTree = _generateMerkleTree(leafs);

manager.setManageRoot(address(this), manageTree[manageTree.length - 1][0]);

ManageLeaf[] memory manageLeafs = new ManageLeaf[](1);
manageLeafs[0] = leafs[2];

bytes32[][] memory manageProofs = _getProofsUsingTree(manageLeafs, manageTree);

address[] memory targets = new address[](1);
targets[0] = getAddress(sourceChain, "pumpStaking");

bytes[] memory targetData = new bytes[](1);
targetData[0] = abi.encodeWithSignature("unstakeRequest(uint256)", 10e8);

address[] memory decodersAndSanitizers = new address[](1);
decodersAndSanitizers[0] = rawDataDecoderAndSanitizer;

uint256[] memory values = new uint256[](1);

manager.manageVaultWithMerkleVerification(manageProofs, decodersAndSanitizers, targets, targetData, values);

assertEq(
getERC20(sourceChain, "pumpBTC").balanceOf(address(boringVault)), 0, "BoringVault should have zero pumpBTC"
);

skip(10 days);

// Operator deposits claimable WBTC into PumpStaking.
getERC20(sourceChain, "WBTC").approve(getAddress(sourceChain, "pumpStaking"), 10e8);
ps.deposit(10e8);

uint256 beforeClaim = vm.snapshot();

// Finish claim using claimSlot(uint8)
manageLeafs[0] = leafs[3];
manageProofs = _getProofsUsingTree(manageLeafs, manageTree);
targets[0] = getAddress(sourceChain, "pumpStaking");
targetData[0] = abi.encodeWithSignature("claimSlot(uint8)", 2);
decodersAndSanitizers[0] = rawDataDecoderAndSanitizer;
values[0] = 0;
manager.manageVaultWithMerkleVerification(manageProofs, decodersAndSanitizers, targets, targetData, values);

assertEq(
getERC20(sourceChain, "WBTC").balanceOf(address(boringVault)), 10e8, "BoringVault should have 10e8 wBTC"
);

vm.revertTo(beforeClaim);

// Finish claim using claimAll()
manageLeafs[0] = leafs[4];
manageProofs = _getProofsUsingTree(manageLeafs, manageTree);
targets[0] = getAddress(sourceChain, "pumpStaking");
targetData[0] = abi.encodeWithSignature("claimAll()");
decodersAndSanitizers[0] = rawDataDecoderAndSanitizer;
values[0] = 0;
manager.manageVaultWithMerkleVerification(manageProofs, decodersAndSanitizers, targets, targetData, values);

assertEq(
getERC20(sourceChain, "WBTC").balanceOf(address(boringVault)), 10e8, "BoringVault should have 10e8 wBTC"
);
}

function testPumpStakingIntegrationUnstakeInstant() external {
// Spoof PumpStaking owner to allow for usntaking.
PumpStaking ps = PumpStaking(getAddress(sourceChain, "pumpStaking"));
address owner = ps.owner();
vm.prank(owner);
ps.setOnlyAllowStake(false);

deal(getAddress(sourceChain, "WBTC"), getAddress(sourceChain, "boringVault"), 10e8);

// approve
// Call deposit
// withdraw
// complete withdraw
ManageLeaf[] memory leafs = new ManageLeaf[](8);
_addLeafsForPumpStaking(leafs, getAddress(sourceChain, "pumpStaking"), getERC20(sourceChain, "WBTC"));

bytes32[][] memory manageTree = _generateMerkleTree(leafs);

manager.setManageRoot(address(this), manageTree[manageTree.length - 1][0]);

ManageLeaf[] memory manageLeafs = new ManageLeaf[](3);
manageLeafs[0] = leafs[0];
manageLeafs[1] = leafs[1];
manageLeafs[2] = leafs[5];

bytes32[][] memory manageProofs = _getProofsUsingTree(manageLeafs, manageTree);

address[] memory targets = new address[](3);
targets[0] = getAddress(sourceChain, "WBTC");
targets[1] = getAddress(sourceChain, "pumpStaking");
targets[2] = getAddress(sourceChain, "pumpStaking");

bytes[] memory targetData = new bytes[](3);
targetData[0] = abi.encodeWithSignature(
"approve(address,uint256)", getAddress(sourceChain, "pumpStaking"), type(uint256).max
);
targetData[1] = abi.encodeWithSignature("stake(uint256)", 10e8);
targetData[2] = abi.encodeWithSignature("unstakeInstant(uint256)", 10e8);

address[] memory decodersAndSanitizers = new address[](3);
decodersAndSanitizers[0] = rawDataDecoderAndSanitizer;
decodersAndSanitizers[1] = rawDataDecoderAndSanitizer;
decodersAndSanitizers[2] = rawDataDecoderAndSanitizer;

uint256[] memory values = new uint256[](3);

manager.manageVaultWithMerkleVerification(manageProofs, decodersAndSanitizers, targets, targetData, values);

assertEq(
getERC20(sourceChain, "pumpBTC").balanceOf(address(boringVault)), 0, "BoringVault should have zero pumpBTC"
);

assertEq(
getERC20(sourceChain, "WBTC").balanceOf(address(boringVault)), 9.7e8, "BoringVault should have 10e8 wBTC"
);
}

// ========================================= HELPER FUNCTIONS =========================================

function _startFork(string memory rpcKey, uint256 blockNumber) internal returns (uint256 forkId) {
forkId = vm.createFork(vm.envString(rpcKey), blockNumber);
vm.selectFork(forkId);
}
}

interface PumpStaking {
function owner() external view returns (address);
function setOnlyAllowStake(bool _onlyAllowStake) external;
function deposit(uint256 amount) external;
function setOperator(address newOperator) external;
}
4 changes: 4 additions & 0 deletions test/resources/ChainValues.sol
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,7 @@ contract ChainValues {
values[mainnet]["WBETH"] = 0xa2E3356610840701BDf5611a53974510Ae27E2e1.toBytes32();
values[mainnet]["UNIETH"] = 0xF1376bceF0f78459C0Ed0ba5ddce976F1ddF51F4.toBytes32();
values[mainnet]["CBETH"] = 0xBe9895146f7AF43049ca1c1AE358B0541Ea49704.toBytes32();
values[mainnet]["pumpBTC"] = 0xF469fBD2abcd6B9de8E169d128226C0Fc90a012e.toBytes32();
values[mainnet]["USD0"] = 0x73A15FeD60Bf67631dC6cd7Bc5B6e8da8190aCF5.toBytes32();
values[mainnet]["USD0_plus"] = 0x35D8949372D46B7a3D5A56006AE77B215fc69bC0.toBytes32();

Expand Down Expand Up @@ -702,6 +703,9 @@ contract ChainValues {

// Merkl
values[mainnet]["merklDistributor"] = 0x3Ef3D8bA38EBe18DB133cEc108f4D14CE00Dd9Ae.toBytes32();

// Pump Staking
values[mainnet]["pumpStaking"] = 0x1fCca65fb6Ae3b2758b9b2B394CB227eAE404e1E.toBytes32();
}

function _addBaseValues() private {
Expand Down
Loading