@dynamic-labs/waas-sui 4.25.2 → 4.25.4

package/CHANGELOG.md

@@ -1,4 +1,29 @@
 
+### [4.25.4](https://github.com/dynamic-labs/dynamic-auth/compare/v4.25.3...v4.25.4) (2025-07-25)
+
+
+### Features
+
+* add passkey as a sign in method ([#9210](https://github.com/dynamic-labs/dynamic-auth/issues/9210)) ([90a6938](https://github.com/dynamic-labs/dynamic-auth/commit/90a69389cf89d7092519e58b0052d85c53a19905))
+
+
+### Bug Fixes
+
+* **QNTM-3416:** fix rainbow wallet connection on base network ([#9227](https://github.com/dynamic-labs/dynamic-auth/issues/9227)) ([d6dcdc3](https://github.com/dynamic-labs/dynamic-auth/commit/d6dcdc3ead194d7580bb8387432dfd15555158de))
+
+### [4.25.3](https://github.com/dynamic-labs/dynamic-auth/compare/v4.25.2...v4.25.3) (2025-07-22)
+
+
+### Features
+
+* add permanentlyDeleteUser for react native ([#8838](https://github.com/dynamic-labs/dynamic-auth/issues/8838)) ([1e00fd1](https://github.com/dynamic-labs/dynamic-auth/commit/1e00fd166f2cef55cd0526a66d9bb3123d10504f))
+
+
+### Bug Fixes
+
+* chain id on dynamic waas for confirmation view and clean up tests ([#9204](https://github.com/dynamic-labs/dynamic-auth/issues/9204)) ([4cebd8b](https://github.com/dynamic-labs/dynamic-auth/commit/4cebd8b63c9327d936ebe1c2305eb4e8f5f69aeb))
+* remove errant calls to fetchMultichainTokenBalances from useEffect ([#9207](https://github.com/dynamic-labs/dynamic-auth/issues/9207)) ([84bacfc](https://github.com/dynamic-labs/dynamic-auth/commit/84bacfc1ddd6b18fde9ba1baed310531b3ec0adc))
+
 ### [4.25.2](https://github.com/dynamic-labs/dynamic-auth/compare/v4.25.1...v4.25.2) (2025-07-18)
 
 

package/package.cjs

@@ -3,6 +3,6 @@
 
 Object.defineProperty(exports, '__esModule', { value: true });
 
-var version = "4.25.2";
+var version = "4.25.4";
 
 exports.version = version;

package/package.js

@@ -1,4 +1,4 @@
 'use client'
-var version = "4.25.2";
+var version = "4.25.4";
 
 export { version };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dynamic-labs/waas-sui",
-  "version": "4.25.2",
+  "version": "4.25.4",
   "description": "A React SDK for implementing wallet web3 authentication and authorization to your website.",
   "author": "Dynamic Labs, Inc.",
   "license": "MIT",
@@ -18,18 +18,18 @@
   },
   "homepage": "https://www.dynamic.xyz/",
   "dependencies": {
-    "@dynamic-labs-wallet/browser-wallet-client": "0.0.120",
+    "@dynamic-labs-wallet/browser-wallet-client": "0.0.124",
     "@mysten/sui": "1.24.0",
     "@mysten/wallet-standard": "0.13.29",
-    "@dynamic-labs/assert-package-version": "4.25.2",
-    "@dynamic-labs/logger": "4.25.2",
-    "@dynamic-labs/rpc-providers": "4.25.2",
-    "@dynamic-labs/sui-core": "4.25.2",
-    "@dynamic-labs/types": "4.25.2",
-    "@dynamic-labs/utils": "4.25.2",
-    "@dynamic-labs/waas": "4.25.2",
-    "@dynamic-labs/wallet-book": "4.25.2",
-    "@dynamic-labs/wallet-connector-core": "4.25.2"
+    "@dynamic-labs/assert-package-version": "4.25.4",
+    "@dynamic-labs/logger": "4.25.4",
+    "@dynamic-labs/rpc-providers": "4.25.4",
+    "@dynamic-labs/sui-core": "4.25.4",
+    "@dynamic-labs/types": "4.25.4",
+    "@dynamic-labs/utils": "4.25.4",
+    "@dynamic-labs/waas": "4.25.4",
+    "@dynamic-labs/wallet-book": "4.25.4",
+    "@dynamic-labs/wallet-connector-core": "4.25.4"
   },
   "peerDependencies": {}
 }

package/src/connector/DynamicWaasSuiConnector.cjs

@@ -11,21 +11,6 @@ var waas = require('@dynamic-labs/waas');
 var walletConnectorCore = require('@dynamic-labs/wallet-connector-core');
 var WaasSuiWallet = require('../wallet/WaasSuiWallet.cjs');
 
-// This will need to be moved to a base class once we have base classes for waas connectors
-class WaasExportHandler {
-    constructor() {
-        this.iframeStamper = null;
-    }
-    setIframeStamper(iframe) {
-        this.iframeStamper = iframe;
-    }
-    clear() {
-        if (this.iframeStamper) {
-            this.iframeStamper.remove();
-            this.iframeStamper = null;
-        }
-    }
-}
 class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConnector) {
     constructor(props) {
         super('Dynamic Waas', props);
@@ -34,7 +19,6 @@ class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConn
         this.name = 'Dynamic Waas';
         this.overrideKey = 'dynamicwaas';
         this.isEmbeddedWallet = true;
-        this.__exportHandler = new WaasExportHandler();
     }
     connect() {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
@@ -103,6 +87,9 @@ class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConn
             var _a, _b;
             const walletClient = yield this.getWaasWalletClient();
             const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
+            if (!signedSessionId) {
+                throw new utils.DynamicError('Signed session ID is required');
+            }
             const targetWallet = yield walletClient.getWallet({
                 accountAddress: expectedAddress,
                 authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
@@ -117,26 +104,14 @@ class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConn
             }
         });
     }
-    createWalletAccount() {
-        return _tslib.__awaiter(this, arguments, void 0, function* ({ thresholdSignatureScheme = 'TWO_OF_TWO', } = {}) {
-            var _a, _b;
-            const walletClient = yield this.getWaasWalletClient();
-            const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
-            const createdWallet = yield walletClient.createWalletAccount({
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                signedSessionId,
-                thresholdSignatureScheme: thresholdSignatureScheme,
-            });
-            return createdWallet;
-        });
-    }
+    // Override importPrivateKey to use the proper type casting for SUI
     importPrivateKey(_a) {
         return _tslib.__awaiter(this, arguments, void 0, function* ({ privateKey, thresholdSignatureScheme = 'TWO_OF_TWO', }) {
             var _b, _c;
             const walletClient = yield this.getWaasWalletClient();
             const signedSessionId = yield ((_b = this.getSignedSessionId) === null || _b === void 0 ? void 0 : _b.call(this));
             if (!signedSessionId) {
-                throw new Error('Signed session ID is required');
+                throw new utils.DynamicError('Signed session ID is required');
             }
             yield walletClient.importPrivateKey({
                 authToken: (_c = this.getAuthToken) === null || _c === void 0 ? void 0 : _c.call(this),
@@ -148,16 +123,21 @@ class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConn
     }
     signMessage(message) {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
-            var _a, _b;
+            var _a, _b, _c;
             const waasSuiClient = yield this.getWaasWalletClient();
             if (!this.activeAccountAddress) {
                 throw new Error('Active account address is required');
             }
             const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
+            if (!signedSessionId) {
+                throw new utils.DynamicError('Signed session ID is required');
+            }
+            const mfaToken = yield ((_b = this.getMfaToken) === null || _b === void 0 ? void 0 : _b.call(this));
             const signedMessage = yield waasSuiClient.signMessage({
                 accountAddress: this.activeAccountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
+                authToken: (_c = this.getAuthToken) === null || _c === void 0 ? void 0 : _c.call(this),
                 message,
+                mfaToken,
                 signedSessionId,
             });
             return signedMessage;
@@ -165,7 +145,7 @@ class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConn
     }
     createAndSignTransaction(transaction) {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
-            var _a, _b;
+            var _a, _b, _c;
             const networkName = yield this.getNetworkName();
             const client$1 = new client.SuiClient({ url: client.getFullnodeUrl(networkName) });
             const txBytes = yield transaction.build({ client: client$1 });
@@ -175,8 +155,13 @@ class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConn
                 throw new Error('Active account address is required');
             }
             const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
+            if (!signedSessionId) {
+                throw new utils.DynamicError('Signed session ID is required');
+            }
+            const mfaToken = yield ((_b = this.getMfaToken) === null || _b === void 0 ? void 0 : _b.call(this));
             const signedTx = yield walletClient.signTransaction({
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
+                authToken: (_c = this.getAuthToken) === null || _c === void 0 ? void 0 : _c.call(this),
+                mfaToken,
                 senderAddress: this.activeAccountAddress,
                 signedSessionId,
                 transaction: txString,
@@ -238,105 +223,18 @@ class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConn
             return client;
         });
     }
-    exportPrivateKey() {
-        return _tslib.__awaiter(this, arguments, void 0, function* ({ accountAddress, displayContainer, password, } = {}) {
-            var _a, _b;
-            const walletClient = yield this.getWaasWalletClient();
-            const targetAccountAddress = accountAddress || this.activeAccountAddress;
-            if (!targetAccountAddress) {
-                throw new Error('Account address is required');
-            }
-            if (!displayContainer) {
-                throw new Error('Missing display container for export private key');
-            }
-            this.__exportHandler.setIframeStamper(displayContainer);
-            const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
-            yield walletClient.exportPrivateKey({
-                accountAddress: targetAccountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                displayContainer,
-                password,
-                signedSessionId,
-            });
-        });
-    }
-    getExportHandler() {
-        return this.__exportHandler;
-    }
+    // Chain-specific override for exportClientKeyshares - SUI needs to handle optional account address
     exportClientKeyshares(_a) {
+        const _super = Object.create(null, {
+            exportClientKeyshares: { get: () => super.exportClientKeyshares }
+        });
         return _tslib.__awaiter(this, arguments, void 0, function* ({ accountAddress, password, }) {
-            var _b, _c;
-            const waasSuiClient = yield this.getWaasWalletClient();
             const targetAccountAddress = accountAddress || this.activeAccountAddress;
             if (!targetAccountAddress) {
                 throw new Error('Account address is required');
             }
-            const signedSessionId = yield ((_b = this.getSignedSessionId) === null || _b === void 0 ? void 0 : _b.call(this));
-            yield waasSuiClient.exportClientKeyshares({
+            return _super.exportClientKeyshares.call(this, {
                 accountAddress: targetAccountAddress,
-                authToken: (_c = this.getAuthToken) === null || _c === void 0 ? void 0 : _c.call(this),
-                password,
-                signedSessionId,
-            });
-        });
-    }
-    backupKeySharesToGoogleDrive(_a) {
-        return _tslib.__awaiter(this, arguments, void 0, function* ({ accountAddress, password, }) {
-            var _b;
-            if (!accountAddress) {
-                throw new Error('Account address is required');
-            }
-            const walletClient = yield this.getWaasWalletClient();
-            return walletClient.backupKeySharesToGoogleDrive({
-                accountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                password,
-            });
-        });
-    }
-    refreshWalletAccountShares(_a) {
-        return _tslib.__awaiter(this, arguments, void 0, function* ({ accountAddress, password, }) {
-            var _b;
-            if (!accountAddress) {
-                throw new Error('Account address is required');
-            }
-            const walletClient = yield this.getWaasWalletClient();
-            return walletClient.refreshWalletAccountShares({
-                accountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                password,
-            });
-        });
-    }
-    updatePassword(_a) {
-        return _tslib.__awaiter(this, arguments, void 0, function* ({ accountAddress, existingPassword, newPassword, }) {
-            var _b;
-            if (!accountAddress) {
-                throw new Error('Account address is required');
-            }
-            const walletClient = yield this.getWaasWalletClient();
-            return walletClient.updatePassword({
-                accountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                existingPassword,
-                newPassword,
-            });
-        });
-    }
-    signRawMessage(_a) {
-        return _tslib.__awaiter(this, arguments, void 0, function* ({ accountAddress, message, password, }) {
-            var _b;
-            if (!accountAddress) {
-                throw new Error('Account address is required');
-            }
-            if (message.length !== 64) {
-                throw new Error('Message must be 64 characters long');
-            }
-            const walletClient = yield this.getWaasWalletClient();
-            return walletClient.signRawMessage({
-                accountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                message,
                 password,
             });
         });
@@ -363,14 +261,12 @@ class DynamicWaasSuiConnector extends waas.withDynamicWaas(suiCore.SuiWalletConn
         });
     }
     endSession() {
+        const _super = Object.create(null, {
+            endSession: { get: () => super.endSession }
+        });
         return _tslib.__awaiter(this, void 0, void 0, function* () {
-            const waasSuiClient = yield this.getWaasWalletClient();
-            if (!waasSuiClient) {
-                return;
-            }
-            yield waasSuiClient.cleanup();
+            yield _super.endSession.call(this);
             this.activeAccountAddress = undefined;
-            this.dynamicWaasClient = undefined;
         });
     }
 }

package/src/connector/DynamicWaasSuiConnector.d.ts

@@ -12,19 +12,71 @@ declare const DynamicWaasSuiConnector_base: (abstract new (...args: any[]) => {
     overrideKey: string;
     isEmbeddedWallet: boolean;
     getSignedSessionId?: (() => Promise<string>) | undefined;
+    getMfaToken?: (() => Promise<string | undefined>) | undefined;
     getAuthToken?: (() => string) | undefined;
     environmentId?: string | undefined;
     baseApiUrl?: string | undefined;
     relayUrl?: string | undefined;
     dynamicWaasClient: DynamicWalletClient | undefined;
     chainName: string;
+    __exportHandler: import("@dynamic-labs/waas").WaasExportHandler;
+    validateActiveWallet(expectedAddress: string): Promise<void>;
     setGetAuthTokenFunction(getAuthToken: () => string): void;
+    setGetMfaTokenFunction(getMfaToken: () => Promise<string | undefined>): void;
     setEnvironmentId(environmentId: string): void;
     setBaseApiUrl(baseApiUrl: string): void;
     setRelayUrl(relayUrl: string): void;
     setGetSignedSessionIdFunction(getSignedSessionId: () => Promise<string>): void;
+    delegateKeyShares({ accountAddress, password, }: {
+        accountAddress: string;
+        password?: string | undefined;
+    }): Promise<void>;
     createDynamicWaasClient(): Promise<DynamicWalletClient>;
     getWaasWalletClient(): Promise<DynamicWalletClient>;
+    createWalletAccount({ thresholdSignatureScheme, }?: {
+        thresholdSignatureScheme?: string | undefined;
+    } | undefined): Promise<{
+        chainName: string;
+        accountAddress: string;
+        publicKeyHex: string;
+        rawPublicKey: string | Uint8Array | undefined;
+    }>;
+    importPrivateKey({ privateKey, thresholdSignatureScheme, }: {
+        privateKey: string;
+        thresholdSignatureScheme?: string | undefined;
+    }): Promise<void>;
+    exportPrivateKey({ accountAddress, displayContainer, password, }?: {
+        accountAddress?: string | undefined;
+        displayContainer?: HTMLIFrameElement | undefined;
+        password?: string | undefined;
+    } | undefined): Promise<void>;
+    getExportHandler(): {
+        clear: () => void;
+    };
+    exportClientKeyshares({ accountAddress, password, }: {
+        accountAddress: string;
+        password?: string | undefined;
+    }): Promise<void>;
+    backupKeySharesToGoogleDrive({ accountAddress, password, }: {
+        accountAddress: string;
+        password?: string | undefined;
+    }): Promise<void>;
+    refreshWalletAccountShares({ accountAddress, password, }: {
+        accountAddress: string;
+        password?: string | undefined;
+    }): Promise<void>;
+    updatePassword({ accountAddress, existingPassword, newPassword, }: {
+        accountAddress: string;
+        existingPassword: string;
+        newPassword: string;
+    }): Promise<void>;
+    signRawMessage({ accountAddress, message, password, }: {
+        accountAddress: string;
+        message: string;
+        password?: string | undefined;
+    }): Promise<string>;
+    endSession(): Promise<void>;
+    getActiveAccountAddress(): Promise<string | undefined>;
 }) & typeof SuiWalletConnector;
 export declare class DynamicWaasSuiConnector extends DynamicWaasSuiConnector_base implements IDynamicWaasConnector {
     ChainWallet: typeof WaasSuiWallet;
@@ -40,19 +92,10 @@ export declare class DynamicWaasSuiConnector extends DynamicWaasSuiConnector_bas
     name: string;
     overrideKey: string;
     isEmbeddedWallet: boolean;
-    private __exportHandler;
     activeAccountAddress: string | undefined;
     private setActiveAccountAddress;
     getActiveAccountAddress(): Promise<string | undefined>;
     validateActiveWallet(expectedAddress: string): Promise<void>;
-    createWalletAccount({ thresholdSignatureScheme, }?: {
-        thresholdSignatureScheme?: string;
-    }): Promise<{
-        chainName: string;
-        accountAddress: string;
-        publicKeyHex: string;
-        rawPublicKey: string | Uint8Array | undefined;
-    }>;
     importPrivateKey({ privateKey, thresholdSignatureScheme, }: {
         privateKey: string;
         thresholdSignatureScheme?: string;
@@ -66,36 +109,10 @@ export declare class DynamicWaasSuiConnector extends DynamicWaasSuiConnector_bas
         accountAddress: string;
     }): Promise<DynamicWalletClient>;
     getWalletClient(): Promise<DynamicWalletClient | undefined>;
-    exportPrivateKey({ accountAddress, displayContainer, password, }?: {
-        accountAddress?: string;
-        displayContainer?: HTMLIFrameElement;
-        password?: string;
-    }): Promise<void>;
-    getExportHandler(): {
-        clear: () => void;
-    };
     exportClientKeyshares({ accountAddress, password, }: {
         accountAddress: string;
         password?: string;
     }): Promise<void>;
-    backupKeySharesToGoogleDrive({ accountAddress, password, }: {
-        accountAddress: string;
-        password?: string;
-    }): Promise<void>;
-    refreshWalletAccountShares({ accountAddress, password, }: {
-        accountAddress: string;
-        password?: string;
-    }): Promise<void>;
-    updatePassword({ accountAddress, existingPassword, newPassword, }: {
-        accountAddress: string;
-        existingPassword: string;
-        newPassword: string;
-    }): Promise<void>;
-    signRawMessage({ accountAddress, message, password, }: {
-        accountAddress: string;
-        message: string;
-        password?: string;
-    }): Promise<string>;
     /** Function used to create transactions in the SDK interface */
     createUiTransaction(from: string): Promise<IUITransaction>;
     endSession(): Promise<void>;

package/src/connector/DynamicWaasSuiConnector.js

@@ -7,21 +7,6 @@ import { withDynamicWaas } from '@dynamic-labs/waas';
 import { isSameAddress } from '@dynamic-labs/wallet-connector-core';
 import { WaasSuiWallet } from '../wallet/WaasSuiWallet.js';
 
-// This will need to be moved to a base class once we have base classes for waas connectors
-class WaasExportHandler {
-    constructor() {
-        this.iframeStamper = null;
-    }
-    setIframeStamper(iframe) {
-        this.iframeStamper = iframe;
-    }
-    clear() {
-        if (this.iframeStamper) {
-            this.iframeStamper.remove();
-            this.iframeStamper = null;
-        }
-    }
-}
 class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
     constructor(props) {
         super('Dynamic Waas', props);
@@ -30,7 +15,6 @@ class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
         this.name = 'Dynamic Waas';
         this.overrideKey = 'dynamicwaas';
         this.isEmbeddedWallet = true;
-        this.__exportHandler = new WaasExportHandler();
     }
     connect() {
         return __awaiter(this, void 0, void 0, function* () {
@@ -99,6 +83,9 @@ class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
             var _a, _b;
             const walletClient = yield this.getWaasWalletClient();
             const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
+            if (!signedSessionId) {
+                throw new DynamicError('Signed session ID is required');
+            }
             const targetWallet = yield walletClient.getWallet({
                 accountAddress: expectedAddress,
                 authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
@@ -113,26 +100,14 @@ class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
             }
         });
     }
-    createWalletAccount() {
-        return __awaiter(this, arguments, void 0, function* ({ thresholdSignatureScheme = 'TWO_OF_TWO', } = {}) {
-            var _a, _b;
-            const walletClient = yield this.getWaasWalletClient();
-            const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
-            const createdWallet = yield walletClient.createWalletAccount({
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                signedSessionId,
-                thresholdSignatureScheme: thresholdSignatureScheme,
-            });
-            return createdWallet;
-        });
-    }
+    // Override importPrivateKey to use the proper type casting for SUI
     importPrivateKey(_a) {
         return __awaiter(this, arguments, void 0, function* ({ privateKey, thresholdSignatureScheme = 'TWO_OF_TWO', }) {
             var _b, _c;
             const walletClient = yield this.getWaasWalletClient();
             const signedSessionId = yield ((_b = this.getSignedSessionId) === null || _b === void 0 ? void 0 : _b.call(this));
             if (!signedSessionId) {
-                throw new Error('Signed session ID is required');
+                throw new DynamicError('Signed session ID is required');
             }
             yield walletClient.importPrivateKey({
                 authToken: (_c = this.getAuthToken) === null || _c === void 0 ? void 0 : _c.call(this),
@@ -144,16 +119,21 @@ class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
     }
     signMessage(message) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b;
+            var _a, _b, _c;
             const waasSuiClient = yield this.getWaasWalletClient();
             if (!this.activeAccountAddress) {
                 throw new Error('Active account address is required');
             }
             const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
+            if (!signedSessionId) {
+                throw new DynamicError('Signed session ID is required');
+            }
+            const mfaToken = yield ((_b = this.getMfaToken) === null || _b === void 0 ? void 0 : _b.call(this));
             const signedMessage = yield waasSuiClient.signMessage({
                 accountAddress: this.activeAccountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
+                authToken: (_c = this.getAuthToken) === null || _c === void 0 ? void 0 : _c.call(this),
                 message,
+                mfaToken,
                 signedSessionId,
             });
             return signedMessage;
@@ -161,7 +141,7 @@ class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
     }
     createAndSignTransaction(transaction) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b;
+            var _a, _b, _c;
             const networkName = yield this.getNetworkName();
             const client = new SuiClient({ url: getFullnodeUrl(networkName) });
             const txBytes = yield transaction.build({ client });
@@ -171,8 +151,13 @@ class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
                 throw new Error('Active account address is required');
             }
             const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
+            if (!signedSessionId) {
+                throw new DynamicError('Signed session ID is required');
+            }
+            const mfaToken = yield ((_b = this.getMfaToken) === null || _b === void 0 ? void 0 : _b.call(this));
             const signedTx = yield walletClient.signTransaction({
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
+                authToken: (_c = this.getAuthToken) === null || _c === void 0 ? void 0 : _c.call(this),
+                mfaToken,
                 senderAddress: this.activeAccountAddress,
                 signedSessionId,
                 transaction: txString,
@@ -234,105 +219,18 @@ class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
             return client;
         });
     }
-    exportPrivateKey() {
-        return __awaiter(this, arguments, void 0, function* ({ accountAddress, displayContainer, password, } = {}) {
-            var _a, _b;
-            const walletClient = yield this.getWaasWalletClient();
-            const targetAccountAddress = accountAddress || this.activeAccountAddress;
-            if (!targetAccountAddress) {
-                throw new Error('Account address is required');
-            }
-            if (!displayContainer) {
-                throw new Error('Missing display container for export private key');
-            }
-            this.__exportHandler.setIframeStamper(displayContainer);
-            const signedSessionId = yield ((_a = this.getSignedSessionId) === null || _a === void 0 ? void 0 : _a.call(this));
-            yield walletClient.exportPrivateKey({
-                accountAddress: targetAccountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                displayContainer,
-                password,
-                signedSessionId,
-            });
-        });
-    }
-    getExportHandler() {
-        return this.__exportHandler;
-    }
+    // Chain-specific override for exportClientKeyshares - SUI needs to handle optional account address
     exportClientKeyshares(_a) {
+        const _super = Object.create(null, {
+            exportClientKeyshares: { get: () => super.exportClientKeyshares }
+        });
         return __awaiter(this, arguments, void 0, function* ({ accountAddress, password, }) {
-            var _b, _c;
-            const waasSuiClient = yield this.getWaasWalletClient();
             const targetAccountAddress = accountAddress || this.activeAccountAddress;
             if (!targetAccountAddress) {
                 throw new Error('Account address is required');
             }
-            const signedSessionId = yield ((_b = this.getSignedSessionId) === null || _b === void 0 ? void 0 : _b.call(this));
-            yield waasSuiClient.exportClientKeyshares({
+            return _super.exportClientKeyshares.call(this, {
                 accountAddress: targetAccountAddress,
-                authToken: (_c = this.getAuthToken) === null || _c === void 0 ? void 0 : _c.call(this),
-                password,
-                signedSessionId,
-            });
-        });
-    }
-    backupKeySharesToGoogleDrive(_a) {
-        return __awaiter(this, arguments, void 0, function* ({ accountAddress, password, }) {
-            var _b;
-            if (!accountAddress) {
-                throw new Error('Account address is required');
-            }
-            const walletClient = yield this.getWaasWalletClient();
-            return walletClient.backupKeySharesToGoogleDrive({
-                accountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                password,
-            });
-        });
-    }
-    refreshWalletAccountShares(_a) {
-        return __awaiter(this, arguments, void 0, function* ({ accountAddress, password, }) {
-            var _b;
-            if (!accountAddress) {
-                throw new Error('Account address is required');
-            }
-            const walletClient = yield this.getWaasWalletClient();
-            return walletClient.refreshWalletAccountShares({
-                accountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                password,
-            });
-        });
-    }
-    updatePassword(_a) {
-        return __awaiter(this, arguments, void 0, function* ({ accountAddress, existingPassword, newPassword, }) {
-            var _b;
-            if (!accountAddress) {
-                throw new Error('Account address is required');
-            }
-            const walletClient = yield this.getWaasWalletClient();
-            return walletClient.updatePassword({
-                accountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                existingPassword,
-                newPassword,
-            });
-        });
-    }
-    signRawMessage(_a) {
-        return __awaiter(this, arguments, void 0, function* ({ accountAddress, message, password, }) {
-            var _b;
-            if (!accountAddress) {
-                throw new Error('Account address is required');
-            }
-            if (message.length !== 64) {
-                throw new Error('Message must be 64 characters long');
-            }
-            const walletClient = yield this.getWaasWalletClient();
-            return walletClient.signRawMessage({
-                accountAddress,
-                authToken: (_b = this.getAuthToken) === null || _b === void 0 ? void 0 : _b.call(this),
-                message,
                 password,
             });
         });
@@ -359,14 +257,12 @@ class DynamicWaasSuiConnector extends withDynamicWaas(SuiWalletConnector) {
         });
     }
     endSession() {
+        const _super = Object.create(null, {
+            endSession: { get: () => super.endSession }
+        });
         return __awaiter(this, void 0, void 0, function* () {
-            const waasSuiClient = yield this.getWaasWalletClient();
-            if (!waasSuiClient) {
-                return;
-            }
-            yield waasSuiClient.cleanup();
+            yield _super.endSession.call(this);
             this.activeAccountAddress = undefined;
-            this.dynamicWaasClient = undefined;
         });
     }
 }