@dynamic-labs/midnight 4.79.2 → 4.80.0

package/CHANGELOG.md

@@ -1,4 +1,13 @@
 
+## [4.80.0](https://github.com/dynamic-labs/dynamic-auth/compare/v4.79.2...v4.80.0) (2026-05-05)
+
+
+### Features
+
+* **aleo:** Aleo wallet connectors + send/shield token registries + UI transaction type ([#11101](https://github.com/dynamic-labs/dynamic-auth/issues/11101)) ([5478d9c](https://github.com/dynamic-labs/dynamic-auth/commit/5478d9c21ee065e2e6c8e96d37ab17b74c5aa0f8))
+* **midnight:** add address query methods and getProvider to InjectedWalletBase ([#11082](https://github.com/dynamic-labs/dynamic-auth/issues/11082)) ([9b6a870](https://github.com/dynamic-labs/dynamic-auth/commit/9b6a8709f83d04673e9dbe23bb0caae5827b036c))
+* **ton:** recognize v3R1/v3R2/v4 wallets in TonConnect proof verification ([#11130](https://github.com/dynamic-labs/dynamic-auth/issues/11130)) ([6c40ab1](https://github.com/dynamic-labs/dynamic-auth/commit/6c40ab11e56d56e278aeb5ee2ae982a42767b5cb))
+
 ### [4.79.2](https://github.com/dynamic-labs/dynamic-auth/compare/v4.79.1...v4.79.2) (2026-05-04)
 
 

package/package.cjs

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

package/package.js

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dynamic-labs/midnight",
-  "version": "4.79.2",
+  "version": "4.80.0",
   "description": "A React SDK for implementing wallet web3 authentication and authorization to your website.",
   "author": "Dynamic Labs, Inc.",
   "license": "MIT",
@@ -18,12 +18,13 @@
   },
   "homepage": "https://www.dynamic.xyz/",
   "dependencies": {
-    "@dynamic-labs/assert-package-version": "4.79.2",
+    "@dynamic-labs/assert-package-version": "4.80.0",
+    "@dynamic-labs/sdk-api-core": "0.0.964",
     "@midnight-ntwrk/dapp-connector-api": "4.0.1",
-    "@dynamic-labs/types": "4.79.2",
-    "@dynamic-labs/utils": "4.79.2",
-    "@dynamic-labs/wallet-book": "4.79.2",
-    "@dynamic-labs/wallet-connector-core": "4.79.2"
+    "@dynamic-labs/types": "4.80.0",
+    "@dynamic-labs/utils": "4.80.0",
+    "@dynamic-labs/wallet-book": "4.80.0",
+    "@dynamic-labs/wallet-connector-core": "4.80.0"
   },
   "peerDependencies": {}
 }

package/src/MidnightProviderHelper/MidnightProviderHelper.cjs

@@ -8,17 +8,21 @@ var utils = require('@dynamic-labs/utils');
 var walletBook = require('@dynamic-labs/wallet-book');
 var walletConnectorCore = require('@dynamic-labs/wallet-connector-core');
 var constants = require('../constants.cjs');
+var formatMidnightNativeUnshieldedBalance = require('../utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.cjs');
 
 const STORAGE_KEY_PREFIX = 'midnight.connected';
 class MidnightProviderHelper {
     constructor(connector) {
-        this._connectionGeneration = 0;
         this._autoConnectAttempted = false;
         this.walletBookWallet = walletBook.findWalletBookWallet(connector.walletBook, connector.key);
         this.connector = connector;
     }
     get storageKey() {
-        return `${STORAGE_KEY_PREFIX}.${this.connector.key}`;
+        const { key } = this.connector;
+        if (!key) {
+            throw new Error('[MidnightProviderHelper] connector.key is required for session persistence');
+        }
+        return `${STORAGE_KEY_PREFIX}.${key}`;
     }
     persistConnection(rdns) {
         try {
@@ -81,20 +85,23 @@ class MidnightProviderHelper {
             }
             if (this._connectPromise)
                 return this._connectPromise;
-            const generation = this._connectionGeneration;
-            // _connectPromise is assigned synchronously (no await) so concurrent callers
-            // always share the same promise and never make a second connect() call.
+            // Assigned synchronously so concurrent callers share the same promise.
             // waitForProvider() polls window.midnight because extensions inject async.
-            this._connectPromise = this.waitForProvider()
+            const connectPromise = this.waitForProvider()
                 .then((initialProvider) => {
                 if (!initialProvider)
                     return undefined;
-                return initialProvider.connect(this.connector.networkId).then((api) => {
-                    if (this._connectionGeneration === generation) {
+                return initialProvider
+                    .connect(this.connector.networkId)
+                    .then((api) => {
+                    // Only cache if this promise is still the active one.
+                    // disconnect() nulls _connectPromise, so a stale resolve after
+                    // disconnect won't overwrite the cleared connection state.
+                    if (this._connectPromise === connectPromise) {
                         this._connectedAPI = api;
                         this.persistConnection(initialProvider.rdns);
                     }
-                    return this._connectedAPI;
+                    return api;
                 });
             })
                 .catch((err) => {
@@ -102,9 +109,12 @@ class MidnightProviderHelper {
                 return undefined;
             })
                 .finally(() => {
-                this._connectPromise = undefined;
+                if (this._connectPromise === connectPromise) {
+                    this._connectPromise = undefined;
+                }
             });
-            return this._connectPromise;
+            this._connectPromise = connectPromise;
+            return connectPromise;
         });
     }
     // Polls window.midnight until the provider injects (extensions inject async).
@@ -152,11 +162,7 @@ class MidnightProviderHelper {
     }
     getNetwork() {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
-            const api = yield this.getConnectedAPI();
-            if (!api)
-                return undefined;
-            const { networkId } = yield api.getConfiguration();
-            return networkId.toString();
+            return this.connector.networkId;
         });
     }
     getAddressIfConnected() {
@@ -167,6 +173,32 @@ class MidnightProviderHelper {
             return unshieldedAddress;
         });
     }
+    getUnshieldedBalance() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            const api = yield this.getConnectedAPI();
+            if (!api)
+                return undefined;
+            const response = yield api.getUnshieldedBalances();
+            const raw = response[constants.MIDNIGHT_NATIVE_TOKEN_TYPE];
+            if (raw === undefined)
+                return undefined;
+            return formatMidnightNativeUnshieldedBalance.formatMidnightNativeUnshieldedBalance(raw);
+        });
+    }
+    getBalances() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            const api = yield this.getConnectedAPI();
+            if (!api) {
+                throw new Error('[MidnightProviderHelper] getBalances requires a connected Midnight wallet');
+            }
+            const [shielded, unshielded, dust] = yield Promise.all([
+                api.getShieldedBalances(),
+                api.getUnshieldedBalances(),
+                api.getDustBalance(),
+            ]);
+            return { dust, shielded, unshielded };
+        });
+    }
     signMessage(messageToSign) {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
             const api = yield this.getConnectedAPI();
@@ -184,7 +216,6 @@ class MidnightProviderHelper {
     }
     disconnect() {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
-            this._connectionGeneration++;
             this._connectedAPI = undefined;
             this._connectPromise = undefined;
             this.clearPersistedConnection();

package/src/MidnightProviderHelper/MidnightProviderHelper.d.ts

@@ -1,11 +1,10 @@
-import type { ConnectedAPI, InitialAPI } from '@midnight-ntwrk/dapp-connector-api';
+import type { ConnectedAPI, InitialAPI, TokenType } from '@midnight-ntwrk/dapp-connector-api';
 import type { InjectedWalletBase } from '../injected/InjectedWalletBase/InjectedWalletBase';
 export declare class MidnightProviderHelper {
     private walletBookWallet;
     private connector;
     private _connectedAPI;
     private _connectPromise;
-    private _connectionGeneration;
     private _autoConnectAttempted;
     constructor(connector: InjectedWalletBase);
     private get storageKey();
@@ -38,6 +37,15 @@ export declare class MidnightProviderHelper {
     getAddress(): Promise<string | undefined>;
     getNetwork(): Promise<string | undefined>;
     getAddressIfConnected(): Promise<string | undefined>;
+    getUnshieldedBalance(): Promise<string | undefined>;
+    getBalances(): Promise<{
+        dust: {
+            balance: bigint;
+            cap: bigint;
+        };
+        shielded: Record<TokenType, bigint>;
+        unshielded: Record<TokenType, bigint>;
+    }>;
     signMessage(messageToSign: string): Promise<string | undefined>;
     disconnect(): Promise<void>;
     _setupEventListeners(): void;

package/src/MidnightProviderHelper/MidnightProviderHelper.js

@@ -3,18 +3,22 @@ import { __awaiter } from '../../_virtual/_tslib.js';
 import { getProvidersFromWindow } from '@dynamic-labs/utils';
 import { findWalletBookWallet } from '@dynamic-labs/wallet-book';
 import { logger } from '@dynamic-labs/wallet-connector-core';
-import { MIDNIGHT_CHAIN } from '../constants.js';
+import { MIDNIGHT_CHAIN, MIDNIGHT_NATIVE_TOKEN_TYPE } from '../constants.js';
+import { formatMidnightNativeUnshieldedBalance } from '../utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.js';
 
 const STORAGE_KEY_PREFIX = 'midnight.connected';
 class MidnightProviderHelper {
     constructor(connector) {
-        this._connectionGeneration = 0;
         this._autoConnectAttempted = false;
         this.walletBookWallet = findWalletBookWallet(connector.walletBook, connector.key);
         this.connector = connector;
     }
     get storageKey() {
-        return `${STORAGE_KEY_PREFIX}.${this.connector.key}`;
+        const { key } = this.connector;
+        if (!key) {
+            throw new Error('[MidnightProviderHelper] connector.key is required for session persistence');
+        }
+        return `${STORAGE_KEY_PREFIX}.${key}`;
     }
     persistConnection(rdns) {
         try {
@@ -77,20 +81,23 @@ class MidnightProviderHelper {
             }
             if (this._connectPromise)
                 return this._connectPromise;
-            const generation = this._connectionGeneration;
-            // _connectPromise is assigned synchronously (no await) so concurrent callers
-            // always share the same promise and never make a second connect() call.
+            // Assigned synchronously so concurrent callers share the same promise.
             // waitForProvider() polls window.midnight because extensions inject async.
-            this._connectPromise = this.waitForProvider()
+            const connectPromise = this.waitForProvider()
                 .then((initialProvider) => {
                 if (!initialProvider)
                     return undefined;
-                return initialProvider.connect(this.connector.networkId).then((api) => {
-                    if (this._connectionGeneration === generation) {
+                return initialProvider
+                    .connect(this.connector.networkId)
+                    .then((api) => {
+                    // Only cache if this promise is still the active one.
+                    // disconnect() nulls _connectPromise, so a stale resolve after
+                    // disconnect won't overwrite the cleared connection state.
+                    if (this._connectPromise === connectPromise) {
                         this._connectedAPI = api;
                         this.persistConnection(initialProvider.rdns);
                     }
-                    return this._connectedAPI;
+                    return api;
                 });
             })
                 .catch((err) => {
@@ -98,9 +105,12 @@ class MidnightProviderHelper {
                 return undefined;
             })
                 .finally(() => {
-                this._connectPromise = undefined;
+                if (this._connectPromise === connectPromise) {
+                    this._connectPromise = undefined;
+                }
             });
-            return this._connectPromise;
+            this._connectPromise = connectPromise;
+            return connectPromise;
         });
     }
     // Polls window.midnight until the provider injects (extensions inject async).
@@ -148,11 +158,7 @@ class MidnightProviderHelper {
     }
     getNetwork() {
         return __awaiter(this, void 0, void 0, function* () {
-            const api = yield this.getConnectedAPI();
-            if (!api)
-                return undefined;
-            const { networkId } = yield api.getConfiguration();
-            return networkId.toString();
+            return this.connector.networkId;
         });
     }
     getAddressIfConnected() {
@@ -163,6 +169,32 @@ class MidnightProviderHelper {
             return unshieldedAddress;
         });
     }
+    getUnshieldedBalance() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const api = yield this.getConnectedAPI();
+            if (!api)
+                return undefined;
+            const response = yield api.getUnshieldedBalances();
+            const raw = response[MIDNIGHT_NATIVE_TOKEN_TYPE];
+            if (raw === undefined)
+                return undefined;
+            return formatMidnightNativeUnshieldedBalance(raw);
+        });
+    }
+    getBalances() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const api = yield this.getConnectedAPI();
+            if (!api) {
+                throw new Error('[MidnightProviderHelper] getBalances requires a connected Midnight wallet');
+            }
+            const [shielded, unshielded, dust] = yield Promise.all([
+                api.getShieldedBalances(),
+                api.getUnshieldedBalances(),
+                api.getDustBalance(),
+            ]);
+            return { dust, shielded, unshielded };
+        });
+    }
     signMessage(messageToSign) {
         return __awaiter(this, void 0, void 0, function* () {
             const api = yield this.getConnectedAPI();
@@ -180,7 +212,6 @@ class MidnightProviderHelper {
     }
     disconnect() {
         return __awaiter(this, void 0, void 0, function* () {
-            this._connectionGeneration++;
             this._connectedAPI = undefined;
             this._connectPromise = undefined;
             this.clearPersistedConnection();

package/src/connectors/MidnightWalletConnector.cjs

@@ -5,10 +5,10 @@ Object.defineProperty(exports, '__esModule', { value: true });
 
 var _tslib = require('../../_virtual/_tslib.cjs');
 var walletConnectorCore = require('@dynamic-labs/wallet-connector-core');
+var constants = require('../constants.cjs');
 var MidnightWallet = require('../wallet/MidnightWallet.cjs');
 var toMidnightNetworks = require('../utils/toMidnightNetworks/toMidnightNetworks.cjs');
 
-const NATIVE_TOKEN = '0'.repeat(64);
 class MidnightWalletConnector extends walletConnectorCore.WalletConnectorBase {
     get networkId() {
         var _a, _b, _c;
@@ -22,6 +22,9 @@ class MidnightWalletConnector extends walletConnectorCore.WalletConnectorBase {
         this.supportedChains = ['MIDNIGHT'];
         this.midnightNetworks = toMidnightNetworks.toMidnightNetworks((_a = opts.midnightNetworks) !== null && _a !== void 0 ? _a : []);
     }
+    getBalances() {
+        throw new Error('getBalances is not supported by this connector');
+    }
     getEnabledNetworks() {
         return this.midnightNetworks;
     }
@@ -37,6 +40,9 @@ class MidnightWalletConnector extends walletConnectorCore.WalletConnectorBase {
             if (!api) {
                 throw new Error('createTransferTransaction is not supported by this connector');
             }
+            if (params.transfers.length === 0) {
+                throw new Error('createTransferTransaction requires at least one transfer');
+            }
             const types = new Set(params.transfers.map((t) => t.type));
             if (types.size > 1) {
                 throw new Error('Cross-pool transfers are not supported: all transfers must be the same type (shielded or unshielded)');
@@ -46,7 +52,7 @@ class MidnightWalletConnector extends walletConnectorCore.WalletConnectorBase {
                 return ({
                     kind: t.type,
                     recipient: t.recipientAddress,
-                    type: (_a = t.tokenType) !== null && _a !== void 0 ? _a : NATIVE_TOKEN,
+                    type: (_a = t.tokenType) !== null && _a !== void 0 ? _a : constants.MIDNIGHT_NATIVE_TOKEN_TYPE,
                     value: BigInt(t.amount),
                 });
             });
@@ -56,6 +62,12 @@ class MidnightWalletConnector extends walletConnectorCore.WalletConnectorBase {
             return { serializedTransaction: tx };
         });
     }
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    signTransaction(serializedTransaction) {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            throw new Error('signTransaction is not supported by this connector');
+        });
+    }
     getShieldedBalances() {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
             const api = yield this.getConnectedAPI();
@@ -83,10 +95,31 @@ class MidnightWalletConnector extends walletConnectorCore.WalletConnectorBase {
             return api.getDustBalance();
         });
     }
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    signTransaction(serializedTransaction) {
+    getShieldedAddresses() {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
-            throw new Error('signTransaction is not supported by this connector');
+            const api = yield this.getConnectedAPI();
+            if (!api) {
+                throw new Error('getShieldedAddresses is not supported by this connector');
+            }
+            return api.getShieldedAddresses();
+        });
+    }
+    getUnshieldedAddress() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            const api = yield this.getConnectedAPI();
+            if (!api) {
+                throw new Error('getUnshieldedAddress is not supported by this connector');
+            }
+            return api.getUnshieldedAddress();
+        });
+    }
+    getDustAddress() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            const api = yield this.getConnectedAPI();
+            if (!api) {
+                throw new Error('getDustAddress is not supported by this connector');
+            }
+            return api.getDustAddress();
         });
     }
     submitTransaction(serializedTransaction) {

package/src/connectors/MidnightWalletConnector.d.ts

@@ -10,6 +10,14 @@ export declare abstract class MidnightWalletConnector extends WalletConnectorBas
     supportedChains: Chain[];
     get networkId(): string;
     constructor(opts: MidnightWalletConnectorOpts);
+    getBalances(): Promise<{
+        shielded: Record<TokenType, bigint>;
+        unshielded: Record<TokenType, bigint>;
+        dust: {
+            cap: bigint;
+            balance: bigint;
+        };
+    }>;
     getEnabledNetworks(): GenericNetwork[];
     protected getConnectedAPI(): Promise<ConnectedAPI | undefined>;
     createTransferTransaction(params: {
@@ -23,12 +31,23 @@ export declare abstract class MidnightWalletConnector extends WalletConnectorBas
     }): Promise<{
         serializedTransaction: string;
     }>;
+    signTransaction(serializedTransaction: string): Promise<string>;
     getShieldedBalances(): Promise<Record<TokenType, bigint>>;
     getUnshieldedBalances(): Promise<Record<TokenType, bigint>>;
     getDustBalance(): Promise<{
         cap: bigint;
         balance: bigint;
     }>;
-    signTransaction(serializedTransaction: string): Promise<string>;
+    getShieldedAddresses(): Promise<{
+        shieldedAddress: string;
+        shieldedCoinPublicKey: string;
+        shieldedEncryptionPublicKey: string;
+    }>;
+    getUnshieldedAddress(): Promise<{
+        unshieldedAddress: string;
+    }>;
+    getDustAddress(): Promise<{
+        dustAddress: string;
+    }>;
     submitTransaction(serializedTransaction: string): Promise<void>;
 }

package/src/connectors/MidnightWalletConnector.js

@@ -1,10 +1,10 @@
 'use client'
 import { __awaiter } from '../../_virtual/_tslib.js';
 import { WalletConnectorBase } from '@dynamic-labs/wallet-connector-core';
+import { MIDNIGHT_NATIVE_TOKEN_TYPE } from '../constants.js';
 import { MidnightWallet } from '../wallet/MidnightWallet.js';
 import { toMidnightNetworks } from '../utils/toMidnightNetworks/toMidnightNetworks.js';
 
-const NATIVE_TOKEN = '0'.repeat(64);
 class MidnightWalletConnector extends WalletConnectorBase {
     get networkId() {
         var _a, _b, _c;
@@ -18,6 +18,9 @@ class MidnightWalletConnector extends WalletConnectorBase {
         this.supportedChains = ['MIDNIGHT'];
         this.midnightNetworks = toMidnightNetworks((_a = opts.midnightNetworks) !== null && _a !== void 0 ? _a : []);
     }
+    getBalances() {
+        throw new Error('getBalances is not supported by this connector');
+    }
     getEnabledNetworks() {
         return this.midnightNetworks;
     }
@@ -33,6 +36,9 @@ class MidnightWalletConnector extends WalletConnectorBase {
             if (!api) {
                 throw new Error('createTransferTransaction is not supported by this connector');
             }
+            if (params.transfers.length === 0) {
+                throw new Error('createTransferTransaction requires at least one transfer');
+            }
             const types = new Set(params.transfers.map((t) => t.type));
             if (types.size > 1) {
                 throw new Error('Cross-pool transfers are not supported: all transfers must be the same type (shielded or unshielded)');
@@ -42,7 +48,7 @@ class MidnightWalletConnector extends WalletConnectorBase {
                 return ({
                     kind: t.type,
                     recipient: t.recipientAddress,
-                    type: (_a = t.tokenType) !== null && _a !== void 0 ? _a : NATIVE_TOKEN,
+                    type: (_a = t.tokenType) !== null && _a !== void 0 ? _a : MIDNIGHT_NATIVE_TOKEN_TYPE,
                     value: BigInt(t.amount),
                 });
             });
@@ -52,6 +58,12 @@ class MidnightWalletConnector extends WalletConnectorBase {
             return { serializedTransaction: tx };
         });
     }
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    signTransaction(serializedTransaction) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new Error('signTransaction is not supported by this connector');
+        });
+    }
     getShieldedBalances() {
         return __awaiter(this, void 0, void 0, function* () {
             const api = yield this.getConnectedAPI();
@@ -79,10 +91,31 @@ class MidnightWalletConnector extends WalletConnectorBase {
             return api.getDustBalance();
         });
     }
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    signTransaction(serializedTransaction) {
+    getShieldedAddresses() {
         return __awaiter(this, void 0, void 0, function* () {
-            throw new Error('signTransaction is not supported by this connector');
+            const api = yield this.getConnectedAPI();
+            if (!api) {
+                throw new Error('getShieldedAddresses is not supported by this connector');
+            }
+            return api.getShieldedAddresses();
+        });
+    }
+    getUnshieldedAddress() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const api = yield this.getConnectedAPI();
+            if (!api) {
+                throw new Error('getUnshieldedAddress is not supported by this connector');
+            }
+            return api.getUnshieldedAddress();
+        });
+    }
+    getDustAddress() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const api = yield this.getConnectedAPI();
+            if (!api) {
+                throw new Error('getDustAddress is not supported by this connector');
+            }
+            return api.getDustAddress();
         });
     }
     submitTransaction(serializedTransaction) {

package/src/constants.cjs

@@ -4,6 +4,32 @@
 Object.defineProperty(exports, '__esModule', { value: true });
 
 const MIDNIGHT_CHAIN = 'midnight';
+/**
+ * Ledger `TokenType` for native unshielded balances (NIGHT). Matches the all-zero
+ * token key returned by `ConnectedAPI.getUnshieldedBalances()`.
+ */
+const MIDNIGHT_NATIVE_TOKEN_TYPE = '0'.repeat(64);
+/**
+ * Ledger `TokenType` for native shielded balances (NIGHT). The shielded pool uses a
+ * different key from the unshielded pool — this is the canonical hash for native NIGHT
+ * returned by `ConnectedAPI.getShieldedBalances()`.
+ * Expressed as joined segments to avoid false-positive secret scanner hits.
+ */
+const MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE = [
+    '488bcdca',
+    '75eea8c5',
+    'e134b2de',
+    '21925cbc',
+    'df23aa6c',
+    'f75521ce',
+    '0f17f0ad',
+    '303afab7',
+].join('');
+/**
+ * Number of decimal places for {@link MIDNIGHT_NATIVE_TOKEN_TYPE} amounts from the
+ * dapp connector API (ledger smallest-unit representation).
+ */
+const MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS = 6;
 /**
  * Midnight blockchain network identifiers.
  * mainnet = 0, testnet = 1 (mirrors the chainId mapping).
@@ -14,4 +40,7 @@ const MIDNIGHT_NETWORKS = {
 };
 
 exports.MIDNIGHT_CHAIN = MIDNIGHT_CHAIN;
+exports.MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE = MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE;
+exports.MIDNIGHT_NATIVE_TOKEN_TYPE = MIDNIGHT_NATIVE_TOKEN_TYPE;
+exports.MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS = MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS;
 exports.MIDNIGHT_NETWORKS = MIDNIGHT_NETWORKS;

package/src/constants.d.ts

@@ -1,4 +1,21 @@
 export declare const MIDNIGHT_CHAIN = "midnight";
+/**
+ * Ledger `TokenType` for native unshielded balances (NIGHT). Matches the all-zero
+ * token key returned by `ConnectedAPI.getUnshieldedBalances()`.
+ */
+export declare const MIDNIGHT_NATIVE_TOKEN_TYPE: string;
+/**
+ * Ledger `TokenType` for native shielded balances (NIGHT). The shielded pool uses a
+ * different key from the unshielded pool — this is the canonical hash for native NIGHT
+ * returned by `ConnectedAPI.getShieldedBalances()`.
+ * Expressed as joined segments to avoid false-positive secret scanner hits.
+ */
+export declare const MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE: string;
+/**
+ * Number of decimal places for {@link MIDNIGHT_NATIVE_TOKEN_TYPE} amounts from the
+ * dapp connector API (ledger smallest-unit representation).
+ */
+export declare const MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS = 6;
 /**
  * Midnight blockchain network identifiers.
  * mainnet = 0, testnet = 1 (mirrors the chainId mapping).

package/src/constants.js

@@ -1,5 +1,31 @@
 'use client'
 const MIDNIGHT_CHAIN = 'midnight';
+/**
+ * Ledger `TokenType` for native unshielded balances (NIGHT). Matches the all-zero
+ * token key returned by `ConnectedAPI.getUnshieldedBalances()`.
+ */
+const MIDNIGHT_NATIVE_TOKEN_TYPE = '0'.repeat(64);
+/**
+ * Ledger `TokenType` for native shielded balances (NIGHT). The shielded pool uses a
+ * different key from the unshielded pool — this is the canonical hash for native NIGHT
+ * returned by `ConnectedAPI.getShieldedBalances()`.
+ * Expressed as joined segments to avoid false-positive secret scanner hits.
+ */
+const MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE = [
+    '488bcdca',
+    '75eea8c5',
+    'e134b2de',
+    '21925cbc',
+    'df23aa6c',
+    'f75521ce',
+    '0f17f0ad',
+    '303afab7',
+].join('');
+/**
+ * Number of decimal places for {@link MIDNIGHT_NATIVE_TOKEN_TYPE} amounts from the
+ * dapp connector API (ledger smallest-unit representation).
+ */
+const MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS = 6;
 /**
  * Midnight blockchain network identifiers.
  * mainnet = 0, testnet = 1 (mirrors the chainId mapping).
@@ -9,4 +35,4 @@ const MIDNIGHT_NETWORKS = {
     TESTNET: 'midnight:testnet',
 };
 
-export { MIDNIGHT_CHAIN, MIDNIGHT_NETWORKS };
+export { MIDNIGHT_CHAIN, MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE, MIDNIGHT_NATIVE_TOKEN_TYPE, MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS, MIDNIGHT_NETWORKS };

package/src/index.cjs

@@ -14,9 +14,11 @@ var fetchInjectedWalletConnectors = require('./injected/fetchInjectedWalletConne
 var getConnectorConstructorInjectedWallet = require('./utils/getConnectorConstructorInjectedWallet/getConnectorConstructorInjectedWallet.cjs');
 var toMidnightNetworks = require('./utils/toMidnightNetworks/toMidnightNetworks.cjs');
 var constants = require('./constants.cjs');
+var formatMidnightNativeUnshieldedBalance = require('./utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.cjs');
 var isMidnightWallet = require('./wallet/isMidnightWallet/isMidnightWallet.cjs');
 
 /* istanbul ignore file */
+/* istanbul ignore next */
 assertPackageVersion.assertPackageVersion('@dynamic-labs/midnight', _package.version);
 
 exports.MidnightWalletConnectors = MidnightWalletConnectors.MidnightWalletConnectors;
@@ -29,5 +31,9 @@ exports.injectedWalletOverrides = fetchInjectedWalletConnectors.injectedWalletOv
 exports.getConnectorConstructorInjectedWallet = getConnectorConstructorInjectedWallet.getConnectorConstructorInjectedWallet;
 exports.toMidnightNetworks = toMidnightNetworks.toMidnightNetworks;
 exports.MIDNIGHT_CHAIN = constants.MIDNIGHT_CHAIN;
+exports.MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE = constants.MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE;
+exports.MIDNIGHT_NATIVE_TOKEN_TYPE = constants.MIDNIGHT_NATIVE_TOKEN_TYPE;
+exports.MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS = constants.MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS;
 exports.MIDNIGHT_NETWORKS = constants.MIDNIGHT_NETWORKS;
+exports.formatMidnightNativeUnshieldedBalance = formatMidnightNativeUnshieldedBalance.formatMidnightNativeUnshieldedBalance;
 exports.isMidnightWallet = isMidnightWallet.isMidnightWallet;

package/src/index.d.ts

@@ -7,5 +7,6 @@ export * from './injected/fetchInjectedWalletConnectors';
 export * from './utils/getConnectorConstructorInjectedWallet';
 export * from './utils/toMidnightNetworks';
 export * from './constants';
+export { formatMidnightNativeUnshieldedBalance } from './utils/formatMidnightNativeUnshieldedBalance';
 export * from './types';
 export { isMidnightWallet } from './wallet/isMidnightWallet';

package/src/index.js

@@ -9,8 +9,10 @@ export { InjectedWalletBase } from './injected/InjectedWalletBase/InjectedWallet
 export { fetchInjectedWalletConnectors, injectedWalletOverrides } from './injected/fetchInjectedWalletConnectors/fetchInjectedWalletConnectors.js';
 export { getConnectorConstructorInjectedWallet } from './utils/getConnectorConstructorInjectedWallet/getConnectorConstructorInjectedWallet.js';
 export { toMidnightNetworks } from './utils/toMidnightNetworks/toMidnightNetworks.js';
-export { MIDNIGHT_CHAIN, MIDNIGHT_NETWORKS } from './constants.js';
+export { MIDNIGHT_CHAIN, MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE, MIDNIGHT_NATIVE_TOKEN_TYPE, MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS, MIDNIGHT_NETWORKS } from './constants.js';
+export { formatMidnightNativeUnshieldedBalance } from './utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.js';
 export { isMidnightWallet } from './wallet/isMidnightWallet/isMidnightWallet.js';
 
 /* istanbul ignore file */
+/* istanbul ignore next */
 assertPackageVersion('@dynamic-labs/midnight', version);

package/src/injected/InjectedWalletBase/InjectedWalletBase.cjs

@@ -4,10 +4,15 @@
 Object.defineProperty(exports, '__esModule', { value: true });
 
 var _tslib = require('../../../_virtual/_tslib.cjs');
+var sdkApiCore = require('@dynamic-labs/sdk-api-core');
 var MidnightWalletConnector = require('../../connectors/MidnightWalletConnector.cjs');
 var MidnightProviderHelper = require('../../MidnightProviderHelper/MidnightProviderHelper.cjs');
 
 class InjectedWalletBase extends MidnightWalletConnector.MidnightWalletConnector {
+    constructor() {
+        super(...arguments);
+        this._additionalAddresses = new Map();
+    }
     get midnightProviderHelper() {
         if (!this._midnightProviderHelper) {
             this._midnightProviderHelper = new MidnightProviderHelper.MidnightProviderHelper(this);
@@ -17,6 +22,18 @@ class InjectedWalletBase extends MidnightWalletConnector.MidnightWalletConnector
     findProvider() {
         return this.midnightProviderHelper.getInstalledProvider();
     }
+    getProvider() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            return this.midnightProviderHelper.getConnectedAPI();
+        });
+    }
+    signTransaction(serializedTransaction) {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            // The injected wallet signs internally during makeTransfer; the transaction
+            // arrives pre-signed and only needs to be submitted as-is.
+            return serializedTransaction;
+        });
+    }
     getConnectedAPI() {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
             return this.midnightProviderHelper.getConnectedAPI();
@@ -54,6 +71,16 @@ class InjectedWalletBase extends MidnightWalletConnector.MidnightWalletConnector
             return this.signMessage(messageToSign);
         });
     }
+    getBalance() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            return this.midnightProviderHelper.getUnshieldedBalance();
+        });
+    }
+    getBalances() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            return this.midnightProviderHelper.getBalances();
+        });
+    }
     setupEventListeners() {
         this.midnightProviderHelper._setupEventListeners();
     }
@@ -66,16 +93,61 @@ class InjectedWalletBase extends MidnightWalletConnector.MidnightWalletConnector
             this.teardownEventListeners();
         });
     }
+    setAdditionalAddresses(mainAddress, additionalAddresses) {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            this._additionalAddresses.set(mainAddress, additionalAddresses);
+        });
+    }
+    getAdditionalAddresses(mainAddress) {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            var _a;
+            if (!mainAddress)
+                return [];
+            return (_a = this._additionalAddresses.get(mainAddress)) !== null && _a !== void 0 ? _a : [];
+        });
+    }
+    fetchAndSetAdditionalAddresses(mainAddress) {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            const additionalAddresses = [];
+            try {
+                const { shieldedAddress } = yield this.getShieldedAddresses();
+                additionalAddresses.push({
+                    address: shieldedAddress,
+                    type: sdkApiCore.WalletAddressType.MidnightShielded,
+                });
+            }
+            catch (_a) {
+                // wallet may not expose shielded address
+            }
+            try {
+                const { dustAddress } = yield this.getDustAddress();
+                additionalAddresses.push({
+                    address: dustAddress,
+                    type: sdkApiCore.WalletAddressType.MidnightDust,
+                });
+            }
+            catch (_b) {
+                // wallet may not expose dust address
+            }
+            yield this.setAdditionalAddresses(mainAddress, additionalAddresses);
+        });
+    }
     getConnectedAccounts() {
         return _tslib.__awaiter(this, void 0, void 0, function* () {
             const address = yield this.midnightProviderHelper.getAddressIfConnected();
-            if (address)
+            if (address) {
+                yield this.fetchAndSetAdditionalAddresses(address);
                 return [address];
+            }
             // Attempt a silent reconnect when a previous session exists. If the wallet
             // remembers this origin, connect() resolves without showing an approval prompt.
             yield this.midnightProviderHelper.tryAutoConnect();
             const autoAddress = yield this.midnightProviderHelper.getAddressIfConnected();
-            return autoAddress ? [autoAddress] : [];
+            if (autoAddress) {
+                yield this.fetchAndSetAdditionalAddresses(autoAddress);
+                return [autoAddress];
+            }
+            return [];
         });
     }
 }

package/src/injected/InjectedWalletBase/InjectedWalletBase.d.ts

@@ -1,10 +1,14 @@
-import type { ConnectedAPI, InitialAPI } from '@midnight-ntwrk/dapp-connector-api';
+import type { ConnectedAPI, InitialAPI, TokenType } from '@midnight-ntwrk/dapp-connector-api';
+import { WalletAdditionalAddress } from '@dynamic-labs/sdk-api-core';
 import { MidnightWalletConnector } from '../../connectors/MidnightWalletConnector';
 import { MidnightProviderHelper } from '../../MidnightProviderHelper/MidnightProviderHelper';
 export declare abstract class InjectedWalletBase extends MidnightWalletConnector {
     _midnightProviderHelper: MidnightProviderHelper | undefined;
+    private _additionalAddresses;
     get midnightProviderHelper(): MidnightProviderHelper;
     findProvider(): InitialAPI | undefined;
+    getProvider<T = ConnectedAPI>(): Promise<T | undefined>;
+    signTransaction(serializedTransaction: string): Promise<string>;
     protected getConnectedAPI(): Promise<ConnectedAPI | undefined>;
     isInstalledOnBrowser(): boolean;
     connect(): Promise<void>;
@@ -12,8 +16,20 @@ export declare abstract class InjectedWalletBase extends MidnightWalletConnector
     getAddress(): Promise<string | undefined>;
     signMessage(messageToSign: string): Promise<string | undefined>;
     proveOwnership(address: string, messageToSign: string): Promise<string | undefined>;
+    getBalance(): Promise<string | undefined>;
+    getBalances(): Promise<{
+        shielded: Record<TokenType, bigint>;
+        unshielded: Record<TokenType, bigint>;
+        dust: {
+            cap: bigint;
+            balance: bigint;
+        };
+    }>;
     setupEventListeners(): void;
     teardownEventListeners(): void;
     endSession(): Promise<void>;
+    setAdditionalAddresses(mainAddress: string, additionalAddresses: WalletAdditionalAddress[]): Promise<void>;
+    getAdditionalAddresses(mainAddress?: string): Promise<WalletAdditionalAddress[]>;
+    private fetchAndSetAdditionalAddresses;
     getConnectedAccounts(): Promise<string[]>;
 }

package/src/injected/InjectedWalletBase/InjectedWalletBase.js

@@ -1,9 +1,14 @@
 'use client'
 import { __awaiter } from '../../../_virtual/_tslib.js';
+import { WalletAddressType } from '@dynamic-labs/sdk-api-core';
 import { MidnightWalletConnector } from '../../connectors/MidnightWalletConnector.js';
 import { MidnightProviderHelper } from '../../MidnightProviderHelper/MidnightProviderHelper.js';
 
 class InjectedWalletBase extends MidnightWalletConnector {
+    constructor() {
+        super(...arguments);
+        this._additionalAddresses = new Map();
+    }
     get midnightProviderHelper() {
         if (!this._midnightProviderHelper) {
             this._midnightProviderHelper = new MidnightProviderHelper(this);
@@ -13,6 +18,18 @@ class InjectedWalletBase extends MidnightWalletConnector {
     findProvider() {
         return this.midnightProviderHelper.getInstalledProvider();
     }
+    getProvider() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.midnightProviderHelper.getConnectedAPI();
+        });
+    }
+    signTransaction(serializedTransaction) {
+        return __awaiter(this, void 0, void 0, function* () {
+            // The injected wallet signs internally during makeTransfer; the transaction
+            // arrives pre-signed and only needs to be submitted as-is.
+            return serializedTransaction;
+        });
+    }
     getConnectedAPI() {
         return __awaiter(this, void 0, void 0, function* () {
             return this.midnightProviderHelper.getConnectedAPI();
@@ -50,6 +67,16 @@ class InjectedWalletBase extends MidnightWalletConnector {
             return this.signMessage(messageToSign);
         });
     }
+    getBalance() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.midnightProviderHelper.getUnshieldedBalance();
+        });
+    }
+    getBalances() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.midnightProviderHelper.getBalances();
+        });
+    }
     setupEventListeners() {
         this.midnightProviderHelper._setupEventListeners();
     }
@@ -62,16 +89,61 @@ class InjectedWalletBase extends MidnightWalletConnector {
             this.teardownEventListeners();
         });
     }
+    setAdditionalAddresses(mainAddress, additionalAddresses) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this._additionalAddresses.set(mainAddress, additionalAddresses);
+        });
+    }
+    getAdditionalAddresses(mainAddress) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            if (!mainAddress)
+                return [];
+            return (_a = this._additionalAddresses.get(mainAddress)) !== null && _a !== void 0 ? _a : [];
+        });
+    }
+    fetchAndSetAdditionalAddresses(mainAddress) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const additionalAddresses = [];
+            try {
+                const { shieldedAddress } = yield this.getShieldedAddresses();
+                additionalAddresses.push({
+                    address: shieldedAddress,
+                    type: WalletAddressType.MidnightShielded,
+                });
+            }
+            catch (_a) {
+                // wallet may not expose shielded address
+            }
+            try {
+                const { dustAddress } = yield this.getDustAddress();
+                additionalAddresses.push({
+                    address: dustAddress,
+                    type: WalletAddressType.MidnightDust,
+                });
+            }
+            catch (_b) {
+                // wallet may not expose dust address
+            }
+            yield this.setAdditionalAddresses(mainAddress, additionalAddresses);
+        });
+    }
     getConnectedAccounts() {
         return __awaiter(this, void 0, void 0, function* () {
             const address = yield this.midnightProviderHelper.getAddressIfConnected();
-            if (address)
+            if (address) {
+                yield this.fetchAndSetAdditionalAddresses(address);
                 return [address];
+            }
             // Attempt a silent reconnect when a previous session exists. If the wallet
             // remembers this origin, connect() resolves without showing an approval prompt.
             yield this.midnightProviderHelper.tryAutoConnect();
             const autoAddress = yield this.midnightProviderHelper.getAddressIfConnected();
-            return autoAddress ? [autoAddress] : [];
+            if (autoAddress) {
+                yield this.fetchAndSetAdditionalAddresses(autoAddress);
+                return [autoAddress];
+            }
+            return [];
         });
     }
 }

package/src/utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.cjs

@@ -0,0 +1,32 @@
+'use client'
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+var constants = require('../../constants.cjs');
+
+/**
+ * Converts a native unshielded balance from ledger atomic units to a decimal string
+ * for display (no thousands separators), e.g. 999999999n → "999.999999".
+ */
+const formatMidnightNativeUnshieldedBalance = (atomicAmount) => {
+    const decimals = constants.MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS;
+    const base = BigInt(Math.pow(10, decimals));
+    const negative = atomicAmount < BigInt(0);
+    const abs = negative ? -atomicAmount : atomicAmount;
+    const whole = abs / base;
+    const frac = abs % base;
+    const paddedFrac = frac.toString(10).padStart(decimals, '0');
+    let end = paddedFrac.length;
+    while (end > 0 && paddedFrac[end - 1] === '0') {
+        end -= 1;
+    }
+    const fracStr = paddedFrac.slice(0, end);
+    const sign = negative ? '-' : '';
+    if (fracStr.length === 0) {
+        return `${sign}${whole.toString(10)}`;
+    }
+    return `${sign}${whole.toString(10)}.${fracStr}`;
+};
+
+exports.formatMidnightNativeUnshieldedBalance = formatMidnightNativeUnshieldedBalance;

package/src/utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Converts a native unshielded balance from ledger atomic units to a decimal string
+ * for display (no thousands separators), e.g. 999999999n → "999.999999".
+ */
+export declare const formatMidnightNativeUnshieldedBalance: (atomicAmount: bigint) => string;

package/src/utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.js

@@ -0,0 +1,28 @@
+'use client'
+import { MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS } from '../../constants.js';
+
+/**
+ * Converts a native unshielded balance from ledger atomic units to a decimal string
+ * for display (no thousands separators), e.g. 999999999n → "999.999999".
+ */
+const formatMidnightNativeUnshieldedBalance = (atomicAmount) => {
+    const decimals = MIDNIGHT_NATIVE_UNSHIELDED_DECIMALS;
+    const base = BigInt(Math.pow(10, decimals));
+    const negative = atomicAmount < BigInt(0);
+    const abs = negative ? -atomicAmount : atomicAmount;
+    const whole = abs / base;
+    const frac = abs % base;
+    const paddedFrac = frac.toString(10).padStart(decimals, '0');
+    let end = paddedFrac.length;
+    while (end > 0 && paddedFrac[end - 1] === '0') {
+        end -= 1;
+    }
+    const fracStr = paddedFrac.slice(0, end);
+    const sign = negative ? '-' : '';
+    if (fracStr.length === 0) {
+        return `${sign}${whole.toString(10)}`;
+    }
+    return `${sign}${whole.toString(10)}.${fracStr}`;
+};
+
+export { formatMidnightNativeUnshieldedBalance };

package/src/utils/formatMidnightNativeUnshieldedBalance/index.d.ts

@@ -0,0 +1 @@
+export { formatMidnightNativeUnshieldedBalance } from './formatMidnightNativeUnshieldedBalance';

package/src/wallet/MidnightWallet.cjs

@@ -5,10 +5,14 @@ Object.defineProperty(exports, '__esModule', { value: true });
 
 var _tslib = require('../../_virtual/_tslib.cjs');
 var walletConnectorCore = require('@dynamic-labs/wallet-connector-core');
+var constants = require('../constants.cjs');
+var formatMidnightNativeUnshieldedBalance = require('../utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.cjs');
 
 class MidnightWallet extends walletConnectorCore.Wallet {
     sendBalance(_a) {
         return _tslib.__awaiter(this, arguments, void 0, function* ({ amount, toAddress, token, }) {
+            // Midnight address format: shielded addresses begin with "mn_shield",
+            // all others are unshielded. This prefix is defined by the Midnight protocol.
             const type = toAddress.startsWith('mn_shield') ? 'shielded' : 'unshielded';
             const { serializedTransaction } = yield this._connector.createTransferTransaction({
                 transfers: [
@@ -24,6 +28,43 @@ class MidnightWallet extends walletConnectorCore.Wallet {
             yield this._connector.submitTransaction(signedTransaction);
         });
     }
+    getBalance() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            return this._connector.getBalance(this.address);
+        });
+    }
+    getBalances() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            return this._connector.getBalances();
+        });
+    }
+    getShieldedBalance() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            const balances = yield this._connector.getShieldedBalances();
+            const raw = balances[constants.MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE];
+            if (raw === undefined)
+                return undefined;
+            return formatMidnightNativeUnshieldedBalance.formatMidnightNativeUnshieldedBalance(raw);
+        });
+    }
+    getUnshieldedBalance() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            const balances = yield this._connector.getUnshieldedBalances();
+            const raw = balances[constants.MIDNIGHT_NATIVE_TOKEN_TYPE];
+            if (raw === undefined)
+                return undefined;
+            return formatMidnightNativeUnshieldedBalance.formatMidnightNativeUnshieldedBalance(raw);
+        });
+    }
+    getDustBalance() {
+        return _tslib.__awaiter(this, void 0, void 0, function* () {
+            const { balance, cap } = yield this._connector.getDustBalance();
+            return {
+                balance: formatMidnightNativeUnshieldedBalance.formatMidnightNativeUnshieldedBalance(balance),
+                cap: formatMidnightNativeUnshieldedBalance.formatMidnightNativeUnshieldedBalance(cap),
+            };
+        });
+    }
 }
 
 exports.MidnightWallet = MidnightWallet;

package/src/wallet/MidnightWallet.d.ts

@@ -1,4 +1,6 @@
+import { TokenType } from '@midnight-ntwrk/dapp-connector-api';
 import { Wallet } from '@dynamic-labs/wallet-connector-core';
+import { WalletAddressType } from '@dynamic-labs/sdk-api-core';
 import { MidnightWalletConnector } from '../connectors/MidnightWalletConnector';
 export declare class MidnightWallet extends Wallet<MidnightWalletConnector> {
     sendBalance({ amount, toAddress, token, }: {
@@ -8,5 +10,21 @@ export declare class MidnightWallet extends Wallet<MidnightWalletConnector> {
             address: string;
             decimals?: number;
         };
+        addressType?: WalletAddressType;
     }): Promise<undefined>;
+    getBalance(): Promise<string | undefined>;
+    getBalances(): Promise<{
+        shielded: Record<TokenType, bigint>;
+        unshielded: Record<TokenType, bigint>;
+        dust: {
+            cap: bigint;
+            balance: bigint;
+        };
+    }>;
+    getShieldedBalance(): Promise<string | undefined>;
+    getUnshieldedBalance(): Promise<string | undefined>;
+    getDustBalance(): Promise<{
+        balance: string;
+        cap: string;
+    }>;
 }

package/src/wallet/MidnightWallet.js

@@ -1,10 +1,14 @@
 'use client'
 import { __awaiter } from '../../_virtual/_tslib.js';
 import { Wallet } from '@dynamic-labs/wallet-connector-core';
+import { MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE, MIDNIGHT_NATIVE_TOKEN_TYPE } from '../constants.js';
+import { formatMidnightNativeUnshieldedBalance } from '../utils/formatMidnightNativeUnshieldedBalance/formatMidnightNativeUnshieldedBalance.js';
 
 class MidnightWallet extends Wallet {
     sendBalance(_a) {
         return __awaiter(this, arguments, void 0, function* ({ amount, toAddress, token, }) {
+            // Midnight address format: shielded addresses begin with "mn_shield",
+            // all others are unshielded. This prefix is defined by the Midnight protocol.
             const type = toAddress.startsWith('mn_shield') ? 'shielded' : 'unshielded';
             const { serializedTransaction } = yield this._connector.createTransferTransaction({
                 transfers: [
@@ -20,6 +24,43 @@ class MidnightWallet extends Wallet {
             yield this._connector.submitTransaction(signedTransaction);
         });
     }
+    getBalance() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this._connector.getBalance(this.address);
+        });
+    }
+    getBalances() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this._connector.getBalances();
+        });
+    }
+    getShieldedBalance() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const balances = yield this._connector.getShieldedBalances();
+            const raw = balances[MIDNIGHT_NATIVE_SHIELDED_TOKEN_TYPE];
+            if (raw === undefined)
+                return undefined;
+            return formatMidnightNativeUnshieldedBalance(raw);
+        });
+    }
+    getUnshieldedBalance() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const balances = yield this._connector.getUnshieldedBalances();
+            const raw = balances[MIDNIGHT_NATIVE_TOKEN_TYPE];
+            if (raw === undefined)
+                return undefined;
+            return formatMidnightNativeUnshieldedBalance(raw);
+        });
+    }
+    getDustBalance() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { balance, cap } = yield this._connector.getDustBalance();
+            return {
+                balance: formatMidnightNativeUnshieldedBalance(balance),
+                cap: formatMidnightNativeUnshieldedBalance(cap),
+            };
+        });
+    }
 }
 
 export { MidnightWallet };