Add wallet model support for Dedicated ETH Staking

src/coinbase/address/wallet_address.ts

@@ -1,6 +1,6 @@
 import { Decimal } from "decimal.js";
 import { ethers } from "ethers";
-import { Address as AddressModel, StakingOperationStatusEnum } from "../../client";
+import { Address as AddressModel } from "../../client";
 import { Address } from "../address";
 import { Asset } from "../asset";
 import { Coinbase } from "../coinbase";
@@ -253,7 +253,7 @@ export class WalletAddress extends Address {
   /**
    * Trades the given amount of the given Asset for another Asset. Only same-network Trades are supported.
    *
-   * @param options = The options to create the Trade.
+   * @param options - = The options to create the Trade.
    * @param options.amount - The amount of the From Asset to send.
    * @param options.fromAssetId - The ID of the Asset to trade from.
    * @param options.toAssetId - The ID of the Asset to trade to.
@@ -503,32 +503,34 @@ export class WalletAddress extends Address {
       options,
     );
 
-    // NOTE: Staking does not yet support server signers at this point.
-    await stakingOperation.sign(this.key!);
-    for (let i = 0; i < stakingOperation.getTransactions().length; i++) {
-      const transaction = stakingOperation.getTransactions()[0];
-      if (!transaction.isSigned()) {
-        continue;
+    const startTime = Date.now();
+
+    // Loop until the staking operation is complete.
+    while (!stakingOperation.isTerminalState() && Date.now() - startTime < timeoutSeconds * 1000) {
+      // Loop through any unsigned transactions that are available, sign and broadcast them.
+      for (let i = 0; i < stakingOperation.getTransactions().length; i++) {
+        const transaction = stakingOperation.getTransactions()[i];
+
+        if (!transaction.isSigned()) {
+          await transaction.sign(this.key!);
+
+          stakingOperation = await this.broadcastStakingOperationRequest(
+            stakingOperation,
+            transaction.getSignedPayload()!.slice(2),
+            i,
+          );
+        }
       }
-      stakingOperation = await this.broadcastStakingOperationRequest(
-        stakingOperation,
-        transaction.getSignedPayload()!.slice(2),
-        i,
-      );
-    }
 
-    const startTime = Date.now();
-    while (Date.now() - startTime < timeoutSeconds * 1000) {
       await stakingOperation.reload();
-      const status = stakingOperation.getStatus();
-      if (
-        status === StakingOperationStatusEnum.Complete ||
-        status === StakingOperationStatusEnum.Failed
-      ) {
+
+      if (stakingOperation.isTerminalState()) {
         return stakingOperation;
       }
+
       await delay(intervalSeconds);
     }
+
     throw new Error("Staking Operation timed out");
   }
 

src/coinbase/staking_operation.ts

@@ -108,6 +108,15 @@ export class StakingOperation {
     return this.transactions;
   }
 
+  /**
+   * Return a human-readable string representation of the StakingOperation object.
+   *
+   * @returns The string representation of the StakingOperation object.
+   */
+  public toString(): string {
+    return JSON.stringify(this.model, null, 2);
+  }
+
   /**
    * Get signed voluntary exit messages for native eth unstaking
    *
@@ -133,6 +142,24 @@ export class StakingOperation {
    * @returns Whether the Staking operation is in a terminal State
    */
   isTerminalState(): boolean {
+    return this.isComplete() || this.isFailed();
+  }
+
+  /**
+   * Returns whether the Staking operation is in a failed state.
+   *
+   * @returns Whether the Staking operation is in a failed state.
+   */
+  isFailed(): boolean {
+    return this.getStatus() === StakingOperationStatusEnum.Failed;
+  }
+
+  /**
+   * Returns whether the Staking operation is in a complete state.
+   *
+   * @returns Whether the Staking operation is in a complete state.
+   */
+  isComplete(): boolean {
     return this.getStatus() === StakingOperationStatusEnum.Complete;
   }
 

src/coinbase/transaction.ts

@@ -9,7 +9,6 @@ import { parseUnsignedPayload } from "./utils";
 export class Transaction {
   private model: TransactionModel;
   private raw?: ethers.Transaction;
-  private signed: boolean | undefined;
 
   /**
    * Transactions should be constructed via higher level abstractions like Trade or Transfer.
@@ -136,7 +135,6 @@ export class Transaction {
   async sign(key: ethers.Wallet) {
     const signedPayload = await key!.signTransaction(this.rawTransaction());
     this.model.signed_payload = signedPayload;
-    this.signed = true;
     // Removes the '0x' prefix as required by the API.
     return signedPayload.slice(2);
   }
@@ -147,7 +145,7 @@ export class Transaction {
    * @returns if the transaction has been signed.
    */
   isSigned(): boolean | undefined {
-    return this.signed;
+    return !!this.model.signed_payload;
   }
 
   /**