polkamarkets-js 3.2.0 → 3.3.0

package/scripts/UpdateMarket.s.sol

@@ -0,0 +1,55 @@
+// SPDX-License-Identifier: GPL-2.0
+pragma solidity ^0.8.26;
+
+import {Script} from "forge-std/Script.sol";
+import {console} from "forge-std/console.sol";
+// Import your contracts here
+import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
+import {PredictionMarketV3_3, IWETH} from "../contracts/PredictionMarketV3_3.sol";
+import {IPredictionMarketV3Manager} from "../contracts/IPredictionMarketV3Manager.sol";
+import {PredictionMarketV3Manager, FantasyERC20} from "../contracts/PredictionMarketV3Manager.sol";
+import {IRealityETH_ERC20} from "../contracts/IRealityETH_ERC20.sol";
+
+contract UpdateMarket is Script {
+    function run() public {
+
+        vm.startBroadcast();
+
+        address payable PREDICTION_MARKET_V3_3 = payable(address(0x3e0F5F8F5Fb043aBFA475C0308417Bf72c463289));
+        address PREDICTION_MARKET_V3_MANAGER = address(0x68dDc91CCC06e63d74905D901A30edEA7C77EebE);
+
+        PredictionMarketV3_3.MarketState state = PredictionMarketV3_3(PREDICTION_MARKET_V3_3).getMarketState(486);
+        console2.log("state: ", state);
+
+        PredictionMarketV3_3.MarketUpdateDescription memory description = PredictionMarketV3_3.MarketUpdateDescription({
+            closesAtTimestamp: 1762066800,
+            balance: 456538064447348172635074428,
+            liquidity: 120000000000000000000000000,
+            sharesAvailable: 1272835104225672313766121101,
+            state: PredictionMarketV3_3.MarketState.open,
+            resolution: PredictionMarketV3_3.MarketResolution({
+                resolved: false,
+                realitio: IRealityETH_ERC20(0x71D76Ff3C2729071B2c52A667197d9f715029F2c),
+                outcomeId: 115792089237316195423570985008687907853269984665640564039457584007913129639935,
+                questionId: 0x21d6c6c195572d82a7d0a7ba677251dfd3839b21e09df9b39ae36ec99fa6744d,
+                realitioTimeout: 3600
+            }),
+            feesPoolWeight: 0,
+            feesTreasury: address(0xBc30e9765Dc8c735206c76DE96d369754eBbcc1f),
+            feesDistributor: address(0xBc30e9765Dc8c735206c76DE96d369754eBbcc1f),
+            buyFees: PredictionMarketV3_3.Fees({fee: 0, treasuryFee: 0, distributorFee: 0}),
+            sellFees: PredictionMarketV3_3.Fees({fee: 0, treasuryFee: 0, distributorFee: 0}),
+            outcomeCount: 5,
+            token: IERC20(0x0b07cf011B6e2b7E0803b892d97f751659940F23),
+            manager: IPredictionMarketV3Manager(0x2EbD002d755bdFadb9e5549aA00a80bd7fa408AE),
+            creator: address(0xBc30e9765Dc8c735206c76DE96d369754eBbcc1f),
+            paused: false
+        });
+
+        PredictionMarketV3_3(PREDICTION_MARKET_V3_3).updateMarket(486, description);
+        PredictionMarketV3_3(PREDICTION_MARKET_V3_3).updateMarketResolution(486, description);
+
+        state = PredictionMarketV3_3(PREDICTION_MARKET_V3_3).getMarketState(486);
+        console2.log("new state: ", state);
+    }
+}

package/src/models/PredictionMarketV3Contract.js

@@ -1,6 +1,7 @@
 const predictionV3 = require("../interfaces").predictionV3;
 const PredictionMarketV2Contract = require("./PredictionMarketV2Contract");
 const PredictionMarketV3QuerierContract = require("./PredictionMarketV3QuerierContract");
+const _ = require('lodash');
 
 const Numbers = require('../utils/Numbers');
 
@@ -217,6 +218,8 @@ class PredictionMarketV3Contract extends PredictionMarketV2Contract {
         ];
       }));
 
+      const liquidityFeesClaimed = _.sumBy(_.filter(events, { action: 'Claim Fees' }), 'value');
+
       const item = {
         liquidity: {
           shares: Numbers.fromDecimalsNumber(marketData.liquidityShares, decimals),
@@ -230,7 +233,8 @@ class PredictionMarketV3Contract extends PredictionMarketV2Contract {
           liquidityClaimed: marketData.liquidityClaimed,
           voidedWinningsToClaim: marketData.voidedSharesToClaim,
           voidedWinningsClaimed: false,
-          liquidityFees: 0 // discontinued
+          liquidityFees: 0, // discontinued
+          liquidityFeesClaimed
         }
       };
 

package/test/UpdateTest.t.sol

@@ -0,0 +1,55 @@
+// SPDX-License-Identifier: GPL-2.0
+pragma solidity ^0.8.0;
+
+import "forge-std/console2.sol";
+import "../contracts/PredictionMarketV3_3.sol";
+
+import {Test} from "forge-std/Test.sol";
+import {console} from "forge-std/console.sol";
+
+// forge test --match-contract POC -vvvvv --fork-url $RPC_URL --fork-block-number 13292408 --zksync
+contract UpdateTest is Test {
+    function test_POC() public {
+        address predictionMarketV3 = 0x3e0F5F8F5Fb043aBFA475C0308417Bf72c463289;
+        address admin = 0x0a70f6F09caa802D8d62dE9C7EdacE1d3b5Ca160;
+
+        vm.startPrank(admin);
+
+        address payable PREDICTION_MARKET_V3_3 = payable(address(0x3e0F5F8F5Fb043aBFA475C0308417Bf72c463289));
+
+        (PredictionMarketV3_3.MarketState state, uint256 closesAt, uint256 liquidity, uint256 balance, uint256 sharesAvailable, int256 resolvedOutcomeId) = PredictionMarketV3_3(payable(predictionMarketV3)).getMarketData(486);
+        console.log("state: ", uint256(state));
+
+        PredictionMarketV3_3.MarketUpdateDescription memory description = PredictionMarketV3_3.MarketUpdateDescription({
+            closesAtTimestamp: 1762066800,
+            balance: 456538064447348172635074428,
+            liquidity: 120000000000000000000000000,
+            sharesAvailable: 1272835104225672313766121101,
+            state: PredictionMarketV3_3.MarketState.open,
+            resolution: PredictionMarketV3_3.MarketResolution({
+                resolved: false,
+                realitio: IRealityETH_ERC20(0x71D76Ff3C2729071B2c52A667197d9f715029F2c),
+                outcomeId: 115792089237316195423570985008687907853269984665640564039457584007913129639935,
+                questionId: 0x21d6c6c195572d82a7d0a7ba677251dfd3839b21e09df9b39ae36ec99fa6744d,
+                realitioTimeout: 3600
+            }),
+            feesPoolWeight: 0,
+            feesTreasury: address(0xBc30e9765Dc8c735206c76DE96d369754eBbcc1f),
+            feesDistributor: address(0xBc30e9765Dc8c735206c76DE96d369754eBbcc1f),
+            buyFees: PredictionMarketV3_3.Fees({fee: 0, treasuryFee: 0, distributorFee: 0}),
+            sellFees: PredictionMarketV3_3.Fees({fee: 0, treasuryFee: 0, distributorFee: 0}),
+            outcomeCount: 5,
+            token: IERC20(0x0b07cf011B6e2b7E0803b892d97f751659940F23),
+            manager: IPredictionMarketV3Manager(0x2EbD002d755bdFadb9e5549aA00a80bd7fa408AE),
+            creator: address(0xBc30e9765Dc8c735206c76DE96d369754eBbcc1f),
+            paused: false
+        });
+
+        PredictionMarketV3_3(PREDICTION_MARKET_V3_3).updateMarket(486, description);
+        PredictionMarketV3_3(PREDICTION_MARKET_V3_3).updateMarketResolution(486, description);
+
+        (state, closesAt, liquidity, balance, sharesAvailable, resolvedOutcomeId) = PredictionMarketV3_3(PREDICTION_MARKET_V3_3).getMarketData(486);
+        console2.log("new state: ", uint256(state));
+
+    }
+}

package/test/WhaleTest.t.sol

@@ -0,0 +1,188 @@
+// SPDX-License-Identifier: GPL-2.0
+pragma solidity ^0.8.0;
+
+import "forge-std/console2.sol";
+
+import {Test} from "forge-std/Test.sol";
+import {ERC20, IERC20} from "openzeppelin-contracts/contracts/token/ERC20/ERC20.sol";
+
+// forge test --match-contract POC -vvvvv --fork-url $RPC_URL --fork-block-number 13292408 --zksync
+contract WhaleTest is Test {
+    function test_POC() public {
+        address predictionMarketV3 = 0x39E66eE6b2ddaf4DEfDEd3038E0162180dbeF340;
+        address predictionMarketV3Manager = 0xDeFb36c47754D2e37d44b8b8C647D4D643e03bAd;
+        address usdc = 0x176211869cA2b568f2A7D4EE941E073a821EE1ff;
+        address admin = 0x7836D709E750F6A2D6ff9362457028e8Cd901EB3;
+        address whale = 0x374D7860c4f2f604De0191298dD393703Cce84f3;
+
+        vm.deal(address(this), 200000e6); // Small eth amount needed to trick contract accounting, 0.000000000001 eth would be enough to steal 10_000 usdc
+
+        // take usdc from the whale
+        vm.startPrank(whale);
+        IERC20(usdc).transfer(admin, 250000e6);
+        vm.stopPrank();
+
+        console2.log("admin's usdc balance at the start: ", IERC20(usdc).balanceOf(admin));
+
+
+        vm.startPrank(admin);
+        uint256[] memory distribution = new uint256[](2);
+        distribution[0] = 100e6;
+        distribution[1] = 100e6;
+
+        IERC20(usdc).approve(predictionMarketV3, type(uint256).max);
+
+
+        uint256 marketId = PredictionMarketV3(predictionMarketV3).createMarket(CreateMarketDescription({
+            value: 10_000e6,
+            closesAt: uint32(block.timestamp + 3600),
+            outcomes: 2,
+            token: IERC20(usdc),
+            distribution: distribution,
+            question: "test",
+            image: "test",
+            arbitrator: address(this),
+            buyFees: Fees({fee: 1e16, treasuryFee: 0, distributorFee: 0}),
+            sellFees: Fees({fee: 0, treasuryFee: 0, distributorFee: 0}),
+            treasury: address(0),
+            distributor: address(0),
+            realitioTimeout: 3600,
+            manager: IPredictionMarketV3Manager(predictionMarketV3Manager)
+        }));
+
+        for (uint256 i = 0; i < 100; i++) {
+          PredictionMarketV3(predictionMarketV3).buy(marketId, 0, 0, 1000e6);
+          PredictionMarketV3(predictionMarketV3).buy(marketId, 1, 0, 1000e6);
+        }
+
+        (uint256 liquidity, uint256[] memory outcomes) = PredictionMarketV3(predictionMarketV3).getUserMarketShares(marketId, admin);
+        console2.log("liquidity: ", liquidity);
+        console2.log("outcome 0: ", outcomes[0]);
+        console2.log("outcome 1: ", outcomes[1]);
+
+        vm.stopPrank();
+
+        console2.log("admin's usdc balance at the end: ", IERC20(usdc).balanceOf(admin));
+    }
+}
+
+interface IPredictionMarketV3Manager {
+    function getERC20RealitioAddress(IERC20 token) external view returns (address);
+    function isAllowedToCreateMarket(address, address) external view returns (bool);
+    function askQuestionERC20(uint256 template_id, string calldata question, address arbitrator, uint32 timeout, uint32 opening_ts, uint256 nonce, uint256 tokens) external returns (bytes32);
+}
+
+contract Manager is IPredictionMarketV3Manager {
+    function getERC20RealitioAddress(IERC20 token) external view returns (address) {
+        return address(this);
+    }
+
+    function isAllowedToCreateMarket(address, address) external view returns (bool) {
+        return true;
+    }
+
+    function askQuestionERC20 (uint256 template_id, string calldata question, address arbitrator, uint32 timeout, uint32 opening_ts, uint256 nonce, uint256 tokens) external returns (bytes32) {
+        return bytes32(uint256(1));
+    }
+
+}
+
+interface PredictionMarketV3 {
+    function buy(uint256 marketId, uint256 outcomeId, uint256 minOutcomeSharesToBuy, uint256 value) external;
+    function sell(uint256 marketId, uint256 outcomeId, uint256 value, uint256 maxOutcomeSharesToSell) external ;
+    function createMarket(CreateMarketDescription calldata desc) external returns (uint256);
+    function getUserMarketShares(uint256 marketId, address user) external view returns (uint256 liquidity, uint256[] memory outcomes);
+}
+
+
+  enum MarketState {
+    open,
+    closed,
+    resolved
+  }
+  enum MarketAction {
+    buy,
+    sell,
+    addLiquidity,
+    removeLiquidity,
+    claimWinnings,
+    claimLiquidity,
+    claimFees,
+    claimVoided
+  }
+
+
+  struct Market {
+    // market details
+    uint256 closesAtTimestamp;
+    uint256 balance; // total stake
+    uint256 liquidity; // stake held
+    uint256 sharesAvailable; // shares held (all outcomes)
+    mapping(address => uint256) liquidityShares;
+    mapping(address => bool) liquidityClaims; // wether user has claimed liquidity earnings
+    MarketState state; // resolution variables
+    MarketResolution resolution; // fees
+    MarketFees fees;
+    // market outcomes
+    uint256 outcomeCount;
+    mapping(uint256 => MarketOutcome) outcomes;
+    IERC20 token; // ERC20 token market will use for trading
+    IPredictionMarketV3Manager manager; // manager contract
+    address creator; // market creator
+    bool paused; // market paused, no trading allowed
+  }
+
+  struct Fees {
+    uint256 fee; // fee % taken from every transaction
+    uint256 treasuryFee; // fee % taken from every transaction to a treasury address
+    uint256 distributorFee; // fee % taken from every transaction to a distributor address
+  }
+
+  struct MarketFees {
+    uint256 poolWeight; // internal var used to ensure pro-rate fee distribution
+    mapping(address => uint256) claimed;
+    address treasury; // address to send treasury fees to
+    address distributor; // fee % taken from every transaction to a treasury address
+    Fees buyFees; // fees for buy transactions
+    Fees sellFees; // fees for sell transactions
+  }
+
+  struct MarketResolution {
+    bool resolved;
+    address realitio;
+    uint256 outcomeId;
+    bytes32 questionId; // realitio questionId
+    // realitio
+    uint256 realitioTimeout;
+  }
+
+  struct MarketOutcome {
+    uint256 marketId;
+    uint256 id;
+    Shares shares;
+  }
+
+  struct Shares {
+    uint256 total; // number of shares
+    uint256 available; // available shares
+    mapping(address => uint256) holders;
+    mapping(address => bool) claims; // wether user has claimed winnings
+    mapping(address => bool) voidedClaims; // wether user has claimed voided market shares
+  }
+
+  struct CreateMarketDescription {
+    uint256 value;
+    uint32 closesAt;
+    uint256 outcomes;
+    IERC20 token;
+    uint256[] distribution;
+    string question;
+    string image;
+    address arbitrator;
+    Fees buyFees;
+    Fees sellFees;
+    address treasury;
+    address distributor;
+    uint32 realitioTimeout;
+    IPredictionMarketV3Manager manager;
+  }