@buildonspark/issuer-sdk 0.0.26 → 0.0.28

package/dist/index.cjs

@@ -37,10 +37,7 @@ if (typeof global === "undefined") {
 var import_lrc20_sdk2 = require("@buildonspark/lrc20-sdk");
 var import_spark_sdk = require("@buildonspark/spark-sdk");
 var import_address = require("@buildonspark/spark-sdk/address");
-var import_utils5 = require("@buildonspark/spark-sdk/utils");
-var import_utils6 = require("@noble/curves/abstract/utils");
-var import_bip39 = require("@scure/bip39");
-var import_english = require("@scure/bip39/wordlists/english");
+var import_utils5 = require("@noble/curves/abstract/utils");
 
 // src/services/freeze.ts
 var import_utils2 = require("@buildonspark/spark-sdk/utils");
@@ -335,6 +332,7 @@ function convertTokenActivityToHexEncoded(rawTransactions) {
 }
 
 // src/issuer-spark-wallet.ts
+var import_address2 = require("@buildonspark/spark-sdk/address");
 var BURN_ADDRESS = "02".repeat(33);
 var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkWallet {
   issuerTokenTransactionService;
@@ -342,7 +340,7 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkW
   tokenPublicKeyInfo;
   static async initialize(options) {
     const wallet = new _IssuerSparkWallet(options.options);
-    const initResponse = await wallet.initIssuerWallet(options.mnemonicOrSeed);
+    const initResponse = await wallet.initWallet(options.mnemonicOrSeed);
     return {
       wallet,
       ...initResponse
@@ -359,33 +357,6 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkW
       this.connectionManager
     );
   }
-  async initIssuerWallet(mnemonicOrSeed) {
-    const initResponse = await super.initWallet(mnemonicOrSeed);
-    mnemonicOrSeed = mnemonicOrSeed || initResponse?.mnemonic;
-    if (mnemonicOrSeed) {
-      let seed;
-      if (typeof mnemonicOrSeed !== "string") {
-        seed = mnemonicOrSeed;
-      } else {
-        if ((0, import_bip39.validateMnemonic)(mnemonicOrSeed, import_english.wordlist)) {
-          seed = await this.config.signer.mnemonicToSeed(mnemonicOrSeed);
-        } else {
-          seed = (0, import_utils6.hexToBytes)(mnemonicOrSeed);
-        }
-      }
-      const hdkey = (0, import_utils5.getMasterHDKeyFromSeed)(seed);
-      const accountType = this.config.getNetwork() === import_utils5.Network.REGTEST ? 0 : 1;
-      const identityKey = hdkey.derive(`m/8797555'/${accountType}'/0'`);
-      const network = this.config.getNetwork();
-      this.lrc20Wallet = new import_lrc20_sdk2.LRCWallet(
-        (0, import_utils6.bytesToHex)(identityKey.privateKey),
-        import_utils5.LRC_WALLET_NETWORK[network],
-        import_utils5.LRC_WALLET_NETWORK_TYPE[network],
-        this.config.lrc20ApiConfig
-      );
-    }
-    return initResponse;
-  }
   async getIssuerTokenBalance() {
     const publicKey = await super.getIdentityPublicKey();
     const balanceObj = await this.getBalance();
@@ -402,7 +373,7 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkW
     if (this.tokenPublicKeyInfo) {
       return convertToTokenPubKeyInfoResponse(this.tokenPublicKeyInfo);
     }
-    const tokenPublicKey = (0, import_utils6.bytesToHex)(this.lrc20Wallet.pubkey);
+    const tokenPublicKey = (0, import_utils5.bytesToHex)(this.lrc20Wallet.pubkey);
     const rawTokenPubkeyInfo = await this.lrc20Wallet.getTokenPubkeyInfo(tokenPublicKey);
     this.tokenPublicKeyInfo = rawTokenPubkeyInfo;
     if (!rawTokenPubkeyInfo) {
@@ -416,7 +387,7 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkW
   async mintTokens(tokenAmount) {
     var tokenPublicKey = await super.getIdentityPublicKey();
     const tokenTransaction = await this.issuerTokenTransactionService.constructMintTokenTransaction(
-      (0, import_utils6.hexToBytes)(tokenPublicKey),
+      (0, import_utils5.hexToBytes)(tokenPublicKey),
       tokenAmount
     );
     return await this.issuerTokenTransactionService.broadcastTokenTransaction(
@@ -438,11 +409,15 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkW
   async freezeTokens(ownerPublicKey) {
     await this.syncTokenLeaves();
     const tokenPublicKey = await super.getIdentityPublicKey();
+    const decodedOwnerPubkey = (0, import_address2.decodeSparkAddress)(
+      ownerPublicKey,
+      this.config.getNetworkType()
+    );
     const response = await this.tokenFreezeService.freezeTokens(
-      (0, import_utils6.hexToBytes)(ownerPublicKey),
-      (0, import_utils6.hexToBytes)(tokenPublicKey)
+      (0, import_utils5.hexToBytes)(decodedOwnerPubkey),
+      (0, import_utils5.hexToBytes)(tokenPublicKey)
     );
-    const tokenAmount = (0, import_utils6.bytesToNumberBE)(response.impactedTokenAmount);
+    const tokenAmount = (0, import_utils5.bytesToNumberBE)(response.impactedTokenAmount);
     return {
       impactedLeafIds: response.impactedLeafIds,
       impactedTokenAmount: tokenAmount
@@ -451,11 +426,15 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkW
   async unfreezeTokens(ownerPublicKey) {
     await this.syncTokenLeaves();
     const tokenPublicKey = await super.getIdentityPublicKey();
+    const decodedOwnerPubkey = (0, import_address2.decodeSparkAddress)(
+      ownerPublicKey,
+      this.config.getNetworkType()
+    );
     const response = await this.tokenFreezeService.unfreezeTokens(
-      (0, import_utils6.hexToBytes)(ownerPublicKey),
-      (0, import_utils6.hexToBytes)(tokenPublicKey)
+      (0, import_utils5.hexToBytes)(decodedOwnerPubkey),
+      (0, import_utils5.hexToBytes)(tokenPublicKey)
     );
-    const tokenAmount = (0, import_utils6.bytesToNumberBE)(response.impactedTokenAmount);
+    const tokenAmount = (0, import_utils5.bytesToNumberBE)(response.impactedTokenAmount);
     return {
       impactedLeafIds: response.impactedLeafIds,
       impactedTokenAmount: tokenAmount
@@ -464,7 +443,7 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkW
   async getTokenActivity(pageSize = 100, cursor, operationTypes, beforeTimestamp, afterTimestamp) {
     const lrc20Client = await this.lrc20ConnectionManager.createLrc20Client();
     const transactions = await lrc20Client.listTransactions({
-      tokenPublicKey: (0, import_utils6.hexToBytes)(await super.getIdentityPublicKey()),
+      tokenPublicKey: (0, import_utils5.hexToBytes)(await super.getIdentityPublicKey()),
       cursor,
       pageSize,
       beforeTimestamp,
@@ -476,8 +455,8 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends import_spark_sdk.SparkW
   async getIssuerTokenActivity(pageSize = 100, cursor, operationTypes, beforeTimestamp, afterTimestamp) {
     const lrc20Client = await this.lrc20ConnectionManager.createLrc20Client();
     const transactions = await lrc20Client.listTransactions({
-      tokenPublicKey: (0, import_utils6.hexToBytes)(await super.getIdentityPublicKey()),
-      ownerPublicKey: (0, import_utils6.hexToBytes)(await super.getIdentityPublicKey()),
+      tokenPublicKey: (0, import_utils5.hexToBytes)(await super.getIdentityPublicKey()),
+      ownerPublicKey: (0, import_utils5.hexToBytes)(await super.getIdentityPublicKey()),
       cursor,
       pageSize,
       beforeTimestamp,

package/dist/index.d.cts

@@ -77,7 +77,6 @@ declare class IssuerSparkWallet extends SparkWallet implements IssuerWalletInter
         wallet: IssuerSparkWallet;
     }>;
     private constructor();
-    private initIssuerWallet;
     getIssuerTokenBalance(): Promise<{
         balance: bigint;
     }>;

package/dist/index.d.ts

@@ -77,7 +77,6 @@ declare class IssuerSparkWallet extends SparkWallet implements IssuerWalletInter
         wallet: IssuerSparkWallet;
     }>;
     private constructor();
-    private initIssuerWallet;
     getIssuerTokenBalance(): Promise<{
         balance: bigint;
     }>;

package/dist/index.js

@@ -2,25 +2,16 @@ import "./chunk-GB7N6I5O.js";
 
 // src/issuer-spark-wallet.ts
 import {
-  LRCWallet,
   TokenPubkey,
   TokenPubkeyAnnouncement
 } from "@buildonspark/lrc20-sdk";
 import { SparkWallet } from "@buildonspark/spark-sdk";
 import { encodeSparkAddress } from "@buildonspark/spark-sdk/address";
-import {
-  getMasterHDKeyFromSeed,
-  LRC_WALLET_NETWORK,
-  LRC_WALLET_NETWORK_TYPE,
-  Network
-} from "@buildonspark/spark-sdk/utils";
 import {
   bytesToHex as bytesToHex2,
   bytesToNumberBE as bytesToNumberBE2,
   hexToBytes
 } from "@noble/curves/abstract/utils";
-import { validateMnemonic } from "@scure/bip39";
-import { wordlist } from "@scure/bip39/wordlists/english";
 
 // src/services/freeze.ts
 import { validateResponses } from "@buildonspark/spark-sdk/utils";
@@ -315,6 +306,7 @@ function convertTokenActivityToHexEncoded(rawTransactions) {
 }
 
 // src/issuer-spark-wallet.ts
+import { decodeSparkAddress } from "@buildonspark/spark-sdk/address";
 var BURN_ADDRESS = "02".repeat(33);
 var IssuerSparkWallet = class _IssuerSparkWallet extends SparkWallet {
   issuerTokenTransactionService;
@@ -322,7 +314,7 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends SparkWallet {
   tokenPublicKeyInfo;
   static async initialize(options) {
     const wallet = new _IssuerSparkWallet(options.options);
-    const initResponse = await wallet.initIssuerWallet(options.mnemonicOrSeed);
+    const initResponse = await wallet.initWallet(options.mnemonicOrSeed);
     return {
       wallet,
       ...initResponse
@@ -339,33 +331,6 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends SparkWallet {
       this.connectionManager
     );
   }
-  async initIssuerWallet(mnemonicOrSeed) {
-    const initResponse = await super.initWallet(mnemonicOrSeed);
-    mnemonicOrSeed = mnemonicOrSeed || initResponse?.mnemonic;
-    if (mnemonicOrSeed) {
-      let seed;
-      if (typeof mnemonicOrSeed !== "string") {
-        seed = mnemonicOrSeed;
-      } else {
-        if (validateMnemonic(mnemonicOrSeed, wordlist)) {
-          seed = await this.config.signer.mnemonicToSeed(mnemonicOrSeed);
-        } else {
-          seed = hexToBytes(mnemonicOrSeed);
-        }
-      }
-      const hdkey = getMasterHDKeyFromSeed(seed);
-      const accountType = this.config.getNetwork() === Network.REGTEST ? 0 : 1;
-      const identityKey = hdkey.derive(`m/8797555'/${accountType}'/0'`);
-      const network = this.config.getNetwork();
-      this.lrc20Wallet = new LRCWallet(
-        bytesToHex2(identityKey.privateKey),
-        LRC_WALLET_NETWORK[network],
-        LRC_WALLET_NETWORK_TYPE[network],
-        this.config.lrc20ApiConfig
-      );
-    }
-    return initResponse;
-  }
   async getIssuerTokenBalance() {
     const publicKey = await super.getIdentityPublicKey();
     const balanceObj = await this.getBalance();
@@ -418,8 +383,12 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends SparkWallet {
   async freezeTokens(ownerPublicKey) {
     await this.syncTokenLeaves();
     const tokenPublicKey = await super.getIdentityPublicKey();
+    const decodedOwnerPubkey = decodeSparkAddress(
+      ownerPublicKey,
+      this.config.getNetworkType()
+    );
     const response = await this.tokenFreezeService.freezeTokens(
-      hexToBytes(ownerPublicKey),
+      hexToBytes(decodedOwnerPubkey),
       hexToBytes(tokenPublicKey)
     );
     const tokenAmount = bytesToNumberBE2(response.impactedTokenAmount);
@@ -431,8 +400,12 @@ var IssuerSparkWallet = class _IssuerSparkWallet extends SparkWallet {
   async unfreezeTokens(ownerPublicKey) {
     await this.syncTokenLeaves();
     const tokenPublicKey = await super.getIdentityPublicKey();
+    const decodedOwnerPubkey = decodeSparkAddress(
+      ownerPublicKey,
+      this.config.getNetworkType()
+    );
     const response = await this.tokenFreezeService.unfreezeTokens(
-      hexToBytes(ownerPublicKey),
+      hexToBytes(decodedOwnerPubkey),
       hexToBytes(tokenPublicKey)
     );
     const tokenAmount = bytesToNumberBE2(response.impactedTokenAmount);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@buildonspark/issuer-sdk",
-  "version": "0.0.26",
+  "version": "0.0.28",
   "description": "Spark Issuer SDK for token issuance",
   "license": "Apache-2.0",
   "module": "./dist/index.js",
@@ -30,7 +30,6 @@
   ],
   "scripts": {
     "build": "yarn tsc && tsup",
-    "prepare": "yarn run build",
     "build:watch": "yarn build --watch --clean=false",
     "clean": "rm -rf dist",
     "circular-deps": "madge --circular --extensions ts,tsx src",
@@ -48,14 +47,15 @@
     "test-cmd": "node --experimental-vm-modules $(yarn bin jest) --no-cache --runInBand",
     "test": "echo \"Error: no tests yet\"",
     "test:integration": "yarn test-cmd src/tests/integration/*.test.ts",
+    "test:stress": "yarn test-cmd src/tests/stress/*.test.ts",
     "types:watch": "tsc-absolute --watch",
     "types": "tsc",
     "generate:proto": "PATH=\"$PATH:./node_modules/.bin\" protoc --ts_proto_out=./src/proto --ts_proto_opt=outputServices=nice-grpc,useExactTypes=false,outputServices=generic-definitions,oneof=unions,importSuffix=.js --proto_path=../../protos spark.proto mock.proto spark_authn.proto"
   },
   "dependencies": {
     "@bufbuild/protobuf": "^2.2.5",
-    "@buildonspark/lrc20-sdk": "0.0.24",
-    "@buildonspark/spark-sdk": "0.0.26",
+    "@buildonspark/lrc20-sdk": "0.0.26",
+    "@buildonspark/spark-sdk": "0.0.28",
     "@noble/curves": "^1.8.0",
     "@scure/bip39": "^1.5.4",
     "@scure/btc-signer": "^1.5.0",

package/src/issuer-spark-wallet.ts

@@ -2,9 +2,12 @@ import {
   LRCWallet,
   TokenPubkey,
   TokenPubkeyAnnouncement,
-  TokenPubkeyInfo
+  TokenPubkeyInfo,
 } from "@buildonspark/lrc20-sdk";
-import { ListAllTokenTransactionsCursor, OperationType } from "@buildonspark/lrc20-sdk/proto/rpc/v1/types";
+import {
+  ListAllTokenTransactionsCursor,
+  OperationType,
+} from "@buildonspark/lrc20-sdk/proto/rpc/v1/types";
 import { SparkWallet, SparkWalletProps } from "@buildonspark/spark-sdk";
 import { encodeSparkAddress } from "@buildonspark/spark-sdk/address";
 import { LeafWithPreviousTransactionData } from "@buildonspark/spark-sdk/proto/spark";
@@ -30,12 +33,14 @@ import {
   convertTokenActivityToHexEncoded,
   convertToTokenPubKeyInfoResponse,
 } from "./utils/type-mappers.js";
+import { decodeSparkAddress } from "@buildonspark/spark-sdk/address";
 
 const BURN_ADDRESS = "02".repeat(33);
 
 export class IssuerSparkWallet
   extends SparkWallet
-  implements IssuerWalletInterface {
+  implements IssuerWalletInterface
+{
   private issuerTokenTransactionService: IssuerTokenTransactionService;
   private tokenFreezeService: TokenFreezeService;
   private tokenPublicKeyInfo?: TokenPubkeyInfo;
@@ -43,7 +48,7 @@ export class IssuerSparkWallet
   public static async initialize(options: SparkWalletProps) {
     const wallet = new IssuerSparkWallet(options.options);
 
-    const initResponse = await wallet.initIssuerWallet(options.mnemonicOrSeed);
+    const initResponse = await wallet.initWallet(options.mnemonicOrSeed);
     return {
       wallet,
       ...initResponse,
@@ -62,39 +67,6 @@ export class IssuerSparkWallet
     );
   }
 
-  private async initIssuerWallet(mnemonicOrSeed?: string | Uint8Array) {
-    const initResponse = await super.initWallet(mnemonicOrSeed);
-
-    // TODO: Remove this in subsequent PRs when LRC20Wallet has a proper signer interface
-    mnemonicOrSeed = mnemonicOrSeed || initResponse?.mnemonic;
-    if (mnemonicOrSeed) {
-      let seed: Uint8Array;
-      if (typeof mnemonicOrSeed !== "string") {
-        seed = mnemonicOrSeed;
-      } else {
-        if (validateMnemonic(mnemonicOrSeed, wordlist)) {
-          seed = await this.config.signer.mnemonicToSeed(mnemonicOrSeed);
-        } else {
-          seed = hexToBytes(mnemonicOrSeed);
-        }
-      }
-
-      const hdkey = getMasterHDKeyFromSeed(seed);
-      const accountType = this.config.getNetwork() === Network.REGTEST ? 0 : 1;
-      const identityKey = hdkey.derive(`m/8797555'/${accountType}'/0'`);
-
-      const network = this.config.getNetwork();
-      this.lrc20Wallet = new LRCWallet(
-        bytesToHex(identityKey.privateKey!),
-        LRC_WALLET_NETWORK[network],
-        LRC_WALLET_NETWORK_TYPE[network],
-        this.config.lrc20ApiConfig
-      );
-    }
-
-    return initResponse;
-  }
-
   public async getIssuerTokenBalance(): Promise<{
     balance: bigint;
   }> {
@@ -115,11 +87,9 @@ export class IssuerSparkWallet
     if (this.tokenPublicKeyInfo) {
       return convertToTokenPubKeyInfoResponse(this.tokenPublicKeyInfo);
     }
-
     const tokenPublicKey = bytesToHex(this.lrc20Wallet!.pubkey);
     const rawTokenPubkeyInfo =
       await this.lrc20Wallet!.getTokenPubkeyInfo(tokenPublicKey);
-
     this.tokenPublicKeyInfo = rawTokenPubkeyInfo;
     if (!rawTokenPubkeyInfo) {
       return null;
@@ -166,9 +136,12 @@ export class IssuerSparkWallet
   ): Promise<{ impactedLeafIds: string[]; impactedTokenAmount: bigint }> {
     await this.syncTokenLeaves();
     const tokenPublicKey = await super.getIdentityPublicKey();
-
+    const decodedOwnerPubkey = decodeSparkAddress(
+      ownerPublicKey,
+      this.config.getNetworkType(),
+    );
     const response = await this.tokenFreezeService!.freezeTokens(
-      hexToBytes(ownerPublicKey),
+      hexToBytes(decodedOwnerPubkey),
       hexToBytes(tokenPublicKey),
     );
 
@@ -186,9 +159,12 @@ export class IssuerSparkWallet
   ): Promise<{ impactedLeafIds: string[]; impactedTokenAmount: bigint }> {
     await this.syncTokenLeaves();
     const tokenPublicKey = await super.getIdentityPublicKey();
-
+    const decodedOwnerPubkey = decodeSparkAddress(
+      ownerPublicKey,
+      this.config.getNetworkType(),
+    );
     const response = await this.tokenFreezeService!.unfreezeTokens(
-      hexToBytes(ownerPublicKey),
+      hexToBytes(decodedOwnerPubkey),
       hexToBytes(tokenPublicKey),
     );
     const tokenAmount = bytesToNumberBE(response.impactedTokenAmount);

package/src/proto/spark.ts

@@ -204,14 +204,14 @@ export interface NodeSignatures {
   refundTxSignature: Uint8Array;
 }
 
-export interface StartTreeCreationRequest {
+export interface StartDepositTreeCreationRequest {
   identityPublicKey: Uint8Array;
   onChainUtxo: UTXO | undefined;
   rootTxSigningJob: SigningJob | undefined;
   refundTxSigningJob: SigningJob | undefined;
 }
 
-export interface StartTreeCreationResponse {
+export interface StartDepositTreeCreationResponse {
   treeId: string;
   rootNodeSignatureShares: NodeSignatureShares | undefined;
 }
@@ -2259,7 +2259,7 @@ export const NodeSignatures: MessageFns<NodeSignatures> = {
   },
 };
 
-function createBaseStartTreeCreationRequest(): StartTreeCreationRequest {
+function createBaseStartDepositTreeCreationRequest(): StartDepositTreeCreationRequest {
   return {
     identityPublicKey: new Uint8Array(0),
     onChainUtxo: undefined,
@@ -2268,8 +2268,8 @@ function createBaseStartTreeCreationRequest(): StartTreeCreationRequest {
   };
 }
 
-export const StartTreeCreationRequest: MessageFns<StartTreeCreationRequest> = {
-  encode(message: StartTreeCreationRequest, writer: BinaryWriter = new BinaryWriter()): BinaryWriter {
+export const StartDepositTreeCreationRequest: MessageFns<StartDepositTreeCreationRequest> = {
+  encode(message: StartDepositTreeCreationRequest, writer: BinaryWriter = new BinaryWriter()): BinaryWriter {
     if (message.identityPublicKey.length !== 0) {
       writer.uint32(10).bytes(message.identityPublicKey);
     }
@@ -2285,10 +2285,10 @@ export const StartTreeCreationRequest: MessageFns<StartTreeCreationRequest> = {
     return writer;
   },
 
-  decode(input: BinaryReader | Uint8Array, length?: number): StartTreeCreationRequest {
+  decode(input: BinaryReader | Uint8Array, length?: number): StartDepositTreeCreationRequest {
     const reader = input instanceof BinaryReader ? input : new BinaryReader(input);
     let end = length === undefined ? reader.len : reader.pos + length;
-    const message = createBaseStartTreeCreationRequest();
+    const message = createBaseStartDepositTreeCreationRequest();
     while (reader.pos < end) {
       const tag = reader.uint32();
       switch (tag >>> 3) {
@@ -2333,7 +2333,7 @@ export const StartTreeCreationRequest: MessageFns<StartTreeCreationRequest> = {
     return message;
   },
 
-  fromJSON(object: any): StartTreeCreationRequest {
+  fromJSON(object: any): StartDepositTreeCreationRequest {
     return {
       identityPublicKey: isSet(object.identityPublicKey)
         ? bytesFromBase64(object.identityPublicKey)
@@ -2344,7 +2344,7 @@ export const StartTreeCreationRequest: MessageFns<StartTreeCreationRequest> = {
     };
   },
 
-  toJSON(message: StartTreeCreationRequest): unknown {
+  toJSON(message: StartDepositTreeCreationRequest): unknown {
     const obj: any = {};
     if (message.identityPublicKey.length !== 0) {
       obj.identityPublicKey = base64FromBytes(message.identityPublicKey);
@@ -2361,11 +2361,11 @@ export const StartTreeCreationRequest: MessageFns<StartTreeCreationRequest> = {
     return obj;
   },
 
-  create(base?: DeepPartial<StartTreeCreationRequest>): StartTreeCreationRequest {
-    return StartTreeCreationRequest.fromPartial(base ?? {});
+  create(base?: DeepPartial<StartDepositTreeCreationRequest>): StartDepositTreeCreationRequest {
+    return StartDepositTreeCreationRequest.fromPartial(base ?? {});
   },
-  fromPartial(object: DeepPartial<StartTreeCreationRequest>): StartTreeCreationRequest {
-    const message = createBaseStartTreeCreationRequest();
+  fromPartial(object: DeepPartial<StartDepositTreeCreationRequest>): StartDepositTreeCreationRequest {
+    const message = createBaseStartDepositTreeCreationRequest();
     message.identityPublicKey = object.identityPublicKey ?? new Uint8Array(0);
     message.onChainUtxo = (object.onChainUtxo !== undefined && object.onChainUtxo !== null)
       ? UTXO.fromPartial(object.onChainUtxo)
@@ -2380,12 +2380,12 @@ export const StartTreeCreationRequest: MessageFns<StartTreeCreationRequest> = {
   },
 };
 
-function createBaseStartTreeCreationResponse(): StartTreeCreationResponse {
+function createBaseStartDepositTreeCreationResponse(): StartDepositTreeCreationResponse {
   return { treeId: "", rootNodeSignatureShares: undefined };
 }
 
-export const StartTreeCreationResponse: MessageFns<StartTreeCreationResponse> = {
-  encode(message: StartTreeCreationResponse, writer: BinaryWriter = new BinaryWriter()): BinaryWriter {
+export const StartDepositTreeCreationResponse: MessageFns<StartDepositTreeCreationResponse> = {
+  encode(message: StartDepositTreeCreationResponse, writer: BinaryWriter = new BinaryWriter()): BinaryWriter {
     if (message.treeId !== "") {
       writer.uint32(10).string(message.treeId);
     }
@@ -2395,10 +2395,10 @@ export const StartTreeCreationResponse: MessageFns<StartTreeCreationResponse> =
     return writer;
   },
 
-  decode(input: BinaryReader | Uint8Array, length?: number): StartTreeCreationResponse {
+  decode(input: BinaryReader | Uint8Array, length?: number): StartDepositTreeCreationResponse {
     const reader = input instanceof BinaryReader ? input : new BinaryReader(input);
     let end = length === undefined ? reader.len : reader.pos + length;
-    const message = createBaseStartTreeCreationResponse();
+    const message = createBaseStartDepositTreeCreationResponse();
     while (reader.pos < end) {
       const tag = reader.uint32();
       switch (tag >>> 3) {
@@ -2427,7 +2427,7 @@ export const StartTreeCreationResponse: MessageFns<StartTreeCreationResponse> =
     return message;
   },
 
-  fromJSON(object: any): StartTreeCreationResponse {
+  fromJSON(object: any): StartDepositTreeCreationResponse {
     return {
       treeId: isSet(object.treeId) ? globalThis.String(object.treeId) : "",
       rootNodeSignatureShares: isSet(object.rootNodeSignatureShares)
@@ -2436,7 +2436,7 @@ export const StartTreeCreationResponse: MessageFns<StartTreeCreationResponse> =
     };
   },
 
-  toJSON(message: StartTreeCreationResponse): unknown {
+  toJSON(message: StartDepositTreeCreationResponse): unknown {
     const obj: any = {};
     if (message.treeId !== "") {
       obj.treeId = message.treeId;
@@ -2447,11 +2447,11 @@ export const StartTreeCreationResponse: MessageFns<StartTreeCreationResponse> =
     return obj;
   },
 
-  create(base?: DeepPartial<StartTreeCreationResponse>): StartTreeCreationResponse {
-    return StartTreeCreationResponse.fromPartial(base ?? {});
+  create(base?: DeepPartial<StartDepositTreeCreationResponse>): StartDepositTreeCreationResponse {
+    return StartDepositTreeCreationResponse.fromPartial(base ?? {});
   },
-  fromPartial(object: DeepPartial<StartTreeCreationResponse>): StartTreeCreationResponse {
-    const message = createBaseStartTreeCreationResponse();
+  fromPartial(object: DeepPartial<StartDepositTreeCreationResponse>): StartDepositTreeCreationResponse {
+    const message = createBaseStartDepositTreeCreationResponse();
     message.treeId = object.treeId ?? "";
     message.rootNodeSignatureShares =
       (object.rootNodeSignatureShares !== undefined && object.rootNodeSignatureShares !== null)
@@ -10993,11 +10993,11 @@ export const SparkServiceDefinition = {
       responseStream: false,
       options: {},
     },
-    start_tree_creation: {
-      name: "start_tree_creation",
-      requestType: StartTreeCreationRequest,
+    start_deposit_tree_creation: {
+      name: "start_deposit_tree_creation",
+      requestType: StartDepositTreeCreationRequest,
       requestStream: false,
-      responseType: StartTreeCreationResponse,
+      responseType: StartDepositTreeCreationResponse,
       responseStream: false,
       options: {},
     },
@@ -11242,10 +11242,10 @@ export interface SparkServiceImplementation<CallContextExt = {}> {
     request: GenerateDepositAddressRequest,
     context: CallContext & CallContextExt,
   ): Promise<DeepPartial<GenerateDepositAddressResponse>>;
-  start_tree_creation(
-    request: StartTreeCreationRequest,
+  start_deposit_tree_creation(
+    request: StartDepositTreeCreationRequest,
     context: CallContext & CallContextExt,
-  ): Promise<DeepPartial<StartTreeCreationResponse>>;
+  ): Promise<DeepPartial<StartDepositTreeCreationResponse>>;
   finalize_node_signatures(
     request: FinalizeNodeSignaturesRequest,
     context: CallContext & CallContextExt,
@@ -11367,10 +11367,10 @@ export interface SparkServiceClient<CallOptionsExt = {}> {
     request: DeepPartial<GenerateDepositAddressRequest>,
     options?: CallOptions & CallOptionsExt,
   ): Promise<GenerateDepositAddressResponse>;
-  start_tree_creation(
-    request: DeepPartial<StartTreeCreationRequest>,
+  start_deposit_tree_creation(
+    request: DeepPartial<StartDepositTreeCreationRequest>,
     options?: CallOptions & CallOptionsExt,
-  ): Promise<StartTreeCreationResponse>;
+  ): Promise<StartDepositTreeCreationResponse>;
   finalize_node_signatures(
     request: DeepPartial<FinalizeNodeSignaturesRequest>,
     options?: CallOptions & CallOptionsExt,

package/src/tests/integration/spark.test.ts

@@ -7,13 +7,10 @@ import {
 } from "../../../../spark-sdk/src/services/wallet-config.js";
 import { BitcoinFaucet } from "../../../../spark-sdk/src/tests/utils/test-faucet.js";
 import { IssuerSparkWallet } from "../../issuer-spark-wallet.js";
+import { filterTokenBalanceForTokenPublicKey } from "@buildonspark/spark-sdk/utils";
 
 describe("token integration test", () => {
-  // Skip all tests if running in GitHub Actions
-  process.env.GITHUB_ACTIONS ? it.skip : it;
-
-  // Increase timeout for all tests in this suite
-  jest.setTimeout(60000);
+  jest.setTimeout(80000);
 
   it("should issue a single token with ECDSA", async () => {
     const tokenAmount: bigint = 1000n;
@@ -191,8 +188,9 @@ describe("token integration test", () => {
     expect(sourceBalance.balance).toEqual(0n);
 
     const tokenPublicKey = await issuerWallet.getIdentityPublicKey();
-    const destinationBalance = await getSparkWalletTokenBalanceOrZero(
-      destinationWallet,
+    const balanceObj = await destinationWallet.getBalance();
+    const destinationBalance = filterTokenBalanceForTokenPublicKey(
+      balanceObj?.tokenBalances,
       tokenPublicKey,
     );
     expect(destinationBalance.balance).toEqual(tokenAmount);
@@ -219,8 +217,9 @@ describe("token integration test", () => {
     expect(sourceBalance.balance).toEqual(0n);
 
     const tokenPublicKey = await issuerWallet.getIdentityPublicKey();
-    const destinationBalance = await getSparkWalletTokenBalanceOrZero(
-      destinationWallet,
+    const balanceObj = await destinationWallet.getBalance();
+    const destinationBalance = filterTokenBalanceForTokenPublicKey(
+      balanceObj?.tokenBalances,
       tokenPublicKey,
     );
     expect(destinationBalance.balance).toEqual(tokenAmount);
@@ -253,10 +252,10 @@ describe("token integration test", () => {
     });
     const sourceBalance = await issuerWallet.getIssuerTokenBalance();
     expect(sourceBalance.balance).toEqual(0n);
-
     const tokenPublicKey = await issuerWallet.getIdentityPublicKey();
-    const destinationBalance = await getSparkWalletTokenBalanceOrZero(
-      destinationWallet,
+    const balanceObj = await destinationWallet.getBalance();
+    const destinationBalance = filterTokenBalanceForTokenPublicKey(
+      balanceObj?.tokenBalances,
       tokenPublicKey,
     );
     expect(destinationBalance.balance).toEqual(tokenAmount);
@@ -284,18 +283,17 @@ describe("token integration test", () => {
       tokenPublicKey: await issuerWallet.getIdentityPublicKey(),
       receiverSparkAddress: userWalletPublicKey,
     });
-
     const issuerBalanceAfterTransfer =
       await issuerWallet.getIssuerTokenBalance();
     expect(issuerBalanceAfterTransfer.balance).toEqual(0n);
 
     const tokenPublicKey = await issuerWallet.getIdentityPublicKey();
-    const userBalanceAfterTransfer = await getSparkWalletTokenBalanceOrZero(
-      userWallet,
+    const userBalanceObj = await userWallet.getBalance();
+    const userBalanceAfterTransfer = filterTokenBalanceForTokenPublicKey(
+      userBalanceObj?.tokenBalances,
       tokenPublicKey,
     );
     expect(userBalanceAfterTransfer.balance).toEqual(tokenAmount);
-
     // Freeze tokens
     const freezeResponse = await issuerWallet.freezeTokens(userWalletPublicKey);
     expect(freezeResponse.impactedLeafIds.length).toBeGreaterThan(0);
@@ -336,8 +334,9 @@ describe("token integration test", () => {
     expect(issuerBalanceAfterTransfer.balance).toEqual(0n);
 
     const tokenPublicKey = await issuerWallet.getIdentityPublicKey();
-    const userBalanceAfterTransfer = await getSparkWalletTokenBalanceOrZero(
-      userWallet,
+    const userBalanceObj = await userWallet.getBalance();
+    const userBalanceAfterTransfer = filterTokenBalanceForTokenPublicKey(
+      userBalanceObj?.tokenBalances,
       tokenPublicKey,
     );
     expect(userBalanceAfterTransfer.balance).toEqual(tokenAmount);
@@ -413,32 +412,31 @@ describe("token integration test", () => {
     const issuerBalanceAfterTransfer =
       await issuerWallet.getIssuerTokenBalance();
     expect(issuerBalanceAfterTransfer.balance).toEqual(0n);
-
     const tokenPublicKeyHex = await issuerWallet.getIdentityPublicKey();
     const userWalletPublicKeyHex = await userWallet.getSparkAddress();
-    const userBalanceAfterTransfer = await getSparkWalletTokenBalanceOrZero(
-      userWallet,
+    const userBalanceObj = await userWallet.getBalance();
+    const userBalanceAfterTransfer = filterTokenBalanceForTokenPublicKey(
+      userBalanceObj?.tokenBalances,
       tokenPublicKeyHex,
     );
     expect(userBalanceAfterTransfer.balance).toEqual(tokenAmount);
-
     await userWallet.transferTokens({
       tokenPublicKey: tokenPublicKeyHex,
       tokenAmount,
-      receiverSparkAddress: userWalletPublicKeyHex,
+      receiverSparkAddress: await issuerWallet.getSparkAddress(),
     });
 
-    const userBalanceAfterTransferBack = await getSparkWalletTokenBalanceOrZero(
-      userWallet,
+    const userBalanceObjAfterTransferBack = await userWallet.getBalance();
+    const userBalanceAfterTransferBack = filterTokenBalanceForTokenPublicKey(
+      userBalanceObjAfterTransferBack?.tokenBalances,
       tokenPublicKeyHex,
     );
+
     expect(userBalanceAfterTransferBack.balance).toEqual(0n);
 
     const issuerTokenBalance = await issuerWallet.getIssuerTokenBalance();
     expect(issuerTokenBalance.balance).toEqual(tokenAmount);
-
     await issuerWallet.burnTokens(tokenAmount);
-
     const issuerTokenBalanceAfterBurn =
       await issuerWallet.getIssuerTokenBalance();
     expect(issuerTokenBalanceAfterBurn.balance).toEqual(0n);
@@ -474,9 +472,9 @@ describe("token integration test", () => {
     expect(issuerBalanceAfterTransfer.balance).toEqual(0n);
 
     const tokenPublicKeyHex = await issuerWallet.getIdentityPublicKey();
-    const userWalletPublicKeyHex = await userWallet.getSparkAddress();
-    const userBalanceAfterTransfer = await getSparkWalletTokenBalanceOrZero(
-      userWallet,
+    const userBalanceObj = await userWallet.getBalance();
+    const userBalanceAfterTransfer = filterTokenBalanceForTokenPublicKey(
+      userBalanceObj?.tokenBalances,
       tokenPublicKeyHex,
     );
     expect(userBalanceAfterTransfer.balance).toEqual(tokenAmount);
@@ -484,11 +482,12 @@ describe("token integration test", () => {
     await userWallet.transferTokens({
       tokenPublicKey: tokenPublicKeyHex,
       tokenAmount,
-      receiverSparkAddress: userWalletPublicKeyHex,
+      receiverSparkAddress: await issuerWallet.getSparkAddress(),
     });
 
-    const userBalanceAfterTransferBack = await getSparkWalletTokenBalanceOrZero(
-      userWallet,
+    const userBalanceObjAfterTransferBack = await userWallet.getBalance();
+    const userBalanceAfterTransferBack = filterTokenBalanceForTokenPublicKey(
+      userBalanceObjAfterTransferBack?.tokenBalances,
       tokenPublicKeyHex,
     );
     expect(userBalanceAfterTransferBack.balance).toEqual(0n);
@@ -503,18 +502,3 @@ describe("token integration test", () => {
     expect(issuerTokenBalanceAfterBurn.balance).toEqual(0n);
   });
 });
-
-async function getSparkWalletTokenBalanceOrZero(
-  sparkWallet: SparkWallet,
-  publicKey: string,
-): Promise<{ balance: bigint }> {
-  const balanceObj = await sparkWallet.getBalance();
-  if (!balanceObj.tokenBalances || !balanceObj.tokenBalances.has(publicKey)) {
-    return {
-      balance: 0n,
-    };
-  }
-  return {
-    balance: balanceObj.tokenBalances.get(publicKey)!.balance,
-  };
-}

package/src/tests/stress/transfers.test.ts

@@ -0,0 +1,92 @@
+import { IssuerSparkWallet } from "../../issuer-spark-wallet.js";
+import { jest } from "@jest/globals";
+import { LOCAL_WALLET_CONFIG_ECDSA } from "../../../../spark-sdk/src/services/wallet-config.js";
+import { SparkWallet } from "@buildonspark/spark-sdk";
+import { filterTokenBalanceForTokenPublicKey } from "@buildonspark/spark-sdk/utils";
+
+const TEST_TIMEOUT = 80000; // 80 seconds
+
+describe("Stress test for token transfers", () => {
+  jest.setTimeout(TEST_TIMEOUT);
+
+  let timeoutReached = false;
+  let timeoutId: NodeJS.Timeout;
+
+  beforeEach(() => {
+    timeoutReached = false;
+    timeoutId = setTimeout(() => {
+      timeoutReached = true;
+    }, TEST_TIMEOUT);
+  });
+
+  afterEach(() => {
+    clearTimeout(timeoutId);
+  });
+
+  it("[Spark] wallets should successfully complete multiple token transactions in rapid succession (5 TPS)", async () => {
+    // Iteration: IssuerSparkWallet -> SparkWallet -> IssuerSparkWallet
+    // 2 transactions per iteration
+    // (2 transactions * 200 iterations) / 80 seconds = 5 TPS
+    const MAX_ITERATIONS = 200;
+    const TOKEN_AMOUNT: bigint = 1000n;
+
+    const { wallet: issuerWallet } = await IssuerSparkWallet.initialize({
+      options: LOCAL_WALLET_CONFIG_ECDSA,
+    });
+    const { wallet: userWallet } = await SparkWallet.initialize({
+      options: LOCAL_WALLET_CONFIG_ECDSA,
+    });
+
+    await issuerWallet.mintTokens(TOKEN_AMOUNT);
+    const tokenPublicKey = await issuerWallet.getIdentityPublicKey();
+    const userWalletSparkAddress = await userWallet.getSparkAddress();
+    const issuerWalletSparkAddress = await issuerWallet.getSparkAddress();
+
+    for (let i = 0; i < MAX_ITERATIONS; i++) {
+      if (timeoutReached) {
+        console.log(
+          "Timeout reached, stopping iterations at idx: " +
+            i +
+            " of " +
+            MAX_ITERATIONS,
+        );
+        break;
+      }
+      try {
+        // Transfer tokens from issuer to user
+        await issuerWallet.transferTokens({
+          tokenPublicKey,
+          tokenAmount: TOKEN_AMOUNT,
+          receiverSparkAddress: userWalletSparkAddress,
+        });
+        const issuerBalance = await issuerWallet.getIssuerTokenBalance();
+        const userBalanceObj = await userWallet.getBalance();
+        const userBalance = filterTokenBalanceForTokenPublicKey(
+          userBalanceObj?.tokenBalances,
+          tokenPublicKey,
+        );
+        expect(issuerBalance.balance).toEqual(0n);
+        expect(userBalance.balance).toEqual(TOKEN_AMOUNT);
+
+        // Transfer tokens from user to issuer
+        await userWallet.transferTokens({
+          tokenPublicKey,
+          tokenAmount: TOKEN_AMOUNT,
+          receiverSparkAddress: issuerWalletSparkAddress,
+        });
+        const userBalanceObjAfterTransferBack = await userWallet.getBalance();
+        const userBalanceAfterTransferBack =
+          filterTokenBalanceForTokenPublicKey(
+            userBalanceObjAfterTransferBack?.tokenBalances,
+            tokenPublicKey,
+          );
+        const issuerBalanceAfterTransferBack =
+          await issuerWallet.getIssuerTokenBalance();
+        expect(userBalanceAfterTransferBack.balance).toEqual(0n);
+        expect(issuerBalanceAfterTransferBack.balance).toEqual(TOKEN_AMOUNT);
+      } catch (error: any) {
+        throw new Error(`Test failed on iteration ${i}: ${error}`);
+      }
+    }
+  });
+});