Merge pull request #48 from bugout-dev/terminus-controller

Terminus controller

Author: zomglings

contracts/terminus/TerminusPermissions.sol

@@ -0,0 +1,41 @@
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * Authors: Moonstream Engineering (engineering@moonstream.to)
+ * GitHub: https://github.com/bugout-dev/dao
+ *
+ */
+
+import "@openzeppelin-contracts/contracts/access/Ownable.sol";
+import "./TerminusFacet.sol";
+
+pragma solidity ^0.8.9;
+
+abstract contract TerminusPermissions {
+    function _holdsPoolToken(
+        address terminusAddress,
+        uint256 poolId,
+        uint256 _amount
+    ) internal view returns (bool) {
+        TerminusFacet terminus = TerminusFacet(terminusAddress);
+        return terminus.balanceOf(msg.sender, poolId) >= _amount;
+    }
+
+    modifier holdsPoolToken(address terminusAddress, uint256 poolId) {
+        require(
+            _holdsPoolToken(terminusAddress, poolId, 1),
+            "TerminusPermissions.holdsPoolToken: Sender doens't hold  pool tokens"
+        );
+        _;
+    }
+
+    modifier spendsPoolToken(address terminusAddress, uint256 poolId) {
+        require(
+            _holdsPoolToken(terminusAddress, poolId, 1),
+            "TerminusPermissions.spendsPoolToken: Sender doens't hold  pool tokens"
+        );
+        TerminusFacet terminusContract = TerminusFacet(terminusAddress);
+        terminusContract.burn(msg.sender, poolId, 1);
+        _;
+    }
+}

contracts/terminus/TokenDrainerFacet.sol

@@ -0,0 +1,63 @@
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * Authors: Moonstream Engineering (engineering@moonstream.to)
+ * GitHub: https://github.com/bugout-dev/dao
+ *
+ */
+pragma solidity ^0.8.9;
+
+import "@openzeppelin-contracts/contracts/token/ERC1155/IERC1155Receiver.sol";
+import "@openzeppelin-contracts/contracts/token/ERC721/IERC721Receiver.sol";
+import "@openzeppelin-contracts/contracts/token/ERC20/IERC20.sol";
+import "@openzeppelin-contracts/contracts/token/ERC1155/IERC1155.sol";
+
+import "../diamond/libraries/LibDiamond.sol";
+
+contract TokenDrainerFacet {
+    function drainERC20(address tokenAddress, address receiverAddress)
+        external
+    {
+        uint256 balance = IERC20(tokenAddress).balanceOf(address(this));
+        withdrawERC20(tokenAddress, balance, receiverAddress);
+    }
+
+    function withdrawERC20(
+        address tokenAddress,
+        uint256 amount,
+        address receiverAddress
+    ) public {
+        LibDiamond.enforceIsContractOwner();
+        IERC20 token = IERC20(tokenAddress);
+        token.transfer(receiverAddress, amount);
+    }
+
+    function drainERC1155(
+        address tokenAddress,
+        uint256 tokenId,
+        address receiverAddress
+    ) external {
+        uint256 balance = IERC1155(tokenAddress).balanceOf(
+            address(this),
+            tokenId
+        );
+        withdrawERC1155(tokenAddress, tokenId, balance, receiverAddress);
+    }
+
+    function withdrawERC1155(
+        address tokenAddress,
+        uint256 tokenId,
+        uint256 amount,
+        address receiverAddress
+    ) public {
+        LibDiamond.enforceIsContractOwner();
+        IERC1155 token = IERC1155(tokenAddress);
+        token.safeTransferFrom(
+            address(this),
+            receiverAddress,
+            tokenId,
+            amount,
+            ""
+        );
+    }
+}

contracts/terminus/controller/LibTerminusController.sol

@@ -0,0 +1,36 @@
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * Authors: Moonstream Engineering (engineering@moonstream.to)
+ * GitHub: https://github.com/bugout-dev/dao
+ *
+ */
+
+pragma solidity ^0.8.9;
+
+struct TerminusPool {
+    address terminusAddress;
+    uint256 poolId;
+}
+
+library LibTerminusController {
+    bytes32 constant TERMINUS_CONTROLLER_STORAGE_POSITION =
+        keccak256("moonstreamdao.eth.storage.terminus.controller");
+
+    struct TerminusControllerStorage {
+        address terminusAddress;
+        TerminusPool terminusMainAdminPool;
+        mapping(uint256 => TerminusPool) poolController;
+    }
+
+    function terminusControllerStorage()
+        internal
+        pure
+        returns (TerminusControllerStorage storage es)
+    {
+        bytes32 position = TERMINUS_CONTROLLER_STORAGE_POSITION;
+        assembly {
+            es.slot := position
+        }
+    }
+}