Add crank script

Author: metaproph3t

Anchor.toml

@@ -26,6 +26,7 @@ clob = "yarn run tsc scripts/clob.ts --esModuleInterop --resolveJsonModule && no
 autocrat_v0 = "yarn run ts-node scripts/autocratV0.ts"
 airdrop = "yarn run ts-node scripts/airdrop.ts"
 place_order = "yarn run ts-node scripts/placeOrder.ts"
+crank = "yarn run ts-node scripts/crank.ts"
 test = "yarn run ts-mocha -p ./tsconfig.json -t 1000000 tests/*.ts"
 
 [test]

scripts/crank.ts

@@ -0,0 +1,45 @@
+import "./main";
+import { autocratProgram, openbook, openbookTwap } from "./main";
+import * as anchor from "@coral-xyz/anchor";
+const { PublicKey, Keypair, SystemProgram } = anchor.web3;
+const { BN, Program } = anchor;
+const {execSync} = require('child_process');
+
+import {
+  OpenBookV2Client,
+  PlaceOrderArgs,
+  Side,
+  OrderType,
+  SelfTradeBehavior,
+} from "@openbook-dex/openbook-v2";
+
+const twapMarketList = ["7TXCqw4iJ8mD9vMAkjEiDTuZLETW9DuBM48J2kPkoVoR", "AwMimsGw2ShWXTeC6JkMguGoT73vky44Z1DaP1fpBxQY"];
+
+let twapMarketPks = twapMarketList.map(twapMarket => new PublicKey(twapMarket));
+
+async function crank(twapMarkets: anchor.web3.PublicKey[]) {
+    let markets = [];
+    for (let twapMarket of twapMarkets) {
+        markets.push((await openbookTwap.account.twapMarket.fetch(twapMarket)).market);
+    }
+
+    let storedMarkets = [];
+    for (let market of markets) {
+        await storedMarkets.push(openbook.getMarketAccount(market));
+    }
+
+    while (true) {
+        for (let i in markets) {
+            let accountsToConsume = await openbook.getAccountsToConsume(storedMarkets[i]);
+            console.log(accountsToConsume);
+
+            if (accountsToConsume.length > 0) {
+                await openbook.consumeEvents(markets[i], storedMarkets[i], new BN(5), accountsToConsume);
+            }
+
+            execSync('sleep 10');
+        }
+    }
+}
+
+crank(twapMarketPks);

scripts/placeOrder.ts

@@ -1,14 +1,106 @@
 import "./main";
 import { autocratProgram } from "./main";
 
+import {
+  OpenBookV2Client,
+  PlaceOrderArgs,
+  Side,
+  OrderType,
+  SelfTradeBehavior,
+} from "@openbook-dex/openbook-v2";
+
 // Put your own proposal number here
 const PROPOSAL_NUMBER = 1;
+// If false, it will do the fail market obviously.
+const PASS_MARKET = true;
 
 async function placeOrder() {
     let proposals = await autocratProgram.account.proposal.all();
 
     let proposal = proposals.filter((proposal) => proposal.account.number == PROPOSAL_NUMBER)[0].account;
-    console.log(proposal);
+
+    let twapMarket = PASS_MARKET ? proposal.openbookTwapPassMarket : proposal.openbookTwapFailMarket;
+    let market = PASS_MARKET ? proposal.openbookPassMarket : proposal.openbookFailMarket;
+
+    // let market = (await openbookTwap.account.twapMarket.fetch(twapMarket)).market;
+
+  let buyArgs: PlaceOrderArgs = {
+    side: Side.Bid,
+    priceLots: new BN(9_000), // 1 USDC for 1 META
+    maxBaseLots: new BN(10),
+    maxQuoteLotsIncludingFees: new BN(10 * 1000_000), // 10 USDC
+    clientOrderId: new BN(1),
+    orderType: OrderType.Limit,
+    expiryTimestamp: new BN(0),
+    selfTradeBehavior: SelfTradeBehavior.DecrementTake,
+    limit: 255,
+  };
+
+  let sellArgs: PlaceOrderArgs = {
+    side: Side.Ask,
+    priceLots: new BN(12_000), // 1.2 USDC for 1 META
+    maxBaseLots: new BN(10),
+    maxQuoteLotsIncludingFees: new BN(10 * 12_000),
+    clientOrderId: new BN(2),
+    orderType: OrderType.Limit,
+    expiryTimestamp: new BN(0),
+    selfTradeBehavior: SelfTradeBehavior.DecrementTake,
+    limit: 255,
+  };
+
+  const storedMarket = await openbook.getMarketAccount(market);
+  let openOrdersAccount = new anchor.web3.PublicKey("CxDQ5RSYebF6mRLDrXYn1An7bawe6S3iyaU5rZBjz4Xs");
+  // let openOrdersAccount = await openbook.createOpenOrders(
+  //   payer,
+  //   market,
+  //   new BN(1),
+  //   "oo"
+  // );
+  // console.log(openOrdersAccount);
+  // let openOrdersAccount = await openbook.createOpenOrders(market, new BN(4), "oo2");
+
+  const userBaseAccount = await token.getOrCreateAssociatedTokenAccount(
+    provider.connection,
+    payer,
+    storedMarket.baseMint,
+    payer.publicKey
+  );
+  const userQuoteAccount = await token.getOrCreateAssociatedTokenAccount(
+    provider.connection,
+    payer,
+    storedMarket.quoteMint,
+    payer.publicKey
+  );
+
+  await openbookTwap.methods
+    .placeOrder(buyArgs)
+    .accounts({
+      asks: storedMarket.asks,
+      bids: storedMarket.bids,
+      marketVault: storedMarket.marketQuoteVault,
+      eventHeap: storedMarket.eventHeap,
+      market,
+      openOrdersAccount,
+      userTokenAccount: userQuoteAccount.address,
+      twapMarket,
+      openbookProgram: OPENBOOK_PROGRAM_ID,
+    })
+    .rpc();
+
+  await openbookTwap.methods
+    .placeOrder(sellArgs)
+    .accounts({
+      asks: storedMarket.asks,
+      bids: storedMarket.bids,
+      marketVault: storedMarket.marketBaseVault,
+      eventHeap: storedMarket.eventHeap,
+      market,
+      openOrdersAccount,
+      userTokenAccount: userBaseAccount.address,
+      twapMarket,
+      openbookProgram: OPENBOOK_PROGRAM_ID,
+    })
+    .rpc();
 }
 
 placeOrder();