mirror of
https://github.com/FuzzingLabs/fuzzforge_ai.git
synced 2026-06-30 23:35:31 +02:00
AFredefon
71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
"""Dependency injection helpers for FuzzForge MCP."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from typing import TYPE_CHECKING, cast
|
|
|
|
from fastmcp.server.dependencies import get_context
|
|
from fuzzforge_runner import Runner, Settings
|
|
|
|
from fuzzforge_mcp.exceptions import FuzzForgeMCPError
|
|
|
|
if TYPE_CHECKING:
|
|
from fastmcp import Context
|
|
|
|
|
|
# Track the current active project path (set by init_project)
|
|
_current_project_path: Path | None = None
|
|
|
|
|
|
def set_current_project_path(project_path: Path) -> None:
|
|
"""Set the current project path.
|
|
|
|
Called by init_project to track which project is active.
|
|
|
|
:param project_path: Path to the project directory.
|
|
|
|
"""
|
|
global _current_project_path
|
|
_current_project_path = project_path
|
|
|
|
|
|
def get_settings() -> Settings:
|
|
"""Get MCP server settings from context.
|
|
|
|
:return: Settings instance.
|
|
:raises FuzzForgeMCPError: If settings not available.
|
|
|
|
"""
|
|
context: Context = get_context()
|
|
if context.request_context is None:
|
|
message: str = "Request context not available"
|
|
raise FuzzForgeMCPError(message)
|
|
return cast("Settings", context.request_context.lifespan_context)
|
|
|
|
|
|
def get_project_path() -> Path:
|
|
"""Get the current project path.
|
|
|
|
Returns the project path set by init_project, or falls back to
|
|
the current working directory if no project has been initialized.
|
|
|
|
:return: Path to the current project.
|
|
|
|
"""
|
|
global _current_project_path
|
|
if _current_project_path is not None:
|
|
return _current_project_path
|
|
# Fall back to current working directory (where the AI agent is working)
|
|
return Path.cwd()
|
|
|
|
|
|
def get_runner() -> Runner:
|
|
"""Get a configured Runner instance.
|
|
|
|
:return: Runner instance configured from MCP settings.
|
|
|
|
"""
|
|
settings: Settings = get_settings()
|
|
return Runner(settings)
|