@openzeppelin/contracts-mcp 0.6.4 → 0.6.6

package/src/tron/tools/erc20.ts

@@ -0,0 +1,60 @@
+import type { McpServer, RegisteredTool } from '@modelcontextprotocol/sdk/server/mcp.js';
+import type { ERC20Options } from '@openzeppelin/wizard';
+import { buildGeneric, printContract, tronPrintProfile, sanitizeTronOptions } from '@openzeppelin/wizard';
+import { makeDetailedPrompt } from '../../utils';
+import { solidityERC20Schema } from '@openzeppelin/wizard-common/schemas';
+import { tronPrompts } from '@openzeppelin/wizard-common';
+import { registerWizardAppTool, wizardAppPrintResult } from '../../apps/register';
+
+export function registerTronTRC20(server: McpServer): RegisteredTool {
+  return registerWizardAppTool(
+    server,
+    'tron-trc20',
+    {
+      description: makeDetailedPrompt(tronPrompts.TRC20),
+      inputSchema: solidityERC20Schema,
+      title: 'TRON TRC20',
+    },
+    async ({
+      name,
+      symbol,
+      decimals,
+      burnable,
+      pausable,
+      premint,
+      premintChainId,
+      mintable,
+      callback,
+      permit,
+      votes,
+      flashmint,
+      crossChainBridging,
+      crossChainLinkAllowOverride,
+      access,
+      upgradeable,
+      info,
+    }) => {
+      const opts: ERC20Options = {
+        name,
+        symbol,
+        decimals,
+        burnable,
+        pausable,
+        premint,
+        premintChainId,
+        mintable,
+        callback,
+        permit,
+        votes,
+        flashmint,
+        crossChainBridging,
+        crossChainLinkAllowOverride,
+        access,
+        upgradeable,
+        info,
+      };
+      const tronOpts = sanitizeTronOptions({ kind: 'ERC20' as const, ...opts });
+      return wizardAppPrintResult(tronOpts, () => printContract(buildGeneric(tronOpts), tronPrintProfile), 'solidity');
+    },
+  );
+}

package/src/tron/tools/erc721.ts

@@ -0,0 +1,58 @@
+import type { McpServer, RegisteredTool } from '@modelcontextprotocol/sdk/server/mcp.js';
+import type { ERC721Options } from '@openzeppelin/wizard';
+import { buildGeneric, printContract, tronPrintProfile, sanitizeTronOptions } from '@openzeppelin/wizard';
+import { makeDetailedPrompt } from '../../utils';
+import { solidityERC721Schema } from '@openzeppelin/wizard-common/schemas';
+import { tronPrompts } from '@openzeppelin/wizard-common';
+import { registerWizardAppTool, wizardAppPrintResult } from '../../apps/register';
+
+export function registerTronTRC721(server: McpServer): RegisteredTool {
+  return registerWizardAppTool(
+    server,
+    'tron-trc721',
+    {
+      description: makeDetailedPrompt(tronPrompts.TRC721),
+      inputSchema: solidityERC721Schema,
+      title: 'TRON TRC721',
+    },
+    async ({
+      name,
+      symbol,
+      baseUri,
+      enumerable,
+      uriStorage,
+      burnable,
+      pausable,
+      mintable,
+      incremental,
+      votes,
+      crossChainBridging,
+      crossChainLinkAllowOverride,
+      access,
+      upgradeable,
+      namespacePrefix,
+      info,
+    }) => {
+      const opts: ERC721Options = {
+        name,
+        symbol,
+        baseUri,
+        enumerable,
+        uriStorage,
+        burnable,
+        pausable,
+        mintable,
+        incremental,
+        votes,
+        crossChainBridging,
+        crossChainLinkAllowOverride,
+        access,
+        upgradeable,
+        namespacePrefix,
+        info,
+      };
+      const tronOpts = sanitizeTronOptions({ kind: 'ERC721' as const, ...opts });
+      return wizardAppPrintResult(tronOpts, () => printContract(buildGeneric(tronOpts), tronPrintProfile), 'solidity');
+    },
+  );
+}

package/src/tron/tools/governor.ts

@@ -0,0 +1,69 @@
+import type { McpServer, RegisteredTool } from '@modelcontextprotocol/sdk/server/mcp.js';
+import type { GovernorOptions } from '@openzeppelin/wizard';
+import {
+  buildGeneric,
+  printContract,
+  tronPrintProfile,
+  TRON_DEFAULT_BLOCK_TIME,
+  sanitizeTronOptions,
+} from '@openzeppelin/wizard';
+import { makeDetailedPrompt } from '../../utils';
+import { tronGovernorSchema } from '@openzeppelin/wizard-common/schemas';
+import { tronPrompts } from '@openzeppelin/wizard-common';
+import { registerWizardAppTool, wizardAppPrintResult } from '../../apps/register';
+
+export function registerTronGovernor(server: McpServer): RegisteredTool {
+  return registerWizardAppTool(
+    server,
+    'tron-governor',
+    {
+      description: makeDetailedPrompt(tronPrompts.Governor),
+      inputSchema: tronGovernorSchema,
+      title: 'TRON Governor',
+    },
+    async ({
+      name,
+      delay,
+      period,
+      votes,
+      clockMode,
+      timelock,
+      blockTime,
+      decimals,
+      proposalThreshold,
+      quorumMode,
+      quorumPercent,
+      quorumAbsolute,
+      storage,
+      settings,
+      crossChainExecution,
+      upgradeable,
+      info,
+    }) => {
+      const opts: GovernorOptions = {
+        name,
+        delay,
+        period,
+        votes,
+        clockMode,
+        timelock,
+        // TRON produces blocks every ~3s (vs ~12s on Ethereum); apply that
+        // default when the caller hasn't supplied one so the generated
+        // voting delay/period in blocks matches TRON's chain.
+        blockTime: blockTime ?? TRON_DEFAULT_BLOCK_TIME,
+        decimals,
+        proposalThreshold,
+        quorumMode,
+        quorumPercent,
+        quorumAbsolute,
+        storage,
+        settings,
+        crossChainExecution,
+        upgradeable,
+        info,
+      };
+      const tronOpts = sanitizeTronOptions({ kind: 'Governor' as const, ...opts });
+      return wizardAppPrintResult(tronOpts, () => printContract(buildGeneric(tronOpts), tronPrintProfile), 'solidity');
+    },
+  );
+}

package/src/tron/tools.ts

@@ -0,0 +1,19 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { registerTronTRC20 } from './tools/erc20.js';
+import { registerTronTRC721 } from './tools/erc721.js';
+import { registerTronTRC1155 } from './tools/erc1155.js';
+import { registerTronGovernor } from './tools/governor.js';
+import { registerTronCustom } from './tools/custom.js';
+
+// TRON tools reuse the Solidity schemas (options are identical) and render the
+// output through `tronPrintProfile` so token standards become TRC* and imports
+// resolve from `@openzeppelin/tron-contracts`. Excluded on purpose: Account
+// (ERC-4337 EntryPoint out of scope) and Stablecoin / RealWorldAsset (they
+// depend on @openzeppelin/community-contracts, which is not ported to TRON).
+export function registerTronTools(server: McpServer) {
+  registerTronTRC20(server);
+  registerTronTRC721(server);
+  registerTronTRC1155(server);
+  registerTronGovernor(server);
+  registerTronCustom(server);
+}