@openzeppelin/contracts-mcp 0.4.3 → 0.4.5

package/README.md

@@ -19,6 +19,7 @@ Provides tools to generate smart contract source code for the following language
 | cairo | erc20, erc721, erc1155, account, multisig, governor, vesting, custom |
 | stellar | fungible, stablecoin, non-fungible |
 | stylus | erc20, erc721, erc1155 |
+| uniswap-hooks | hooks (tool name is just `uniswap-hooks`) |
 
 
 ## Installation

package/dist/index.d.ts

@@ -2,4 +2,5 @@ export { registerSolidityTools } from './solidity/tools';
 export { registerCairoTools } from './cairo/tools';
 export { registerStellarTools } from './stellar/tools';
 export { registerStylusTools } from './stylus/tools';
+export { registerUniswapHooksTools } from './uniswap-hooks/tools';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AACvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AACvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.registerStylusTools = exports.registerStellarTools = exports.registerCairoTools = exports.registerSolidityTools = void 0;
+exports.registerUniswapHooksTools = exports.registerStylusTools = exports.registerStellarTools = exports.registerCairoTools = exports.registerSolidityTools = void 0;
 var tools_1 = require("./solidity/tools");
 Object.defineProperty(exports, "registerSolidityTools", { enumerable: true, get: function () { return tools_1.registerSolidityTools; } });
 var tools_2 = require("./cairo/tools");
@@ -9,4 +9,6 @@ var tools_3 = require("./stellar/tools");
 Object.defineProperty(exports, "registerStellarTools", { enumerable: true, get: function () { return tools_3.registerStellarTools; } });
 var tools_4 = require("./stylus/tools");
 Object.defineProperty(exports, "registerStylusTools", { enumerable: true, get: function () { return tools_4.registerStylusTools; } });
+var tools_5 = require("./uniswap-hooks/tools");
+Object.defineProperty(exports, "registerUniswapHooksTools", { enumerable: true, get: function () { return tools_5.registerUniswapHooksTools; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,0CAAyD;AAAhD,8GAAA,qBAAqB,OAAA;AAC9B,uCAAmD;AAA1C,2GAAA,kBAAkB,OAAA;AAC3B,yCAAuD;AAA9C,6GAAA,oBAAoB,OAAA;AAC7B,wCAAqD;AAA5C,4GAAA,mBAAmB,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,0CAAyD;AAAhD,8GAAA,qBAAqB,OAAA;AAC9B,uCAAmD;AAA1C,2GAAA,kBAAkB,OAAA;AAC3B,yCAAuD;AAA9C,6GAAA,oBAAoB,OAAA;AAC7B,wCAAqD;AAA5C,4GAAA,mBAAmB,OAAA;AAC5B,+CAAkE;AAAzD,kHAAA,yBAAyB,OAAA"}

package/dist/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAOpE,wBAAgB,YAAY,cAsB3B"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAQpE,wBAAgB,YAAY,cAuB3B"}

package/dist/server.js

@@ -6,6 +6,7 @@ const tools_js_1 = require("./solidity/tools.js");
 const tools_js_2 = require("./cairo/tools.js");
 const tools_js_3 = require("./stellar/tools.js");
 const tools_js_4 = require("./stylus/tools.js");
+const tools_js_5 = require("./uniswap-hooks/tools.js");
 const package_json_1 = require("../package.json");
 function createServer() {
     const server = new mcp_js_1.McpServer({
@@ -13,7 +14,7 @@ function createServer() {
         version: package_json_1.version,
     }, {
         instructions: `\
-Tools are provided for different smart contract languages: Solidity, Starknet's Cairo, Stellar Soroban, and Arbitrum Stylus.
+Tools are provided for different smart contract languages and frameworks: Solidity, Starknet's Cairo, Stellar Soroban, Arbitrum Stylus, and Uniswap v4 Hooks.
 Each tool generates a smart contract using recommended best practices with OpenZeppelin Contracts libraries, and returns the source code. The tools do not write to disk.
 If the user requests to create a new smart contract, use the appropriate tool to generate the contract.
 If the user asks to modify an existing smart contract, use these tools to determine the recommended patterns. Toggle the options in a tool to determine how different features affect the code, then apply the same types of changes to the user's contract.
@@ -23,6 +24,7 @@ If the user asks to modify an existing smart contract, use these tools to determ
     (0, tools_js_2.registerCairoTools)(server);
     (0, tools_js_3.registerStellarTools)(server);
     (0, tools_js_4.registerStylusTools)(server);
+    (0, tools_js_5.registerUniswapHooksTools)(server);
     return server;
 }
 //# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -1 +1 @@
-{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":";;AAOA,oCAsBC;AA7BD,oEAAoE;AACpE,kDAA4D;AAC5D,+CAAsD;AACtD,iDAA0D;AAC1D,gDAAwD;AACxD,kDAA0C;AAE1C,SAAgB,YAAY;IAC1B,MAAM,MAAM,GAAG,IAAI,kBAAS,CAC1B;QACE,IAAI,EAAE,+BAA+B;QACrC,OAAO,EAAP,sBAAO;KACR,EACD;QACE,YAAY,EAAE;;;;;CAKnB;KACI,CACF,CAAC;IAEF,IAAA,gCAAqB,EAAC,MAAM,CAAC,CAAC;IAC9B,IAAA,6BAAkB,EAAC,MAAM,CAAC,CAAC;IAC3B,IAAA,+BAAoB,EAAC,MAAM,CAAC,CAAC;IAC7B,IAAA,8BAAmB,EAAC,MAAM,CAAC,CAAC;IAE5B,OAAO,MAAM,CAAC;AAChB,CAAC"}
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":";;AAQA,oCAuBC;AA/BD,oEAAoE;AACpE,kDAA4D;AAC5D,+CAAsD;AACtD,iDAA0D;AAC1D,gDAAwD;AACxD,uDAAqE;AACrE,kDAA0C;AAE1C,SAAgB,YAAY;IAC1B,MAAM,MAAM,GAAG,IAAI,kBAAS,CAC1B;QACE,IAAI,EAAE,+BAA+B;QACrC,OAAO,EAAP,sBAAO;KACR,EACD;QACE,YAAY,EAAE;;;;;CAKnB;KACI,CACF,CAAC;IAEF,IAAA,gCAAqB,EAAC,MAAM,CAAC,CAAC;IAC9B,IAAA,6BAAkB,EAAC,MAAM,CAAC,CAAC;IAC3B,IAAA,+BAAoB,EAAC,MAAM,CAAC,CAAC;IAC7B,IAAA,8BAAmB,EAAC,MAAM,CAAC,CAAC;IAC5B,IAAA,oCAAyB,EAAC,MAAM,CAAC,CAAC;IAElC,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/uniswap-hooks/schemas.d.ts

@@ -0,0 +1,106 @@
+import { z } from 'zod';
+export declare const commonSchema: {
+    readonly access: z.ZodOptional<z.ZodUnion<[z.ZodUnion<[z.ZodLiteral<"ownable">, z.ZodLiteral<"roles">]>, z.ZodLiteral<"managed">]>>;
+    readonly info: z.ZodOptional<z.ZodObject<{
+        securityContact: z.ZodOptional<z.ZodString>;
+        license: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        securityContact?: string | undefined;
+        license?: string | undefined;
+    }, {
+        securityContact?: string | undefined;
+        license?: string | undefined;
+    }>>;
+};
+export declare const hooksSchema: {
+    readonly access: z.ZodOptional<z.ZodUnion<[z.ZodUnion<[z.ZodLiteral<"ownable">, z.ZodLiteral<"roles">]>, z.ZodLiteral<"managed">]>>;
+    readonly info: z.ZodOptional<z.ZodObject<{
+        securityContact: z.ZodOptional<z.ZodString>;
+        license: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        securityContact?: string | undefined;
+        license?: string | undefined;
+    }, {
+        securityContact?: string | undefined;
+        license?: string | undefined;
+    }>>;
+    readonly hook: z.ZodEnum<["BaseHook" | "BaseAsyncSwap" | "BaseCustomAccounting" | "BaseCustomCurve" | "BaseDynamicFee" | "BaseOverrideFee" | "BaseDynamicAfterFee" | "BaseHookFee" | "AntiSandwichHook" | "LiquidityPenaltyHook" | "LimitOrderHook" | "ReHypothecationHook" | "BaseOracleHook" | "OracleHookWithV3Adapters", ...("BaseHook" | "BaseAsyncSwap" | "BaseCustomAccounting" | "BaseCustomCurve" | "BaseDynamicFee" | "BaseOverrideFee" | "BaseDynamicAfterFee" | "BaseHookFee" | "AntiSandwichHook" | "LiquidityPenaltyHook" | "LimitOrderHook" | "ReHypothecationHook" | "BaseOracleHook" | "OracleHookWithV3Adapters")[]]>;
+    readonly name: z.ZodString;
+    readonly pausable: z.ZodBoolean;
+    readonly currencySettler: z.ZodBoolean;
+    readonly safeCast: z.ZodBoolean;
+    readonly transientStorage: z.ZodBoolean;
+    readonly shares: z.ZodObject<{
+        options: z.ZodUnion<[z.ZodLiteral<false>, z.ZodEnum<["ERC20", "ERC6909", "ERC1155"]>]>;
+        name: z.ZodOptional<z.ZodString>;
+        symbol: z.ZodOptional<z.ZodString>;
+        uri: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        options: false | "ERC20" | "ERC1155" | "ERC6909";
+        symbol?: string | undefined;
+        name?: string | undefined;
+        uri?: string | undefined;
+    }, {
+        options: false | "ERC20" | "ERC1155" | "ERC6909";
+        symbol?: string | undefined;
+        name?: string | undefined;
+        uri?: string | undefined;
+    }>;
+    readonly permissions: z.ZodObject<{
+        beforeInitialize: z.ZodBoolean;
+        afterInitialize: z.ZodBoolean;
+        beforeAddLiquidity: z.ZodBoolean;
+        beforeRemoveLiquidity: z.ZodBoolean;
+        afterAddLiquidity: z.ZodBoolean;
+        afterRemoveLiquidity: z.ZodBoolean;
+        beforeSwap: z.ZodBoolean;
+        afterSwap: z.ZodBoolean;
+        beforeDonate: z.ZodBoolean;
+        afterDonate: z.ZodBoolean;
+        beforeSwapReturnDelta: z.ZodBoolean;
+        afterSwapReturnDelta: z.ZodBoolean;
+        afterAddLiquidityReturnDelta: z.ZodBoolean;
+        afterRemoveLiquidityReturnDelta: z.ZodBoolean;
+    }, "strip", z.ZodTypeAny, {
+        beforeInitialize: boolean;
+        afterInitialize: boolean;
+        beforeAddLiquidity: boolean;
+        beforeRemoveLiquidity: boolean;
+        afterAddLiquidity: boolean;
+        afterRemoveLiquidity: boolean;
+        beforeSwap: boolean;
+        afterSwap: boolean;
+        beforeDonate: boolean;
+        afterDonate: boolean;
+        beforeSwapReturnDelta: boolean;
+        afterSwapReturnDelta: boolean;
+        afterAddLiquidityReturnDelta: boolean;
+        afterRemoveLiquidityReturnDelta: boolean;
+    }, {
+        beforeInitialize: boolean;
+        afterInitialize: boolean;
+        beforeAddLiquidity: boolean;
+        beforeRemoveLiquidity: boolean;
+        afterAddLiquidity: boolean;
+        afterRemoveLiquidity: boolean;
+        beforeSwap: boolean;
+        afterSwap: boolean;
+        beforeDonate: boolean;
+        afterDonate: boolean;
+        beforeSwapReturnDelta: boolean;
+        afterSwapReturnDelta: boolean;
+        afterAddLiquidityReturnDelta: boolean;
+        afterRemoveLiquidityReturnDelta: boolean;
+    }>;
+    readonly inputs: z.ZodObject<{
+        blockNumberOffset: z.ZodNumber;
+        maxAbsTickDelta: z.ZodNumber;
+    }, "strip", z.ZodTypeAny, {
+        blockNumberOffset: number;
+        maxAbsTickDelta: number;
+    }, {
+        blockNumberOffset: number;
+        maxAbsTickDelta: number;
+    }>;
+};
+//# sourceMappingURL=schemas.d.ts.map

package/dist/uniswap-hooks/schemas.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemas.d.ts","sourceRoot":"","sources":["../../src/uniswap-hooks/schemas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AA2BxB,eAAO,MAAM,YAAY;;;;;;;;;;;;CAcS,CAAC;AAuCnC,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAWU,CAAC"}

package/dist/uniswap-hooks/schemas.js

@@ -0,0 +1,78 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.hooksSchema = exports.commonSchema = void 0;
+const zod_1 = require("zod");
+const wizard_common_1 = require("@openzeppelin/wizard-common");
+const wizard_uniswap_hooks_1 = require("@openzeppelin/wizard-uniswap-hooks");
+/**
+ * Static type assertions to ensure schemas satisfy the Wizard API types. Not called at runtime.
+ */
+function _typeAssertions() {
+    const _assertions = {
+        Hooks: zod_1.z.object(exports.hooksSchema).parse({}),
+    };
+}
+const hookEnum = zod_1.z.enum(wizard_uniswap_hooks_1.HooksNames);
+const sharesOptionsSchema = zod_1.z.union([zod_1.z.literal(false), zod_1.z.enum(wizard_uniswap_hooks_1.ShareOptions)]);
+exports.commonSchema = {
+    access: zod_1.z
+        .literal('ownable')
+        .or(zod_1.z.literal('roles'))
+        .or(zod_1.z.literal('managed'))
+        .optional()
+        .describe(wizard_common_1.solidityCommonDescriptions.access),
+    info: zod_1.z
+        .object({
+        securityContact: zod_1.z.string().optional().describe(wizard_common_1.infoDescriptions.securityContact),
+        license: zod_1.z.string().optional().describe(wizard_common_1.infoDescriptions.license),
+    })
+        .optional()
+        .describe(wizard_common_1.infoDescriptions.info),
+};
+const sharesSchema = zod_1.z
+    .object({
+    options: sharesOptionsSchema.describe(wizard_common_1.uniswapHooksSharesDescriptions.options),
+    name: zod_1.z.string().optional().describe(wizard_common_1.uniswapHooksSharesDescriptions.name),
+    symbol: zod_1.z.string().optional().describe(wizard_common_1.uniswapHooksSharesDescriptions.symbol),
+    uri: zod_1.z.string().optional().describe(wizard_common_1.uniswapHooksSharesDescriptions.uri),
+})
+    .describe(wizard_common_1.uniswapHooksDescriptions.shares);
+const permissionsSchema = zod_1.z
+    .object({
+    beforeInitialize: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.beforeInitialize),
+    afterInitialize: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.afterInitialize),
+    beforeAddLiquidity: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.beforeAddLiquidity),
+    beforeRemoveLiquidity: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.beforeRemoveLiquidity),
+    afterAddLiquidity: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.afterAddLiquidity),
+    afterRemoveLiquidity: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.afterRemoveLiquidity),
+    beforeSwap: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.beforeSwap),
+    afterSwap: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.afterSwap),
+    beforeDonate: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.beforeDonate),
+    afterDonate: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.afterDonate),
+    beforeSwapReturnDelta: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.beforeSwapReturnDelta),
+    afterSwapReturnDelta: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.afterSwapReturnDelta),
+    afterAddLiquidityReturnDelta: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksPermissionDescriptions.afterAddLiquidityReturnDelta),
+    afterRemoveLiquidityReturnDelta: zod_1.z
+        .boolean()
+        .describe(wizard_common_1.uniswapHooksPermissionDescriptions.afterRemoveLiquidityReturnDelta),
+})
+    .describe(wizard_common_1.uniswapHooksDescriptions.permissions);
+const inputsSchema = zod_1.z
+    .object({
+    blockNumberOffset: zod_1.z.number().describe(wizard_common_1.uniswapHooksInputsDescriptions.blockNumberOffset),
+    maxAbsTickDelta: zod_1.z.number().describe(wizard_common_1.uniswapHooksInputsDescriptions.maxAbsTickDelta),
+})
+    .describe(wizard_common_1.uniswapHooksDescriptions.inputs);
+exports.hooksSchema = {
+    hook: hookEnum.describe(wizard_common_1.uniswapHooksDescriptions.hook),
+    name: zod_1.z.string().describe(wizard_common_1.commonDescriptions.name),
+    pausable: zod_1.z.boolean().describe(wizard_common_1.commonDescriptions.pausable),
+    currencySettler: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksDescriptions.currencySettler),
+    safeCast: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksDescriptions.safeCast),
+    transientStorage: zod_1.z.boolean().describe(wizard_common_1.uniswapHooksDescriptions.transientStorage),
+    shares: sharesSchema,
+    permissions: permissionsSchema,
+    inputs: inputsSchema,
+    ...exports.commonSchema,
+};
+//# sourceMappingURL=schemas.js.map

package/dist/uniswap-hooks/schemas.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemas.js","sourceRoot":"","sources":["../../src/uniswap-hooks/schemas.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AACxB,+DAQqC;AACrC,6EAAkG;AAElG;;GAEG;AACH,SAAS,eAAe;IACtB,MAAM,WAAW,GAEb;QACF,KAAK,EAAE,OAAC,CAAC,MAAM,CAAC,mBAAW,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;KACvC,CAAC;AACJ,CAAC;AAED,MAAM,QAAQ,GAAG,OAAC,CAAC,IAAI,CAAC,iCAAU,CAAC,CAAC;AAEpC,MAAM,mBAAmB,GAAG,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,OAAC,CAAC,IAAI,CAAC,mCAAY,CAAC,CAAC,CAAC,CAAC;AAEjE,QAAA,YAAY,GAAG;IAC1B,MAAM,EAAE,OAAC;SACN,OAAO,CAAC,SAAS,CAAC;SAClB,EAAE,CAAC,OAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SACtB,EAAE,CAAC,OAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;SACxB,QAAQ,EAAE;SACV,QAAQ,CAAC,0CAA0B,CAAC,MAAM,CAAC;IAC9C,IAAI,EAAE,OAAC;SACJ,MAAM,CAAC;QACN,eAAe,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,gCAAgB,CAAC,eAAe,CAAC;QACjF,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,gCAAgB,CAAC,OAAO,CAAC;KAClE,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,gCAAgB,CAAC,IAAI,CAAC;CACF,CAAC;AAEnC,MAAM,YAAY,GAAG,OAAC;KACnB,MAAM,CAAC;IACN,OAAO,EAAE,mBAAmB,CAAC,QAAQ,CAAC,8CAA8B,CAAC,OAAO,CAAC;IAC7E,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8B,CAAC,IAAI,CAAC;IACzE,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8B,CAAC,MAAM,CAAC;IAC7E,GAAG,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8B,CAAC,GAAG,CAAC;CACxE,CAAC;KACD,QAAQ,CAAC,wCAAwB,CAAC,MAAM,CAAC,CAAC;AAE7C,MAAM,iBAAiB,GAAG,OAAC;KACxB,MAAM,CAAC;IACN,gBAAgB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,gBAAgB,CAAC;IAC3F,eAAe,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,eAAe,CAAC;IACzF,kBAAkB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,kBAAkB,CAAC;IAC/F,qBAAqB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,qBAAqB,CAAC;IACrG,iBAAiB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,iBAAiB,CAAC;IAC7F,oBAAoB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,oBAAoB,CAAC;IACnG,UAAU,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,UAAU,CAAC;IAC/E,SAAS,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,SAAS,CAAC;IAC7E,YAAY,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,YAAY,CAAC;IACnF,WAAW,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,WAAW,CAAC;IACjF,qBAAqB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,qBAAqB,CAAC;IACrG,oBAAoB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,oBAAoB,CAAC;IACnG,4BAA4B,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kDAAkC,CAAC,4BAA4B,CAAC;IACnH,+BAA+B,EAAE,OAAC;SAC/B,OAAO,EAAE;SACT,QAAQ,CAAC,kDAAkC,CAAC,+BAA+B,CAAC;CAChF,CAAC;KACD,QAAQ,CAAC,wCAAwB,CAAC,WAAW,CAAC,CAAC;AAElD,MAAM,YAAY,GAAG,OAAC;KACnB,MAAM,CAAC;IACN,iBAAiB,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8CAA8B,CAAC,iBAAiB,CAAC;IACxF,eAAe,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8CAA8B,CAAC,eAAe,CAAC;CACrF,CAAC;KACD,QAAQ,CAAC,wCAAwB,CAAC,MAAM,CAAC,CAAC;AAEhC,QAAA,WAAW,GAAG;IACzB,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,wCAAwB,CAAC,IAAI,CAAC;IACtD,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kCAAkB,CAAC,IAAI,CAAC;IAClD,QAAQ,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,kCAAkB,CAAC,QAAQ,CAAC;IAC3D,eAAe,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,wCAAwB,CAAC,eAAe,CAAC;IAC/E,QAAQ,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,wCAAwB,CAAC,QAAQ,CAAC;IACjE,gBAAgB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,wCAAwB,CAAC,gBAAgB,CAAC;IACjF,MAAM,EAAE,YAAY;IACpB,WAAW,EAAE,iBAAiB;IAC9B,MAAM,EAAE,YAAY;IACpB,GAAG,oBAAY;CACiB,CAAC"}

package/dist/uniswap-hooks/tools/hooks.d.ts

@@ -0,0 +1,3 @@
+import type { McpServer, RegisteredTool } from '@modelcontextprotocol/sdk/server/mcp.js';
+export declare function registerUniswapHooks(server: McpServer): RegisteredTool;
+//# sourceMappingURL=hooks.d.ts.map

package/dist/uniswap-hooks/tools/hooks.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hooks.d.ts","sourceRoot":"","sources":["../../../src/uniswap-hooks/tools/hooks.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,yCAAyC,CAAC;AAMzF,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,SAAS,GAAG,cAAc,CA0CtE"}

package/dist/uniswap-hooks/tools/hooks.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerUniswapHooks = registerUniswapHooks;
+const wizard_uniswap_hooks_1 = require("@openzeppelin/wizard-uniswap-hooks");
+const schemas_1 = require("../schemas");
+const utils_1 = require("../../utils");
+const wizard_common_1 = require("@openzeppelin/wizard-common");
+function registerUniswapHooks(server) {
+    return server.tool('uniswap-hooks', (0, utils_1.makeDetailedPrompt)(wizard_common_1.uniswapHooksPrompts.Hooks), schemas_1.hooksSchema, async ({ hook, name, pausable, currencySettler, safeCast, transientStorage, shares, permissions, inputs, access, info, }) => {
+        const opts = {
+            hook,
+            name,
+            pausable,
+            currencySettler,
+            safeCast,
+            transientStorage,
+            shares,
+            permissions,
+            inputs,
+            access,
+            info,
+        };
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: (0, utils_1.safePrintSolidityCodeBlock)(() => wizard_uniswap_hooks_1.hooks.print(opts)),
+                },
+            ],
+        };
+    });
+}
+//# sourceMappingURL=hooks.js.map

package/dist/uniswap-hooks/tools/hooks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hooks.js","sourceRoot":"","sources":["../../../src/uniswap-hooks/tools/hooks.ts"],"names":[],"mappings":";;AAMA,oDA0CC;AA/CD,6EAA8E;AAC9E,wCAAyC;AACzC,uCAA6E;AAC7E,+DAAkE;AAElE,SAAgB,oBAAoB,CAAC,MAAiB;IACpD,OAAO,MAAM,CAAC,IAAI,CAChB,eAAe,EACf,IAAA,0BAAkB,EAAC,mCAAmB,CAAC,KAAK,CAAC,EAC7C,qBAAW,EACX,KAAK,EAAE,EACL,IAAI,EACJ,IAAI,EACJ,QAAQ,EACR,eAAe,EACf,QAAQ,EACR,gBAAgB,EAChB,MAAM,EACN,WAAW,EACX,MAAM,EACN,MAAM,EACN,IAAI,GACL,EAAE,EAAE;QACH,MAAM,IAAI,GAAiB;YACzB,IAAI;YACJ,IAAI;YACJ,QAAQ;YACR,eAAe;YACf,QAAQ;YACR,gBAAgB;YAChB,MAAM;YACN,WAAW;YACX,MAAM;YACN,MAAM;YACN,IAAI;SACL,CAAC;QAEF,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAA,kCAA0B,EAAC,GAAG,EAAE,CAAC,4BAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;iBAC1D;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;AACJ,CAAC"}

package/dist/uniswap-hooks/tools.d.ts

@@ -0,0 +1,3 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+export declare function registerUniswapHooksTools(server: McpServer): void;
+//# sourceMappingURL=tools.d.ts.map

package/dist/uniswap-hooks/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../src/uniswap-hooks/tools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAGzE,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,SAAS,QAE1D"}

package/dist/uniswap-hooks/tools.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerUniswapHooksTools = registerUniswapHooksTools;
+const hooks_js_1 = require("./tools/hooks.js");
+function registerUniswapHooksTools(server) {
+    (0, hooks_js_1.registerUniswapHooks)(server);
+}
+//# sourceMappingURL=tools.js.map

package/dist/uniswap-hooks/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/uniswap-hooks/tools.ts"],"names":[],"mappings":";;AAGA,8DAEC;AAJD,+CAAwD;AAExD,SAAgB,yBAAyB,CAAC,MAAiB;IACzD,IAAA,+BAAoB,EAAC,MAAM,CAAC,CAAC;AAC/B,CAAC"}

package/package.json

@@ -1,14 +1,12 @@
 {
   "name": "@openzeppelin/contracts-mcp",
-  "version": "0.4.3",
+  "version": "0.4.5",
   "description": "OpenZeppelin Contracts MCP Server",
   "license": "AGPL-3.0-only",
   "repository": "https://github.com/OpenZeppelin/contracts-wizard",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
-  "bin": {
-    "contracts-mcp": "dist/cli.js"
-  },
+  "bin": "dist/cli.js",
   "ts:main": "src/index.ts",
   "files": [
     "LICENSE",
@@ -26,16 +24,17 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.11.1",
-    "@openzeppelin/wizard-common": "^0.3.2",
-    "@openzeppelin/wizard": "^0.10.3",
+    "@openzeppelin/wizard-common": "^0.3.3",
+    "@openzeppelin/wizard": "^0.10.4",
     "@openzeppelin/wizard-stylus": "^0.2.0",
     "@openzeppelin/wizard-stellar": "^0.4.4",
-    "@openzeppelin/wizard-cairo": "^2.1.0"
+    "@openzeppelin/wizard-cairo": "^2.1.0",
+    "@openzeppelin/wizard-uniswap-hooks": "^0.1.0"
   },
   "devDependencies": {
     "@types/node": "^20.0.0",
-    "typescript": "^5.0.0",
     "ava": "^6.0.0",
-    "ts-node": "^10.4.0"
+    "ts-node": "^10.4.0",
+    "typescript": "^5.0.0"
   }
 }

package/src/index.ts

@@ -2,3 +2,4 @@ export { registerSolidityTools } from './solidity/tools';
 export { registerCairoTools } from './cairo/tools';
 export { registerStellarTools } from './stellar/tools';
 export { registerStylusTools } from './stylus/tools';
+export { registerUniswapHooksTools } from './uniswap-hooks/tools';

package/src/server.ts

@@ -3,6 +3,7 @@ import { registerSolidityTools } from './solidity/tools.js';
 import { registerCairoTools } from './cairo/tools.js';
 import { registerStellarTools } from './stellar/tools.js';
 import { registerStylusTools } from './stylus/tools.js';
+import { registerUniswapHooksTools } from './uniswap-hooks/tools.js';
 import { version } from '../package.json';
 
 export function createServer() {
@@ -13,7 +14,7 @@ export function createServer() {
     },
     {
       instructions: `\
-Tools are provided for different smart contract languages: Solidity, Starknet's Cairo, Stellar Soroban, and Arbitrum Stylus.
+Tools are provided for different smart contract languages and frameworks: Solidity, Starknet's Cairo, Stellar Soroban, Arbitrum Stylus, and Uniswap v4 Hooks.
 Each tool generates a smart contract using recommended best practices with OpenZeppelin Contracts libraries, and returns the source code. The tools do not write to disk.
 If the user requests to create a new smart contract, use the appropriate tool to generate the contract.
 If the user asks to modify an existing smart contract, use these tools to determine the recommended patterns. Toggle the options in a tool to determine how different features affect the code, then apply the same types of changes to the user's contract.
@@ -25,6 +26,7 @@ If the user asks to modify an existing smart contract, use these tools to determ
   registerCairoTools(server);
   registerStellarTools(server);
   registerStylusTools(server);
+  registerUniswapHooksTools(server);
 
   return server;
 }

package/src/uniswap-hooks/schemas.ts

@@ -0,0 +1,92 @@
+import { z } from 'zod';
+import {
+  commonDescriptions,
+  infoDescriptions,
+  uniswapHooksDescriptions,
+  uniswapHooksSharesDescriptions,
+  uniswapHooksPermissionDescriptions,
+  uniswapHooksInputsDescriptions,
+  solidityCommonDescriptions,
+} from '@openzeppelin/wizard-common';
+import { HooksNames, ShareOptions, type KindedOptions } from '@openzeppelin/wizard-uniswap-hooks';
+
+/**
+ * Static type assertions to ensure schemas satisfy the Wizard API types. Not called at runtime.
+ */
+function _typeAssertions() {
+  const _assertions: {
+    [K in keyof KindedOptions]: Omit<KindedOptions[K], 'kind'>;
+  } = {
+    Hooks: z.object(hooksSchema).parse({}),
+  };
+}
+
+const hookEnum = z.enum(HooksNames);
+
+const sharesOptionsSchema = z.union([z.literal(false), z.enum(ShareOptions)]);
+
+export const commonSchema = {
+  access: z
+    .literal('ownable')
+    .or(z.literal('roles'))
+    .or(z.literal('managed'))
+    .optional()
+    .describe(solidityCommonDescriptions.access),
+  info: z
+    .object({
+      securityContact: z.string().optional().describe(infoDescriptions.securityContact),
+      license: z.string().optional().describe(infoDescriptions.license),
+    })
+    .optional()
+    .describe(infoDescriptions.info),
+} as const satisfies z.ZodRawShape;
+
+const sharesSchema = z
+  .object({
+    options: sharesOptionsSchema.describe(uniswapHooksSharesDescriptions.options),
+    name: z.string().optional().describe(uniswapHooksSharesDescriptions.name),
+    symbol: z.string().optional().describe(uniswapHooksSharesDescriptions.symbol),
+    uri: z.string().optional().describe(uniswapHooksSharesDescriptions.uri),
+  })
+  .describe(uniswapHooksDescriptions.shares);
+
+const permissionsSchema = z
+  .object({
+    beforeInitialize: z.boolean().describe(uniswapHooksPermissionDescriptions.beforeInitialize),
+    afterInitialize: z.boolean().describe(uniswapHooksPermissionDescriptions.afterInitialize),
+    beforeAddLiquidity: z.boolean().describe(uniswapHooksPermissionDescriptions.beforeAddLiquidity),
+    beforeRemoveLiquidity: z.boolean().describe(uniswapHooksPermissionDescriptions.beforeRemoveLiquidity),
+    afterAddLiquidity: z.boolean().describe(uniswapHooksPermissionDescriptions.afterAddLiquidity),
+    afterRemoveLiquidity: z.boolean().describe(uniswapHooksPermissionDescriptions.afterRemoveLiquidity),
+    beforeSwap: z.boolean().describe(uniswapHooksPermissionDescriptions.beforeSwap),
+    afterSwap: z.boolean().describe(uniswapHooksPermissionDescriptions.afterSwap),
+    beforeDonate: z.boolean().describe(uniswapHooksPermissionDescriptions.beforeDonate),
+    afterDonate: z.boolean().describe(uniswapHooksPermissionDescriptions.afterDonate),
+    beforeSwapReturnDelta: z.boolean().describe(uniswapHooksPermissionDescriptions.beforeSwapReturnDelta),
+    afterSwapReturnDelta: z.boolean().describe(uniswapHooksPermissionDescriptions.afterSwapReturnDelta),
+    afterAddLiquidityReturnDelta: z.boolean().describe(uniswapHooksPermissionDescriptions.afterAddLiquidityReturnDelta),
+    afterRemoveLiquidityReturnDelta: z
+      .boolean()
+      .describe(uniswapHooksPermissionDescriptions.afterRemoveLiquidityReturnDelta),
+  })
+  .describe(uniswapHooksDescriptions.permissions);
+
+const inputsSchema = z
+  .object({
+    blockNumberOffset: z.number().describe(uniswapHooksInputsDescriptions.blockNumberOffset),
+    maxAbsTickDelta: z.number().describe(uniswapHooksInputsDescriptions.maxAbsTickDelta),
+  })
+  .describe(uniswapHooksDescriptions.inputs);
+
+export const hooksSchema = {
+  hook: hookEnum.describe(uniswapHooksDescriptions.hook),
+  name: z.string().describe(commonDescriptions.name),
+  pausable: z.boolean().describe(commonDescriptions.pausable),
+  currencySettler: z.boolean().describe(uniswapHooksDescriptions.currencySettler),
+  safeCast: z.boolean().describe(uniswapHooksDescriptions.safeCast),
+  transientStorage: z.boolean().describe(uniswapHooksDescriptions.transientStorage),
+  shares: sharesSchema,
+  permissions: permissionsSchema,
+  inputs: inputsSchema,
+  ...commonSchema,
+} as const satisfies z.ZodRawShape;

package/src/uniswap-hooks/tools/hooks.ts

@@ -0,0 +1,49 @@
+import type { McpServer, RegisteredTool } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { hooks, type HooksOptions } from '@openzeppelin/wizard-uniswap-hooks';
+import { hooksSchema } from '../schemas';
+import { makeDetailedPrompt, safePrintSolidityCodeBlock } from '../../utils';
+import { uniswapHooksPrompts } from '@openzeppelin/wizard-common';
+
+export function registerUniswapHooks(server: McpServer): RegisteredTool {
+  return server.tool(
+    'uniswap-hooks',
+    makeDetailedPrompt(uniswapHooksPrompts.Hooks),
+    hooksSchema,
+    async ({
+      hook,
+      name,
+      pausable,
+      currencySettler,
+      safeCast,
+      transientStorage,
+      shares,
+      permissions,
+      inputs,
+      access,
+      info,
+    }) => {
+      const opts: HooksOptions = {
+        hook,
+        name,
+        pausable,
+        currencySettler,
+        safeCast,
+        transientStorage,
+        shares,
+        permissions,
+        inputs,
+        access,
+        info,
+      };
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: safePrintSolidityCodeBlock(() => hooks.print(opts)),
+          },
+        ],
+      };
+    },
+  );
+}

package/src/uniswap-hooks/tools.ts

@@ -0,0 +1,6 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { registerUniswapHooks } from './tools/hooks.js';
+
+export function registerUniswapHooksTools(server: McpServer) {
+  registerUniswapHooks(server);
+}