@centrifuge/sdk 0.0.0-alpha.52 → 0.0.0-alpha.54

package/dist/entities/MerkleProofManager.js

@@ -1,5 +1,7 @@
+import { map } from 'rxjs';
 import { encodeFunctionData, encodePacked, keccak256, parseAbiItem, toFunctionSelector, toHex } from 'viem';
 import { ABI } from '../abi/index.js';
+import { NULL_ADDRESS } from '../constants.js';
 import { MessageType } from '../types/transaction.js';
 import { BigIntWrapper } from '../utils/BigInt.js';
 import { addressToBytes32 } from '../utils/index.js';
@@ -24,61 +26,119 @@ export class MerkleProofManager extends Entity {
         this.pool = network.pool;
         this.address = address.toLowerCase();
     }
-    async getPolicyRoot() {
-        const { SimpleMerkleTree } = await import('@openzeppelin/merkle-tree');
-        const poolId = 281474976710657n;
-        const scId = '0x31000000000000000000000000000000';
-        const erc20 = '0xa0Cb889707d426A7A386870A03bc70d1b0697598';
-        const manager = '0xA4AD4f68d0b91CFD19687c881e50f3A00242828c';
-        const leaves = [
-            {
-                decoder: '0x03A6a84cD762D9707A21605b548aaaB891562aAb',
-                target: '0x1B237b1866D34C8619D52F9A5047a4Ab976e0426',
-                valueNonZero: false,
-                selector: toFunctionSelector('function withdraw(uint64,bytes16,address,uint256,address,uint128)'),
-                addresses: encodePacked(['uint64', 'bytes16', 'address', 'address'], [poolId, scId, erc20, manager]),
-            },
-            {
-                decoder: '0x03A6a84cD762D9707A21605b548aaaB891562aAb',
-                target: '0x1B237b1866D34C8619D52F9A5047a4Ab976e0426',
-                valueNonZero: false,
-                selector: toFunctionSelector('function deposit(uint64,bytes16,address,uint256,uint128)'),
-                addresses: encodePacked(['uint64', 'bytes16', 'address'], [poolId, scId, erc20]),
-            },
-        ];
-        console.log('withdrawsel', toFunctionSelector('function withdraw(uint64,bytes16,address,uint256,address,uint128)'));
-        console.log('addr', encodePacked(['uint64', 'bytes16', 'address', 'address'], [poolId, scId, erc20, manager]));
-        console.log('0x000100000000000131000000000000000000000000000000a0cb889707d426a7a386870a03bc70d1b0697598a4ad4f68d0b91cfd19687c881e50f3a00242828c' ===
-            encodePacked(['uint64', 'bytes16', 'address', 'address'], [poolId, scId, erc20, manager]));
-        console.log('depositsel', toFunctionSelector('function deposit(uint64,bytes16,address,uint256,uint128)'));
-        console.log('addr', encodePacked(['uint64', 'bytes16', 'address'], [poolId, scId, erc20]));
-        const leavesHashes = leaves.map((leaf) => keccak256(encodePacked(['address', 'address', 'bool', 'bytes4', 'bytes'], [leaf.decoder, leaf.target, leaf.valueNonZero, leaf.selector, leaf.addresses])));
-        const tree = SimpleMerkleTree.of(leavesHashes);
-        console.log('tree', tree, tree.root);
+    policies(strategist) {
+        return this._query(null, () => this.pool
+            .metadata()
+            .pipe(map((metadata) => metadata?.merkleProofManager?.[this.chainId]?.[strategist.toLowerCase()] ?? [])));
     }
-    setPolicies(strategist, policies, actions = []) {
+    setPolicies(strategist, policyInputs) {
         const self = this;
         return this._transact(async function* (ctx) {
-            const [{ hub }, id, poolDetails] = await Promise.all([
+            const [{ hub }, id, poolDetails, { SimpleMerkleTree: SimpleMerkleTreeConstructor }] = await Promise.all([
                 self._root._protocolAddresses(self.pool.chainId),
                 self._root.id(self.chainId),
                 self.pool.details(),
+                import('@openzeppelin/merkle-tree'),
             ]);
-            const rootHash = toHex(0, { size: 32 });
+            const { metadata, shareClasses } = poolDetails;
+            const client = self._root.getClient(self.chainId);
+            // Get the encoded args from chain by calling the decoder contract.
+            async function getEncodedArgs(policy) {
+                const abi = parseAbiItem(policy.abi);
+                const args = policy.args.map((arg, i) => {
+                    if (arg !== null)
+                        return arg;
+                    const { type } = abi.inputs[i] || {};
+                    if (!type) {
+                        throw new Error(`No type found for argument ${i} in abi item "${policy.abi}, args: ${policy.args}"`);
+                    }
+                    if (type === 'address') {
+                        return NULL_ADDRESS;
+                    }
+                    else if (type.includes('int')) {
+                        return 0;
+                    }
+                    else if (type === 'bytes') {
+                        return '0x';
+                    }
+                    else if (type === 'bool') {
+                        return false;
+                    }
+                    else if (type === 'string') {
+                        return '';
+                    }
+                    else if (type.startsWith('bytes')) {
+                        return toHex(0, { size: parseInt(type.slice(5)) });
+                    }
+                    else if (type.includes('[')) {
+                        return [];
+                    }
+                    throw new Error(`Unsupported type "${type}" for abi item "${policy.abi}"`);
+                });
+                abi.outputs = [
+                    {
+                        name: 'addressesFound',
+                        type: 'bytes',
+                        internalType: 'bytes',
+                    },
+                ];
+                const encoded = await client.readContract({
+                    address: policy.decoder,
+                    abi: [abi],
+                    functionName: abi.name,
+                    args,
+                });
+                return encoded;
+            }
+            const policies = await Promise.all(policyInputs.map(async (input) => {
+                const argsEncoded = await getEncodedArgs(input);
+                return {
+                    ...input,
+                    assetId: input.assetId?.toString(),
+                    valueNonZero: input.valueNonZero ?? false,
+                    argsEncoded,
+                };
+            }));
+            let rootHash;
+            if (policies.length === 0) {
+                rootHash = toHex(0, { size: 32 });
+            }
+            else {
+                const tree = getMerkleTree(SimpleMerkleTreeConstructor, policies);
+                rootHash = tree.root;
+            }
+            const newMetadata = {
+                ...metadata,
+                merkleProofManager: {
+                    ...metadata?.merkleProofManager,
+                    [self.chainId]: {
+                        ...metadata?.merkleProofManager?.[self.chainId],
+                        [strategist.toLowerCase()]: { policies: policies },
+                    },
+                },
+            };
+            const cid = await self._root.config.pinJson(newMetadata);
             yield* wrapTransaction('Set policies', ctx, {
                 contract: hub,
-                data: encodeFunctionData({
-                    abi: ABI.Hub,
-                    functionName: 'updateContract',
-                    args: [
-                        self.pool.id.raw,
-                        toHex(0, { size: 16 }),
-                        id,
-                        addressToBytes32(self.address),
-                        encodePacked(['uint8', 'bytes32', 'bytes32'], [/* UpdateContractType.Policy */ 4, addressToBytes32(strategist), rootHash]),
-                        0n,
-                    ],
-                }),
+                data: [
+                    encodeFunctionData({
+                        abi: ABI.Hub,
+                        functionName: 'setPoolMetadata',
+                        args: [self.pool.id.raw, toHex(cid)],
+                    }),
+                    encodeFunctionData({
+                        abi: ABI.Hub,
+                        functionName: 'updateContract',
+                        args: [
+                            self.pool.id.raw,
+                            shareClasses[0].shareClass.id.raw,
+                            id,
+                            addressToBytes32(self.address),
+                            encodePacked(['uint8', 'bytes32', 'bytes32'], [/* UpdateContractType.Policy */ 4, addressToBytes32(strategist), rootHash]),
+                            0n,
+                        ],
+                    }),
+                ],
                 messages: { [id]: [MessageType.UpdateContract] },
             });
         }, this.pool.chainId);
@@ -93,48 +153,58 @@ export class MerkleProofManager extends Entity {
     execute(calls) {
         const self = this;
         return this._transact(async function* (ctx) {
-            const [{ hub }, id, poolDetails] = await Promise.all([
-                self._root._protocolAddresses(self.pool.chainId),
-                self._root.id(self.chainId),
-                self.pool.details(),
+            const [metadata, { SimpleMerkleTree: SimpleMerkleTreeConstructor }] = await Promise.all([
+                self.pool.metadata(),
+                import('@openzeppelin/merkle-tree'),
             ]);
-            const addressesEncoded = await Promise.all(calls.map(async (call) => {
-                const { policy, values } = call;
+            const policiesForStrategist = metadata?.merkleProofManager?.[self.chainId]?.[ctx.signingAddress.toLowerCase()]?.policies;
+            if (!policiesForStrategist) {
+                throw new Error(`No policies found for strategist "${ctx.signingAddress}" on chain "${self.chainId}"`);
+            }
+            const tree = getMerkleTree(SimpleMerkleTreeConstructor, policiesForStrategist);
+            const formattedCalls = calls.map((call) => {
+                const { policy, inputs, value } = call;
                 const abi = parseAbiItem(policy.abi);
-                const args = [...policy.addresses];
-                values.forEach((value, i) => {
-                    const argIndex = policy.inputs[i];
+                const args = [...policy.args];
+                const inputIndeces = args.map((arg, i) => (arg === null ? i : -1)).filter((i) => i >= 0);
+                inputs.forEach((value, i) => {
+                    const argIndex = inputIndeces[i];
                     args[argIndex] = value instanceof BigIntWrapper ? value.toBigInt() : value;
                 });
-                const dataObj = {
+                const proof = getMerkleProof(tree, policy);
+                const targetData = encodeFunctionData({
                     abi: [abi],
                     functionName: abi.name,
                     args,
-                };
-                const data = encodeFunctionData(dataObj);
-                const result = await ctx.publicClient.readContract({
-                    address: policy.decoder,
-                    data,
                 });
-                const encoded = encodePacked(['address', 'address', 'bool', 'bytes4', 'bytes'], [
-                    policy.decoder,
-                    policy.target,
-                    policy.valueNonZero,
-                    toFunctionSelector(policy.abi),
-                    encodePacked(policy.inputs.map((input, i) => {
-                        if (input.type === 'address') {
-                            return values[i];
-                        }
-                        else if (input.type === 'uint') {
-                            return BigInt(values[i]);
-                        }
-                        throw new Error(`Unsupported input type: ${input.type}`);
-                    })),
-                ]);
-                return encoded;
-            }));
-            const leavesHashes = calls.map((leaf) => keccak256(encodePacked(['address', 'address', 'bool', 'bytes4', 'bytes'], [leaf.decoder, leaf.target, leaf.valueNonZero, leaf.selector, leaf.addresses])));
+                return {
+                    decoder: policy.decoder,
+                    target: policy.target,
+                    targetData,
+                    value: value ?? 0n,
+                    proof,
+                };
+            });
+            yield* wrapTransaction('Execute calls', ctx, {
+                contract: self.address,
+                data: encodeFunctionData({
+                    abi: ABI.MerkleProofManager,
+                    functionName: 'execute',
+                    args: [formattedCalls],
+                }),
+                value: calls.reduce((acc, call) => acc + (call.value ?? 0n), 0n),
+            });
         }, this.chainId);
     }
 }
+export function toHashedPolicyLeaf(policy) {
+    return keccak256(encodePacked(['address', 'address', 'bool', 'bytes4', 'bytes'], [policy.decoder, policy.target, policy.valueNonZero, toFunctionSelector(policy.abi), policy.argsEncoded]));
+}
+export function getMerkleTree(MerkleTreeConstructor, policies) {
+    const leaves = policies.map(toHashedPolicyLeaf);
+    return MerkleTreeConstructor.of(leaves);
+}
+export function getMerkleProof(tree, leaf) {
+    return tree.getProof(toHashedPolicyLeaf(leaf));
+}
 //# sourceMappingURL=MerkleProofManager.js.map

package/dist/entities/MerkleProofManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"MerkleProofManager.js","sourceRoot":"","sources":["../../src/entities/MerkleProofManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,kBAAkB,EAAE,KAAK,EAAE,MAAM,MAAM,CAAA;AAC3G,OAAO,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAA;AAIrC,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAA;AACrD,OAAO,EAAW,aAAa,EAAS,MAAM,oBAAoB,CAAA;AAClE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAA;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAA;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AAIpC;;GAEG;AACH,MAAM,OAAO,kBAAmB,SAAQ,MAAM;IAUnC;IATT,IAAI,CAAM;IACV,OAAO,CAAQ;IACf;;OAEG;IACH,OAAO,CAAW;IAClB,gBAAgB;IAChB,YACE,KAAiB,EACV,OAAoB,EAC3B,OAAkB;QAElB,KAAK,CAAC,KAAK,EAAE,CAAC,oBAAoB,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;QAH1E,YAAO,GAAP,OAAO,CAAa;QAI3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAA;QAC9B,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAA;QACxB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,WAAW,EAAe,CAAA;IACnD,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAA;QACtE,MAAM,MAAM,GAAG,gBAAgB,CAAA;QAC/B,MAAM,IAAI,GAAG,oCAAoC,CAAA;QACjD,MAAM,KAAK,GAAG,4CAA4C,CAAA;QAC1D,MAAM,OAAO,GAAG,4CAA4C,CAAA;QAC5D,MAAM,MAAM,GAAG;YACb;gBACE,OAAO,EAAE,4CAA4C;gBACrD,MAAM,EAAE,4CAA4C;gBACpD,YAAY,EAAE,KAAK;gBACnB,QAAQ,EAAE,kBAAkB,CAAC,mEAAmE,CAAC;gBACjG,SAAS,EAAE,YAAY,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;aACrG;YACD;gBACE,OAAO,EAAE,4CAA4C;gBACrD,MAAM,EAAE,4CAA4C;gBACpD,YAAY,EAAE,KAAK;gBACnB,QAAQ,EAAE,kBAAkB,CAAC,0DAA0D,CAAC;gBACxF,SAAS,EAAE,YAAY,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;aACjF;SACF,CAAA;QAED,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,kBAAkB,CAAC,mEAAmE,CAAC,CAAC,CAAA;QACnH,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,CAAA;QAC9G,OAAO,CAAC,GAAG,CACT,oIAAoI;YAClI,YAAY,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAC5F,CAAA;QACD,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,kBAAkB,CAAC,0DAA0D,CAAC,CAAC,CAAA;QACzG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAA;QAC1F,MAAM,YAAY,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACvC,SAAS,CACP,YAAY,CACV,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,EACjD,CAAC,IAAI,CAAC,OAAc,EAAE,IAAI,CAAC,MAAa,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,CAC5F,CACF,CACF,CAAA;QAED,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,CAAC,YAAY,CAAC,CAAA;QAC9C,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;IACtC,CAAC;IAED,WAAW,CAAC,UAAqB,EAAE,QAA6B,EAAE,UAA+B,EAAE;QACjG,MAAM,IAAI,GAAG,IAAI,CAAA;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,SAAS,CAAC,EAAE,GAAG;YACxC,MAAM,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACnD,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC;gBAChD,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;aACpB,CAAC,CAAA;YAEF,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAA;YAEvC,KAAK,CAAC,CAAC,eAAe,CAAC,cAAc,EAAE,GAAG,EAAE;gBAC1C,QAAQ,EAAE,GAAG;gBACb,IAAI,EAAE,kBAAkB,CAAC;oBACvB,GAAG,EAAE,GAAG,CAAC,GAAG;oBACZ,YAAY,EAAE,gBAAgB;oBAC9B,IAAI,EAAE;wBACJ,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG;wBAChB,KAAK,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;wBACtB,EAAE;wBACF,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC;wBAC9B,YAAY,CACV,CAAC,OAAO,EAAE,SAAS,EAAE,SAAS,CAAC,EAC/B,CAAC,+BAA+B,CAAC,CAAC,EAAE,gBAAgB,CAAC,UAAU,CAAC,EAAE,QAAQ,CAAC,CAC5E;wBACD,EAAE;qBACH;iBACF,CAAC;gBACF,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE;aACjD,CAAC,CAAA;QACJ,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;IACvB,CAAC;IAED;;;OAGG;IACH,iBAAiB,CAAC,UAAqB;QACrC,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAA;IACzC,CAAC;IAED,OAAO,CACL,KAGG;QAEH,MAAM,IAAI,GAAG,IAAI,CAAA;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,SAAS,CAAC,EAAE,GAAG;YACxC,MAAM,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACnD,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC;gBAChD,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;aACpB,CAAC,CAAA;YAEF,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,GAAG,CACxC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;gBACvB,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAA;gBAC/B,MAAM,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;gBACpC,MAAM,IAAI,GAAiC,CAAC,GAAG,MAAM,CAAC,SAAS,CAAQ,CAAA;gBACvE,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;oBAC1B,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAE,CAAA;oBAClC,IAAI,CAAC,QAAQ,CAAC,GAAG,KAAK,YAAY,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,KAAK,CAAA;gBAC5E,CAAC,CAAC,CAAA;gBACF,MAAM,OAAO,GAAG;oBACd,GAAG,EAAE,CAAC,GAAG,CAAC;oBACV,YAAY,EAAG,GAAW,CAAC,IAAI;oBAC/B,IAAI;iBACL,CAAA;gBACD,MAAM,IAAI,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAA;gBACxC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,YAAY,CAAC,YAAY,CAAC;oBACjD,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,IAAI;iBACL,CAAC,CAAA;gBACF,MAAM,OAAO,GAAG,YAAY,CAC1B,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,EACjD;oBACE,MAAM,CAAC,OAAO;oBACd,MAAM,CAAC,MAAM;oBACb,MAAM,CAAC,YAAY;oBACnB,kBAAkB,CAAC,MAAM,CAAC,GAAG,CAAC;oBAC9B,YAAY,CACV,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;wBAC7B,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;4BAC7B,OAAO,MAAM,CAAC,CAAC,CAAc,CAAA;wBAC/B,CAAC;6BAAM,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BACjC,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAoB,CAAC,CAAA;wBAC7C,CAAC;wBACD,MAAM,IAAI,KAAK,CAAC,2BAA2B,KAAK,CAAC,IAAI,EAAE,CAAC,CAAA;oBAC1D,CAAC,CAAC,CACH;iBACF,CACF,CAAA;gBACD,OAAO,OAAO,CAAA;YAChB,CAAC,CAAC,CACH,CAAA;YAED,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACtC,SAAS,CACP,YAAY,CACV,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,EACjD,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,QAAqB,EAAE,IAAI,CAAC,SAAS,CAAC,CAC3F,CACF,CACF,CAAA;QACH,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;IAClB,CAAC;CACF"}
+{"version":3,"file":"MerkleProofManager.js","sourceRoot":"","sources":["../../src/entities/MerkleProofManager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,GAAG,EAAE,MAAM,MAAM,CAAA;AAC1B,OAAO,EAAe,kBAAkB,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,kBAAkB,EAAE,KAAK,EAAE,MAAM,MAAM,CAAA;AACxH,OAAO,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAA;AAErC,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAA;AAG9C,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAA;AACrD,OAAO,EAAW,aAAa,EAAS,MAAM,oBAAoB,CAAA;AAClE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAA;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAA;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AAIpC;;GAEG;AACH,MAAM,OAAO,kBAAmB,SAAQ,MAAM;IAUnC;IATT,IAAI,CAAM;IACV,OAAO,CAAQ;IACf;;OAEG;IACH,OAAO,CAAW;IAClB,gBAAgB;IAChB,YACE,KAAiB,EACV,OAAoB,EAC3B,OAAkB;QAElB,KAAK,CAAC,KAAK,EAAE,CAAC,oBAAoB,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;QAH1E,YAAO,GAAP,OAAO,CAAa;QAI3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAA;QAC9B,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAA;QACxB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,WAAW,EAAe,CAAA;IACnD,CAAC;IAED,QAAQ,CAAC,UAAqB;QAC5B,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE,CAC5B,IAAI,CAAC,IAAI;aACN,QAAQ,EAAE;aACV,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,EAAE,kBAAkB,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAS,CAAC,IAAI,EAAE,CAAC,CAAC,CAClH,CAAA;IACH,CAAC;IAED,WAAW,CAAC,UAAqB,EAAE,YAAsC;QACvE,MAAM,IAAI,GAAG,IAAI,CAAA;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,SAAS,CAAC,EAAE,GAAG;YACxC,MAAM,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,gBAAgB,EAAE,2BAA2B,EAAE,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACtG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC;gBAChD,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;gBACnB,MAAM,CAAC,2BAA2B,CAAC;aACpC,CAAC,CAAA;YACF,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,WAAW,CAAA;YAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAEjD,mEAAmE;YACnE,KAAK,UAAU,cAAc,CAAC,MAA8B;gBAC1D,MAAM,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAgB,CAAA;gBACnD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;oBACtC,IAAI,GAAG,KAAK,IAAI;wBAAE,OAAO,GAAG,CAAA;oBAC5B,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;oBACpC,IAAI,CAAC,IAAI,EAAE,CAAC;wBACV,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,iBAAiB,MAAM,CAAC,GAAG,WAAW,MAAM,CAAC,IAAI,GAAG,CAAC,CAAA;oBACtG,CAAC;oBACD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;wBACvB,OAAO,YAAY,CAAA;oBACrB,CAAC;yBAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;wBAChC,OAAO,CAAC,CAAA;oBACV,CAAC;yBAAM,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;wBAC5B,OAAO,IAAI,CAAA;oBACb,CAAC;yBAAM,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;wBAC3B,OAAO,KAAK,CAAA;oBACd,CAAC;yBAAM,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;wBAC7B,OAAO,EAAE,CAAA;oBACX,CAAC;yBAAM,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;wBACpC,OAAO,KAAK,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;oBACpD,CAAC;yBAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;wBAC9B,OAAO,EAAE,CAAA;oBACX,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,qBAAqB,IAAI,mBAAmB,MAAM,CAAC,GAAG,GAAG,CAAC,CAAA;gBAC5E,CAAC,CAAC,CAAA;gBAEF,GAAG,CAAC,OAAO,GAAG;oBACZ;wBACE,IAAI,EAAE,gBAAgB;wBACtB,IAAI,EAAE,OAAO;wBACb,YAAY,EAAE,OAAO;qBACtB;iBACF,CAAA;gBAED,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC;oBACxC,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,GAAG,EAAE,CAAC,GAAG,CAAC;oBACV,YAAY,EAAE,GAAG,CAAC,IAAI;oBACtB,IAAI;iBACL,CAAC,CAAA;gBAEF,OAAO,OAA2B,CAAA;YACpC,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,CAChC,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;gBAC/B,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,CAAA;gBAC/C,OAAO;oBACL,GAAG,KAAK;oBACR,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,QAAQ,EAAE;oBAClC,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,KAAK;oBACzC,WAAW;iBACgB,CAAA;YAC/B,CAAC,CAAC,CACH,CAAA;YAED,IAAI,QAAQ,CAAA;YACZ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,QAAQ,GAAG,KAAK,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAA;YACnC,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,GAAG,aAAa,CAAC,2BAA2B,EAAE,QAAQ,CAAC,CAAA;gBAEjE,QAAQ,GAAG,IAAI,CAAC,IAAiB,CAAA;YACnC,CAAC;YAED,MAAM,WAAW,GAAG;gBAClB,GAAG,QAAQ;gBACX,kBAAkB,EAAE;oBAClB,GAAG,QAAQ,EAAE,kBAAkB;oBAC/B,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;wBACd,GAAG,QAAQ,EAAE,kBAAkB,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;wBAC/C,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,QAAsC,EAAE;qBACjF;iBACF;aACF,CAAA;YAED,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAA;YAExD,KAAK,CAAC,CAAC,eAAe,CAAC,cAAc,EAAE,GAAG,EAAE;gBAC1C,QAAQ,EAAE,GAAG;gBACb,IAAI,EAAE;oBACJ,kBAAkB,CAAC;wBACjB,GAAG,EAAE,GAAG,CAAC,GAAG;wBACZ,YAAY,EAAE,iBAAiB;wBAC/B,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;qBACrC,CAAC;oBACF,kBAAkB,CAAC;wBACjB,GAAG,EAAE,GAAG,CAAC,GAAG;wBACZ,YAAY,EAAE,gBAAgB;wBAC9B,IAAI,EAAE;4BACJ,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG;4BAChB,YAAY,CAAC,CAAC,CAAE,CAAC,UAAU,CAAC,EAAE,CAAC,GAAG;4BAClC,EAAE;4BACF,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC;4BAC9B,YAAY,CACV,CAAC,OAAO,EAAE,SAAS,EAAE,SAAS,CAAC,EAC/B,CAAC,+BAA+B,CAAC,CAAC,EAAE,gBAAgB,CAAC,UAAU,CAAC,EAAE,QAAQ,CAAC,CAC5E;4BACD,EAAE;yBACH;qBACF,CAAC;iBACH;gBACD,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE;aACjD,CAAC,CAAA;QACJ,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;IACvB,CAAC;IAED;;;OAGG;IACH,iBAAiB,CAAC,UAAqB;QACrC,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAA;IACzC,CAAC;IAED,OAAO,CACL,KAIG;QAEH,MAAM,IAAI,GAAG,IAAI,CAAA;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,SAAS,CAAC,EAAE,GAAG;YACxC,MAAM,CAAC,QAAQ,EAAE,EAAE,gBAAgB,EAAE,2BAA2B,EAAE,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACtF,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACpB,MAAM,CAAC,2BAA2B,CAAC;aACpC,CAAC,CAAA;YAEF,MAAM,qBAAqB,GACzB,QAAQ,EAAE,kBAAkB,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,WAAW,EAAS,CAAC,EAAE,QAAQ,CAAA;YAEnG,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC3B,MAAM,IAAI,KAAK,CAAC,qCAAqC,GAAG,CAAC,cAAc,eAAe,IAAI,CAAC,OAAO,GAAG,CAAC,CAAA;YACxG,CAAC;YAED,MAAM,IAAI,GAAG,aAAa,CAAC,2BAA2B,EAAE,qBAAqB,CAAC,CAAA;YAE9E,MAAM,cAAc,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBACxC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAA;gBAEtC,MAAM,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;gBACpC,MAAM,IAAI,GAAiC,CAAC,GAAG,MAAM,CAAC,IAAI,CAAQ,CAAA;gBAClE,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;gBACxF,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;oBAC1B,MAAM,QAAQ,GAAG,YAAY,CAAC,CAAC,CAAE,CAAA;oBACjC,IAAI,CAAC,QAAQ,CAAC,GAAG,KAAK,YAAY,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,KAAK,CAAA;gBAC5E,CAAC,CAAC,CAAA;gBAEF,MAAM,KAAK,GAAG,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;gBAE1C,MAAM,UAAU,GAAG,kBAAkB,CAAC;oBACpC,GAAG,EAAE,CAAC,GAAG,CAAC;oBACV,YAAY,EAAG,GAAW,CAAC,IAAI;oBAC/B,IAAI;iBACL,CAAC,CAAA;gBACF,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,UAAU;oBACV,KAAK,EAAE,KAAK,IAAI,EAAE;oBAClB,KAAK;iBACN,CAAA;YACH,CAAC,CAAC,CAAA;YACF,KAAK,CAAC,CAAC,eAAe,CAAC,eAAe,EAAE,GAAG,EAAE;gBAC3C,QAAQ,EAAE,IAAI,CAAC,OAAO;gBACtB,IAAI,EAAE,kBAAkB,CAAC;oBACvB,GAAG,EAAE,GAAG,CAAC,kBAAkB;oBAC3B,YAAY,EAAE,SAAS;oBACvB,IAAI,EAAE,CAAC,cAAc,CAAC;iBACvB,CAAC;gBACF,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;aACjE,CAAC,CAAA;QACJ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;IAClB,CAAC;CACF;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAyB;IAC1D,OAAO,SAAS,CACd,YAAY,CACV,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,EACjD,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE,kBAAkB,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC,CACzG,CACF,CAAA;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,qBAA8C,EAAE,QAA6B;IACzG,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAA;IAE/C,OAAO,qBAAqB,CAAC,EAAE,CAAC,MAAM,CAAC,CAAA;AACzC,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAsB,EAAE,IAAuB;IAC5E,OAAO,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAgB,CAAA;AAC/D,CAAC"}

package/dist/entities/MerkleProofManager.test.js

@@ -1,27 +1,241 @@
+import { SimpleMerkleTree } from '@openzeppelin/merkle-tree';
+import { expect } from 'chai';
+import { of } from 'rxjs';
+import sinon from 'sinon';
+import { encodePacked } from 'viem';
+import { ABI } from '../abi/index.js';
+import { Centrifuge } from '../Centrifuge.js';
 import { context } from '../tests/setup.js';
+import { randomAddress } from '../tests/utils.js';
+import { makeThenable } from '../utils/rx.js';
 import { AssetId, PoolId, ShareClassId } from '../utils/types.js';
-import { MerkleProofManager } from './MerkleProofManager.js';
+import { getMerkleTree, MerkleProofManager } from './MerkleProofManager.js';
 import { Pool } from './Pool.js';
 import { PoolNetwork } from './PoolNetwork.js';
 const chainId = 11155111;
 const centId = 1;
-// Pool with async vault, permissioned redeem
 const poolId = PoolId.from(centId, 1);
 const scId = ShareClassId.from(poolId, 1);
 const assetId = AssetId.from(centId, 1);
+const someErc20 = '0x3aaaa86458d576BafCB1B7eD290434F0696dA65c';
 const fundManager = '0x423420Ae467df6e90291fd0252c0A8a637C1e03f';
 const mpmAddress = '0x9E14250c4C53bdcA1437F7EDa25B0D9ca46CfFE2';
+const vaultDecoder = '0x8E5bE47D081F53033eb7C9DB3ad31BaF67F15585';
 describe('MerkleProofManager', () => {
+    let pool;
     let merkleProofManager;
+    let mockPolicies;
+    const strategist = randomAddress();
     before(async () => {
         const { centrifuge } = context;
-        const pool = new Pool(centrifuge, poolId.raw, chainId);
+        const { balanceSheet, vaultRouter } = await centrifuge._protocolAddresses(chainId);
+        pool = new Pool(centrifuge, poolId.raw, chainId);
         const poolNetwork = new PoolNetwork(centrifuge, pool, chainId);
         merkleProofManager = new MerkleProofManager(centrifuge, poolNetwork, mpmAddress);
+        const decoder = vaultDecoder;
+        const target = balanceSheet;
+        const randomUser = randomAddress();
+        mockPolicies = [
+            {
+                assetId: undefined,
+                decoder,
+                target: someErc20,
+                abi: 'function approve(address,uint256)',
+                valueNonZero: false,
+                args: [vaultRouter, null],
+                argsEncoded: encodePacked(['address'], [vaultRouter]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function cancelDepositRequest(uint256, address controller)',
+                valueNonZero: false,
+                args: [null, randomUser],
+                argsEncoded: encodePacked(['address'], [randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function cancelRedeemRequest(uint256, address controller)',
+                valueNonZero: false,
+                args: [null, randomUser],
+                argsEncoded: encodePacked(['address'], [randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function claimCancelDepositRequest(uint256, address receiver, address controller)',
+                valueNonZero: false,
+                args: [null, randomUser, randomUser],
+                argsEncoded: encodePacked(['address', 'address'], [randomUser, randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function claimCancelRedeemRequest(uint256, address receiver, address controller)',
+                valueNonZero: false,
+                args: [null, randomUser, randomUser],
+                argsEncoded: encodePacked(['address', 'address'], [randomUser, randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function deposit(uint64 poolId, bytes16 scId, address asset, uint256, uint128)',
+                valueNonZero: false,
+                args: [poolId.toString(), scId.raw, someErc20, null, null],
+                argsEncoded: encodePacked(['uint64', 'bytes16', 'address'], [poolId.raw, scId.raw, someErc20]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function deposit(uint256, address receiver)',
+                valueNonZero: false,
+                args: [null, randomUser],
+                argsEncoded: encodePacked(['address'], [randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function mint(uint256, address receiver)',
+                valueNonZero: false,
+                args: [null, randomUser],
+                argsEncoded: encodePacked(['address'], [randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function redeem(uint256, address receiver, address owner)',
+                valueNonZero: false,
+                args: [null, randomUser, randomUser],
+                argsEncoded: encodePacked(['address', 'address'], [randomUser, randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function requestDeposit(uint256, address controller, address owner)',
+                valueNonZero: false,
+                args: [null, randomUser, randomUser],
+                argsEncoded: encodePacked(['address', 'address'], [randomUser, randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function requestRedeem(uint256, address controller, address owner)',
+                valueNonZero: false,
+                args: [null, randomUser, randomUser],
+                argsEncoded: encodePacked(['address', 'address'], [randomUser, randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function withdraw(uint256, address receiver, address owner)',
+                valueNonZero: false,
+                args: [null, randomUser, randomUser],
+                argsEncoded: encodePacked(['address', 'address'], [randomUser, randomUser]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function withdraw(uint64,bytes16,address,uint256,address,uint128)',
+                valueNonZero: false,
+                args: [poolId.toString(), scId.toString(), someErc20, null, randomUser, null],
+                argsEncoded: encodePacked(['uint64', 'bytes16', 'address', 'address'], [poolId.raw, scId.raw, someErc20, randomUser]),
+            },
+        ];
+        context.tenderlyFork.impersonateAddress = fundManager;
+        context.centrifuge.setSigner(context.tenderlyFork.signer);
+        await pool.updateBalanceSheetManagers([{ chainId, address: mpmAddress, canManage: true }]);
     });
-    it('works', async () => {
+    it('correctly constructs the merkle tree', async () => {
         const expectedRootHash = '0x282444929cdaf1a1b1b4f69f682f30978c826397dd82c8e876db37c3a69fe4e5';
-        await merkleProofManager.getPolicyRoot();
+        const poolId = 281474976710657n;
+        const scId = '0x31000000000000000000000000000000';
+        const erc20 = '0xa0Cb889707d426A7A386870A03bc70d1b0697598';
+        const manager = '0xA4AD4f68d0b91CFD19687c881e50f3A00242828c';
+        const decoder = '0x03A6a84cD762D9707A21605b548aaaB891562aAb';
+        const target = '0x1B237b1866D34C8619D52F9A5047a4Ab976e0426';
+        const policies = [
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function withdraw(uint64,bytes16,address,uint256,address,uint128)',
+                valueNonZero: false,
+                args: [poolId.toString(), scId, erc20, null, manager, null],
+                argsEncoded: encodePacked(['uint64', 'bytes16', 'address', 'address'], [poolId, scId, erc20, manager]),
+            },
+            {
+                assetId: assetId.toString(),
+                decoder,
+                target,
+                abi: 'function deposit(uint64,bytes16,address,uint256,uint128)',
+                valueNonZero: false,
+                args: [poolId.toString(), scId, erc20, null],
+                argsEncoded: encodePacked(['uint64', 'bytes16', 'address'], [poolId, scId, erc20]),
+            },
+        ];
+        const tree = getMerkleTree(SimpleMerkleTree, policies);
+        expect(tree.root).to.equal(expectedRootHash);
+    });
+    it('sets policies', async () => {
+        const centrifugeWithPin = new Centrifuge({
+            environment: 'testnet',
+            pinJson: async (data) => {
+                expect(data.merkleProofManager[chainId][strategist].policies).to.deep.equal(mockPolicies);
+                // Simulate pinning JSON and returning a CID
+                return 'abc';
+            },
+            rpcUrls: {
+                11155111: context.tenderlyFork.rpcUrl,
+            },
+        });
+        const mpm = new MerkleProofManager(centrifugeWithPin, merkleProofManager.network, mpmAddress);
+        context.tenderlyFork.impersonateAddress = fundManager;
+        centrifugeWithPin.setSigner(context.tenderlyFork.signer);
+        await mpm.setPolicies(strategist, mockPolicies.map((p) => ({ ...p, argsEncoded: undefined })));
+        const tree = getMerkleTree(SimpleMerkleTree, mockPolicies);
+        const rootHash = await centrifugeWithPin.getClient(chainId).readContract({
+            address: mpmAddress,
+            abi: ABI.MerkleProofManager,
+            functionName: 'policy',
+            args: [strategist],
+        });
+        expect(rootHash).to.equal(tree.root);
+    });
+    it('can execute calls', async () => {
+        const { vaultRouter } = await context.centrifuge._protocolAddresses(chainId);
+        await context.tenderlyFork.fundAccountEth(strategist, 10n ** 18n);
+        context.tenderlyFork.impersonateAddress = strategist;
+        context.centrifuge.setSigner(context.tenderlyFork.signer);
+        const mock = sinon.stub(merkleProofManager.pool, 'metadata');
+        mock.returns(makeThenable(of({
+            merkleProofManager: {
+                [chainId]: {
+                    [strategist.toLowerCase()]: { policies: mockPolicies },
+                },
+            },
+        })));
+        await merkleProofManager.execute([{ policy: mockPolicies[0], inputs: [123000000n] }]);
+        const balance = await context.centrifuge.getClient(chainId).readContract({
+            address: someErc20,
+            abi: ABI.Currency,
+            functionName: 'allowance',
+            args: [mpmAddress, vaultRouter],
+        });
+        expect(balance).to.equal(123000000n);
+        mock.restore();
     });
 });
 //# sourceMappingURL=MerkleProofManager.test.js.map

package/dist/entities/MerkleProofManager.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"MerkleProofManager.test.js","sourceRoot":"","sources":["../../src/entities/MerkleProofManager.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAA;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAA;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAA;AAC5D,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAA;AAE9C,MAAM,OAAO,GAAG,QAAQ,CAAA;AACxB,MAAM,MAAM,GAAG,CAAC,CAAA;AAEhB,6CAA6C;AAC7C,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;AACrC,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;AACzC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;AAEvC,MAAM,WAAW,GAAG,4CAA4C,CAAA;AAEhE,MAAM,UAAU,GAAG,4CAA4C,CAAA;AAE/D,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,kBAAsC,CAAA;IAC1C,MAAM,CAAC,KAAK,IAAI,EAAE;QAChB,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,CAAA;QAC9B,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,CAAA;QACtD,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,UAAU,EAAE,IAAI,EAAE,OAAO,CAAC,CAAA;QAC9D,kBAAkB,GAAG,IAAI,kBAAkB,CAAC,UAAU,EAAE,WAAW,EAAE,UAAU,CAAC,CAAA;IAClF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;QACrB,MAAM,gBAAgB,GAAG,oEAAoE,CAAA;QAC7F,MAAM,kBAAkB,CAAC,aAAa,EAAE,CAAA;IAC1C,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
+{"version":3,"file":"MerkleProofManager.test.js","sourceRoot":"","sources":["../../src/entities/MerkleProofManager.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAA;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAA;AAC7B,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAA;AACzB,OAAO,KAAK,MAAM,OAAO,CAAA;AACzB,OAAO,EAAE,YAAY,EAAE,MAAM,MAAM,CAAA;AACnC,OAAO,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAA;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAA;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAA;AAEjD,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAA;AACjE,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAA;AAC3E,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAA;AAE9C,MAAM,OAAO,GAAG,QAAQ,CAAA;AACxB,MAAM,MAAM,GAAG,CAAC,CAAA;AAEhB,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;AACrC,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;AACzC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;AAEvC,MAAM,SAAS,GAAG,4CAA4C,CAAA;AAE9D,MAAM,WAAW,GAAG,4CAA4C,CAAA;AAChE,MAAM,UAAU,GAAG,4CAA4C,CAAA;AAC/D,MAAM,YAAY,GAAG,4CAA4C,CAAA;AAEjE,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,IAAU,CAAA;IACd,IAAI,kBAAsC,CAAA;IAC1C,IAAI,YAAiC,CAAA;IACrC,MAAM,UAAU,GAAG,aAAa,EAAE,CAAA;IAClC,MAAM,CAAC,KAAK,IAAI,EAAE;QAChB,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,CAAA;QAC9B,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,MAAM,UAAU,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAA;QAClF,IAAI,GAAG,IAAI,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,CAAA;QAChD,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,UAAU,EAAE,IAAI,EAAE,OAAO,CAAC,CAAA;QAC9D,kBAAkB,GAAG,IAAI,kBAAkB,CAAC,UAAU,EAAE,WAAW,EAAE,UAAU,CAAC,CAAA;QAChF,MAAM,OAAO,GAAG,YAAY,CAAA;QAC5B,MAAM,MAAM,GAAG,YAAY,CAAA;QAC3B,MAAM,UAAU,GAAG,aAAa,EAAE,CAAA;QAElC,YAAY,GAAG;YACb;gBACE,OAAO,EAAE,SAAS;gBAClB,OAAO;gBACP,MAAM,EAAE,SAAS;gBACjB,GAAG,EAAE,mCAAmC;gBACxC,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC;gBACzB,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;aACtD;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,4DAA4D;gBACjE,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC;gBACxB,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;aACrD;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,2DAA2D;gBAChE,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC;gBACxB,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;aACrD;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,mFAAmF;gBACxF,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;gBACpC,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;aAC5E;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,kFAAkF;gBACvF,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;gBACpC,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;aAC5E;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,gFAAgF;gBACrF,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC;gBAC1D,WAAW,EAAE,YAAY,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;aAC/F;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,6CAA6C;gBAClD,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC;gBACxB,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;aACrD;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,0CAA0C;gBAC/C,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC;gBACxB,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;aACrD;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,2DAA2D;gBAChE,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;gBACpC,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;aAC5E;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,qEAAqE;gBAC1E,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;gBACpC,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;aAC5E;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,oEAAoE;gBACzE,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;gBACpC,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;aAC5E;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,6DAA6D;gBAClE,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;gBACpC,WAAW,EAAE,YAAY,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;aAC5E;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,mEAAmE;gBACxE,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,CAAC;gBAC7E,WAAW,EAAE,YAAY,CACvB,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,EAC3C,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,SAAS,EAAE,UAAU,CAAC,CAC9C;aACF;SACF,CAAA;QAED,OAAO,CAAC,YAAY,CAAC,kBAAkB,GAAG,WAAW,CAAA;QACrD,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;QAEzD,MAAM,IAAI,CAAC,0BAA0B,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;IAC5F,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,gBAAgB,GAAG,oEAAoE,CAAA;QAC7F,MAAM,MAAM,GAAG,gBAAgB,CAAA;QAC/B,MAAM,IAAI,GAAG,oCAAoC,CAAA;QACjD,MAAM,KAAK,GAAG,4CAA4C,CAAA;QAC1D,MAAM,OAAO,GAAG,4CAA4C,CAAA;QAC5D,MAAM,OAAO,GAAG,4CAA4C,CAAA;QAC5D,MAAM,MAAM,GAAG,4CAA4C,CAAA;QAE3D,MAAM,QAAQ,GAAwB;YACpC;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,mEAAmE;gBACxE,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC;gBAC3D,WAAW,EAAE,YAAY,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;aACvG;YACD;gBACE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC3B,OAAO;gBACP,MAAM;gBACN,GAAG,EAAE,0DAA0D;gBAC/D,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC;gBAC5C,WAAW,EAAE,YAAY,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;aACnF;SACF,CAAA;QAED,MAAM,IAAI,GAAG,aAAa,CAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAA;QACtD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,iBAAiB,GAAG,IAAI,UAAU,CAAC;YACvC,WAAW,EAAE,SAAS;YACtB,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;gBACtB,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAA;gBACzF,4CAA4C;gBAC5C,OAAO,KAAK,CAAA;YACd,CAAC;YACD,OAAO,EAAE;gBACP,QAAQ,EAAE,OAAO,CAAC,YAAY,CAAC,MAAM;aACtC;SACF,CAAC,CAAA;QACF,MAAM,GAAG,GAAG,IAAI,kBAAkB,CAAC,iBAAiB,EAAE,kBAAkB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAA;QAE7F,OAAO,CAAC,YAAY,CAAC,kBAAkB,GAAG,WAAW,CAAA;QACrD,iBAAiB,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;QAExD,MAAM,GAAG,CAAC,WAAW,CACnB,UAAU,EACV,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC,CAC5D,CAAA;QAED,MAAM,IAAI,GAAG,aAAa,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAA;QAE1D,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC;YACvE,OAAO,EAAE,UAAU;YACnB,GAAG,EAAE,GAAG,CAAC,kBAAkB;YAC3B,YAAY,EAAE,QAAQ;YACtB,IAAI,EAAE,CAAC,UAAU,CAAC;SACnB,CAAC,CAAA;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;QACjC,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAA;QAE5E,MAAM,OAAO,CAAC,YAAY,CAAC,cAAc,CAAC,UAAU,EAAE,GAAG,IAAI,GAAG,CAAC,CAAA;QACjE,OAAO,CAAC,YAAY,CAAC,kBAAkB,GAAG,UAAU,CAAA;QACpD,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;QAEzD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;QAC5D,IAAI,CAAC,OAAO,CACV,YAAY,CACV,EAAE,CAAC;YACD,kBAAkB,EAAE;gBAClB,CAAC,OAAO,CAAC,EAAE;oBACT,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE;iBACvD;aACF;SACK,CAAC,CACV,CACF,CAAA;QAED,MAAM,kBAAkB,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAE,EAAE,MAAM,EAAE,CAAC,UAAY,CAAC,EAAE,CAAC,CAAC,CAAA;QAExF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC;YACvE,OAAO,EAAE,SAAS;YAClB,GAAG,EAAE,GAAG,CAAC,QAAQ;YACjB,YAAY,EAAE,WAAW;YACzB,IAAI,EAAE,CAAC,UAAU,EAAE,WAAW,CAAC;SAChC,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,UAAY,CAAC,CAAA;QAEtC,IAAI,CAAC,OAAO,EAAE,CAAA;IAChB,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/entities/PoolNetwork.d.ts

@@ -2,6 +2,7 @@ import { HexString } from '../types/index.js';
 import { AssetId, ShareClassId } from '../utils/types.js';
 import { BalanceSheet } from './BalanceSheet.js';
 import { Entity } from './Entity.js';
+import { MerkleProofManager } from './MerkleProofManager.js';
 import type { Pool } from './Pool.js';
 import { ShareClass } from './ShareClass.js';
 /**
@@ -48,6 +49,7 @@ export declare class PoolNetwork extends Entity {
      * and doesn't indicate whether any vaults have been deployed.
      */
     isActive(): import("../index.js").Query<boolean>;
+    merkleProofManager(): import("../index.js").Query<MerkleProofManager>;
     /**
      * Enable and deploy share classes/vaults.
      * @param shareClasses - An array of share classes to enable

package/dist/entities/PoolNetwork.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"PoolNetwork.d.ts","sourceRoot":"","sources":["../../src/entities/PoolNetwork.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAA;AAK7C,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAA;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAA;AAChD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AACpC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAE5C;;GAEG;AACH,qBAAa,WAAY,SAAQ,MAAM;IAI5B,IAAI,EAAE,IAAI;IACV,OAAO,EAAE,MAAM;IAKxB;;;;OAIG;IACH,OAAO;;;;;;;;;IA8BP,YAAY,CAAC,IAAI,EAAE,YAAY;IAM/B;;;OAGG;IACH,aAAa,CAAC,IAAI,EAAE,YAAY;IAMhC;;;;;OAKG;IACH,MAAM,CAAC,IAAI,EAAE,YAAY,EAAE,eAAe,UAAQ;IASlD;;;;OAIG;IACH,KAAK,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,GAAG,OAAO;IAgBpD;;;OAGG;IACH,QAAQ;IAgCR;;;;OAIG;IACH,MAAM,CACJ,YAAY,GAAE;QAAE,EAAE,EAAE,YAAY,CAAC;QAAC,IAAI,EAAE,SAAS,CAAA;KAAE,EAAO,EAC1D,MAAM,GAAE;QAAE,YAAY,EAAE,YAAY,CAAC;QAAC,OAAO,EAAE,OAAO,CAAC;QAAC,IAAI,EAAE,OAAO,GAAG,aAAa,CAAA;KAAE,EAAO;IAuJhG;;;OAGG;IACH,YAAY,CAAC,MAAM,EAAE;QAAE,YAAY,EAAE,YAAY,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,EAAE;CA6GxE"}
+{"version":3,"file":"PoolNetwork.d.ts","sourceRoot":"","sources":["../../src/entities/PoolNetwork.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAA;AAK7C,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAA;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAA;AAChD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AACpC,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAA;AAC5D,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAE5C;;GAEG;AACH,qBAAa,WAAY,SAAQ,MAAM;IAI5B,IAAI,EAAE,IAAI;IACV,OAAO,EAAE,MAAM;IAKxB;;;;OAIG;IACH,OAAO;;;;;;;;;IA8BP,YAAY,CAAC,IAAI,EAAE,YAAY;IAM/B;;;OAGG;IACH,aAAa,CAAC,IAAI,EAAE,YAAY;IAMhC;;;;;OAKG;IACH,MAAM,CAAC,IAAI,EAAE,YAAY,EAAE,eAAe,UAAQ;IASlD;;;;OAIG;IACH,KAAK,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,GAAG,OAAO;IAgBpD;;;OAGG;IACH,QAAQ;IAgCR,kBAAkB;IAOlB;;;;OAIG;IACH,MAAM,CACJ,YAAY,GAAE;QAAE,EAAE,EAAE,YAAY,CAAC;QAAC,IAAI,EAAE,SAAS,CAAA;KAAE,EAAO,EAC1D,MAAM,GAAE;QAAE,YAAY,EAAE,YAAY,CAAC;QAAC,OAAO,EAAE,OAAO,CAAC;QAAC,IAAI,EAAE,OAAO,GAAG,aAAa,CAAA;KAAE,EAAO;IAuJhG;;;OAGG;IACH,YAAY,CAAC,MAAM,EAAE;QAAE,YAAY,EAAE,YAAY,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,EAAE;CA6GxE"}

package/dist/entities/PoolNetwork.js

@@ -8,6 +8,7 @@ import { repeatOnEvents } from '../utils/rx.js';
 import { wrapTransaction } from '../utils/transaction.js';
 import { BalanceSheet } from './BalanceSheet.js';
 import { Entity } from './Entity.js';
+import { MerkleProofManager } from './MerkleProofManager.js';
 import { ShareClass } from './ShareClass.js';
 /**
  * Query and interact with a pool on a specific network.
@@ -107,6 +108,12 @@ export class PoolNetwork extends Entity {
             }, this.chainId));
         })));
     }
+    merkleProofManager() {
+        return this._query(['merkleProofManager'], () => {
+            // TODO: Get Merkle Proof Manager address from indexer
+            return of(new MerkleProofManager(this._root, this, '0x0'));
+        });
+    }
     /**
      * Enable and deploy share classes/vaults.
      * @param shareClasses - An array of share classes to enable