feat: transaction builder

target_chains/solana/sdk/js/pyth_solana_receiver/README.md

@@ -12,28 +12,60 @@ Price update accounts can be closed by whoever wrote them to recover the rent.
 ## Example use
 
 ```ts
-import { Connection, PublicKey } from '@solana/web3.js';
-import { PriceServiceConnection } from '@pythnetwork/price-service-client';
-import { PythSolanaReceiver } from '@pythnetwork/pyth-solana-receiver';
-import { MyFirstPythApp, IDL } from './idl/my_first_pyth_app';
-
-
-const SOL_PRICE_FEED_ID = "0xef0d8b6fda2ceba41da15d4095d1da392a0d2f8ed0c6c7bc0f4cfac8c280b56d"
-const ETH_PRICE_FEED_ID = "0xff61491a931112ddf1bd8147cd1b641375f79f5825126d665480874634fd0ace"
+import { Connection, PublicKey } from "@solana/web3.js";
+import { PriceServiceConnection } from "@pythnetwork/price-service-client";
+import { PythSolanaReceiver } from "@pythnetwork/pyth-solana-receiver";
+import { MyFirstPythApp, IDL } from "./idl/my_first_pyth_app";
 
-const priceServiceConnection = new PriceServiceConnection("https://hermes.pyth.network/", { priceFeedRequestConfig: { binary: true } });
-const priceUpdateData = await priceServiceConnection.getLatestVaas([SOL_PRICE_FEED_ID, ETH_PRICE_FEED_ID]);  // Fetch off-chain price update data
+const SOL_PRICE_FEED_ID =
+  "0xef0d8b6fda2ceba41da15d4095d1da392a0d2f8ed0c6c7bc0f4cfac8c280b56d";
+const ETH_PRICE_FEED_ID =
+  "0xff61491a931112ddf1bd8147cd1b641375f79f5825126d665480874634fd0ace";
 
+const priceServiceConnection = new PriceServiceConnection(
+  "https://hermes.pyth.network/",
+  { priceFeedRequestConfig: { binary: true } }
+);
+const priceUpdateData = await priceServiceConnection.getLatestVaas([
+  SOL_PRICE_FEED_ID,
+  ETH_PRICE_FEED_ID,
+]); // Fetch off-chain price update data
 
-const myFirstPythApp = new Program<MyFirstPythApp>(IDL as MyFirstPythApp, , PublicKey.unique(), {})
-const getInstructions = async (priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>) => { return [{ instruction: await myFirstApp.methods.consume().accounts({ solPriceUpdate: priceFeedIdToPriceUpdateAccount[SOL_PRICE_FEED_ID], ethPriceUpdate: priceFeedIdToPriceUpdateAccount[ETH_PRICE_FEED_ID] }).instruction(), signers: [] }] };
+const myFirstPythApp = new Program<MyFirstPythApp>(
+  IDL as MyFirstPythApp,
+  MY_FIRST_PYTH_APP_PROGRAM_ID,
+  {}
+);
 
-const pythSolanaReceiver = new PythSolanaReceiver({ connection, wallet });
-const transactions = await pythSolanaReceiver.withPriceUpdate(priceUpdateData, getInstructions, {})
-await pythSolanaReceiver.provider.sendAll(transactions);
+const transactionBuilder = pythSolanaReceiver.newTransactionBuilder();
+await transactionBuilder.withPostPriceUpdates(priceUpdateData);
+await transactionBuilder.withPriceConsumerInstructions(
+  async (
+    priceFeedIdToPriceAccount: Record<string, PublicKey>
+  ): Promise<InstructionWithEphemeralSigners[]> => {
+    return [
+      {
+        instruction: await myFirstPythApp.methods
+          .consume()
+          .accounts({
+            solPriceUpdate: priceFeedIdToPriceUpdateAccount[SOL_PRICE_FEED_ID],
+            ethPriceUpdate: priceFeedIdToPriceUpdateAccount[ETH_PRICE_FEED_ID],
+          })
+          .instruction(),
+        signers: [],
+      },
+    ];
+  }
+);
+transactionBuilder.withCloseInstructions();
+await pythSolanaReceiver.provider.sendAll(
+  await transactionBuilder.getVersionedTransactions({
+    computeUnitPriceMicroLamports: 1000000,
+  })
+);
 ```
 
-Or, alternatively:
+Alternatively you can use the instruction builder methods from `PythSolanaReceiver` :
 
 ```ts
 import { PublicKey } from "@solana/web3.js";
@@ -61,7 +93,7 @@ const { postInstructions, closeInstructions, priceFeedIdToPriceUpdateAccount } =
 
 const myFirstPythApp = new Program<MyFirstPythApp>(
   IDL as MyFirstPythApp,
-  PublicKey.unique(),
+  MY_FIRST_PYTH_APP_PROGRAM_ID,
   {}
 );
 const consumerInstruction: InstructionWithEphemeralSigners = {
@@ -77,7 +109,7 @@ const consumerInstruction: InstructionWithEphemeralSigners = {
 
 const transactions = pythSolanaReceiver.batchIntoVersionedTransactions(
   [...postInstructions, consumerInstruction, ...closeInstructions],
-  {}
+  { computeUnitPriceMicroLamports: 1000000 }
 ); // Put all the instructions together
 await pythSolanaReceiver.provider.sendAll(transactions);
 ```

target_chains/solana/sdk/js/pyth_solana_receiver/src/PythSolanaReceiver.ts

@@ -39,12 +39,150 @@ import {
   PriorityFeeConfig,
 } from "@pythnetwork/solana-utils";
 
+/**
+ * A builder class to build transactions that:
+ * - Post price updates (fully or partially verified)
+ * - Consume price updates in a consumer program
+ * - (Optionally) Close price update and encoded vaa accounts to recover the rent
+ *
+ * @example
+ * ```typescript
+ *  const priceUpdateData = await priceServiceConnection.getLatestVaas([
+ *    SOL_PRICE_FEED_ID,
+ *    ETH_PRICE_FEED_ID,
+ *  ]);
+ *
+ * const transactionBuilder = pythSolanaReceiver.newTransactionBuilder();
+ * await transactionBuilder.withPostPriceUpdates(priceUpdateData);
+ * await transactionBuilder.withPriceConsumerInstructions(...)
+ * transactionBuilder.withCloseInstructions();
+ *
+ * await pythSolanaReceiver.provider.sendAll(await transactionBuilder.getVersionedTransactions({computeUnitPriceMicroLamports:1000000}))
+ * ```
+ */
+export class PythTransactionBuilder extends TransactionBuilder {
+  readonly pythSolanaReceiver: PythSolanaReceiver;
+  private closeInstructions: InstructionWithEphemeralSigners[];
+  private priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>;
+
+  constructor(pythSolanaReceiver: PythSolanaReceiver) {
+    super(pythSolanaReceiver.wallet.publicKey, pythSolanaReceiver.connection);
+    this.pythSolanaReceiver = pythSolanaReceiver;
+    this.closeInstructions = [];
+    this.priceFeedIdToPriceUpdateAccount = {};
+  }
+
+  /**
+   * Add instructions to post price updates to the builder.
+   *
+   * @param priceUpdateDataArray the output of the `@pythnetwork/price-service-client`'s `PriceServiceConnection.getLatestVaas`. This is an array of verifiable price updates.
+   */
+  async withPostPriceUpdates(priceUpdateDataArray: string[]) {
+    const {
+      postInstructions,
+      priceFeedIdToPriceUpdateAccount,
+      closeInstructions,
+    } = await this.pythSolanaReceiver.buildPostPriceUpdateInstructions(
+      priceUpdateDataArray
+    );
+    this.closeInstructions.push(...closeInstructions);
+    this.priceFeedIdToPriceUpdateAccount = {
+      ...this.priceFeedIdToPriceUpdateAccount,
+      ...priceFeedIdToPriceUpdateAccount,
+    };
+    this.addInstructions(postInstructions);
+  }
+
+  /**
+   * Add instructions to post partially verified price updates to the builder.
+   *
+   * @param priceUpdateDataArray the output of the `@pythnetwork/price-service-client`'s `PriceServiceConnection.getLatestVaas`. This is an array of verifiable price updates.
+   *
+   * Partially verified price updates are price updates where not all the guardian signatures have been verified. By default this methods checks `DEFAULT_REDUCED_GUARDIAN_SET_SIZE` signatures when posting the VAA.
+   * If you are a on-chain program developer, make sure you understand the risks of consuming partially verified price updates here: {@link https://github.com/pyth-network/pyth-crosschain/blob/main/target_chains/solana/pyth_solana_receiver_state/src/price_update.rs}.
+   *
+   * @example
+   * ```typescript
+   * const priceUpdateData = await priceServiceConnection.getLatestVaas([
+   *    SOL_PRICE_FEED_ID,
+   *    ETH_PRICE_FEED_ID,
+   * ]);
+   *
+   * const transactionBuilder = pythSolanaReceiver.newTransactionBuilder();
+   * await transactionBuilder.withPostPartiallyVerifiedPriceUpdates(priceUpdateData);
+   * await transactionBuilder.withPriceConsumerInstructions(...)
+   * ...
+   * ```
+   */
+  async withPostPartiallyVerifiedPriceUpdates(priceUpdateDataArray: string[]) {
+    const {
+      postInstructions,
+      priceFeedIdToPriceUpdateAccount,
+      closeInstructions,
+    } = await this.pythSolanaReceiver.buildPostPriceUpdateAtomicInstructions(
+      priceUpdateDataArray
+    );
+    this.closeInstructions.push(...closeInstructions);
+    this.priceFeedIdToPriceUpdateAccount = {
+      ...this.priceFeedIdToPriceUpdateAccount,
+      ...priceFeedIdToPriceUpdateAccount,
+    };
+    this.addInstructions(postInstructions);
+  }
+
+  /**
+   * Add instructions that consume price updates to the builder.
+   *
+   * @param getInstructions a function that given a map of price feed IDs to price update accounts, generates a series of instructions. Price updates get posted to ephemeral accounts and this function allows the user to indicate which accounts in their instruction need to be "replaced" with each price update account.
+   *
+   * @example
+   * ```typescript
+   * ...
+   * await transactionBuilder.withPostPriceUpdates(priceUpdateData);
+   * await transactionBuilder.withPriceConsumerInstructions(
+   *   async (
+   *     priceFeedIdToPriceAccount: Record<string, PublicKey>
+   *   ): Promise<InstructionWithEphemeralSigners[]> => {
+   *     return [
+   *       {
+   *         instruction: await myFirstPythApp.methods
+   *           .consume()
+   *           .accounts({
+   *              solPriceUpdate: priceFeedIdToPriceUpdateAccount[SOL_PRICE_FEED_ID],
+   *              ethPriceUpdate: priceFeedIdToPriceUpdateAccount[ETH_PRICE_FEED_ID],
+   *           })
+   *           .instruction(),
+   *         signers: [],
+   *       },
+   *     ];
+   *   }
+   * );
+   * transactionBuilder.withCloseInstructions();
+   * ```
+   */
+  async withPriceConsumerInstructions(
+    getInstructions: (
+      priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>
+    ) => Promise<InstructionWithEphemeralSigners[]>
+  ) {
+    this.addInstructions(
+      await getInstructions(this.priceFeedIdToPriceUpdateAccount)
+    );
+  }
+
+  /**
+   * Add instructions to close the encoded vaa accounts and the price update accounts created by `withPostPriceUpdates` and `withPostPartiallyVerifiedPriceUpdates` to reclaim rent.
+   */
+  withCloseInstructions() {
+    this.addInstructions(this.closeInstructions);
+  }
+}
+
 /**
  * A class to interact with the Pyth Solana Receiver program.
  *
- * This class provides helpful methods to:
- * - Post price updates from Pythnet to the Pyth Solana Receiver program
- * - Consume price updates in a consumer program
+ * This class provides helpful methods to build instructions to interact with the Pyth Solana Receiver program:
+ * - Post price updates (fully or partially verified)
  * - Close price update and encoded vaa accounts to recover rent
  */
 export class PythSolanaReceiver {
@@ -83,65 +221,10 @@ export class PythSolanaReceiver {
   }
 
   /**
-   * Build a series of transactions that post price updates to the Pyth Solana Receiver program, consume them in a consumer program and close the encoded vaa accounts and price update accounts.
-   * @param priceUpdateDataArray the output of the `@pythnetwork/price-service-client`'s `PriceServiceConnection.getLatestVaas`. This is an array of verifiable price updates.
-   * @param getInstructions a function that given a map of price feed IDs to price update accounts, returns a series of instructions to consume the price updates in a consumer program. This function is a way for the user to indicate which accounts in their instruction need to be "replaced" with price update accounts.
-   * @param priorityFeeConfig a configuration for the compute unit price to use for the transactions.
-   * @returns an array of transactions and their corresponding ephemeral signers
+   * Get a new transaction builder to build transactions that interact with the Pyth Solana Receiver program and consume price updates
    */
-  async withPriceUpdate(
-    priceUpdateDataArray: string[],
-    getInstructions: (
-      priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>
-    ) => Promise<InstructionWithEphemeralSigners[]>,
-    priorityFeeConfig?: PriorityFeeConfig
-  ): Promise<{ tx: VersionedTransaction; signers: Signer[] }[]> {
-    const {
-      postInstructions,
-      priceFeedIdToPriceUpdateAccount: priceFeedIdToPriceUpdateAccount,
-      closeInstructions,
-    } = await this.buildPostPriceUpdateInstructions(priceUpdateDataArray);
-    return this.batchIntoVersionedTransactions(
-      [
-        ...postInstructions,
-        ...(await getInstructions(priceFeedIdToPriceUpdateAccount)),
-        ...closeInstructions,
-      ],
-      priorityFeeConfig ?? {}
-    );
-  }
-
-  /**
-   * Build a series of transactions that post partially verified price updates to the Pyth Solana Receiver program, consume them in a consumer program and close the price update accounts.
-   *
-   * Partially verified price updates are price updates where not all the guardian signatures have been verified. By default this methods checks `DEFAULT_REDUCED_GUARDIAN_SET_SIZE` signatures when posting the VAA.
-   * If you are a on-chain program developer, make sure you understand the risks of consuming partially verified price updates here: {@link https://github.com/pyth-network/pyth-crosschain/blob/main/target_chains/solana/pyth_solana_receiver_state/src/price_update.rs}.
-   *
-   * @param priceUpdateDataArray the output of the `@pythnetwork/price-service-client`'s `PriceServiceConnection.getLatestVaas`. This is an array of verifiable price updates.
-   * @param getInstructions a function that given a map of price feed IDs to price update accounts, returns a series of instructions to consume the price updates in a consumer program. This function is a way for the user to indicate which accounts in their instruction need to be "replaced" with price update accounts.
-   * @param priorityFeeConfig a configuration for the compute unit price to use for the transactions.
-   * @returns an array of transactions and their corresponding ephemeral signers
-   */
-  async withPartiallyVerifiedPriceUpdate(
-    priceUpdateDataArray: string[],
-    getInstructions: (
-      priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>
-    ) => Promise<InstructionWithEphemeralSigners[]>,
-    priorityFeeConfig?: PriorityFeeConfig
-  ): Promise<{ tx: VersionedTransaction; signers: Signer[] }[]> {
-    const {
-      postInstructions,
-      priceFeedIdToPriceUpdateAccount,
-      closeInstructions,
-    } = await this.buildPostPriceUpdateAtomicInstructions(priceUpdateDataArray);
-    return this.batchIntoVersionedTransactions(
-      [
-        ...postInstructions,
-        ...(await getInstructions(priceFeedIdToPriceUpdateAccount)),
-        ...closeInstructions,
-      ],
-      priorityFeeConfig ?? {}
-    );
+  newTransactionBuilder(): PythTransactionBuilder {
+    return new PythTransactionBuilder(this);
   }
 
   /**

target_chains/solana/sdk/js/pyth_solana_receiver/src/index.ts

@@ -1,4 +1,7 @@
-export { PythSolanaReceiver } from "./PythSolanaReceiver";
+export {
+  PythSolanaReceiver,
+  PythTransactionBuilder,
+} from "./PythSolanaReceiver";
 export {
   TransactionBuilder,
   InstructionWithEphemeralSigners,