adding prepare order

Author: barnjamin

solana/ts/src/common/index.ts

@@ -44,10 +44,7 @@ export async function reclaimCctpMessageIx(
     const { payer, cctpMessage: messageSentEventData } = accounts;
 
     return messageTransmitter.reclaimEventAccountIx(
-        {
-            payee: payer,
-            messageSentEventData,
-        },
+        { payee: payer, messageSentEventData },
         cctpAttestation,
     );
 }

solana/ts/src/protocol/tokenRouter.ts

@@ -1,6 +1,8 @@
+import * as splToken from "@solana/spl-token";
 import {
     AddressLookupTableAccount,
     Connection,
+    Keypair,
     PublicKey,
     TransactionInstruction,
     TransactionMessage,
@@ -10,15 +12,14 @@ import {
     FastTransfer,
     TokenRouter,
 } from "@wormhole-foundation/example-liquidity-layer-definitions";
-import { Chain, Network, Platform } from "@wormhole-foundation/sdk-base";
+import { Chain, Network, Platform, toChainId } from "@wormhole-foundation/sdk-base";
 import {
     AccountAddress,
     ChainAddress,
     ChainsConfig,
     CircleBridge,
     Contracts,
     UnsignedTransaction,
-    VAA,
 } from "@wormhole-foundation/sdk-definitions";
 import {
     AnySolanaAddress,
@@ -88,8 +89,60 @@ export class SolanaTokenRouter<N extends Network, C extends SolanaChains>
         yield this.createUnsignedTx({ transaction }, "TokenRouter.Initialize");
     }
 
-    getInitialAuctionFee(): Promise<bigint> {
-        throw new Error("Method not implemented.");
+    async *prepareMarketOrder(
+        sender: AnySolanaAddress,
+        amount: bigint,
+        redeemer: ChainAddress<Chain>,
+        minAmountOut?: bigint,
+        redeemerMessage?: Uint8Array,
+        preparedOrder?: Keypair,
+    ) {
+        const payer = new SolanaAddress(sender).unwrap();
+
+        // assume sender token is the usdc mint address
+        const senderToken = splToken.getAssociatedTokenAddressSync(this.mint, payer);
+
+        // Where we'll write the prepared order
+        preparedOrder = preparedOrder ?? Keypair.generate();
+
+        const [approveIx, prepareIx] = await this.prepareMarketOrderIx(
+            {
+                payer,
+                senderToken,
+                preparedOrder: preparedOrder.publicKey,
+            },
+            {
+                amountIn: amount,
+                minAmountOut: minAmountOut !== undefined ? minAmountOut : null,
+                targetChain: toChainId(redeemer.chain),
+                redeemer: Array.from(redeemer.address.toUniversalAddress().toUint8Array()),
+                redeemerMessage: redeemerMessage ? Buffer.from(redeemerMessage) : Buffer.from(""),
+            },
+        );
+
+        // TODO: fix prepareMarketOrderIx to not return null at all
+        const ixs = [];
+        if (approveIx) ixs.push(approveIx);
+        ixs.push(prepareIx);
+
+        const transaction = this.createTx(payer, ixs);
+        yield this.createUnsignedTx(
+            { transaction, signers: [preparedOrder] },
+            "TokenRouter.PrepareMarketOrder",
+        );
+    }
+
+    async *closePreparedOrder(sender: AnySolanaAddress, order: AnySolanaAddress) {
+        const payer = new SolanaAddress(sender).unwrap();
+        const preparedOrder = new SolanaAddress(order).unwrap();
+
+        const ix = await this.closePreparedOrderIx({
+            preparedOrder,
+            preparedBy: payer,
+        });
+
+        const transaction = this.createTx(payer, [ix]);
+        yield this.createUnsignedTx({ transaction }, "TokenRouter.ClosePreparedOrder");
     }
 
     placeMarketOrder(
@@ -101,6 +154,7 @@ export class SolanaTokenRouter<N extends Network, C extends SolanaChains>
     ): AsyncGenerator<UnsignedTransaction<N, C>, any, unknown> {
         throw new Error("Method not implemented.");
     }
+
     placeFastMarketOrder<RC extends Chain>(
         amount: bigint,
         chain: RC,
@@ -113,6 +167,7 @@ export class SolanaTokenRouter<N extends Network, C extends SolanaChains>
     ): AsyncGenerator<UnsignedTransaction<N, C>, any, unknown> {
         throw new Error("Method not implemented.");
     }
+
     redeemFill(
         vaa: FastTransfer.VAA,
         cctp: CircleBridge.Attestation,

solana/ts/src/tokenRouter/state/PreparedOrder.ts

@@ -1,7 +1,5 @@
 import { BN } from "@coral-xyz/anchor";
 import { PublicKey } from "@solana/web3.js";
-import { Keccak } from "sha3";
-import { Uint64, uint64ToBN } from "../../common";
 
 export type OrderType = {
     market?: {