"use client";
 
import { useWallet } from "@zkscatter/sdk/react";
import {
  generateNote,
  generateDepositProof,
  deriveEdDSAKey,
} from "@zkscatter/sdk/zk";
import { ensureAllowance, callDeposit } from "@zkscatter/sdk/contracts";
import { createIndexedDbNoteAdapter } from "@zkscatter/sdk/notes";
import { network } from "@/lib/network";
 
const notes = createIndexedDbNoteAdapter();
 
export function DepositButton({
  token,
  symbol,
  amount,
}: {
  token: string;
  symbol: string;
  amount: bigint;
}) {
  const { signer } = useWallet();
 
  async function deposit() {
    if (!signer) throw new Error("Connect wallet first");
 
    const eddsa = await deriveEdDSAKey(signer);
    const note = generateNote(token, amount, eddsa.publicKey);
 
    const result = await generateDepositProof(note, {
      wasm: "/zk/deposit.wasm",
      zkey: "/zk/deposit_final.zkey",
    });
 
    // ensureAllowance returns one or two PENDING approval txs (USDT
    // requires reset-to-zero + approve(amount)). Wait for each to
    // confirm before depositing — otherwise the deposit can race
    // ahead of the approve and revert with `transfer amount exceeds
    // allowance`.
    const approvals = await ensureAllowance(
      signer,
      token,
      network.contracts.commitmentPool,
      amount,
    );
    await Promise.all(approvals.map((tx) => tx.wait()));
 
    const tx = await callDeposit(
      signer,
      network.contracts.commitmentPool,
      result,
      token,
      amount,
    );
    const receipt = await tx.wait();
 
    await notes.ready();
    await notes.put({
      id: result.commitment.toString(16),
      label: `Deposit ${symbol}`,
      symbol,
      amount: amount.toString(),
      note,
      commitment: result.commitment,
      leafIndex: -1, // populated when the CommitmentInserted event lands
      txHash: receipt?.hash,
      chainId: network.chainId,
      createdAt: Date.now(),
    });
  }
 
  return <button onClick={deposit}>Deposit {symbol}</button>;
}

"use client";
 
import { useEffect } from "react";
import { useWallet } from "@zkscatter/sdk/react";
import { subscribeCommitmentInserted } from "@zkscatter/sdk";
 
export function PoolWatcher() {
  const { readProvider } = useWallet();
 
  useEffect(() => {
    if (!readProvider) return;
    const unsub = subscribeCommitmentInserted(
      readProvider,
      network.contracts.commitmentPool,
      async (row) => {
        const stored = (await notes.loadAll()).find(
          (n) => n.commitment === row.commitment,
        );
        if (stored && stored.leafIndex < 0) {
          await notes.put({ ...stored, leafIndex: row.leafIndex });
        }
      },
    );
    return unsub;
  }, [readProvider]);
 
  return null;
}