import type { McpServer, RegisteredTool } from '@modelcontextprotocol/sdk/server/mcp.js';
import type { ERC721Options } from '@openzeppelin/wizard';
import { erc721 } from '@openzeppelin/wizard';
import { safePrintSolidityCodeBlock, makeDetailedPrompt } from '../../utils';
import { erc721Schema } from '../schemas';
import { solidityPrompts } from '@openzeppelin/wizard-common';
export function registerSolidityERC721(server: McpServer): RegisteredTool {
return server.tool(
'solidity-erc721',
makeDetailedPrompt(solidityPrompts.ERC721),
erc721Schema,
async ({
name,
symbol,
baseUri,
enumerable,
uriStorage,
burnable,
pausable,
mintable,
incremental,
votes,
access,
upgradeable,
namespacePrefix,
info,
}) => {
const opts: ERC721Options = {
name,
symbol,
baseUri,
enumerable,
uriStorage,
burnable,
pausable,
mintable,
incremental,
votes,
access,
upgradeable,
namespacePrefix,
info,
};
return {
content: [
{
type: 'text',
text: safePrintSolidityCodeBlock(() => erc721.print(opts)),
},
],
};
},
);
}
