Roots на практике

mkdir -p ~/mcp-sandbox/shop-frontend/src ~/mcp-sandbox/shop-api/app ~/mcp-sandbox/secret
printf 'def render():\n    # TODO: вынести в компонент\n    pass\n' > ~/mcp-sandbox/shop-frontend/src/page.py
printf 'def pay():\n    # TODO: ретраи\n    # TODO: логирование\n    pass\n' > ~/mcp-sandbox/shop-api/app/billing.py
printf 'DB_PASSWORD=hunter2\n' > ~/mcp-sandbox/secret/.env

import logging
from pathlib import Path
from typing import Annotated
from urllib.parse import unquote, urlparse

from mcp.server import MCPServer
from mcp.server.mcpserver import ListRoots, Resolve
from mcp.types import ListRootsResult

logger = logging.getLogger(__name__)
mcp = MCPServer("project-search")


def workspace_roots() -> ListRoots:
    """Резолвер: попросить у клиента список рабочих папок."""
    return ListRoots()


def root_paths(roots: ListRootsResult) -> list[Path]:
    paths: list[Path] = []
    for root in roots.roots:
        parsed = urlparse(str(root.uri))
        if parsed.scheme == "file":
            paths.append(Path(unquote(parsed.path)).resolve())
    return paths


def inside(path: Path, roots: list[Path]) -> bool:
    resolved = path.resolve()
    return any(resolved == r or r in resolved.parents for r in roots)


@mcp.tool()
async def count_todos(
    roots: Annotated[ListRootsResult, Resolve(workspace_roots)],
) -> str:
    """Посчитать пометки TODO в python-файлах рабочих папок."""
    paths = root_paths(roots)
    if not paths:
        return "Клиент не сообщил рабочих папок."
    report = []
    for base in paths:
        count = sum(f.read_text(errors="ignore").count("TODO") for f in base.rglob("*.py"))
        report.append(f"{base.name}: {count}")
    return "Пометок TODO: " + ", ".join(report)


@mcp.tool()
async def read_file(
    path: str,
    roots: Annotated[ListRootsResult, Resolve(workspace_roots)],
) -> str:
    """Прочитать файл, если он лежит в одной из рабочих папок."""
    allowed = root_paths(roots)
    target = Path(path)
    if not inside(target, allowed):
        logger.warning("read_file: отказ, %s вне рабочих папок", path)
        return f"Отказ: {path} находится вне рабочих папок."
    return target.read_text(errors="ignore")


if __name__ == "__main__":
    mcp.run()

from pathlib import Path

import anyio

from mcp import Client
from mcp.client.session import ClientRequestContext
from mcp.types import ListRootsResult, Root

from server import mcp

SANDBOX = Path.home() / "mcp-sandbox"


async def my_roots(context: ClientRequestContext) -> ListRootsResult:
    """Играет роль хоста: отдаёт две открытые папки."""
    print("--- сервер попросил список корней ---")
    return ListRootsResult(
        roots=[
            Root(uri=(SANDBOX / "shop-frontend").as_uri(), name="shop-frontend"),
            Root(uri=(SANDBOX / "shop-api").as_uri(), name="shop-api"),
        ]
    )


async def main() -> None:
    async with Client(mcp, list_roots_callback=my_roots) as client:
        result = await client.call_tool("count_todos", {})
        print(result.content[0].text)

        ok = await client.call_tool("read_file", {"path": str(SANDBOX / "shop-api/app/billing.py")})
        print(ok.content[0].text.splitlines()[0])

        bad = await client.call_tool("read_file", {"path": str(SANDBOX / "secret/.env")})
        print(bad.content[0].text)

        sneaky = await client.call_tool(
            "read_file", {"path": str(SANDBOX / "shop-api/../secret/.env")}
        )
        print(sneaky.content[0].text)


anyio.run(main)

--- сервер попросил список корней ---
Пометок TODO: shop-frontend: 1, shop-api: 2
--- сервер попросил список корней ---
def pay():
--- сервер попросил список корней ---
Отказ: /Users/anna/mcp-sandbox/secret/.env находится вне рабочих папок.
--- сервер попросил список корней ---
Отказ: /Users/anna/mcp-sandbox/shop-api/../secret/.env находится вне рабочих папок.

import logging
import sys
from pathlib import Path

from mcp.server import MCPServer

logger = logging.getLogger(__name__)
mcp = MCPServer("project-search")

ALLOWED: list[Path] = [Path(arg).resolve() for arg in sys.argv[1:]]


def inside(path: Path) -> bool:
    resolved = path.resolve()
    return any(resolved == r or r in resolved.parents for r in ALLOWED)


@mcp.tool()
async def count_todos() -> str:
    """Посчитать пометки TODO в python-файлах разрешённых папок."""
    if not ALLOWED:
        return "Сервер запущен без разрешённых папок."
    report = [f"{b.name}: {sum(f.read_text(errors='ignore').count('TODO') for f in b.rglob('*.py'))}" for b in ALLOWED]
    return "Пометок TODO: " + ", ".join(report)


@mcp.tool()
async def read_file(path: str) -> str:
    """Прочитать файл из разрешённых папок."""
    target = Path(path)
    if not inside(target):
        return f"Отказ: {path} находится вне разрешённых папок."
    return target.read_text(errors="ignore")


if __name__ == "__main__":
    logger.info("allowed: %s", ALLOWED)
    mcp.run()

import anyio

from mcp import Client, StdioServerParameters

server = StdioServerParameters(
    command="uv",
    args=["run", "server2.py", "/Users/anna/mcp-sandbox/shop-frontend", "/Users/anna/mcp-sandbox/shop-api"],
)


async def main() -> None:
    async with Client(server) as client:
        print((await client.call_tool("count_todos", {})).content[0].text)
        print((await client.call_tool("read_file", {"path": "/Users/anna/mcp-sandbox/secret/.env"})).content[0].text)


anyio.run(main)