mcp-rand
Version:
MCP server providing random generation utilities including UUID, numbers, strings, passwords, Gaussian distribution, dice rolling, and card drawing
66 lines • 2.26 kB
JavaScript
import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
export const toolSpec = {
name: 'roll_dice',
description: 'Roll a set of dice using standard dice notation (e.g., "2d6" for two six-sided dice, "3d6+5" for three six-sided dice plus 5)',
inputSchema: {
type: 'object',
properties: {
dice: {
type: 'array',
items: {
type: 'string',
description: 'Dice notation (e.g., "2d6", "1d20", "4d4")'
},
description: 'Array of dice to roll'
}
},
required: ['dice']
}
};
function parseDiceNotation(notation) {
const match = notation.toLowerCase().match(/^(\d+)d(\d+)([+-]\d+)?$/);
if (!match) {
throw new McpError(ErrorCode.InvalidParams, 'Invalid dice notation');
}
const count = parseInt(match[1]);
const sides = parseInt(match[2]);
const modifier = match[3] ? parseInt(match[3]) : 0; // If no modifier, default to 0
if (count <= 0) {
throw new McpError(ErrorCode.InvalidParams, 'Number of dice must be positive');
}
if (sides <= 0) {
throw new McpError(ErrorCode.InvalidParams, 'Die size must be positive');
}
return { count, sides, modifier };
}
function rollDie(sides) {
return Math.floor(Math.random() * sides) + 1;
}
function rollDiceSet(notation) {
const { count, sides, modifier } = parseDiceNotation(notation);
const rolls = Array.from({ length: count }, () => rollDie(sides));
const rollTotal = rolls.reduce((a, b) => a + b, 0);
const total = rollTotal + modifier;
return {
dice: notation,
rolls,
modifier,
total
};
}
export const rollDiceHandler = async (request) => {
const args = request.params.arguments;
if (!args.dice || args.dice.length === 0) {
throw new McpError(ErrorCode.InvalidParams, 'Must specify at least one die to roll');
}
const results = args.dice.map(dice => rollDiceSet(dice));
return {
content: [
{
type: 'text',
text: JSON.stringify(results, null, 2)
}
]
};
};
//# sourceMappingURL=roll-dice.handler.js.map