feat(price_pusher): solana price pusher (#1408)

* Checkpoint

* Checkpoint

* Checkpoint

* Checkpoint

* fix: pusher

* Checkpoint

* Works

* fix: pass pusher program id

* Add docs

* 0.1.0

* Bump npm package

* Go

* Comment

* Add customizable shard id

* Allow configurable priority fees

Author: guibescos

package-lock.json

@@ -0,0 +1,4 @@
+{
+  "endpoint": "https://api.devnet.solana.com",
+  "keypair-file": "/keypair"
+}

price_pusher/config.solana.testnet.sample.json

@@ -52,6 +52,7 @@
   },
   "dependencies": {
     "@injectivelabs/sdk-ts": "1.10.72",
+    "@pythnetwork/pyth-solana-receiver": "*",
     "@mysten/sui.js": "^0.49.1",
     "@pythnetwork/price-service-client": "*",
     "@pythnetwork/pyth-sdk-solidity": "*",

price_pusher/package.json

@@ -6,6 +6,7 @@ import evm from "./evm/command";
 import aptos from "./aptos/command";
 import sui from "./sui/command";
 import near from "./near/command";
+import solana from "./solana/command";
 
 yargs(hideBin(process.argv))
   .config("config")
@@ -15,4 +16,5 @@ yargs(hideBin(process.argv))
   .command(aptos)
   .command(sui)
   .command(near)
+  .command(solana)
   .help().argv;

price_pusher/src/index.ts

@@ -0,0 +1,115 @@
+import { Options } from "yargs";
+import * as options from "../options";
+import { readPriceConfigFile } from "../price-config";
+import { PriceServiceConnection } from "@pythnetwork/price-service-client";
+import { PythPriceListener } from "../pyth-price-listener";
+import { SolanaPriceListener, SolanaPricePusher } from "./solana";
+import { Controller } from "../controller";
+import { PythSolanaReceiver } from "@pythnetwork/pyth-solana-receiver";
+import NodeWallet from "@coral-xyz/anchor/dist/cjs/nodewallet";
+import { Keypair, Connection } from "@solana/web3.js";
+import fs from "fs";
+import { PublicKey } from "@solana/web3.js";
+
+export default {
+  command: "solana",
+  describe: "run price pusher for solana",
+  builder: {
+    endpoint: {
+      description: "Solana RPC API endpoint",
+      type: "string",
+      required: true,
+    } as Options,
+    "keypair-file": {
+      description: "Path to a keypair file",
+      type: "string",
+      required: true,
+    } as Options,
+    "shard-id": {
+      description: "Shard ID",
+      type: "number",
+      required: true,
+    } as Options,
+    "compute-unit-price-micro-lamports": {
+      description: "Priority fee per compute unit",
+      type: "number",
+      default: 50000,
+    } as Options,
+    ...options.priceConfigFile,
+    ...options.priceServiceEndpoint,
+    ...options.pythContractAddress,
+    ...options.pollingFrequency,
+    ...options.pushingFrequency,
+  },
+  handler: function (argv: any) {
+    const {
+      endpoint,
+      keypairFile,
+      shardId,
+      computeUnitPriceMicroLamports,
+      priceConfigFile,
+      priceServiceEndpoint,
+      pythContractAddress,
+      pushingFrequency,
+      pollingFrequency,
+    } = argv;
+
+    const priceConfigs = readPriceConfigFile(priceConfigFile);
+
+    const priceServiceConnection = new PriceServiceConnection(
+      priceServiceEndpoint,
+      {
+        logger: {
+          // Log only warnings and errors from the price service client
+          info: () => undefined,
+          warn: console.warn,
+          error: console.error,
+          debug: () => undefined,
+          trace: () => undefined,
+        },
+      }
+    );
+
+    const priceItems = priceConfigs.map(({ id, alias }) => ({ id, alias }));
+
+    const pythListener = new PythPriceListener(
+      priceServiceConnection,
+      priceItems
+    );
+
+    const wallet = new NodeWallet(
+      Keypair.fromSecretKey(
+        Uint8Array.from(JSON.parse(fs.readFileSync(keypairFile, "ascii")))
+      )
+    );
+
+    const pythSolanaReceiver = new PythSolanaReceiver({
+      connection: new Connection(endpoint),
+      wallet,
+      pushOracleProgramId: new PublicKey(pythContractAddress),
+    });
+
+    const solanaPricePusher = new SolanaPricePusher(
+      pythSolanaReceiver,
+      priceServiceConnection,
+      shardId,
+      computeUnitPriceMicroLamports
+    );
+    const solanaPriceListener = new SolanaPriceListener(
+      pythSolanaReceiver,
+      shardId,
+      priceItems,
+      { pollingFrequency }
+    );
+
+    const controller = new Controller(
+      priceConfigs,
+      pythListener,
+      solanaPriceListener,
+      solanaPricePusher,
+      { pushingFrequency }
+    );
+
+    controller.start();
+  },
+};

price_pusher/src/solana/command.ts

@@ -0,0 +1,95 @@
+import { PythSolanaReceiver } from "@pythnetwork/pyth-solana-receiver";
+import {
+  ChainPriceListener,
+  IPricePusher,
+  PriceInfo,
+  PriceItem,
+} from "../interface";
+import { DurationInSeconds } from "../utils";
+import { PriceServiceConnection } from "@pythnetwork/price-service-client";
+
+export class SolanaPriceListener extends ChainPriceListener {
+  constructor(
+    private pythSolanaReceiver: PythSolanaReceiver,
+    private shardId: number,
+    priceItems: PriceItem[],
+    config: {
+      pollingFrequency: DurationInSeconds;
+    }
+  ) {
+    super("solana", config.pollingFrequency, priceItems);
+  }
+
+  async getOnChainPriceInfo(priceId: string): Promise<PriceInfo | undefined> {
+    try {
+      const priceFeedAccount =
+        await this.pythSolanaReceiver.fetchPriceFeedAccount(
+          this.shardId,
+          Buffer.from(priceId, "hex")
+        );
+      console.log(
+        `Polled a Solana on chain price for feed ${this.priceIdToAlias.get(
+          priceId
+        )} (${priceId}).`
+      );
+      if (priceFeedAccount) {
+        return {
+          conf: priceFeedAccount.priceMessage.conf.toString(),
+          price: priceFeedAccount.priceMessage.price.toString(),
+          publishTime: priceFeedAccount.priceMessage.publishTime.toNumber(),
+        };
+      } else {
+        return undefined;
+      }
+    } catch (e) {
+      console.error(`Polling on-chain price for ${priceId} failed. Error:`);
+      console.error(e);
+      return undefined;
+    }
+  }
+}
+
+export class SolanaPricePusher implements IPricePusher {
+  constructor(
+    private pythSolanaReceiver: PythSolanaReceiver,
+    private priceServiceConnection: PriceServiceConnection,
+    private shardId: number,
+    private computeUnitPriceMicroLamports: number
+  ) {}
+
+  async updatePriceFeed(
+    priceIds: string[],
+    pubTimesToPush: number[]
+  ): Promise<void> {
+    if (priceIds.length === 0) {
+      return;
+    }
+
+    let priceFeedUpdateData;
+    try {
+      priceFeedUpdateData = await this.priceServiceConnection.getLatestVaas(
+        priceIds
+      );
+    } catch (e: any) {
+      console.error(new Date(), "getPriceFeedsUpdateData failed:", e);
+      return;
+    }
+
+    const transactionBuilder = this.pythSolanaReceiver.newTransactionBuilder({
+      closeUpdateAccounts: false,
+    });
+    transactionBuilder.addUpdatePriceFeed(priceFeedUpdateData, this.shardId);
+
+    try {
+      await this.pythSolanaReceiver.provider.sendAll(
+        await transactionBuilder.buildVersionedTransactions({
+          computeUnitPriceMicroLamports: this.computeUnitPriceMicroLamports,
+        })
+      );
+      console.log(new Date(), "updatePriceFeed successful");
+    } catch (e: any) {
+      console.error(new Date(), "updatePriceFeed failed", e);
+      return;
+    }
+  }
+}

price_pusher/src/solana/solana.ts

@@ -89,10 +89,14 @@ pub struct UpdatePriceFeed<'info> {
     #[account(mut)]
     pub payer:                Signer<'info>,
     pub pyth_solana_receiver: Program<'info, PythSolanaReceiver>,
+    /// CHECK: Checked by CPI into the Pyth Solana Receiver
     pub encoded_vaa:          AccountInfo<'info>,
+    /// CHECK: Checked by CPI into the Pyth Solana Receiver
     pub config:               AccountInfo<'info>,
+    /// CHECK: Checked by CPI into the Pyth Solana Receiver
     #[account(mut)]
     pub treasury:             AccountInfo<'info>,
+    /// CHECK: This account's seeds are checked
     #[account(mut, seeds = [&shard_id.to_le_bytes(), &feed_id], bump)]
     pub price_feed_account:   AccountInfo<'info>,
     pub system_program:       Program<'info, System>,

target_chains/solana/programs/pyth-push-oracle/src/lib.rs

@@ -12,4 +12,4 @@ pub mod price_update;
 
 declare_id!("rec5EKMGg6MxZYaMdyBfgwp4d5rB9T1VQH5pJv5LtFJ");
 
-pub const PYTH_PUSH_ORACLE_ID: Pubkey = pubkey!("F9SP6tBXw9Af7BYauo7Y2R5Es2mpv8FP5aNCXMihp6Za");
+pub const PYTH_PUSH_ORACLE_ID: Pubkey = pubkey!("pythWSnswVUd12oZpeFP8e9CVaEqJg25g1Vtc2biRsT");

target_chains/solana/pyth_solana_receiver_sdk/src/lib.rs

@@ -1,6 +1,6 @@
 {
   "name": "@pythnetwork/pyth-solana-receiver",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Pyth solana receiver SDK",
   "homepage": "https://pyth.network",
   "main": "lib/index.js",