server.py

"""
text-utils MCP Server
======================
A minimal FastMCP server exposing two text-processing tools:
word_count and char_count.

Trimmed down intentionally for a first real deployment -- fewer moving
parts to debug while learning the DigitalOcean / systemd / Caddy pipeline.
More tools can be added back later once the deployment itself is solid.

Run locally (STDIO transport, for use with Claude Desktop / Claude Code):
    uv run server.py

Run over HTTP (for remote/droplet deployment):
    uv run server.py --http
"""
import re
import sys
from mcp.server.fastmcp import FastMCP

mcp = FastMCP("text-utils")

@mcp.tool()
def word_count(text: str) -> int:
    """Count the number of words in the given text (whitespace-separated)."""
    return len(text.split())

@mcp.tool()
def char_count(text: str, include_spaces: bool = True) -> int:
    """
    Count the number of characters in the given text.

    Args:
        text: The text to measure.
        include_spaces: If False, whitespace characters are excluded from the count.
    """
    if include_spaces:
        return len(text)
    return len(re.sub(r"\s", "", text))

if __name__ == "__main__":
    if "--http" in sys.argv:
        # Streamable HTTP transport, for droplet / remote deployment.
        # Binds to 0.0.0.0:8000 by default so Caddy (running on the same
        # box) can reverse-proxy to it over localhost.
        mcp.run(transport="streamable-http")
    else:
        # STDIO transport, for local Claude Desktop / Claude Code use.
        mcp.run()