dexe-mcp 0.3.0 → 0.4.0

package/dist/tools/merkle.js

@@ -0,0 +1,174 @@
+import { z } from "zod";
+import { isAddress, getAddress } from "ethers";
+import { buildAddressMerkleTree, buildMerkleTree, computeLeafHash, verifyProof, } from "../lib/merkleTree.js";
+/**
+ * Merkle utility tools — produce roots and proofs compatible with
+ * OpenZeppelin's `StandardMerkleTree` and the on-chain verifier in
+ * `TokenSaleProposalBuy.sol::_checkMerkleProofs`.
+ *
+ * Default leaf encoding is `["address"]` (the only shape used by the
+ * frontend's OTC token-sale flow). Set `leafEncoding` + `entries` for
+ * advanced trees (e.g. `["address","uint256"]`).
+ */
+function errorResult(message) {
+    return { content: [{ type: "text", text: message }], isError: true };
+}
+function jsonResult(value) {
+    return {
+        content: [{ type: "text", text: JSON.stringify(value, null, 2) }],
+        structuredContent: value,
+    };
+}
+function normaliseAddresses(addresses) {
+    const seen = new Set();
+    const result = [];
+    for (const raw of addresses) {
+        if (!isAddress(raw))
+            throw new Error(`Invalid address: ${raw}`);
+        const checksummed = getAddress(raw);
+        if (seen.has(checksummed.toLowerCase()))
+            continue;
+        seen.add(checksummed.toLowerCase());
+        result.push(checksummed);
+    }
+    return result;
+}
+export function registerMerkleTools(server, _ctx) {
+    server.registerTool("dexe_merkle_build", {
+        title: "Build a merkle tree (OZ StandardMerkleTree compatible)",
+        description: "Builds a merkle tree compatible with OpenZeppelin StandardMerkleTree (used by DeXe's TokenSaleProposal merkle whitelists). Default leaf shape is a single address; pass `leafEncoding` + `entries` for richer leaves (e.g. address + amount). Returns root, leaf hashes, and per-input-index proofs.",
+        inputSchema: {
+            addresses: z
+                .array(z.string())
+                .min(1)
+                .optional()
+                .describe("Convenience: addresses for an `address`-only merkle whitelist. Mutually exclusive with `entries`."),
+            entries: z
+                .array(z.array(z.union([z.string(), z.number()])))
+                .min(1)
+                .optional()
+                .describe("Advanced: per-leaf raw values matching `leafEncoding` order. Mutually exclusive with `addresses`."),
+            leafEncoding: z
+                .array(z.string())
+                .min(1)
+                .default(["address"])
+                .describe("ABI types for each leaf column. Default `['address']` matches the OTC frontend."),
+        },
+        outputSchema: {
+            root: z.string(),
+            leafHashes: z.array(z.string()),
+            proofs: z.array(z.array(z.string())),
+            addresses: z.array(z.string()).optional(),
+        },
+    }, async ({ addresses, entries, leafEncoding = ["address"] }) => {
+        try {
+            if (!addresses && !entries) {
+                return errorResult("Must provide either `addresses` or `entries`.");
+            }
+            if (addresses && entries) {
+                return errorResult("Provide either `addresses` or `entries`, not both.");
+            }
+            if (addresses) {
+                const normalised = normaliseAddresses(addresses);
+                const result = buildAddressMerkleTree(normalised);
+                return jsonResult({
+                    root: result.root,
+                    leafHashes: result.leafHashes,
+                    proofs: result.proofs,
+                    addresses: normalised,
+                });
+            }
+            const result = buildMerkleTree(entries, leafEncoding);
+            return jsonResult({
+                root: result.root,
+                leafHashes: result.leafHashes,
+                proofs: result.proofs,
+            });
+        }
+        catch (err) {
+            return errorResult(err instanceof Error ? err.message : String(err));
+        }
+    });
+    server.registerTool("dexe_merkle_proof", {
+        title: "Compute a merkle proof for one address (or leaf)",
+        description: "Builds the same tree as `dexe_merkle_build` and returns the proof for a single target. Useful for buyer-side flows where the full whitelist is known but only one proof is needed. Default shape: address-only.",
+        inputSchema: {
+            addresses: z
+                .array(z.string())
+                .min(1)
+                .optional()
+                .describe("Address-only whitelist (mutually exclusive with `entries`)."),
+            entries: z
+                .array(z.array(z.union([z.string(), z.number()])))
+                .min(1)
+                .optional(),
+            leafEncoding: z.array(z.string()).min(1).default(["address"]),
+            target: z.string().describe("Address (when using `addresses`)."),
+            targetEntry: z
+                .array(z.union([z.string(), z.number()]))
+                .optional()
+                .describe("Raw leaf values when using `entries`."),
+        },
+        outputSchema: {
+            root: z.string(),
+            leaf: z.string(),
+            proof: z.array(z.string()),
+            verified: z.boolean(),
+        },
+    }, async ({ addresses, entries, leafEncoding = ["address"], target, targetEntry }) => {
+        try {
+            if (!addresses && !entries) {
+                return errorResult("Must provide either `addresses` or `entries`.");
+            }
+            if (addresses) {
+                if (!isAddress(target))
+                    return errorResult(`Invalid target address: ${target}`);
+                const normalised = normaliseAddresses(addresses);
+                const checksummed = getAddress(target);
+                const idx = normalised.findIndex((a) => a.toLowerCase() === checksummed.toLowerCase());
+                if (idx === -1) {
+                    return errorResult(`Target ${checksummed} not present in whitelist.`);
+                }
+                const tree = buildAddressMerkleTree(normalised);
+                const leaf = computeLeafHash([checksummed], ["address"]);
+                const proof = tree.proofs[idx] ?? [];
+                return jsonResult({
+                    root: tree.root,
+                    leaf,
+                    proof,
+                    verified: verifyProof(proof, tree.root, leaf),
+                });
+            }
+            if (!entries) {
+                return errorResult("Must provide `entries` when not using `addresses`.");
+            }
+            if (!targetEntry) {
+                return errorResult("Provide `targetEntry` when using `entries`.");
+            }
+            const tree = buildMerkleTree(entries, leafEncoding);
+            const leaf = computeLeafHash(targetEntry, leafEncoding);
+            const idx = tree.leafHashes.findIndex((h) => h.toLowerCase() === leaf.toLowerCase());
+            if (idx === -1) {
+                return errorResult("Target entry leaf not in tree.");
+            }
+            let proof = [];
+            for (let i = 0; i < entries.length; i++) {
+                const candidateLeaf = computeLeafHash(entries[i], leafEncoding);
+                if (candidateLeaf.toLowerCase() === leaf.toLowerCase()) {
+                    proof = tree.proofs[i] ?? [];
+                    break;
+                }
+            }
+            return jsonResult({
+                root: tree.root,
+                leaf,
+                proof,
+                verified: verifyProof(proof, tree.root, leaf),
+            });
+        }
+        catch (err) {
+            return errorResult(err instanceof Error ? err.message : String(err));
+        }
+    });
+}
+//# sourceMappingURL=merkle.js.map

package/dist/tools/merkle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"merkle.js","sourceRoot":"","sources":["../../src/tools/merkle.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAG/C,OAAO,EACL,sBAAsB,EACtB,eAAe,EACf,eAAe,EACf,WAAW,GACZ,MAAM,sBAAsB,CAAC;AAE9B;;;;;;;;GAQG;AAEH,SAAS,WAAW,CAAC,OAAe;IAClC,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAChF,CAAC;AAED,SAAS,UAAU,CAAoC,KAAQ;IAC7D,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAC1E,iBAAiB,EAAE,KAAK;KACzB,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,SAA4B;IACtD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAC5B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,GAAG,EAAE,CAAC,CAAC;QAChE,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;YAAE,SAAS;QAClD,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC3B,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAiB,EAAE,IAAiB;IACtE,MAAM,CAAC,YAAY,CACjB,mBAAmB,EACnB;QACE,KAAK,EAAE,wDAAwD;QAC/D,WAAW,EACT,sSAAsS;QACxS,WAAW,EAAE;YACX,SAAS,EAAE,CAAC;iBACT,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;iBACjB,GAAG,CAAC,CAAC,CAAC;iBACN,QAAQ,EAAE;iBACV,QAAQ,CACP,mGAAmG,CACpG;YACH,OAAO,EAAE,CAAC;iBACP,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC;iBACjD,GAAG,CAAC,CAAC,CAAC;iBACN,QAAQ,EAAE;iBACV,QAAQ,CACP,mGAAmG,CACpG;YACH,YAAY,EAAE,CAAC;iBACZ,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;iBACjB,GAAG,CAAC,CAAC,CAAC;iBACN,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC;iBACpB,QAAQ,CACP,iFAAiF,CAClF;SACJ;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YAC/B,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACpC,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;SAC1C;KACF,EACD,KAAK,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY,GAAG,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE;QAC3D,IAAI,CAAC;YACH,IAAI,CAAC,SAAS,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC3B,OAAO,WAAW,CAAC,+CAA+C,CAAC,CAAC;YACtE,CAAC;YACD,IAAI,SAAS,IAAI,OAAO,EAAE,CAAC;gBACzB,OAAO,WAAW,CAAC,oDAAoD,CAAC,CAAC;YAC3E,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,UAAU,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;gBACjD,MAAM,MAAM,GAAG,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAClD,OAAO,UAAU,CAAC;oBAChB,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,SAAS,EAAE,UAAU;iBACtB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,MAAM,GAAG,eAAe,CAAC,OAA0C,EAAE,YAAY,CAAC,CAAC;YACzF,OAAO,UAAU,CAAC;gBAChB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;aACtB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QACvE,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,mBAAmB,EACnB;QACE,KAAK,EAAE,kDAAkD;QACzD,WAAW,EACT,iNAAiN;QACnN,WAAW,EAAE;YACX,SAAS,EAAE,CAAC;iBACT,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;iBACjB,GAAG,CAAC,CAAC,CAAC;iBACN,QAAQ,EAAE;iBACV,QAAQ,CAAC,6DAA6D,CAAC;YAC1E,OAAO,EAAE,CAAC;iBACP,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC;iBACjD,GAAG,CAAC,CAAC,CAAC;iBACN,QAAQ,EAAE;YACb,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC;YAC7D,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;YAChE,WAAW,EAAE,CAAC;iBACX,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;iBACxC,QAAQ,EAAE;iBACV,QAAQ,CAAC,uCAAuC,CAAC;SACrD;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YAC1B,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE;SACtB;KACF,EACD,KAAK,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY,GAAG,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE;QAChF,IAAI,CAAC;YACH,IAAI,CAAC,SAAS,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC3B,OAAO,WAAW,CAAC,+CAA+C,CAAC,CAAC;YACtE,CAAC;YACD,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;oBAAE,OAAO,WAAW,CAAC,2BAA2B,MAAM,EAAE,CAAC,CAAC;gBAChF,MAAM,UAAU,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;gBACjD,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC;gBACvC,MAAM,GAAG,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,WAAW,CAAC,WAAW,EAAE,CAAC,CAAC;gBACvF,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;oBACf,OAAO,WAAW,CAAC,UAAU,WAAW,4BAA4B,CAAC,CAAC;gBACxE,CAAC;gBACD,MAAM,IAAI,GAAG,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,IAAI,GAAG,eAAe,CAAC,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;gBACzD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;gBACrC,OAAO,UAAU,CAAC;oBAChB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI;oBACJ,KAAK;oBACL,QAAQ,EAAE,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC;iBAC9C,CAAC,CAAC;YACL,CAAC;YACD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,WAAW,CAAC,oDAAoD,CAAC,CAAC;YAC3E,CAAC;YACD,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,OAAO,WAAW,CAAC,6CAA6C,CAAC,CAAC;YACpE,CAAC;YACD,MAAM,IAAI,GAAG,eAAe,CAC1B,OAA0C,EAC1C,YAAY,CACb,CAAC;YACF,MAAM,IAAI,GAAG,eAAe,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YACxD,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YACrF,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;gBACf,OAAO,WAAW,CAAC,gCAAgC,CAAC,CAAC;YACvD,CAAC;YACD,IAAI,KAAK,GAAa,EAAE,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxC,MAAM,aAAa,GAAG,eAAe,CACnC,OAAO,CAAC,CAAC,CAAuB,EAChC,YAAY,CACb,CAAC;gBACF,IAAI,aAAa,CAAC,WAAW,EAAE,KAAK,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;oBACvD,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBAC7B,MAAM;gBACR,CAAC;YACH,CAAC;YACD,OAAO,UAAU,CAAC;gBAChB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI;gBACJ,KAAK;gBACL,QAAQ,EAAE,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC;aAC9C,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QACvE,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/dist/tools/otc.d.ts

@@ -0,0 +1,5 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { ToolContext } from "./context.js";
+import { SignerManager } from "../lib/signer.js";
+export declare function registerOtcTools(server: McpServer, ctx: ToolContext, signer: SignerManager): void;
+//# sourceMappingURL=otc.d.ts.map

package/dist/tools/otc.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"otc.d.ts","sourceRoot":"","sources":["../../src/tools/otc.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AA+DjD,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,SAAS,EACjB,GAAG,EAAE,WAAW,EAChB,MAAM,EAAE,aAAa,GACpB,IAAI,CAohBN"}