@gurge/sdk 0.3.150 → 0.3.151

package/common/src/data-structures/event-service/evm/BlockchainEventEmitter.mjs

@@ -1,17 +1,21 @@
-import { Logger as e } from "../../../error-handling/logger.mjs";
-import { getSequence as t, resolveSync as n } from "../../../functions/utils/resolve-sync.utils.mjs";
-import { blockReorgDepth as r } from "../../../constants/reorg-depths.constants.mjs";
-import { getProviderFromContract as i } from "../../../functions/utils/getSignerFromContract.mjs";
-import { BlockchainEvent as a } from "../../BlockchainEvent/BlockchainEvent.mjs";
-import { ethers as o } from "ethers";
+import { getNonLocalhostChainId as e } from "../../../constants/chains.constants.mjs";
+import { Logger as t } from "../../../error-handling/logger.mjs";
+import { ContractType as n } from "../../../types/ethereum-network.types.mjs";
+import { getDepositOnChainUtxosAddress as r } from "../../../constants/contracts.constants.mjs";
+import { getContractWithFetcherByChainId as i } from "../../../functions/web3/getContractMetadata.mjs";
+import { getProviderFromContract as a } from "../../../functions/utils/getSignerFromContract.mjs";
+import { getSequence as o, resolveSync as s } from "../../../functions/utils/resolve-sync.utils.mjs";
+import { blockReorgDepth as c } from "../../../constants/reorg-depths.constants.mjs";
+import { BlockchainEvent as l } from "../../BlockchainEvent/BlockchainEvent.mjs";
+import { ethers as u } from "ethers";
 //#region libs/shared/common/src/data-structures/event-service/evm/BlockchainEventEmitter.ts
-var s = class {
+var d = class {
 	RETRY_MIN_PAGE_SIZE = 100;
 	MAX_RETRIES = 20;
 	_chainId;
 	contract;
-	depositOnChainUtxosContract;
 	_latestBlockNumber;
+	_blockedUtxosContract;
 	_initialBlockNumber;
 	onEventsProcessed;
 	eventProcessors = /* @__PURE__ */ new Set();
@@ -20,8 +24,8 @@ var s = class {
 	isServer;
 	inProgress;
 	eventsFetchingMutex;
-	constructor(e, t, n, r, i, a, o) {
-		this.contract = t, this.depositOnChainUtxosContract = o, this._chainId = e, this._initialBlockNumber = n, this.maxPageSize = a, this.isServer = r, this.inProgress = !1, this.eventsFetchingMutex = i;
+	constructor(e, t, n, r, i, a) {
+		this.contract = t, this._chainId = e, this._initialBlockNumber = n, this.maxPageSize = a, this.isServer = r, this.inProgress = !1, this.eventsFetchingMutex = i;
 	}
 	get latestBlockNumber() {
 		return this._latestBlockNumber ?? this._initialBlockNumber;
@@ -48,50 +52,54 @@ var s = class {
 	requireReady() {
 		if (!this._isReady) throw Error("Not ready");
 	}
-	async getEventsForSingleContract(e, t, n, r = 0) {
+	async getEventsForSingleContract(e, t, n, r = 0, i = "*") {
 		try {
-			return (await e.queryFilter("*", t, n)).filter((e) => e instanceof o.EventLog);
-		} catch (i) {
+			return (await e.queryFilter(i, t, n)).filter((e) => e instanceof u.EventLog);
+		} catch (a) {
 			if (r < this.MAX_RETRIES && n - t > this.RETRY_MIN_PAGE_SIZE) {
-				let i = Math.floor((t + n) / 2), a = await this.getEventsForSingleContract(e, t, i, r + 1), o = await this.getEventsForSingleContract(e, i + 1, n, r + 1);
-				return [...a, ...o];
+				let a = Math.floor((t + n) / 2), o = await this.getEventsForSingleContract(e, t, a, r + 1, i), s = await this.getEventsForSingleContract(e, a + 1, n, r + 1, i);
+				return [...o, ...s];
 			}
-			throw i;
+			throw a;
 		}
 	}
+	get blockedUtxosContract() {
+		if (this._blockedUtxosContract === void 0) {
+			let t = e(this._chainId);
+			this._blockedUtxosContract = r(t) ? i(t, n.DepositOnChainUtxosContract) : null;
+		}
+		return this._blockedUtxosContract ?? void 0;
+	}
 	addEventProcessorFunction(e) {
 		if (this._isReady) throw Error("Can't add event processor after event emitter initialized");
 		this.eventProcessors.add(e);
 	}
-	async getEventsInRange(e, r) {
-		let i = [], o = t(e, r, this.maxPageSize);
-		await n(o.map(({ from: e, to: t }) => async () => {
-			let n = await this.getEventsForSingleContract(this.contract, e, t);
-			i.push(...n);
-		}));
-		let s = this.depositOnChainUtxosContract;
-		return s && await n(o.map(({ from: e, to: t }) => async () => {
-			let n = await this.getEventsForSingleContract(s, e, t);
-			i.push(...n);
-		})), i.map((e) => a.fromEthers(e));
+	async getEventsInRange(e, t) {
+		let n = [];
+		return await s(o(e, t, this.maxPageSize).map(({ from: e, to: t }) => async () => {
+			let r = await this.getEventsForSingleContract(this.contract, e, t);
+			n.push(...r);
+			let i = this.blockedUtxosContract;
+			i && n.push(...await this.getEventsForSingleContract(i, e, t, 0, "BlockedUtxosCreated"));
+		})), n.map((e) => l.fromEthers(e));
 	}
 	async getLastBlockNumberForEventRequest() {
-		let e = await i(this.contract).getBlockNumber();
+		let e = await a(this.contract).getBlockNumber();
 		if (!this.isServer) return e;
-		let { chainId: t } = await i(this.contract).getNetwork();
-		return Math.max(this.latestBlockNumber, e - r[Number(t)] + 1);
+		let { chainId: t } = await a(this.contract).getNetwork();
+		return Math.max(this.latestBlockNumber, e - c[Number(t)] + 1);
 	}
-	async retrieveEvents(t, n = !1) {
+	async retrieveEvents(e, n = !1) {
 		return this.requireReady(), await this.eventsFetchingMutex.runExclusive(async () => {
 			try {
 				if (this.inProgress && !n) return !1;
 				this.inProgress = !0;
-				let e = await this.getLastBlockNumberForEventRequest();
-				if (e < t) return this.inProgress = !1, !1;
-				let r = await this.getEventsInRange(t, e);
-				return await this.processEvents(r, e), this._latestBlockNumber = e, this.inProgress = !1, !0;
-			} catch (t) {
-				return e.error("retrieveEvents error", t), this.inProgress = !1, !1;
+				let t = await this.getLastBlockNumberForEventRequest();
+				if (t < e) return this.inProgress = !1, !1;
+				let r = await this.getEventsInRange(e, t);
+				return await this.processEvents(r, t), this._latestBlockNumber = t, this.inProgress = !1, !0;
+			} catch (e) {
+				return t.error("retrieveEvents error", e), this.inProgress = !1, !1;
 			}
 		});
 	}
@@ -101,4 +109,4 @@ var s = class {
 	}
 };
 //#endregion
-export { s as BlockchainEventEmitter };
+export { d as BlockchainEventEmitter };

package/common/src/data-structures/event-service/evm/ClientBlockchainEventEmitter.cjs

@@ -1 +1 @@
-const e=require(`./BlockchainEventEmitter.cjs`),t=require(`../helpers/ClientEventHelper.cjs`);var n=class extends e.BlockchainEventEmitter{clientEventHelper;constructor(e,n,r,i,a,o,s){super(n,r,i,!1,a,o,s),this.clientEventHelper=new t.ClientEventHelper(n,e)}clear(){this.clientEventHelper.clear()}startUpdateListener(){this.clientEventHelper.startUpdateListener(async e=>{await this.eventsFetchingMutex.runExclusive(async()=>{await this.processEvents(e)})},this.latestBlockNumber+1,e=>this.advanceLatestBlockNumber(e))}};exports.ClientBlockchainEventEmitter=n;
+const e=require(`./BlockchainEventEmitter.cjs`),t=require(`../helpers/ClientEventHelper.cjs`);var n=class extends e.BlockchainEventEmitter{clientEventHelper;constructor(e,n,r,i,a,o){super(n,r,i,!1,a,o),this.clientEventHelper=new t.ClientEventHelper(n,e)}clear(){this.clientEventHelper.clear()}startUpdateListener(){this.clientEventHelper.startUpdateListener(async e=>{await this.eventsFetchingMutex.runExclusive(async()=>{await this.processEvents(e)})},this.latestBlockNumber+1,e=>this.advanceLatestBlockNumber(e))}};exports.ClientBlockchainEventEmitter=n;

package/common/src/data-structures/event-service/evm/ClientBlockchainEventEmitter.d.ts

@@ -5,7 +5,7 @@ import { ChainBalanceFetchingLock } from '../../../mutexes/chainBalanceRwLock';
 import { ClientEventHelper } from '../helpers/ClientEventHelper';
 export declare class ClientBlockchainEventEmitter extends BlockchainEventEmitter {
     clientEventHelper: ClientEventHelper;
-    constructor(eventCategory: EventCategory, chainId: number, contract: ethers.Contract, initialBlockNumber: number, eventsFetchingMutex: ChainBalanceFetchingLock, maxPageSize?: number, depositOnChainUtxosContract?: ethers.Contract);
+    constructor(eventCategory: EventCategory, chainId: number, contract: ethers.Contract, initialBlockNumber: number, eventsFetchingMutex: ChainBalanceFetchingLock, maxPageSize?: number);
     protected clear(): void;
     protected startUpdateListener(): void;
 }

package/common/src/data-structures/event-service/evm/ClientBlockchainEventEmitter.mjs

@@ -3,8 +3,8 @@ import { ClientEventHelper as t } from "../helpers/ClientEventHelper.mjs";
 //#region libs/shared/common/src/data-structures/event-service/evm/ClientBlockchainEventEmitter.ts
 var n = class extends e {
 	clientEventHelper;
-	constructor(e, n, r, i, a, o, s) {
-		super(n, r, i, !1, a, o, s), this.clientEventHelper = new t(n, e);
+	constructor(e, n, r, i, a, o) {
+		super(n, r, i, !1, a, o), this.clientEventHelper = new t(n, e);
 	}
 	clear() {
 		this.clientEventHelper.clear();

package/common/src/data-structures/event-service/solana/AbstractSolanaCommitmentsSnapshotService.cjs

@@ -1 +1 @@
-const e=require(`../../../functions/utils/amounts.utils.cjs`),t=require(`../../../crypto/poseidon.cjs`),n=require(`../../crypto-keys/decodeUTXO.cjs`);require(`../../../crypto/index.cjs`);const r=require(`../../merkle-tree/MerkleTree.cjs`);require(`../../merkle-tree/index.cjs`);const i=require(`../../../functions/utils/merkleTree.utils.cjs`),a=require(`./AbstractSolanaSnapshotService.cjs`);var o=class extends a.AbstractSolanaSnapshotService{constructor(e){super(e,`NewCommitment`)}_merkleTree;_encryptedOutputs;_encryptedOutputsByCommitment=new Map;get merkleTree(){return this._merkleTree}get encryptedOutputs(){return this._encryptedOutputs}serializeSnapshot(e){return{merkleTree:e.payload.merkleTree.toJSON(),latestBlockNumber:e.latestBlockNumber,encryptedOutputs:e.payload.encryptedOutputs.map(e=>({value:e.value,isPositive:String(e.isPositive),isBlocked:e.isBlocked??!1})),encryptedOutputsByCommitment:Array.from(e.payload.encryptedOutputsByCommitment.entries()).map(([e,t])=>({commitment:e.toString(),encryptedOutput:t}))}}deserializeSnapshot(e){let n;n=e.latestBlockNumber&&e.merkleTree?i.contructMerkleTreeFromSerialized(e.merkleTree):r.MerkleTree.create(t.poseidonFunction,0n);let a=e.encryptedOutputs?.map(e=>({value:e.value,isPositive:JSON.parse(e.isPositive),isBlocked:e.isBlocked??!1})),o=new Map;return e.encryptedOutputsByCommitment?.forEach(({commitment:e,encryptedOutput:t})=>{o.set(BigInt(e),t)}),{latestBlockNumber:e.latestBlockNumber??0,payload:{merkleTree:n,encryptedOutputs:a||[],encryptedOutputsByCommitment:o}}}getSnapshotPayload(){return{merkleTree:this._merkleTree,encryptedOutputs:this._encryptedOutputs,encryptedOutputsByCommitment:this._encryptedOutputsByCommitment}}populateSnapshot({payload:{merkleTree:e,encryptedOutputs:t,encryptedOutputsByCommitment:n}}){this._merkleTree=e,this._encryptedOutputs=t,this._encryptedOutputsByCommitment=n}async mapEvent(t){let r=t.getArg(`commitment`),i=t.getArg(`index`),a=t.getArg(`encrypted_output`),o=t.getArg(`on_chain_data`),s=e.advancedToBigInt(r),c=e.advancedToBigInt(i);return a.length===0?{commitment:s,index:c,encryptedOutput:n.encodeSolanaOnChainUtxo(o)}:{commitment:s,index:c,encryptedOutput:`0x${Buffer.from(a).toString(`hex`)}`}}acceptEvent(e,t,r,i){let a=e.index>=0n?e.index:-1n*e.index,o=this._merkleTree.getValue(a);if(o===e.commitment)return!1;this._merkleTree.insert(e.commitment,a);let s=o===void 0?void 0:this._encryptedOutputsByCommitment.get(o);if(o!==void 0&&s!==void 0){let e=this.encryptedOutputs.findIndex(e=>e.value===s);e!==-1&&this.encryptedOutputs.splice(e,1),this._encryptedOutputsByCommitment.delete(o)}if(this._encryptedOutputsByCommitment.set(e.commitment,e.encryptedOutput),this.encryptedOutputs.map(e=>e.value).includes(e.encryptedOutput))return!0;let c={value:e.encryptedOutput,isPositive:!n.isSolanaOnChainUtxo(e.encryptedOutput),isBlocked:i??!1};return this._encryptedOutputs.push(c),!0}};exports.AbstractSolanaCommitmentsSnapshotService=o;
+const e=require(`../../../functions/utils/amounts.utils.cjs`),t=require(`../../../crypto/poseidon.cjs`),n=require(`../../crypto-keys/decodeUTXO.cjs`);require(`../../../crypto/index.cjs`);const r=require(`../../merkle-tree/MerkleTree.cjs`);require(`../../merkle-tree/index.cjs`);const i=require(`../../../functions/utils/merkleTree.utils.cjs`),a=require(`./AbstractSolanaSnapshotService.cjs`);var o=class extends a.AbstractSolanaSnapshotService{constructor(e){super(e,`NewCommitment`)}_merkleTree;_encryptedOutputs;_encryptedOutputsByCommitment=new Map;get merkleTree(){return this._merkleTree}get encryptedOutputs(){return this._encryptedOutputs}serializeSnapshot(e){return{merkleTree:e.payload.merkleTree.toJSON(),latestBlockNumber:e.latestBlockNumber,encryptedOutputs:e.payload.encryptedOutputs.map(e=>({value:e.value,isPositive:String(e.isPositive),isBlocked:e.isBlocked??!1,encryptedBlob:e.encryptedBlob})),encryptedOutputsByCommitment:Array.from(e.payload.encryptedOutputsByCommitment.entries()).map(([e,t])=>({commitment:e.toString(),encryptedOutput:t}))}}deserializeSnapshot(e){let n;n=e.latestBlockNumber&&e.merkleTree?i.contructMerkleTreeFromSerialized(e.merkleTree):r.MerkleTree.create(t.poseidonFunction,0n);let a=e.encryptedOutputs?.map(e=>({value:e.value,isPositive:JSON.parse(e.isPositive),isBlocked:e.isBlocked??!1,encryptedBlob:e.encryptedBlob})),o=new Map;return e.encryptedOutputsByCommitment?.forEach(({commitment:e,encryptedOutput:t})=>{o.set(BigInt(e),t)}),{latestBlockNumber:e.latestBlockNumber??0,payload:{merkleTree:n,encryptedOutputs:a||[],encryptedOutputsByCommitment:o}}}getSnapshotPayload(){return{merkleTree:this._merkleTree,encryptedOutputs:this._encryptedOutputs,encryptedOutputsByCommitment:this._encryptedOutputsByCommitment}}populateSnapshot({payload:{merkleTree:e,encryptedOutputs:t,encryptedOutputsByCommitment:n}}){this._merkleTree=e,this._encryptedOutputs=t,this._encryptedOutputsByCommitment=n}async mapEvent(t){let r=t.getArg(`commitment`),i=t.getArg(`index`),a=t.getArg(`encrypted_output`),o=t.getArg(`on_chain_data`),s=e.advancedToBigInt(r),c=e.advancedToBigInt(i),l=`0x${Buffer.from(a).toString(`hex`)}`;return n.hasSolanaOnChainData(o)?{commitment:s,index:c,encryptedOutput:n.encodeSolanaOnChainUtxo(o),encryptedBlob:a.length>0?l:void 0}:{commitment:s,index:c,encryptedOutput:l}}acceptEvent(e,t,r,i){let a=e.index>=0n?e.index:-1n*e.index,o=this._merkleTree.getValue(a);if(o===e.commitment)return!1;this._merkleTree.insert(e.commitment,a);let s=o===void 0?void 0:this._encryptedOutputsByCommitment.get(o);if(o!==void 0&&s!==void 0){let e=this.encryptedOutputs.findIndex(e=>e.value===s);e!==-1&&this.encryptedOutputs.splice(e,1),this._encryptedOutputsByCommitment.delete(o)}if(this._encryptedOutputsByCommitment.set(e.commitment,e.encryptedOutput),this.encryptedOutputs.map(e=>e.value).includes(e.encryptedOutput))return!0;let c={value:e.encryptedOutput,isPositive:!n.isSolanaOnChainUtxo(e.encryptedOutput),isBlocked:i??!1,encryptedBlob:e.encryptedBlob};return this._encryptedOutputs.push(c),!0}};exports.AbstractSolanaCommitmentsSnapshotService=o;

package/common/src/data-structures/event-service/solana/AbstractSolanaCommitmentsSnapshotService.mjs

@@ -1,13 +1,13 @@
 import { advancedToBigInt as e } from "../../../functions/utils/amounts.utils.mjs";
 import { poseidonFunction as t } from "../../../crypto/poseidon.mjs";
-import { encodeSolanaOnChainUtxo as n, isSolanaOnChainUtxo as r } from "../../crypto-keys/decodeUTXO.mjs";
+import { encodeSolanaOnChainUtxo as n, hasSolanaOnChainData as r, isSolanaOnChainUtxo as i } from "../../crypto-keys/decodeUTXO.mjs";
 import "../../../crypto/index.mjs";
-import { MerkleTree as i } from "../../merkle-tree/MerkleTree.mjs";
+import { MerkleTree as a } from "../../merkle-tree/MerkleTree.mjs";
 import "../../merkle-tree/index.mjs";
-import { contructMerkleTreeFromSerialized as a } from "../../../functions/utils/merkleTree.utils.mjs";
-import { AbstractSolanaSnapshotService as o } from "./AbstractSolanaSnapshotService.mjs";
+import { contructMerkleTreeFromSerialized as o } from "../../../functions/utils/merkleTree.utils.mjs";
+import { AbstractSolanaSnapshotService as s } from "./AbstractSolanaSnapshotService.mjs";
 //#region libs/shared/common/src/data-structures/event-service/solana/AbstractSolanaCommitmentsSnapshotService.ts
-var s = class extends o {
+var c = class extends s {
 	constructor(e) {
 		super(e, "NewCommitment");
 	}
@@ -27,7 +27,8 @@ var s = class extends o {
 			encryptedOutputs: e.payload.encryptedOutputs.map((e) => ({
 				value: e.value,
 				isPositive: String(e.isPositive),
-				isBlocked: e.isBlocked ?? !1
+				isBlocked: e.isBlocked ?? !1,
+				encryptedBlob: e.encryptedBlob
 			})),
 			encryptedOutputsByCommitment: Array.from(e.payload.encryptedOutputsByCommitment.entries()).map(([e, t]) => ({
 				commitment: e.toString(),
@@ -37,20 +38,21 @@ var s = class extends o {
 	}
 	deserializeSnapshot(e) {
 		let n;
-		n = e.latestBlockNumber && e.merkleTree ? a(e.merkleTree) : i.create(t, 0n);
+		n = e.latestBlockNumber && e.merkleTree ? o(e.merkleTree) : a.create(t, 0n);
 		let r = e.encryptedOutputs?.map((e) => ({
 			value: e.value,
 			isPositive: JSON.parse(e.isPositive),
-			isBlocked: e.isBlocked ?? !1
-		})), o = /* @__PURE__ */ new Map();
+			isBlocked: e.isBlocked ?? !1,
+			encryptedBlob: e.encryptedBlob
+		})), i = /* @__PURE__ */ new Map();
 		return e.encryptedOutputsByCommitment?.forEach(({ commitment: e, encryptedOutput: t }) => {
-			o.set(BigInt(e), t);
+			i.set(BigInt(e), t);
 		}), {
 			latestBlockNumber: e.latestBlockNumber ?? 0,
 			payload: {
 				merkleTree: n,
 				encryptedOutputs: r || [],
-				encryptedOutputsByCommitment: o
+				encryptedOutputsByCommitment: i
 			}
 		};
 	}
@@ -65,18 +67,19 @@ var s = class extends o {
 		this._merkleTree = e, this._encryptedOutputs = t, this._encryptedOutputsByCommitment = n;
 	}
 	async mapEvent(t) {
-		let r = t.getArg("commitment"), i = t.getArg("index"), a = t.getArg("encrypted_output"), o = t.getArg("on_chain_data"), s = e(r), c = e(i);
-		return a.length === 0 ? {
-			commitment: s,
-			index: c,
-			encryptedOutput: n(o)
+		let i = t.getArg("commitment"), a = t.getArg("index"), o = t.getArg("encrypted_output"), s = t.getArg("on_chain_data"), c = e(i), l = e(a), u = `0x${Buffer.from(o).toString("hex")}`;
+		return r(s) ? {
+			commitment: c,
+			index: l,
+			encryptedOutput: n(s),
+			encryptedBlob: o.length > 0 ? u : void 0
 		} : {
-			commitment: s,
-			index: c,
-			encryptedOutput: `0x${Buffer.from(a).toString("hex")}`
+			commitment: c,
+			index: l,
+			encryptedOutput: u
 		};
 	}
-	acceptEvent(e, t, n, i) {
+	acceptEvent(e, t, n, r) {
 		let a = e.index >= 0n ? e.index : -1n * e.index, o = this._merkleTree.getValue(a);
 		if (o === e.commitment) return !1;
 		this._merkleTree.insert(e.commitment, a);
@@ -88,11 +91,12 @@ var s = class extends o {
 		if (this._encryptedOutputsByCommitment.set(e.commitment, e.encryptedOutput), this.encryptedOutputs.map((e) => e.value).includes(e.encryptedOutput)) return !0;
 		let c = {
 			value: e.encryptedOutput,
-			isPositive: !r(e.encryptedOutput),
-			isBlocked: i ?? !1
+			isPositive: !i(e.encryptedOutput),
+			isBlocked: r ?? !1,
+			encryptedBlob: e.encryptedBlob
 		};
 		return this._encryptedOutputs.push(c), !0;
 	}
 };
 //#endregion
-export { s as AbstractSolanaCommitmentsSnapshotService };
+export { c as AbstractSolanaCommitmentsSnapshotService };

package/common/src/data-structures/merkle-tree/MerkleTree.cjs

@@ -1 +1 @@
-const e=require(`../../constants/protocol.constants.cjs`),t=require(`../../error-handling/logError.cjs`),n=require(`./MerkleTreeIncompleteError.cjs`);var r=class r{tree;count;index;reverseTree;static createWithData(e,t,n,i,a,o,s){return new r(a,o,s,e,t,n,i)}static create(t,n,i=e.MERKLE_LEVELS){return new r(t,i,n)}createReverseTree(){let e=new Map;for(let t=this.getStartIndex();t<this.getStartIndex()+this.count;t+=1n){let n=this.tree.get(t);n!==void 0&&e.set(n,t)}return e}constructor(e,t,n,r,i,a,o){this.hashFunction=e,this.levels=t,this.defaultNodeValue=n,this.tree=r??new Map,this.count=o??0n,this.index=a??2n**(t-1n),i?this.reverseTree=i:this.reverseTree=this.createReverseTree()}getStartIndex(){return 2n**(this.levels-1n)}logarithm2(e){let t=0n;for(;2n**t<e;)t+=1n;return t}bigIntMax(e,t){return e>t?e:t}forceInsert(e,t){if(t<this.getStartIndex())throw RangeError();this.tree.set(t,e),this.reverseTree.set(e,t);let n=this.index-this.getStartIndex(),r=this.logarithm2(n),i=t;for(let e=1n;e<=r;e+=1n){i/=2n;let e=this.hashFunction(this.tree.get(i*2n)||this.defaultNodeValue,this.tree.get(i*2n+1n)||this.defaultNodeValue);this.tree.set(i,e)}}insert(e,n){let r=!1;this.tree.has(n)&&(r=!0,t.logError(`Editing commitment index: ${n} `)),r||(this.count+=1n),this.index=this.bigIntMax(this.index,n+1n),this.forceInsert(e,n)}remove(e){this.forceInsert(this.defaultNodeValue,e)}completenessCheck(){if(this.count!==this.index-this.getStartIndex())throw new n.MerkleTreeIncompleteError}getRootHash(){this.completenessCheck();for(let e=1n;e<2n**this.levels;e*=2n)if(this.tree.get(e))return this.tree.get(e);return this.defaultNodeValue}getMerkleData(){return this.completenessCheck(),new Map(this.tree)}getValue(e){return this.tree.get(e)}getSiblingIndex(e){return e===1n?1n:e%2n==1n?e-1n:e+1n}getSiblingHashesForVerification(e){this.completenessCheck();let t=this.reverseTree.get(e);if(t===void 0)return Array(25).fill(0n);let n=[];for(;t!==0n;)n.push(this.tree.get(this.getSiblingIndex(t))||this.defaultNodeValue),t/=2n;return n.slice(0,25)}getSiblingSides(e){this.completenessCheck();let t=this.reverseTree.get(e);if(t===void 0)return Array(25).fill(0n);let n=[];for(;t!==0n;){let e=t%2n==0n?0n:1n;n.push(e),t/=2n}return n.slice(0,25)}toJSON(){let{tree:e,reverseTree:t,count:n,index:r}=this;return{tree:Object.fromEntries(e),reverseTree:Object.fromEntries(t),count:n.toString(),index:r.toString()}}toJSONLite(){let{tree:e,count:t,index:n}=this;return{tree:Object.fromEntries(e),count:t.toString(),index:n.toString()}}clone(){return r.createWithData(new Map(this.tree),new Map(this.reverseTree),this.index,this.count,this.hashFunction,this.levels,this.defaultNodeValue)}getIndex(){return this.index}};exports.MerkleTree=r;
+const e=require(`../../constants/protocol.constants.cjs`),t=require(`../../error-handling/logError.cjs`),n=require(`./MerkleTreeIncompleteError.cjs`);var r=class r{tree;count;index;reverseTree;static createWithData(e,t,n,i,a,o,s){return new r(a,o,s,e,t,n,i)}static create(t,n,i=e.MERKLE_LEVELS){return new r(t,i,n)}createReverseTree(){let e=new Map;for(let t=this.getStartIndex();t<this.getStartIndex()+this.count;t+=1n){let n=this.tree.get(t);n!==void 0&&e.set(n,t)}return e}constructor(e,t,n,r,i,a,o){this.hashFunction=e,this.levels=t,this.defaultNodeValue=n,this.tree=r??new Map,this.count=o??0n,this.index=a??2n**(t-1n),i?this.reverseTree=i:this.reverseTree=this.createReverseTree()}getStartIndex(){return 2n**(this.levels-1n)}logarithm2(e){let t=0n;for(;2n**t<e;)t+=1n;return t}bigIntMax(e,t){return e>t?e:t}forceInsert(e,t){if(t<this.getStartIndex())throw RangeError();this.tree.set(t,e),this.reverseTree.set(e,t);let n=this.index-this.getStartIndex(),r=this.logarithm2(n),i=t;for(let e=1n;e<=r;e+=1n){i/=2n;let e=this.hashFunction(this.tree.get(i*2n)||this.defaultNodeValue,this.tree.get(i*2n+1n)||this.defaultNodeValue);this.tree.set(i,e)}}insert(e,n){let r=!1;this.tree.has(n)&&(r=!0,t.logError(`Editing commitment index: ${n} `)),r||(this.count+=1n),this.index=this.bigIntMax(this.index,n+1n),this.forceInsert(e,n)}remove(e){this.forceInsert(this.defaultNodeValue,e)}completenessCheck(){if(this.count!==this.index-this.getStartIndex())throw new n.MerkleTreeIncompleteError}getRootHash(){this.completenessCheck();for(let e=1n;e<2n**this.levels;e*=2n)if(this.tree.get(e))return this.tree.get(e);return this.defaultNodeValue}getMerkleData(){return this.completenessCheck(),new Map(this.tree)}getValue(e){return this.tree.get(e)}contains(e){return this.reverseTree.has(e)}getSiblingIndex(e){return e===1n?1n:e%2n==1n?e-1n:e+1n}getSiblingHashesForVerification(e){this.completenessCheck();let t=this.reverseTree.get(e);if(t===void 0)return Array(25).fill(0n);let n=[];for(;t!==0n;)n.push(this.tree.get(this.getSiblingIndex(t))??this.defaultNodeValue),t/=2n;return n.slice(0,25)}getSiblingSides(e){this.completenessCheck();let t=this.reverseTree.get(e);if(t===void 0)return Array(25).fill(0n);let n=[];for(;t!==0n;){let e=t%2n==0n?0n:1n;n.push(e),t/=2n}return n.slice(0,25)}toJSON(){let{tree:e,reverseTree:t,count:n,index:r}=this;return{tree:Object.fromEntries(e),reverseTree:Object.fromEntries(t),count:n.toString(),index:r.toString()}}toJSONLite(){let{tree:e,count:t,index:n}=this;return{tree:Object.fromEntries(e),count:t.toString(),index:n.toString()}}clone(){return r.createWithData(new Map(this.tree),new Map(this.reverseTree),this.index,this.count,this.hashFunction,this.levels,this.defaultNodeValue)}getIndex(){return this.index}getInsertionSiblingData(e){let t=[],n=[],r=e;for(;r!==0n;)t.push(this.tree.get(this.getSiblingIndex(r))??this.defaultNodeValue),n.push(r%2n==0n?0n:1n),r/=2n;return{siblings:t.slice(0,25),sides:n.slice(0,25)}}};exports.MerkleTree=r;

package/common/src/data-structures/merkle-tree/MerkleTree.d.ts

@@ -63,6 +63,7 @@ export declare class MerkleTree<T> {
     getRootHash(): T | undefined;
     getMerkleData(): Map<bigint, T>;
     getValue(index: bigint): T | undefined;
+    contains(value: T): boolean;
     getSiblingIndex(index: bigint): bigint;
     /**
      * get sibling hashes needed by main.circom
@@ -79,4 +80,14 @@ export declare class MerkleTree<T> {
     toJSONLite(): MerkleTreeJson;
     clone(): MerkleTree<T>;
     getIndex(): bigint;
+    /**
+     * Compute sibling hashes and sides needed to prove insertion at a given leaf index.
+     * Used by the Tron circuit to compute newRootHash off-chain.
+     * The caller is responsible for passing the correct nodeIndex (e.g. tree.getIndex()
+     * for the first insertion, or tree.getIndex() + 1n after simulating the first).
+     */
+    getInsertionSiblingData(nodeIndex: bigint): {
+        siblings: T[];
+        sides: bigint[];
+    };
 }

package/common/src/data-structures/merkle-tree/MerkleTree.mjs

@@ -66,6 +66,9 @@ var r = class r {
 	getValue(e) {
 		return this.tree.get(e);
 	}
+	contains(e) {
+		return this.reverseTree.has(e);
+	}
 	getSiblingIndex(e) {
 		return e === 1n ? 1n : e % 2n == 1n ? e - 1n : e + 1n;
 	}
@@ -74,7 +77,7 @@ var r = class r {
 		let t = this.reverseTree.get(e);
 		if (t === void 0) return Array(25).fill(0n);
 		let n = [];
-		for (; t !== 0n;) n.push(this.tree.get(this.getSiblingIndex(t)) || this.defaultNodeValue), t /= 2n;
+		for (; t !== 0n;) n.push(this.tree.get(this.getSiblingIndex(t)) ?? this.defaultNodeValue), t /= 2n;
 		return n.slice(0, 25);
 	}
 	getSiblingSides(e) {
@@ -111,6 +114,14 @@ var r = class r {
 	getIndex() {
 		return this.index;
 	}
+	getInsertionSiblingData(e) {
+		let t = [], n = [], r = e;
+		for (; r !== 0n;) t.push(this.tree.get(this.getSiblingIndex(r)) ?? this.defaultNodeValue), n.push(r % 2n == 0n ? 0n : 1n), r /= 2n;
+		return {
+			siblings: t.slice(0, 25),
+			sides: n.slice(0, 25)
+		};
+	}
 };
 //#endregion
 export { r as MerkleTree };

package/common/src/data-structures/private-balances/HinkalPrivateBalanceController.cjs

@@ -1 +1 @@
-const e=require(`../../functions/web3/etherFunctions.cjs`),t=require(`../../functions/utils/upToDateState.cjs`),n=require(`../../types/balances.types.cjs`),r=require(`../../error-handling/logger.cjs`),i=require(`../../functions/utils/string.utils.cjs`),a=require(`./BalanceChangedHandler.cjs`),o=require(`../IndexedDB/private-balances-db.cjs`),s=require(`../IndexedDB/prices-db.cjs`),c=require(`./privateBalancesReducer.cjs`);var l=class{privateBalancesState={};privateBalancesListeners=new Set;hinkal;unsubscribeBalanceChange;constructor(e){this.hinkal=e,this.unsubscribeBalanceChange=a.balanceChangedHandler.subscribe(e=>{this.applyRefreshFromBalanceChangeEvent(e).catch(e=>{r.Logger.error(`Private balance refresh from balance change failed:`,e)})})}getSnapshot(){return this.privateBalancesState}onPrivateBalancesWithUSDChange(e){return this.privateBalancesListeners.add(e),()=>this.privateBalancesListeners.delete(e)}dispose(){this.unsubscribeBalanceChange(),this.privateBalancesListeners.clear()}async applyRefreshFromBalanceChangeEvent(e){if(!e){await this.refreshInternal(n.PrivateBalanceUpdateType.FreshWithoutLoader);return}let{updateType:t,balanceChangeState:r,chainIdToUpdate:i,resetCacheBefore:a,updateTokensListBefore:o}=e;await this.refreshInternal(t,r,a,o,i)}dispatch(e){this.privateBalancesState=c.privateBalancesReducer(this.privateBalancesState,e);let t=this.privateBalancesState;this.privateBalancesListeners.forEach(e=>e(t))}async getAddressAndChain(e){try{return{ethAdd:e?.ethereumAddress??await this.hinkal.getEthereumAddress(),userKeys:e?.userKeys??this.hinkal.userKeys}}catch{return{ethAdd:void 0,userKeys:void 0}}}async hasCache(e,t){return await o.privateBalancesStorage.getBalancesByWallet(e,t)!==void 0}async getBalancesWithUsdValues(t,n){try{if(!n)return[];let r=t.map(e=>e.token.erc20TokenAddress);if(!r.length)return[];let i=await s.pricesStorage.getAndUpdatePrices(n,r),a=new Map(r.map((e,t)=>[e,i[t]]));return[...t.map(t=>{try{let n=t.token.erc20TokenAddress,r=a.get(n)??0,i=t.balance,o=Number(e.getAmountInToken(t.token,i))*r;return{...t,usdValue:o}}catch{return{...t,usdValue:0}}})]}catch{return r.Logger.log(`Error fetching private token balances`),[]}}handleCached=async(n,a)=>{try{await t.checkAddressNotUpdated(this.hinkal,n),this.dispatch({type:`SET_BALANCES`,chainId:a,balances:void 0});let r=await o.privateBalancesStorage.getBalancesByWallet(n,a),c=await s.pricesStorage.getPrices();if(await t.checkAddressNotUpdated(this.hinkal,n),!r||(this.dispatch({type:`SET_BALANCES`,chainId:a,balances:r}),!c))return;let l=r.map(t=>{let n=i.getTokenStateKey(t.token.erc20TokenAddress,a);return{...t,usdValue:c[n]?Number(e.getAmountInToken(t.token,t.balance))*c[n]:0}});this.dispatch({type:`SET_USD_BALANCES`,chainId:a,balances:l})}catch(e){r.Logger.error(`Error refreshing cached private balance:`,e)}};handleFresh=async(e,n,i,a,s,c)=>{let l=s;try{await t.checkAddressNotUpdated(this.hinkal,e);let r=await this.hasCache(e,n);await t.checkAddressNotUpdated(this.hinkal,e),l=s||!r,l&&this.dispatch({type:`SET_LOADING`,chainId:n,newValue:!0}),await t.checkAddressNotUpdated(this.hinkal,e),a&&await this.hinkal.resetMerkle([n]);let u=await this.hinkal.getTotalBalance(n,i,e,a,c);await t.checkAddressNotUpdated(this.hinkal,e);let d=u.filter(e=>e.balance>0n);this.dispatch({type:`SET_BALANCES`,chainId:n,balances:d}),await o.privateBalancesStorage.setBalances(e,n,d),await t.checkAddressNotUpdated(this.hinkal,e),l&&(this.dispatch({type:`SET_USD_LOADING`,chainId:n,newValue:!0}),this.dispatch({type:`SET_LOADING`,chainId:n,newValue:!1})),await t.checkAddressNotUpdated(this.hinkal,e);let f=await this.getBalancesWithUsdValues(d,n);await t.checkAddressNotUpdated(this.hinkal,e),f.length>0&&this.dispatch({type:`SET_USD_BALANCES`,chainId:n,balances:f})}catch(e){r.Logger.error(`Error refreshing private balance:`,e)}finally{l&&(this.dispatch({type:`SET_LOADING`,chainId:n,newValue:!1}),this.dispatch({type:`SET_USD_LOADING`,chainId:n,newValue:!1}))}};getSupportedChainsToUpdate(e){let t=this.hinkal.getSupportedChains();return e?t.includes(e)?[e]:[]:t}async refreshInternal(e=n.PrivateBalanceUpdateType.FreshWithoutLoader,t,r=!1,i=!1,a){let{ethAdd:o,userKeys:s}=await this.getAddressAndChain(t);!o||!s?.getShieldedPublicKey()||await Promise.all(this.getSupportedChainsToUpdate(a).map(async t=>{switch(e){case n.PrivateBalanceUpdateType.OnlyCache:await this.handleCached(o,t);break;case n.PrivateBalanceUpdateType.Fresh:await this.handleFresh(o,t,s,r,!0,i);break;case n.PrivateBalanceUpdateType.FreshWithoutLoader:await this.handleFresh(o,t,s,r,!1,i);break;default:break}}))}};exports.HinkalPrivateBalanceController=l;
+const e=require(`../../functions/web3/etherFunctions.cjs`),t=require(`../../functions/utils/upToDateState.cjs`),n=require(`../../error-handling/logger.cjs`),r=require(`../../types/balances.types.cjs`),i=require(`../../functions/utils/string.utils.cjs`),a=require(`../../constants/balance.constants.cjs`),o=require(`./BalanceChangedHandler.cjs`),s=require(`../IndexedDB/private-balances-db.cjs`),c=require(`../IndexedDB/prices-db.cjs`),l=require(`./privateBalancesReducer.cjs`);var u=class{privateBalancesState={};privateBalancesListeners=new Set;hinkal;unsubscribeBalanceChange;pollIntervalId;pollInFlight=!1;constructor(e){this.hinkal=e,this.unsubscribeBalanceChange=o.balanceChangedHandler.subscribe(e=>{this.applyRefreshFromBalanceChangeEvent(e).catch(e=>{n.Logger.error(`Private balance refresh from balance change failed:`,e)})}),this.pollIntervalId=setInterval(()=>this.pollRemoteBalances(),a.REMOTE_BALANCE_POLL_INTERVAL_MS)}getSnapshot(){return this.privateBalancesState}onPrivateBalancesWithUSDChange(e){return this.privateBalancesListeners.add(e),()=>this.privateBalancesListeners.delete(e)}dispose(){clearInterval(this.pollIntervalId),this.unsubscribeBalanceChange(),this.privateBalancesListeners.clear()}pollRemoteBalances(){this.privateBalancesListeners.size!==0&&(!this.hinkal.generateProofRemotely||this.pollInFlight||(this.pollInFlight=!0,this.refreshInternal(r.PrivateBalanceUpdateType.FreshWithoutLoader,void 0,!1,!0).catch(e=>{n.Logger.error(`Remote private balance poll failed:`,e)}).finally(()=>{this.pollInFlight=!1})))}async applyRefreshFromBalanceChangeEvent(e){if(!e){await this.refreshInternal(r.PrivateBalanceUpdateType.FreshWithoutLoader);return}let{updateType:t,balanceChangeState:n,chainIdToUpdate:i,resetCacheBefore:a,updateTokensListBefore:o}=e;await this.refreshInternal(t,n,a,o,i)}dispatch(e){this.privateBalancesState=l.privateBalancesReducer(this.privateBalancesState,e);let t=this.privateBalancesState;this.privateBalancesListeners.forEach(e=>e(t))}async getAddressAndChain(e){try{return{ethAdd:e?.ethereumAddress??await this.hinkal.getEthereumAddress(),userKeys:e?.userKeys??this.hinkal.userKeys}}catch{return{ethAdd:void 0,userKeys:void 0}}}async hasCache(e,t){return await s.privateBalancesStorage.getBalancesByWallet(e,t)!==void 0}async getBalancesWithUsdValues(t,r){try{if(!r)return[];let n=t.map(e=>e.token.erc20TokenAddress);if(!n.length)return[];let i=await c.pricesStorage.getAndUpdatePrices(r,n),a=new Map(n.map((e,t)=>[e,i[t]]));return[...t.map(t=>{try{let n=t.token.erc20TokenAddress,r=a.get(n)??0,i=t.balance,o=Number(e.getAmountInToken(t.token,i))*r;return{...t,usdValue:o}}catch{return{...t,usdValue:0}}})]}catch{return n.Logger.log(`Error fetching private token balances`),[]}}handleCached=async(r,a)=>{try{await t.checkAddressNotUpdated(this.hinkal,r),this.dispatch({type:`SET_BALANCES`,chainId:a,balances:void 0});let n=await s.privateBalancesStorage.getBalancesByWallet(r,a),o=await c.pricesStorage.getPrices();if(await t.checkAddressNotUpdated(this.hinkal,r),!n||(this.dispatch({type:`SET_BALANCES`,chainId:a,balances:n}),!o))return;let l=n.map(t=>{let n=i.getTokenStateKey(t.token.erc20TokenAddress,a);return{...t,usdValue:o[n]?Number(e.getAmountInToken(t.token,t.balance))*o[n]:0}});this.dispatch({type:`SET_USD_BALANCES`,chainId:a,balances:l})}catch(e){n.Logger.error(`Error refreshing cached private balance:`,e)}};applyFreshBalances=async(e,n,r,i)=>{let a=r.filter(e=>e.balance>0n);this.dispatch({type:`SET_BALANCES`,chainId:n,balances:a}),await s.privateBalancesStorage.setBalances(e,n,a),await t.checkAddressNotUpdated(this.hinkal,e),i&&(this.dispatch({type:`SET_USD_LOADING`,chainId:n,newValue:!0}),this.dispatch({type:`SET_LOADING`,chainId:n,newValue:!1})),await t.checkAddressNotUpdated(this.hinkal,e);let o=await this.getBalancesWithUsdValues(a,n);await t.checkAddressNotUpdated(this.hinkal,e),o.length>0&&this.dispatch({type:`SET_USD_BALANCES`,chainId:n,balances:o})};handleFresh=async(e,r,i,a,o,s)=>{let c=new Set;try{await t.checkAddressNotUpdated(this.hinkal,e),await Promise.all(r.map(async t=>{(o||!await this.hasCache(e,t))&&c.add(t)})),await t.checkAddressNotUpdated(this.hinkal,e),c.forEach(e=>this.dispatch({type:`SET_LOADING`,chainId:e,newValue:!0})),a&&await this.hinkal.resetMerkle(r);let n=await this.hinkal.getTotalBalances(r,i,e,a,s);await t.checkAddressNotUpdated(this.hinkal,e),await Promise.all([...n].map(([t,n])=>this.applyFreshBalances(e,t,n,c.has(t))))}catch(e){n.Logger.error(`Error refreshing private balance:`,e)}finally{c.forEach(e=>{this.dispatch({type:`SET_LOADING`,chainId:e,newValue:!1}),this.dispatch({type:`SET_USD_LOADING`,chainId:e,newValue:!1})})}};getSupportedChainsToUpdate(e){let t=this.hinkal.getSupportedChains();return e?t.includes(e)?[e]:[]:t}async refreshInternal(e=r.PrivateBalanceUpdateType.FreshWithoutLoader,t,n=!1,i=!1,a){let{ethAdd:o,userKeys:s}=await this.getAddressAndChain(t);if(!o||!s?.getShieldedPublicKey())return;let c=this.getSupportedChainsToUpdate(a);if(c.length!==0)switch(e){case r.PrivateBalanceUpdateType.OnlyCache:await Promise.all(c.map(e=>this.handleCached(o,e)));break;case r.PrivateBalanceUpdateType.Fresh:await this.handleFresh(o,c,s,n,!0,i);break;case r.PrivateBalanceUpdateType.FreshWithoutLoader:await this.handleFresh(o,c,s,n,!1,i);break;default:break}}};exports.HinkalPrivateBalanceController=u;

package/common/src/data-structures/private-balances/HinkalPrivateBalanceController.d.ts

@@ -5,16 +5,20 @@ export declare class HinkalPrivateBalanceController {
     private readonly privateBalancesListeners;
     private readonly hinkal;
     private readonly unsubscribeBalanceChange;
+    private readonly pollIntervalId;
+    private pollInFlight;
     constructor(hinkal: IHinkal);
     getSnapshot(): PrivateBalancesState;
     onPrivateBalancesWithUSDChange(callback: (state: PrivateBalancesState) => void): () => void;
     dispose(): void;
+    private pollRemoteBalances;
     private applyRefreshFromBalanceChangeEvent;
     private dispatch;
     private getAddressAndChain;
     private hasCache;
     private getBalancesWithUsdValues;
     private handleCached;
+    private applyFreshBalances;
     private handleFresh;
     private getSupportedChainsToUpdate;
     private refreshInternal;