query_solana_stake_pool.ts

// Usage: API_KEY=<> npx tsx ts-test/testnet.ts
import {
    PerChainQueryRequest,
    QueryProxyQueryResponse,
    QueryRequest,
    SolanaAccountQueryRequest,
  } from "@wormhole-foundation/wormhole-query-sdk";
  import axios from "axios";
  import "dotenv/config";
import { logQueryResponseInfo } from "./utils";
  
  const SOLANA_NODE_URL = "https://api.devnet.solana.com";

// only query (up to) the required three fields
    // https://github.com/solana-labs/solana-program-library/blob/b7dd8fee93815b486fce98d3d43d1d0934980226/stake-pool/program/src/state.rs#L87-L97
// and the Clock sysvar data for comparing the epoch
    // https://docs.solana.com/developing/runtime-facilities/sysvars#clock
    // https://docs.rs/solana-program/1.17.17/solana_program/clock/struct.Clock.html
// Solana's getMultipleAccounts rpc spec only allows setting one data slice for all of the accounts, so this will result in some extra bytes
    // https://docs.solana.com/api/http#getmultipleaccounts
// This could also be done in two per-chain queries, though then it may not be done in the same batch call.
export const FIRST_FIELD_BYTE_IDX = 258;
export const SIZE_OF_U64 = 8;
export const DATA_SLICE_OFFSET = 0;
export const DATA_SLICE_LENGTH = FIRST_FIELD_BYTE_IDX + SIZE_OF_U64 * 3;
  
  async function getSolanaSlot(comm: string): Promise<bigint> {
    const response = await axios.post(SOLANA_NODE_URL, {
      jsonrpc: "2.0",
      id: 1,
      method: "getSlot",
      params: [{ commitment: comm, transactionDetails: "none" }],
    });
  
    return response.data.result;
  }
  
  (async () => {
    const QUERY_URL = "https://testnet.query.wormhole.com/v1/query";
    const API_KEY = process.env.API_KEY;
    if (!API_KEY) {
      throw new Error("API_KEY is required");
    }
  
    const currSlot = await getSolanaSlot("finalized");
    const minContextSlot = BigInt(currSlot) + BigInt(2);
  
    console.log(
      `Performing query against Wormhole testnet, currentSlot: `,
      currSlot,
      ` using minContext Slot: `,
      minContextSlot,
      `\n`
    );
  
    // devnet stake pool from
    // solana-program-library % ./target/release/spl-stake-pool --url "https://api.devnet.solana.com" list-all
    const accounts = [
      "DBEr3Z4vdR9WH2jv1hY8Xh1KYQWnBGFjUubvLGdRSvZw", // random stake pool address
      "SysvarC1ock11111111111111111111111111111111",
    ];
  
    const query = new QueryRequest(42, [
      new PerChainQueryRequest(
        1,
        new SolanaAccountQueryRequest(
          "finalized",
          accounts,
          minContextSlot,
          BigInt(DATA_SLICE_OFFSET),
          BigInt(DATA_SLICE_LENGTH)
        )
      ),
    ]);
    const serialized = Buffer.from(query.serialize()).toString("hex");
    const before = performance.now();
    const resp = (
      await axios.post<QueryProxyQueryResponse>(
        QUERY_URL,
        { bytes: serialized },
        { headers: { "X-API-Key": API_KEY } }
      )
    ).data;
    const after = performance.now();
    const logResp = logQueryResponseInfo(resp.bytes);
    if (minContextSlot == logResp.slotNumber) {
      console.log("\nReturned slot matches requested slot.");
    } else {
      console.error(
        "\nSlot mismatch: slotNumber: ",
        logResp.slotNumber,
        ", minContextSlot: ",
        minContextSlot
      );
    }
    console.log(`\nQuery completed in ${(after - before).toFixed(2)}ms.`);
  })();