add reimburse script

package.json

@@ -11,6 +11,7 @@
     "liquidate": "ts-node src/liquidate.ts",
     "partialLiquidate": "ts-node src/partial.ts",
     "test": "ts-node src/test.ts",
+    "reimburse": "ts-node src/reimburse.ts",
     "watch": "tsc --watch",
     "clean": "rm -rf dist",
     "prepare": "run-s clean build",

src/reimburse.ts

@@ -0,0 +1,91 @@
+import {
+  encodeMangoInstruction,
+  findLargestTokenAccountForOwner,
+  IDS,
+  MangoClient,
+  uiToNative,
+} from '@blockworks-foundation/mango-client';
+import {
+  Account,
+  Connection,
+  PublicKey,
+  SYSVAR_CLOCK_PUBKEY,
+  Transaction,
+  TransactionInstruction,
+} from '@solana/web3.js';
+import { homedir } from 'os';
+import fs from 'fs';
+import { TOKEN_PROGRAM_ID } from '@solana/spl-token';
+
+async function runReimburse() {
+  const client = new MangoClient();
+  const cluster = process.env.CLUSTER || 'mainnet-beta';
+  const group_name = process.env.GROUP_NAME || 'BTC_ETH_SOL_SRM_USDC';
+  const clusterUrl = process.env.CLUSTER_URL || IDS.cluster_urls[cluster];
+  const connection = new Connection(clusterUrl, 'singleGossip');
+
+  // The address of the Mango Program on the blockchain
+  const programId = new PublicKey(IDS[cluster].mango_program_id);
+
+  // Address of the MangoGroup
+  const mangoGroupPk = new PublicKey(
+    IDS[cluster].mango_groups[group_name].mango_group_pk,
+  );
+
+  // keypair
+  const keyPairPath =
+    process.env.KEYPAIR || homedir() + '/.config/solana/id.json';
+  const payer = new Account(JSON.parse(fs.readFileSync(keyPairPath, 'utf-8')));
+
+  let mangoGroup = await client.getMangoGroup(connection, mangoGroupPk);
+
+  const tokenWallets = await Promise.all(
+    mangoGroup.tokens.map((mint) =>
+      findLargestTokenAccountForOwner(connection, payer.publicKey, mint).then(
+        (response) => response,
+      ),
+    ),
+  );
+
+  const marginAccounts = [
+    { address: '4iZH4jgK1tdXQ3jwvQE7bsUDmxw1PnkhiBPdgEq5iRoS', amount: 1316.310144},
+    { address: '4WoRSRkvJNenNVk7uB8st9PsjaPY8MZpHtK8NM7pRmzr', amount: 405.5065004},
+    { address: '79k2VqrXFz8vSsuMWtCq2zJX2YGUL3gH5yS2A7gd2WNJ', amount: 534.7915884},
+    { address: '7hudJyNLJKLH14httaFihK3zQ1auF3C1Rex2N79nja9u', amount: 1572.512634},
+    { address: '7PAC5DfMxTYHLZicMrimsdL8XvgTp7gTM6ZbWzArQJXc', amount: 2180.24129},
+    { address: 'CFcm6dtw6xaYkgoxZbRDa6WhLyHvirURT49HZAQGiLX2', amount: 3146.680519},
+    { address: 'GVLZhyZma9tvKiXtKPTeRboju8kzj7MyoY1wPZ93eq79', amount: 928.1663967},
+    { address: 'GvT7iqpnNe1zrXG5G1cNgKDyFYVx3E4Suqx1JmxXZW8m', amount: 2421.853979},
+    { address: 'n3YxMeZvPnMqiNZuecWMdMm7tMszYS2Ve4TnN2bXYPi', amount: 2464.867504},
+    { address: 'x3qpJRZFMHC4SRy2RGnbeahChNEUtd24uFidHMyrkjT', amount: 258.1840228}
+  ]
+
+  for (const ma of marginAccounts) {
+    const marginAccount = new PublicKey(ma.address)
+
+    const quantity = (ma.amount * 1.2) // fill in quantity
+    const tokenIndex = 4
+    const nativeQuantity = uiToNative(quantity, mangoGroup.mintDecimals[tokenIndex])
+    const tokenAcc = tokenWallets[4].publicKey // 4 is index of usdc token acct
+  
+    const keys = [
+      { isSigner: false, isWritable: true, pubkey: mangoGroup.publicKey},
+      { isSigner: false,  isWritable: true, pubkey: marginAccount },
+      { isSigner: true, isWritable: false, pubkey: payer.publicKey },
+      { isSigner: false, isWritable: true,  pubkey: tokenAcc },
+      { isSigner: false, isWritable: true,  pubkey: mangoGroup.vaults[tokenIndex] },
+      { isSigner: false, isWritable: false, pubkey: TOKEN_PROGRAM_ID },
+      { isSigner: false, isWritable: false, pubkey: SYSVAR_CLOCK_PUBKEY }
+    ]
+    const data = encodeMangoInstruction({Deposit: {quantity: nativeQuantity}})
+  
+    const instruction = new TransactionInstruction( { keys, data, programId })
+  
+    const transaction = new Transaction()
+    transaction.add(instruction)
+  
+    await client.sendTransaction(connection, transaction, payer, []);
+  }
+}
+
+runReimburse();