Processor frontend

blockchain/Dockerfile

@@ -1,2 +1,2 @@
-FROM ghcr.io/foundry-rs/foundry
-ENTRYPOINT [ "anvil", "--block-time", "5", "--host", "0.0.0.0" ]
+FROM ghcr.io/foundry-rs/foundry:v1.0.0
+ENTRYPOINT [ "anvil", "--host", "0.0.0.0", "--port", "80", "--chain-id", "74207", "--block-time", "5" ]

blockchain/apps/treasury/contracts/Treasury.sol

@@ -1,4 +1,4 @@
-// SPDX-License-Identifier: UNLICENSED
+// SPDX-License-Identifier: MIT
 pragma solidity ^0.8.9;
 
 import "@openzeppelin/contracts/interfaces/IERC20.sol";
@@ -7,24 +7,44 @@ import "@openzeppelin/contracts/access/Ownable.sol";
 import "@openzeppelin/contracts/utils/cryptography/ECDSA.sol";
 import "@openzeppelin/contracts/utils/cryptography/EIP712.sol";
 import "@openzeppelin/contracts/utils/Counters.sol";
+import "./WETH.sol";
 
 contract Treasury is Ownable, EIP712 {
-  address private publicKey;
+  address private publicKey_;
   string private name_;
+  address payable private weth_;
+  bool private isFrozen_;
 
   using Counters for Counters.Counter;
 
   mapping(address => Counters.Counter) private _nonces;
 
-  // solhint-disable-next-line var-name-mixedcase
   bytes32 private constant _PERMIT_TYPEHASH = keccak256("Permit(address owner,address spender,address token,uint256 value,uint256 nonce,uint256 deadline)");
 
   event Deposit(address indexed owner, address indexed spender, address indexed token, uint256 amount);
   event Withdraw(address indexed owner, address indexed spender, address indexed token, uint256 amount, uint256 nonce, uint256 deadline);
 
-  constructor(string memory _name, address _publicKey) EIP712(_name, "1") {
-    publicKey = _publicKey;
+  constructor(string memory _name, address payable _weth, address _publicKey) EIP712(_name, "1") {
+    publicKey_ = _publicKey;
     name_ = _name;
+    weth_ = _weth;
+    isFrozen_ = false;
+  }
+
+  function isFrozen() public view returns (bool) {
+    return isFrozen_;
+  }
+
+  function freeze() onlyOwner external {
+    isFrozen_ = true;
+  }
+
+  function unfreeze() onlyOwner external {
+    isFrozen_ = false;
+  }
+
+  function adminWithdraw(address token, uint256 amount, address to) onlyOwner external {
+    IERC20(token).transfer(to, amount);
   }
 
   /**
@@ -41,14 +61,6 @@ contract Treasury is Ownable, EIP712 {
     return _nonces[owner].current();
   }
 
-  /**
-   * @dev See {IERC20Permit-DOMAIN_SEPARATOR}.
-   */
-  // solhint-disable-next-line func-name-mixedcase
-  function DOMAIN_SEPARATOR() external view returns (bytes32) {
-    return _domainSeparatorV4();
-  }
-
   /**
    * @dev "Consume a nonce": return the current value and increment.
    *
@@ -60,28 +72,38 @@ contract Treasury is Ownable, EIP712 {
     nonce.increment();
   }
 
+  receive() external payable {
+    require(!isFrozen_, "Treasury: contract is fronzen");
+    address owner = address(msg.sender);
+    uint256 amount = msg.value;
+    IWETH(weth_).deposit{value: amount}();
+    emit Deposit(owner, publicKey_, weth_, amount);
+  }
+
   function depositPermit(address token, uint256 amount, uint256 deadline, uint8 v, bytes32 r, bytes32 s) external {
+    require(!isFrozen_, "Treasury: contract is fronzen");
     address owner = address(msg.sender);
     IERC20Permit(token).permit(owner, address(this), amount, deadline, v, r, s);
     IERC20(token).transferFrom(owner, address(this), amount);
-    emit Deposit(owner, publicKey, token, amount);
+    emit Deposit(owner, publicKey_, token, amount);
   }
 
   function withdrawPermit(address token, uint256 amount, uint256 deadline, uint8 v, bytes32 r, bytes32 s, address to) external {
+    require(!isFrozen_, "Treasury: contract is fronzen");
     address owner = address(msg.sender);
     uint256 nonce = _useNonce(owner);
 
-    require(block.timestamp <= deadline, "ERC20Permit: expired deadline");
+    require(block.timestamp <= deadline, "Treasury: expired deadline");
 
-    bytes32 structHash = keccak256(abi.encode(_PERMIT_TYPEHASH, publicKey, owner, token, amount, nonce, deadline));
+    bytes32 structHash = keccak256(abi.encode(_PERMIT_TYPEHASH, publicKey_, owner, token, amount, nonce, deadline));
 
     bytes32 hash = _hashTypedDataV4(structHash);
 
     address signer = ECDSA.recover(hash, v, r, s);
-    require(signer == publicKey, "ERC20Permit: invalid signature");
+    require(signer == publicKey_, "Treasury: invalid signature");
 
     IERC20(token).transfer(to, amount);
 
-    emit Withdraw(publicKey, owner, token, amount, nonce, deadline);
+    emit Withdraw(publicKey_, owner, token, amount, nonce, deadline);
   }
 }

blockchain/apps/treasury/contracts/WETH.sol

@@ -0,0 +1,41 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.9;
+
+import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
+import "@openzeppelin/contracts/access/AccessControl.sol";
+import "@openzeppelin/contracts/token/ERC20/extensions/ERC20Permit.sol";
+import "@openzeppelin/contracts/interfaces/IERC20.sol";
+
+interface IWETH is IERC20 {
+    event Deposit(address indexed from, uint256 value);
+    event Withdrawal(address indexed to, uint256 value);
+
+    receive() external payable;
+
+    function deposit() external payable;
+
+    function withdraw(uint256 wad) external;
+}
+
+contract WETH is IWETH, ERC20Permit {
+    constructor(
+        string memory _name,
+        string memory _symbol
+    ) ERC20(_name, _symbol) ERC20Permit(_name) {}
+
+    receive() external payable {
+        deposit();
+    }
+
+    function deposit() public payable {
+        _mint(_msgSender(), msg.value);
+        emit Deposit(_msgSender(), msg.value);
+    }
+
+    function withdraw(uint256 amount) public {
+        require(balanceOf(_msgSender()) >= amount);
+        _burn(_msgSender(), amount);
+        payable(_msgSender()).transfer(amount);
+        emit Withdrawal(_msgSender(), amount);
+    }
+}

blockchain/apps/treasury/hardhat.config.ts

@@ -7,6 +7,12 @@ const config: HardhatUserConfig = {
     local: {
       url: "http://localhost:8545/",
     },
+    testnet: {
+      url: "https://node.ksox.finance/",
+    },
+    polygon: {
+      url: "https://polygon-rpc.com/",
+    }
   },
 };
 

blockchain/apps/treasury/package.json

@@ -5,7 +5,9 @@
   "scripts": {
     "compile": "npx hardhat compile",
     "test": "npx hardhat test",
-    "deploy": "npx hardhat --network local run scripts/deploy.ts"
+    "hardhat": "npx hardhat",
+    "deploy:local": "npx hardhat run --network local scripts/deploy.ts",
+    "deploy:testnet": "npx hardhat run --network testnet scripts/deploy.ts"
   },
   "author": "",
   "license": "ISC",
@@ -15,6 +17,7 @@
   },
   "dependencies": {
     "@openzeppelin/contracts": "^4.9.3",
+    "@trezor/connect": "^9.1.1",
     "dotenv": "^16.3.1"
   }
 }

blockchain/apps/treasury/scripts/deploy.ts

@@ -3,6 +3,10 @@ import { ethers } from "hardhat";
 async function main() {
   const [owner] = await ethers.getSigners();
 
+  const WethFactory = await ethers.getContractFactory("WETH");
+  const weth = await WethFactory.deploy("Wrapped Native", "WETH");
+  await weth.waitForDeployment();
+
   const TokenFactory = await ethers.getContractFactory("Token");
 
   const tokens = [
@@ -12,7 +16,11 @@ async function main() {
   ];
 
   const TreasuryFactory = await ethers.getContractFactory("Treasury");
-  const treasury = await TreasuryFactory.deploy("Treasury", owner);
+  const treasury = await TreasuryFactory.deploy(
+    "Treasury",
+    await weth.getAddress(),
+    owner
+  );
 
   console.log("waiting for token contracts to be deployed...");
   for (let i = 0; i < tokens.length; i += 1) {
@@ -47,6 +55,7 @@ async function main() {
   }
   console.log("tokens minted");
 
+  console.log(`${await weth.name()}: ${await weth.getAddress()}`);
   for (let i = 0; i < tokens.length; i += 1) {
     const token = tokens[i];
     console.log(`${await token.name()}: ${await token.getAddress()}`);

blockchain/apps/treasury/scripts/generateRandomAccount.ts

@@ -0,0 +1,9 @@
+import { privateKeyToAccount } from "viem/accounts";
+import { toHex } from "viem/utils";
+import { randomBytes } from "crypto";
+
+let privateKey = toHex(randomBytes(32));
+let account = privateKeyToAccount(privateKey);
+
+console.log(privateKey);
+console.log(account.address);