141 lines
4.2 KiB
Python
141 lines
4.2 KiB
Python
from typing import Any, Dict
|
|
import logging
|
|
import httpx
|
|
from mcp.server.fastmcp import FastMCP
|
|
from dice import Dice
|
|
from player import Player
|
|
from item import Item
|
|
from game import Game
|
|
from npc import NPC
|
|
|
|
mcp = FastMCP("wyvern-castle")
|
|
game = Game()
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
handlers=[
|
|
logging.StreamHandler(),
|
|
]
|
|
)
|
|
|
|
@mcp.tool()
|
|
async def throw_a_dice(n_faces: int) -> Any:
|
|
"""Throw a dice with n faces. If n==2 its a coin toss.
|
|
|
|
Args:
|
|
n_faces: Number of faces of the dice
|
|
"""
|
|
logging.info(f"Throwing a dice with {n_faces} faces")
|
|
dice = Dice()
|
|
|
|
if n_faces < 1:
|
|
raise ValueError("Number of faces must be at least 1")
|
|
elif n_faces == 1:
|
|
return 1
|
|
elif n_faces == 2:
|
|
return dice.head_or_tails()
|
|
else:
|
|
return dice.roll(n_faces)
|
|
|
|
@mcp.tool()
|
|
async def mult(a: int, b: int) -> int:
|
|
"""Multiply two numbers.
|
|
|
|
Args:
|
|
a: First number
|
|
b: Second number
|
|
"""
|
|
logging.info(f"Calling mult with a={a}, b={b}")
|
|
result = a * b
|
|
logging.info(f"mult result: {result}")
|
|
return result
|
|
|
|
@mcp.tool()
|
|
async def add(a: int, b: int) -> int:
|
|
"""Add two numbers.
|
|
|
|
Args:
|
|
a: First number
|
|
b: Second number
|
|
"""
|
|
logging.info(f"Calling add with a={a}, b={b}")
|
|
result = a + b
|
|
logging.info(f"add result: {result}")
|
|
return result
|
|
|
|
@mcp.tool()
|
|
async def create_player(name: str, strength: int, dexterity: int, intelligence: int, wisdom: int, charisma: int, hp: int, armor: int, speed: int, item: str = None) -> Dict[str, Any]:
|
|
"""Create a new player. Need all the stats to function properly. Throw a d20 for every stats you don't have,
|
|
and a d6 for hp, armor and speed.
|
|
|
|
Args:
|
|
name: Name of the player
|
|
strength: Strength of the player
|
|
dexterity: Dexterity of the player
|
|
intelligence: Intelligence of the player
|
|
wisdom: Wisdom of the player
|
|
charisma: Charisma of the player
|
|
hp: Hit points of the player
|
|
armor: Armor class of the player
|
|
speed: Speed of the player
|
|
item: Item carried by the player
|
|
"""
|
|
logging.info(f"Creating player with name={name}")
|
|
player = Player(name, strength, dexterity, intelligence, wisdom, charisma, hp, armor, speed)
|
|
logging.info(f"Created player: {player}")
|
|
return player.serialize_dict()
|
|
|
|
@mcp.tool()
|
|
async def create_npc(name: str, strength: int, dexterity: int, intelligence: int, wisdom: int, charisma: int, hp: int, armor: int, speed: int, item: str = None) -> Dict[str, Any]:
|
|
"""Create a new NPC. Need all the stats to function properly. Throw a d20 for every stats you don't have,
|
|
and a d6 for hp, armor and speed.
|
|
|
|
Args:
|
|
name: Name of the NPC
|
|
strength: Strength of the NPC
|
|
dexterity: Dexterity of the NPC
|
|
intelligence: Intelligence of the NPC
|
|
wisdom: Wisdom of the NPC
|
|
charisma: Charisma of the NPC
|
|
hp: Hit points of the NPC
|
|
armor: Armor class of the NPC
|
|
speed: Speed of the NPC
|
|
item: Item carried by the NPC
|
|
"""
|
|
logging.info(f"Creating NPC with name={name}")
|
|
npc = NPC(name, strength, dexterity, intelligence, wisdom, charisma, hp, armor, speed)
|
|
logging.info(f"Created NPC: {npc}")
|
|
return npc.serialize_dict()
|
|
|
|
@mcp.tool()
|
|
async def create_item(name: str, description: str, bonus: str) -> Dict[str, Any]:
|
|
"""Create a new item.
|
|
|
|
Args:
|
|
name: Name of the item
|
|
description: Description of the item
|
|
bonus: Bonus of the item ex: strength+1,hp+5
|
|
"""
|
|
logging.info(f"Creating item with name={name}")
|
|
item = Item(name, description, bonus)
|
|
logging.info(f"Created item: {item}")
|
|
return item.serialize_dict()
|
|
|
|
@mcp.tool()
|
|
async def add_item_to_player(player_name: str, item_name: str) -> Dict[str, Any]:
|
|
"""Add an item to a player's inventory.
|
|
|
|
Args:
|
|
player_name: The name of the player to add the item to
|
|
item_name: The name of the item to add
|
|
"""
|
|
logging.info(f"Adding item {item_name} to player {player_name}")
|
|
return {"status": "Item added"}
|
|
|
|
def main():
|
|
# Initialize and run the server
|
|
mcp.run(transport="stdio")
|
|
|
|
if __name__ == "__main__":
|
|
main() |