@pythnetwork/pyth-solana-receiver 0.1.0

package/LICENSE

@@ -0,0 +1,13 @@
+Copyright 2023 Pyth Contributors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

package/lib/PythSolanaReceiver.d.ts

@@ -0,0 +1,49 @@
+/// <reference types="node" />
+import { AnchorProvider, Program } from "@coral-xyz/anchor";
+import { Connection, Signer, VersionedTransaction } from "@solana/web3.js";
+import { PythSolanaReceiver as PythSolanaReceiverProgram } from "./idl/pyth_solana_receiver";
+import { WormholeCoreBridgeSolana } from "./idl/wormhole_core_bridge_solana";
+import { PublicKey } from "@solana/web3.js";
+import { Wallet } from "@coral-xyz/anchor/dist/cjs/provider";
+import { InstructionWithEphemeralSigners } from "@pythnetwork/solana-utils";
+import { PriorityFeeConfig } from "@pythnetwork/solana-utils/lib/transaction";
+export declare const DEFAULT_TREASURY_ID = 0;
+export declare class PythSolanaReceiver {
+    readonly connection: Connection;
+    readonly wallet: Wallet;
+    readonly provider: AnchorProvider;
+    readonly receiver: Program<PythSolanaReceiverProgram>;
+    readonly wormhole: Program<WormholeCoreBridgeSolana>;
+    constructor({ connection, wallet, wormholeProgramId, receiverProgramId, }: {
+        connection: Connection;
+        wallet: Wallet;
+        wormholeProgramId?: PublicKey;
+        receiverProgramId?: PublicKey;
+    });
+    withPriceUpdate(priceUpdateDataArray: string[], getInstructions: (priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>) => Promise<InstructionWithEphemeralSigners[]>, priorityFeeConfig?: PriorityFeeConfig): Promise<{
+        tx: VersionedTransaction;
+        signers: Signer[];
+    }[]>;
+    withPartiallyVerifiedPriceUpdate(priceUpdateDataArray: string[], getInstructions: (priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>) => Promise<InstructionWithEphemeralSigners[]>, priorityFeeConfig?: PriorityFeeConfig): Promise<{
+        tx: VersionedTransaction;
+        signers: Signer[];
+    }[]>;
+    buildPostPriceUpdateAtomicInstructions(priceUpdateDataArray: string[]): Promise<{
+        postInstructions: InstructionWithEphemeralSigners[];
+        priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>;
+        cleanupInstructions: InstructionWithEphemeralSigners[];
+    }>;
+    buildPostEncodedVaaInstructions(vaa: Buffer): Promise<{
+        postInstructions: InstructionWithEphemeralSigners[];
+        encodedVaaAddress: PublicKey;
+        cleanupInstructions: InstructionWithEphemeralSigners[];
+    }>;
+    buildPostPriceUpdateInstructions(priceUpdateDataArray: string[]): Promise<{
+        postInstructions: InstructionWithEphemeralSigners[];
+        priceFeedIdToPriceUpdateAccount: Record<string, PublicKey>;
+        cleanupInstructions: InstructionWithEphemeralSigners[];
+    }>;
+    buildCloseEncodedVaaInstruction(encodedVaa: PublicKey): Promise<InstructionWithEphemeralSigners>;
+    buildClosePriceUpdateInstruction(priceUpdateAccount: PublicKey): Promise<InstructionWithEphemeralSigners>;
+}
+//# sourceMappingURL=PythSolanaReceiver.d.ts.map

package/lib/PythSolanaReceiver.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PythSolanaReceiver.d.ts","sourceRoot":"","sources":["../src/PythSolanaReceiver.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAC3E,OAAO,EACL,kBAAkB,IAAI,yBAAyB,EAEhD,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,wBAAwB,EAEzB,MAAM,mCAAmC,CAAC;AAQ3C,OAAO,EAAE,SAAS,EAAW,MAAM,iBAAiB,CAAC;AAUrD,OAAO,EAAE,MAAM,EAAE,MAAM,qCAAqC,CAAC;AAO7D,OAAO,EAEL,+BAA+B,EAChC,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,iBAAiB,EAAE,MAAM,2CAA2C,CAAC;AAE9E,eAAO,MAAM,mBAAmB,IAAI,CAAC;AAErC,qBAAa,kBAAkB;IAC7B,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAChC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAC;IAClC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,yBAAyB,CAAC,CAAC;IACtD,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,wBAAwB,CAAC,CAAC;gBAEzC,EACV,UAAU,EACV,MAAM,EACN,iBAA+C,EAC/C,iBAA+C,GAChD,EAAE;QACD,UAAU,EAAE,UAAU,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC;QACf,iBAAiB,CAAC,EAAE,SAAS,CAAC;QAC9B,iBAAiB,CAAC,EAAE,SAAS,CAAC;KAC/B;IAkBK,eAAe,CACnB,oBAAoB,EAAE,MAAM,EAAE,EAC9B,eAAe,EAAE,CACf,+BAA+B,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,KACvD,OAAO,CAAC,+BAA+B,EAAE,CAAC,EAC/C,iBAAiB,CAAC,EAAE,iBAAiB,GACpC,OAAO,CAAC;QAAE,EAAE,EAAE,oBAAoB,CAAC;QAAC,OAAO,EAAE,MAAM,EAAE,CAAA;KAAE,EAAE,CAAC;IAkBvD,gCAAgC,CACpC,oBAAoB,EAAE,MAAM,EAAE,EAC9B,eAAe,EAAE,CACf,+BAA+B,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,KACvD,OAAO,CAAC,+BAA+B,EAAE,CAAC,EAC/C,iBAAiB,CAAC,EAAE,iBAAiB,GACpC,OAAO,CAAC;QAAE,EAAE,EAAE,oBAAoB,CAAC;QAAC,OAAO,EAAE,MAAM,EAAE,CAAA;KAAE,EAAE,CAAC;IAkBvD,sCAAsC,CAC1C,oBAAoB,EAAE,MAAM,EAAE,GAC7B,OAAO,CAAC;QACT,gBAAgB,EAAE,+BAA+B,EAAE,CAAC;QACpD,+BAA+B,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC3D,mBAAmB,EAAE,+BAA+B,EAAE,CAAC;KACxD,CAAC;IAiDI,+BAA+B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAC1D,gBAAgB,EAAE,+BAA+B,EAAE,CAAC;QACpD,iBAAiB,EAAE,SAAS,CAAC;QAC7B,mBAAmB,EAAE,+BAA+B,EAAE,CAAC;KACxD,CAAC;IAsDI,gCAAgC,CACpC,oBAAoB,EAAE,MAAM,EAAE,GAC7B,OAAO,CAAC;QACT,gBAAgB,EAAE,+BAA+B,EAAE,CAAC;QACpD,+BAA+B,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC3D,mBAAmB,EAAE,+BAA+B,EAAE,CAAC;KACxD,CAAC;IAuDI,+BAA+B,CACnC,UAAU,EAAE,SAAS,GACpB,OAAO,CAAC,+BAA+B,CAAC;IAQrC,gCAAgC,CACpC,kBAAkB,EAAE,SAAS,GAC5B,OAAO,CAAC,+BAA+B,CAAC;CAO5C"}

package/lib/PythSolanaReceiver.js

@@ -0,0 +1,168 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PythSolanaReceiver = exports.DEFAULT_TREASURY_ID = void 0;
+const anchor_1 = require("@coral-xyz/anchor");
+const pyth_solana_receiver_1 = require("./idl/pyth_solana_receiver");
+const wormhole_core_bridge_solana_1 = require("./idl/wormhole_core_bridge_solana");
+const address_1 = require("./address");
+const web3_js_1 = require("@solana/web3.js");
+const price_service_sdk_1 = require("@pythnetwork/price-service-sdk");
+const compute_budget_1 = require("./compute_budget");
+const vaa_1 = require("./vaa");
+const solana_utils_1 = require("@pythnetwork/solana-utils");
+exports.DEFAULT_TREASURY_ID = 0;
+class PythSolanaReceiver {
+    connection;
+    wallet;
+    provider;
+    receiver;
+    wormhole;
+    constructor({ connection, wallet, wormholeProgramId = address_1.DEFAULT_WORMHOLE_PROGRAM_ID, receiverProgramId = address_1.DEFAULT_RECEIVER_PROGRAM_ID, }) {
+        this.connection = connection;
+        this.wallet = wallet;
+        this.provider = new anchor_1.AnchorProvider(this.connection, this.wallet, {
+            commitment: connection.commitment,
+        });
+        this.receiver = new anchor_1.Program(pyth_solana_receiver_1.IDL, receiverProgramId, this.provider);
+        this.wormhole = new anchor_1.Program(wormhole_core_bridge_solana_1.IDL, wormholeProgramId, this.provider);
+    }
+    async withPriceUpdate(priceUpdateDataArray, getInstructions, priorityFeeConfig) {
+        const { postInstructions, priceFeedIdToPriceUpdateAccount: priceFeedIdToPriceUpdateAccount, cleanupInstructions, } = await this.buildPostPriceUpdateInstructions(priceUpdateDataArray);
+        return solana_utils_1.TransactionBuilder.batchIntoVersionedTransactions(this.wallet.publicKey, this.connection, [
+            ...postInstructions,
+            ...(await getInstructions(priceFeedIdToPriceUpdateAccount)),
+            ...cleanupInstructions,
+        ], priorityFeeConfig ?? {});
+    }
+    async withPartiallyVerifiedPriceUpdate(priceUpdateDataArray, getInstructions, priorityFeeConfig) {
+        const { postInstructions, priceFeedIdToPriceUpdateAccount, cleanupInstructions, } = await this.buildPostPriceUpdateAtomicInstructions(priceUpdateDataArray);
+        return solana_utils_1.TransactionBuilder.batchIntoVersionedTransactions(this.wallet.publicKey, this.connection, [
+            ...postInstructions,
+            ...(await getInstructions(priceFeedIdToPriceUpdateAccount)),
+            ...cleanupInstructions,
+        ], priorityFeeConfig ?? {});
+    }
+    async buildPostPriceUpdateAtomicInstructions(priceUpdateDataArray) {
+        const postInstructions = [];
+        const priceFeedIdToPriceUpdateAccount = {};
+        const cleanupInstructions = [];
+        for (const priceUpdateData of priceUpdateDataArray) {
+            const accumulatorUpdateData = (0, price_service_sdk_1.parseAccumulatorUpdateData)(Buffer.from(priceUpdateData, "base64"));
+            const guardianSetIndex = (0, vaa_1.getGuardianSetIndex)(accumulatorUpdateData.vaa);
+            const trimmedVaa = (0, vaa_1.trimSignatures)(accumulatorUpdateData.vaa);
+            for (const update of accumulatorUpdateData.updates) {
+                const priceUpdateKeypair = new web3_js_1.Keypair();
+                postInstructions.push({
+                    instruction: await this.receiver.methods
+                        .postUpdateAtomic({
+                        vaa: trimmedVaa,
+                        merklePriceUpdate: update,
+                        treasuryId: exports.DEFAULT_TREASURY_ID,
+                    })
+                        .accounts({
+                        priceUpdateAccount: priceUpdateKeypair.publicKey,
+                        treasury: (0, address_1.getTreasuryPda)(exports.DEFAULT_TREASURY_ID),
+                        config: (0, address_1.getConfigPda)(),
+                        guardianSet: (0, address_1.getGuardianSetPda)(guardianSetIndex),
+                    })
+                        .instruction(),
+                    signers: [priceUpdateKeypair],
+                    computeUnits: compute_budget_1.POST_UPDATE_ATOMIC_COMPUTE_BUDGET,
+                });
+                priceFeedIdToPriceUpdateAccount["0x" + (0, price_service_sdk_1.parsePriceFeedMessage)(update.message).feedId.toString("hex")] = priceUpdateKeypair.publicKey;
+                cleanupInstructions.push(await this.buildClosePriceUpdateInstruction(priceUpdateKeypair.publicKey));
+            }
+        }
+        return {
+            postInstructions,
+            priceFeedIdToPriceUpdateAccount,
+            cleanupInstructions,
+        };
+    }
+    async buildPostEncodedVaaInstructions(vaa) {
+        const postInstructions = [];
+        const cleanupInstructions = [];
+        const encodedVaaKeypair = new web3_js_1.Keypair();
+        const guardianSetIndex = (0, vaa_1.getGuardianSetIndex)(vaa);
+        postInstructions.push(await (0, vaa_1.buildEncodedVaaCreateInstruction)(this.wormhole, vaa, encodedVaaKeypair));
+        postInstructions.push({
+            instruction: await this.wormhole.methods
+                .initEncodedVaa()
+                .accounts({
+                encodedVaa: encodedVaaKeypair.publicKey,
+            })
+                .instruction(),
+            signers: [],
+        });
+        postInstructions.push(...(await (0, vaa_1.buildWriteEncodedVaaWithSplit)(this.wormhole, vaa, encodedVaaKeypair.publicKey)));
+        postInstructions.push({
+            instruction: await this.wormhole.methods
+                .verifyEncodedVaaV1()
+                .accounts({
+                guardianSet: (0, address_1.getGuardianSetPda)(guardianSetIndex),
+                draftVaa: encodedVaaKeypair.publicKey,
+            })
+                .instruction(),
+            signers: [],
+            computeUnits: compute_budget_1.VERIFY_ENCODED_VAA_COMPUTE_BUDGET,
+        });
+        cleanupInstructions.push(await this.buildCloseEncodedVaaInstruction(encodedVaaKeypair.publicKey));
+        return {
+            postInstructions,
+            encodedVaaAddress: encodedVaaKeypair.publicKey,
+            cleanupInstructions,
+        };
+    }
+    async buildPostPriceUpdateInstructions(priceUpdateDataArray) {
+        const postInstructions = [];
+        const priceFeedIdToPriceUpdateAccount = {};
+        const cleanupInstructions = [];
+        for (const priceUpdateData of priceUpdateDataArray) {
+            const accumulatorUpdateData = (0, price_service_sdk_1.parseAccumulatorUpdateData)(Buffer.from(priceUpdateData, "base64"));
+            const { postInstructions: postEncodedVaaInstructions, encodedVaaAddress: encodedVaa, cleanupInstructions: postEncodedVaaCleanupInstructions, } = await this.buildPostEncodedVaaInstructions(accumulatorUpdateData.vaa);
+            postInstructions.push(...postEncodedVaaInstructions);
+            cleanupInstructions.push(...postEncodedVaaCleanupInstructions);
+            for (const update of accumulatorUpdateData.updates) {
+                const priceUpdateKeypair = new web3_js_1.Keypair();
+                postInstructions.push({
+                    instruction: await this.receiver.methods
+                        .postUpdate({
+                        merklePriceUpdate: update,
+                        treasuryId: exports.DEFAULT_TREASURY_ID,
+                    })
+                        .accounts({
+                        encodedVaa,
+                        priceUpdateAccount: priceUpdateKeypair.publicKey,
+                        treasury: (0, address_1.getTreasuryPda)(exports.DEFAULT_TREASURY_ID),
+                        config: (0, address_1.getConfigPda)(),
+                    })
+                        .instruction(),
+                    signers: [priceUpdateKeypair],
+                    computeUnits: compute_budget_1.POST_UPDATE_COMPUTE_BUDGET,
+                });
+                priceFeedIdToPriceUpdateAccount["0x" + (0, price_service_sdk_1.parsePriceFeedMessage)(update.message).feedId.toString("hex")] = priceUpdateKeypair.publicKey;
+                cleanupInstructions.push(await this.buildClosePriceUpdateInstruction(priceUpdateKeypair.publicKey));
+            }
+        }
+        return {
+            postInstructions,
+            priceFeedIdToPriceUpdateAccount,
+            cleanupInstructions,
+        };
+    }
+    async buildCloseEncodedVaaInstruction(encodedVaa) {
+        const instruction = await this.wormhole.methods
+            .closeEncodedVaa()
+            .accounts({ encodedVaa })
+            .instruction();
+        return { instruction, signers: [] };
+    }
+    async buildClosePriceUpdateInstruction(priceUpdateAccount) {
+        const instruction = await this.receiver.methods
+            .reclaimRent()
+            .accounts({ priceUpdateAccount })
+            .instruction();
+        return { instruction, signers: [] };
+    }
+}
+exports.PythSolanaReceiver = PythSolanaReceiver;

package/lib/address.d.ts

@@ -0,0 +1,7 @@
+import { PublicKey } from "@solana/web3.js";
+export declare const DEFAULT_RECEIVER_PROGRAM_ID: PublicKey;
+export declare const DEFAULT_WORMHOLE_PROGRAM_ID: PublicKey;
+export declare const getGuardianSetPda: (guardianSetIndex: number) => PublicKey;
+export declare const getTreasuryPda: (treasuryId: number) => PublicKey;
+export declare const getConfigPda: () => PublicKey;
+//# sourceMappingURL=address.d.ts.map

package/lib/address.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"address.d.ts","sourceRoot":"","sources":["../src/address.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAE5C,eAAO,MAAM,2BAA2B,WAEvC,CAAC;AACF,eAAO,MAAM,2BAA2B,WAEvC,CAAC;AAEF,eAAO,MAAM,iBAAiB,qBAAsB,MAAM,cAOzD,CAAC;AAEF,eAAO,MAAM,cAAc,eAAgB,MAAM,cAKhD,CAAC;AAEF,eAAO,MAAM,YAAY,iBAKxB,CAAC"}

package/lib/address.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getConfigPda = exports.getTreasuryPda = exports.getGuardianSetPda = exports.DEFAULT_WORMHOLE_PROGRAM_ID = exports.DEFAULT_RECEIVER_PROGRAM_ID = void 0;
+const web3_js_1 = require("@solana/web3.js");
+exports.DEFAULT_RECEIVER_PROGRAM_ID = new web3_js_1.PublicKey("rec5EKMGg6MxZYaMdyBfgwp4d5rB9T1VQH5pJv5LtFJ");
+exports.DEFAULT_WORMHOLE_PROGRAM_ID = new web3_js_1.PublicKey("HDwcJBJXjL9FpJ7UBsYBtaDjsBUhuLCUYoz3zr8SWWaQ");
+const getGuardianSetPda = (guardianSetIndex) => {
+    const guardianSetIndexBuf = Buffer.alloc(4);
+    guardianSetIndexBuf.writeUInt32BE(guardianSetIndex, 0);
+    return web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("GuardianSet"), guardianSetIndexBuf], exports.DEFAULT_WORMHOLE_PROGRAM_ID)[0];
+};
+exports.getGuardianSetPda = getGuardianSetPda;
+const getTreasuryPda = (treasuryId) => {
+    return web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("treasury"), Buffer.from([treasuryId])], exports.DEFAULT_RECEIVER_PROGRAM_ID)[0];
+};
+exports.getTreasuryPda = getTreasuryPda;
+const getConfigPda = () => {
+    return web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("config")], exports.DEFAULT_RECEIVER_PROGRAM_ID)[0];
+};
+exports.getConfigPda = getConfigPda;

package/lib/compute_budget.d.ts

@@ -0,0 +1,4 @@
+export declare const VERIFY_ENCODED_VAA_COMPUTE_BUDGET = 400000;
+export declare const POST_UPDATE_ATOMIC_COMPUTE_BUDGET = 400000;
+export declare const POST_UPDATE_COMPUTE_BUDGET = 200000;
+//# sourceMappingURL=compute_budget.d.ts.map

package/lib/compute_budget.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"compute_budget.d.ts","sourceRoot":"","sources":["../src/compute_budget.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,iCAAiC,SAAS,CAAC;AACxD,eAAO,MAAM,iCAAiC,SAAS,CAAC;AACxD,eAAO,MAAM,0BAA0B,SAAS,CAAC"}

package/lib/compute_budget.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.POST_UPDATE_COMPUTE_BUDGET = exports.POST_UPDATE_ATOMIC_COMPUTE_BUDGET = exports.VERIFY_ENCODED_VAA_COMPUTE_BUDGET = void 0;
+exports.VERIFY_ENCODED_VAA_COMPUTE_BUDGET = 400000;
+exports.POST_UPDATE_ATOMIC_COMPUTE_BUDGET = 400000;
+exports.POST_UPDATE_COMPUTE_BUDGET = 200000;