@kohaku-eth/tornado-cash 0.0.2-alpha.11 → 0.0.2-alpha.12

package/dist/index.d.ts

@@ -39,41 +39,6 @@ interface SecretManagerParams {
 }
 declare function SecretManager({ host: { keystore }, accountIndex }: SecretManagerParams): Promise<ISecretManager>;
 
-interface IRelayerStatusResponse {
-    currentQueue: number;
-    netId: number;
-    rewardAccount: string;
-    version: string;
-    tornadoServiceFee: number;
-    ethPrices: Record<string, string>;
-}
-interface ITornadoWithdrawRequest {
-    proof: `0x${string}`;
-    args: [`0x${string}`, `0x${string}`, `0x${string}`, `0x${string}`, `0x${string}`, `0x${string}`];
-    contract: `0x${string}`;
-}
-interface ITornadoWithdrawResponse {
-    id: string;
-}
-interface IRelayerClient {
-    getStatus(hostname: string): Promise<IRelayerStatusResponse>;
-    withdraw(relayerUrl: string, body: ITornadoWithdrawRequest): Promise<ITornadoWithdrawResponse>;
-}
-
-interface InstanceRegistryConfig extends IScrapeableContract {
-    lastDeployedOnBlock?: bigint;
-}
-interface ProtocolConfigState {
-    chainId: bigint;
-    ensSubdomainKey: string;
-    instanceRegistry: InstanceRegistryConfig;
-    relayerRegistry: IScrapeableContract;
-    aggregator: IContract;
-}
-
-type RootState = ReturnType<ReturnType<typeof combineReducers<typeof reducers>>>;
-type PublicRootState = Omit<RootState, 'userSecrets'>;
-
 interface IBaseEvent {
     blockNumber: bigint;
     transactionHash: bigint;
@@ -86,6 +51,15 @@ interface IDepositEvent extends IPoolEvent {
     leafIndex: number;
     timestamp: bigint;
 }
+interface IIndexedDepositEvent extends IDepositEvent {
+    index: number;
+}
+interface IDepositWithAsset extends IIndexedDepositEvent {
+    assetAddress: Address;
+}
+interface IDepositWithBalance extends IDepositWithAsset {
+    balance: bigint;
+}
 interface IWithdrawalEvent extends IPoolEvent {
     to: Address;
     nullifierHash: NullifierHash;
@@ -111,6 +85,41 @@ interface IAsset {
     symbol: string;
 }
 
+interface IRelayerStatusResponse {
+    currentQueue: number;
+    netId: number;
+    rewardAccount: string;
+    version: string;
+    tornadoServiceFee: number;
+    ethPrices: Record<string, string>;
+}
+interface ITornadoWithdrawRequest {
+    proof: `0x${string}`;
+    args: [`0x${string}`, `0x${string}`, `0x${string}`, `0x${string}`, `0x${string}`, `0x${string}`];
+    contract: `0x${string}`;
+}
+interface ITornadoWithdrawResponse {
+    id: string;
+}
+interface IRelayerClient {
+    getStatus(hostname: string): Promise<IRelayerStatusResponse>;
+    withdraw(relayerUrl: string, body: ITornadoWithdrawRequest): Promise<ITornadoWithdrawResponse>;
+}
+
+interface InstanceRegistryConfig extends IScrapeableContract {
+    lastDeployedOnBlock?: bigint;
+}
+interface ProtocolConfigState {
+    chainId: bigint;
+    ensSubdomainKey: string;
+    instanceRegistry: InstanceRegistryConfig;
+    relayerRegistry: IScrapeableContract;
+    aggregator: IContract;
+}
+
+type RootState = ReturnType<ReturnType<typeof combineReducers<typeof reducers>>>;
+type PublicRootState = Omit<RootState, 'userSecrets'>;
+
 declare const POOL_EVENTS_SIGNATURES: {
     readonly Deposited: {
         readonly name: "Deposit";
@@ -323,16 +332,6 @@ interface IPaymasterBroadcasterClient {
     broadcast(withdrawals: IPaymasterWithdrawalPayload[]): Promise<PaymasterBroadcastResult[]>;
 }
 
-declare function makeLazyProverFactory(artifactsLoader: () => Promise<ITornadoArtifacts>, version?: 'legacy' | 'js'): () => Promise<ITornadoProver>;
-
-interface WorkerInitOptions {
-    protocolConfig: TCProtocolConfig;
-    relayerConfig?: IRelayerFeeConfig;
-    accountIndex: number;
-    paymasterConfig: IChainsPaymastersConfig;
-    proverVersion?: Parameters<typeof makeLazyProverFactory>[1];
-}
-
 type DelegationConfig = {
     mode: 'deterministic';
     path?: string;
@@ -367,7 +366,7 @@ interface ITornadoArtifacts {
     provingKey: ArrayBuffer;
 }
 type TCProtocolConfig = Omit<ProtocolConfigState, 'chainId'>;
-interface TCProtocolParams extends Pick<WorkerInitOptions, 'proverVersion'> {
+interface TCProtocolParams {
     accountIndex?: number;
     paymasterConfig: IChainsPaymastersConfig;
     secretManagerFactory: (params: SecretManagerParams) => Promise<ISecretManager>;
@@ -399,6 +398,15 @@ interface IPaymasterWithdrawParams extends IWithdrawBaseParams {
     delegation?: DelegationConfig;
 }
 type IWithdrawapOperationParams = IRelayerWithdrawParams | IPaymasterWithdrawParams;
+interface IGetNotesParams extends IBaseOperationParams {
+    includeSpent?: boolean;
+    assets?: Address[];
+}
+type TCNote = Pick<IDepositWithBalance, 'commitment' | 'balance' | 'assetAddress' | 'timestamp' | 'leafIndex' | 'pool'> & {
+    /** Fixed pool denomination for this note. */
+    amount: bigint;
+    depositIndex: number;
+};
 type StoreKey = `${string}-${string}`;
 type StoreStorageKey = `tornado-cash-state-${StoreKey}`;
 interface IStateManager {
@@ -419,6 +427,12 @@ interface IStateManager {
      * All assets if not specified.
      */
     getBalances: SpecificAssetBalanceFn<true>;
+    /**
+     * Gets all notes for the account.
+     * @param includeSpent - If true, include notes with zero balance
+     * @param assets - Optional filter by specific assets
+     */
+    getNotes: (params: IGetNotesParams) => Promise<TCNote[]>;
     dumpState: () => Record<StoreStorageKey, PublicRootState>;
 }
 
@@ -431,7 +445,7 @@ type TCBroadcaster = Broadcaster<TCPrivateOperation, ITornadoWithdrawResponse[]>
 interface TCBaseCredential {
     accountIndex: number;
 }
-interface TCPluginParameters extends TCBroadcasterParameters, TCBaseCredential, Pick<TCProtocolParams, 'proverVersion' | 'initialState'> {
+interface TCPluginParameters extends TCBroadcasterParameters, TCBaseCredential, Pick<TCProtocolParams, 'initialState'> {
     protocolConfig: TCProtocolConfig;
     relayerConfig?: IRelayerFeeConfig;
     stateManagerWorkerUrl?: string;
@@ -463,6 +477,7 @@ type TCInstance = PluginInstance<TCAddress, {
         output: TCAssetAmount;
         read: TCAssetBalance;
     };
+    note: TCNote;
     extras: {
         sync(): Promise<void>;
         prepareShield(asset: TCAssetAmount, options: TCPrepareShieldOptions): Promise<TCPublicOperation>;
@@ -480,7 +495,7 @@ type RequireOnly<T, Keys extends keyof T> = Partial<T> & Pick<T, Keys>;
 declare class TornadoCashProtocol implements TCInstance {
     readonly host: Host;
     private stateManager;
-    constructor(host: Host, { accountIndex, initialState, protocolConfig, artifactsLoader, stateManagerWorkerUrl, relayerConfig, paymasterConfig, relayerClientFactory, proverVersion }: RequireOnly<TCProtocolParams, 'protocolConfig'>);
+    constructor(host: Host, { accountIndex, initialState, protocolConfig, artifactsLoader, stateManagerWorkerUrl, relayerConfig, paymasterConfig, relayerClientFactory, }: RequireOnly<TCProtocolParams, 'protocolConfig'>);
     instanceId: () => Promise<"0x1">;
     /**
      * Only process supported assets or error out?
@@ -488,6 +503,12 @@ declare class TornadoCashProtocol implements TCInstance {
      * The assets retain the provided order. If an asset is not supported its balance will be 0
      */
     balance(assets?: ERC20AssetId[]): Promise<TCAssetBalance[]>;
+    /**
+     * Returns all notes for the account.
+     * @param assets - Filter by specific assets (optional, if empty returns all)
+     * @param includeSpent - Include notes with zero balance (default: false)
+     */
+    notes(assets?: ERC20AssetId[], includeSpent?: boolean): Promise<TCNote[]>;
     prepareShield(assets: TCAssetAmount, options?: TCPrepareShieldOptions | `0x${string}`): Promise<TCPublicOperation>;
     prepareUnshield(assets: AssetAmount, to: AccountId, options?: TCRelayerUnshieldOptions): Promise<TCPrivateOperation<'relayer'>>;
     prepareUnshield(assets: AssetAmount, to: AccountId, options?: TCPaymasterUnshieldOptions): Promise<TCPrivateOperation<'paymaster'>>;
@@ -557,4 +578,4 @@ declare const TornadoCashConfigs: {
     };
 };
 
-export { type Commitment, DepositStrategy, E_ADDRESS, type ISecretManager, type Nullifier, SecretManager, type SecretManagerParams, type TCAddress, type TCAssetAmount, type TCAssetBalance, type TCBroadcaster, type TCBroadcasterParameters, type TCInstance, type TCPaymasterUnshieldOptions, type TCPluginParameters, type TCPrepareShieldOptions, type TCPrepareUnshieldOptions, type TCRelayerUnshieldOptions, TornadoCashConfigs, TornadoCashProtocol, TornadoPaymasterConfigs, createTCBroadcaster, createTCPlugin };
+export { type Commitment, DepositStrategy, E_ADDRESS, type ISecretManager, type Nullifier, SecretManager, type SecretManagerParams, type TCAddress, type TCAssetAmount, type TCAssetBalance, type TCBroadcaster, type TCBroadcasterParameters, type TCInstance, type TCNote, type TCPaymasterUnshieldOptions, type TCPluginParameters, type TCPrepareShieldOptions, type TCPrepareUnshieldOptions, type TCRelayerUnshieldOptions, TornadoCashConfigs, TornadoCashProtocol, TornadoPaymasterConfigs, createTCBroadcaster, createTCPlugin };

package/dist/index.js

@@ -8,8 +8,6 @@ import { generatePrivateKey, privateKeyToAccount } from 'viem/accounts';
 import { numberToBytesLE, bytesToNumberLE, concatBytes } from '@noble/curves/utils.js';
 import { keccak_256 } from '@noble/hashes/sha3.js';
 import { pedersenHash as pedersenHash$1, Point } from 'micro-zk-proofs/pedersen.js';
-import 'micro-zk-proofs/mimcsponge.js';
-import 'fixed-merkle-tree';
 
 // src/state/thunks/getDepositPayloadThunk.ts
 
@@ -422,8 +420,7 @@ var TornadoCashProtocol = class {
     stateManagerWorkerUrl,
     relayerConfig,
     paymasterConfig = TornadoPaymasterConfigs,
-    relayerClientFactory = () => new RelayerClient(host),
-    proverVersion
+    relayerClientFactory = () => new RelayerClient(host)
   }) {
     this.host = host;
     this.stateManager = (async () => {
@@ -442,13 +439,14 @@ var TornadoCashProtocol = class {
           proxy(host.storage),
           proxy(initialState),
           proxy(artifactsLoader),
-          { protocolConfig, accountIndex, relayerConfig, paymasterConfig, proverVersion }
+          { protocolConfig, accountIndex, relayerConfig, paymasterConfig }
         ),
         workerReady
       ]);
       return {
         sync: () => remote.sync(),
         getBalances: ((assets) => remote.getBalances(assets)),
+        getNotes: (params) => remote.getNotes(params),
         getDepositPayload: (params) => remote.getDepositPayload(params),
         getWithdrawalPayloads: (params) => remote.getWithdrawalPayloads(params),
         dumpState: (() => remote.dumpState())
@@ -487,6 +485,23 @@ var TornadoCashProtocol = class {
       };
     });
   }
+  /**
+   * Returns all notes for the account.
+   * @param assets - Filter by specific assets (optional, if empty returns all)
+   * @param includeSpent - Include notes with zero balance (default: false)
+   */
+  async notes(assets = [], includeSpent = false) {
+    const stateManager = await this.stateManager;
+    await stateManager.sync();
+    const assetAddresses = assets.map(({ contract }) => {
+      const parsedAddress = BigInt(contract);
+      return parsedAddress === E_ADDRESS_BIGINT ? 0n : parsedAddress;
+    });
+    return stateManager.getNotes({
+      includeSpent,
+      assets: assetAddresses.length > 0 ? assetAddresses : void 0
+    });
+  }
   async prepareShield(assets, options) {
     const { asset, amount } = assets;
     const strategy = typeof options === "string" ? 1 /* MinFee */ : options?.strategy || 1 /* MinFee */;