use crate::connection::ConnectionPool;
use crate::error::{McpError, McpResult};
use crate::protocol::{
InitializeParams, InitializeResult, PromptsCapability, PromptsGetParams, PromptsGetResult,
PromptsListResult, ResourceReadParams, ResourceReadResult, ResourcesCapability,
ResourcesListResult, ServerCapabilities, ServerInfo, Tool, ToolCallParams, ToolCallResult,
ToolContent, ToolsCapability, ToolsListResult, MCP_PROTOCOL_VERSION,
SUPPORTED_PROTOCOL_VERSIONS,
};
use crate::{prompts, resources, tools};
use serde_json::json;
use std::sync::Arc;
use tokio::sync::Mutex;
use tracing::debug;
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ServerState {
Uninitialized,
Ready,
Closed,
}
pub struct McpServer {
state: Arc<Mutex<ServerState>>,
connection_pool: ConnectionPool,
}
impl McpServer {
pub fn new() -> Self {
Self {
state: Arc::new(Mutex::new(ServerState::Uninitialized)),
connection_pool: ConnectionPool::new(),
}
}
pub fn connection_pool(&self) -> &ConnectionPool {
&self.connection_pool
}
pub async fn state(&self) -> ServerState {
*self.state.lock().await
}
pub async fn is_ready(&self) -> bool {
self.state().await == ServerState::Ready
}
#[allow(dead_code)]
pub(crate) async fn set_state(&self, new_state: ServerState) {
*self.state.lock().await = new_state;
}
pub async fn handle_initialize(&self, params: InitializeParams) -> McpResult<InitializeResult> {
if !SUPPORTED_PROTOCOL_VERSIONS.contains(¶ms.protocol_version.as_str()) {
return Err(McpError::UnsupportedProtocol(params.protocol_version));
}
self.set_state(ServerState::Ready).await;
tools::init_connection_pool(self.connection_pool.clone());
crate::architecture::init_session_store();
let capabilities = ServerCapabilities {
tools: Some(ToolsCapability {
list_changed: Some(false), }),
resources: Some(ResourcesCapability {
subscribe: Some(false),
}),
prompts: Some(PromptsCapability {
list_changed: Some(false), }),
};
let server_info = ServerInfo {
name: "aimdb-mcp".to_string(),
version: env!("CARGO_PKG_VERSION").to_string(),
metadata: Some(json!({
"prompts_available": ["schema-help", "troubleshooting"],
})),
};
Ok(InitializeResult {
protocol_version: MCP_PROTOCOL_VERSION.to_string(),
capabilities,
server_info,
})
}
pub async fn handle_tools_list(&self) -> McpResult<ToolsListResult> {
if !self.is_ready().await {
return Err(McpError::NotInitialized);
}
debug!("📋 Listing available tools");
let tools = vec![
Tool {
name: "discover_instances".to_string(),
description: "Discover all running AimDB instances on the system. Scans /tmp/*.sock and /var/run/aimdb/*.sock for AimDB servers.".to_string(),
input_schema: json!({
"type": "object",
"properties": {},
"additionalProperties": false
}),
},
Tool {
name: "list_records".to_string(),
description: "List all records from a specific AimDB instance. Returns metadata including buffer type, capacity, producer/consumer counts, and timestamps.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
}
},
"required": [],
"additionalProperties": false
}),
},
Tool {
name: "get_record".to_string(),
description: "Get the current value of a specific record from an AimDB instance. Returns the record's current JSON value.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
},
"record_name": {
"type": "string",
"description": "Name of the record to retrieve (e.g., server::Temperature)"
}
},
"required": ["record_name"],
"additionalProperties": false
}),
},
Tool {
name: "set_record".to_string(),
description: "Set the value of a writable record in an AimDB instance. Only works for records with write permissions.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
},
"record_name": {
"type": "string",
"description": "Name of the record to update (must be writable)"
},
"value": {
"description": "New value for the record (must match record's type schema)"
}
},
"required": ["record_name", "value"],
"additionalProperties": false
}),
},
Tool {
name: "get_instance_info".to_string(),
description: "Get detailed information about a specific AimDB instance. Returns server version, protocol, permissions, and capabilities.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
}
},
"required": [],
"additionalProperties": false
}),
},
Tool {
name: "query_schema".to_string(),
description: "Get JSON schema and type information for a record.\n\n\
Returns the data structure, field types, and metadata.\n\
Use this before setting record values to understand expected format.\n\n\
Schema is inferred from current value + database metadata.\n\n\
💡 TIP: Field names like 'celsius', 'timestamp', 'sensor_id' carry semantic meaning.\n\
If units or formats are unclear, ask the user for clarification.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
},
"record_name": {
"type": "string",
"description": "Name of the record to query schema for (e.g., server::Temperature)"
},
"include_example": {
"type": "boolean",
"description": "Include current value as example (default: true)",
"default": true
}
},
"required": ["record_name"],
"additionalProperties": false
}),
},
Tool {
name: "drain_record".to_string(),
description: "Drain all pending values from a record since the last drain call. \
Returns values in chronological order. This is a destructive read — \
drained values won't be returned again. Use this for batch analysis \
of accumulated data (e.g., time-series analysis, trend detection). \
The first drain call creates a reader and returns empty (cold start). \
Subsequent calls return all values accumulated since the previous drain.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
},
"record_name": {
"type": "string",
"description": "Name of the record to drain (e.g., temp.berlin)"
},
"limit": {
"type": "integer",
"description": "Maximum number of values to drain. Optional, defaults to all pending.",
"minimum": 1
}
},
"required": ["record_name"],
"additionalProperties": false
}),
},
Tool {
name: "graph_nodes".to_string(),
description: "Get all nodes in the dependency graph. Returns metadata for all records as graph nodes, including origin (source/link/transform/passive), buffer configuration, and connection counts. Useful for understanding database topology and data flow.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
}
},
"required": [],
"additionalProperties": false
}),
},
Tool {
name: "graph_edges".to_string(),
description: "Get all edges in the dependency graph. Returns directed edges representing data flow between records. Shows how data flows from sources through transforms to consumers.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
}
},
"required": [],
"additionalProperties": false
}),
},
Tool {
name: "graph_topo_order".to_string(),
description: "Get the topological ordering of records in the dependency graph. Returns record keys ordered so all dependencies appear before their dependents. Reflects the spawn/initialization order used by AimDB.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
}
},
"required": [],
"additionalProperties": false
}),
},
Tool {
name: "get_architecture".to_string(),
description: "Return the current architecture state from .aimdb/state.toml as structured JSON, including record count, validation summary, and decision log length. Run this first when entering an architecture session.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"state_path": {
"type": "string",
"description": "Path to state.toml (default: .aimdb/state.toml)"
}
},
"additionalProperties": false
}),
},
Tool {
name: "propose_add_record".to_string(),
description: "Propose adding a new record to the architecture. All payload fields are explicit and typed — no guessing required. Present the proposal to the user before calling resolve_proposal.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"name": {
"type": "string",
"description": "PascalCase record name, e.g. \"TemperatureReading\""
},
"description": {
"type": "string",
"description": "Human-readable description of the proposal shown to the user"
},
"buffer": {
"type": "string",
"enum": ["SpmcRing", "SingleLatest", "Mailbox"],
"description": "Buffer semantics: SpmcRing=stream (every value), SingleLatest=state (newest only), Mailbox=command (overwrite)"
},
"capacity": {
"type": "integer",
"description": "Ring buffer capacity — required when buffer=SpmcRing. Use power-of-2, e.g. 256, 512, 1024."
},
"key_prefix": {
"type": "string",
"description": "Optional common key prefix, e.g. \"sensors.temp.\". Default: \"\""
},
"key_variants": {
"type": "array",
"items": { "type": "string" },
"description": "Concrete PascalCase variant names, e.g. [\"Default\"] or [\"Indoor\", \"Outdoor\"]. Default: []"
},
"producers": {
"type": "array",
"items": { "type": "string" },
"description": "Task names that write to this record, e.g. [\"sensor_task\"]."
},
"consumers": {
"type": "array",
"items": { "type": "string" },
"description": "Task names that read from this record, e.g. [\"anomaly_detector\"]."
},
"fields": {
"type": "array",
"description": "Value struct fields",
"items": {
"type": "object",
"properties": {
"name": { "type": "string", "description": "snake_case field name" },
"type": { "type": "string", "description": "Rust primitive: f64, f32, u8, u16, u32, u64, i8, i16, i32, i64, bool, String" },
"description": { "type": "string" }
},
"required": ["name", "type", "description"]
}
},
"connectors": {
"type": "array",
"description": "Connector wiring (MQTT, KNX, etc.)",
"items": {
"type": "object",
"properties": {
"protocol": { "type": "string", "description": "e.g. mqtt, knx" },
"direction": { "type": "string", "enum": ["inbound", "outbound"] },
"url": { "type": "string", "description": "Topic/address template; may contain {variant}" }
},
"required": ["protocol", "direction", "url"]
}
}
},
"required": ["name", "description", "buffer"],
"additionalProperties": false
}),
},
Tool {
name: "propose_modify_buffer".to_string(),
description: "Propose changing the buffer type (and optionally capacity) of an existing record. Present the proposal to the user before calling resolve_proposal.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"record_name": {
"type": "string",
"description": "PascalCase name of the existing record to modify"
},
"description": {
"type": "string",
"description": "Human-readable description of the proposal shown to the user"
},
"buffer": {
"type": "string",
"enum": ["SpmcRing", "SingleLatest", "Mailbox"],
"description": "New buffer type"
},
"capacity": {
"type": "integer",
"description": "Ring capacity — required when buffer=SpmcRing"
}
},
"required": ["record_name", "description", "buffer"],
"additionalProperties": false
}),
},
Tool {
name: "propose_add_connector".to_string(),
description: "Propose adding a connector (MQTT, KNX, etc.) to an existing record. Present the proposal to the user before calling resolve_proposal.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"record_name": {
"type": "string",
"description": "PascalCase name of the existing record to wire up"
},
"description": {
"type": "string",
"description": "Human-readable description of the proposal shown to the user"
},
"protocol": {
"type": "string",
"description": "Connector protocol identifier, e.g. \"mqtt\" or \"knx\""
},
"direction": {
"type": "string",
"enum": ["inbound", "outbound"],
"description": "inbound = broker→DB, outbound = DB→broker"
},
"url": {
"type": "string",
"description": "Topic or address template; use {variant} placeholder for key variants, e.g. \"sensors/temp/{variant}\""
}
},
"required": ["record_name", "description", "protocol", "direction", "url"],
"additionalProperties": false
}),
},
Tool {
name: "propose_modify_fields".to_string(),
description: "Propose replacing the value struct fields of an existing record. This replaces ALL fields — include unchanged fields too. Present the proposal to the user before calling resolve_proposal.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"record_name": {
"type": "string",
"description": "PascalCase name of the existing record to modify"
},
"description": {
"type": "string",
"description": "Human-readable description of the proposal shown to the user"
},
"fields": {
"type": "array",
"description": "Complete replacement field list for the value struct",
"items": {
"type": "object",
"properties": {
"name": { "type": "string", "description": "snake_case field name" },
"type": { "type": "string", "description": "f64, f32, u8, u16, u32, u64, i8, i16, i32, i64, bool, String" },
"description": { "type": "string" }
},
"required": ["name", "type", "description"]
}
}
},
"required": ["record_name", "description", "fields"],
"additionalProperties": false
}),
},
Tool {
name: "propose_modify_key_variants".to_string(),
description: "Propose updating the key variants of an existing record. Use this when adding a record with no variants (e.g. [\"Default\"]) or expanding a fleet (e.g. adding a new device). Present the proposal to the user before calling resolve_proposal.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"record_name": {
"type": "string",
"description": "PascalCase name of the existing record to modify"
},
"description": {
"type": "string",
"description": "Human-readable description of the proposal shown to the user"
},
"key_variants": {
"type": "array",
"items": { "type": "string" },
"description": "Complete replacement list of PascalCase variant names, e.g. [\"Default\"] or [\"ApiServer\", \"Worker\", \"Db\"]. Replaces prior variant list."
},
"key_prefix": {
"type": "string",
"description": "Optional common key prefix. If omitted the existing prefix is preserved."
}
},
"required": ["record_name", "description", "key_variants"],
"additionalProperties": false
}),
},
Tool {
name: "propose_add_task".to_string(),
description: "Propose adding a new task definition. Tasks are async functions that produce, transform, or consume record data. Present the proposal to the user before calling resolve_proposal.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"name": {
"type": "string",
"description": "snake_case task function name, e.g. \"sensor_polling_task\""
},
"description": {
"type": "string",
"description": "Human-readable description of the proposal shown to the user"
},
"task_type": {
"type": "string",
"enum": ["transform", "agent", "source", "tap"],
"description": "Functional role: source (autonomous producer writing to a record), transform (reactive derivation from input records to output record), tap (read-only observer, no output records), agent (LLM reasoning loop). Default: transform"
},
"inputs": {
"type": "array",
"items": {
"type": "object",
"properties": {
"record": { "type": "string", "description": "PascalCase record name to read from" },
"variants": { "type": "array", "items": { "type": "string" }, "description": "Specific variants to consume (empty = all)" }
},
"required": ["record"]
},
"description": "Records this task reads from"
},
"outputs": {
"type": "array",
"items": {
"type": "object",
"properties": {
"record": { "type": "string", "description": "PascalCase record name to write to" },
"variants": { "type": "array", "items": { "type": "string" }, "description": "Specific variants to produce (empty = all)" }
},
"required": ["record"]
},
"description": "Records this task writes to"
}
},
"required": ["name", "description"],
"additionalProperties": false
}),
},
Tool {
name: "propose_add_binary".to_string(),
description: "Propose adding a new binary definition. Binaries are deployable crates that group tasks together and optionally declare external broker connections. Present the proposal to the user before calling resolve_proposal.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"name": {
"type": "string",
"description": "Crate directory name, e.g. \"weather-sentinel-hub\""
},
"description": {
"type": "string",
"description": "Human-readable description of the proposal shown to the user"
},
"tasks": {
"type": "array",
"items": { "type": "string" },
"description": "Task names belonging to this binary (must match [[tasks]] entries)"
},
"external_connectors": {
"type": "array",
"items": {
"type": "object",
"properties": {
"protocol": { "type": "string", "description": "Protocol identifier, e.g. \"mqtt\"" },
"env_var": { "type": "string", "description": "Environment variable for the broker URL" },
"default": { "type": "string", "description": "Default URL when env var is not set" }
},
"required": ["protocol", "env_var"]
},
"description": "Runtime broker connections needed by this binary"
}
},
"required": ["name", "description"],
"additionalProperties": false
}),
},
Tool {
name: "remove_task".to_string(),
description: "Propose removal of an existing task. Creates a pending proposal — call resolve_proposal to confirm. Note: removing a task affects binaries that reference it.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"task_name": {
"type": "string",
"description": "snake_case name of the task to remove"
}
},
"required": ["task_name"],
"additionalProperties": false
}),
},
Tool {
name: "remove_binary".to_string(),
description: "Propose removal of an existing binary. Creates a pending proposal — call resolve_proposal to confirm. Task definitions are preserved; only the binary grouping is removed.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"binary_name": {
"type": "string",
"description": "Name of the binary crate to remove"
}
},
"required": ["binary_name"],
"additionalProperties": false
}),
},
Tool {
name: "resolve_proposal".to_string(),
description: "Resolve a pending proposal. On confirm: applies the change, writes state.toml, generates Mermaid and Rust artefacts. On reject: discards without changes. On revise: discards with a redirect message.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"proposal_id": {
"type": "string",
"description": "The proposal ID returned by any propose_* tool, remove_record, rename_record, remove_task, or remove_binary"
},
"resolution": {
"type": "string",
"enum": ["confirm", "reject", "revise"],
"description": "User decision: confirm applies the change, reject discards it, revise returns a redirect"
},
"redirect": {
"type": "string",
"description": "Message explaining what to revise (only used when resolution=revise)"
},
"state_path": { "type": "string", "description": "Override state.toml path" },
"mermaid_path": { "type": "string", "description": "Override Mermaid output path" },
"rust_path": { "type": "string", "description": "Override Rust output path" }
},
"required": ["proposal_id", "resolution"],
"additionalProperties": false
}),
},
Tool {
name: "remove_record".to_string(),
description: "Propose removal of an existing record. Creates a pending proposal — call resolve_proposal to confirm. Note: removing a record breaks generated type aliases.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"record_name": {
"type": "string",
"description": "PascalCase name of the record to remove"
}
},
"required": ["record_name"],
"additionalProperties": false
}),
},
Tool {
name: "rename_record".to_string(),
description: "Propose renaming a record. Creates a pending proposal — call resolve_proposal to confirm. Note: renames the generated key enum and value struct, breaking existing references.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"old_name": {
"type": "string",
"description": "Current PascalCase record name"
},
"new_name": {
"type": "string",
"description": "New PascalCase record name"
}
},
"required": ["old_name", "new_name"],
"additionalProperties": false
}),
},
Tool {
name: "validate_against_instance".to_string(),
description: "Compare state.toml against a live AimDB instance and return a conflict report. Detects missing records, buffer type mismatches, capacity differences, and connector mismatches.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
},
"state_path": {
"type": "string",
"description": "Path to state.toml (default: .aimdb/state.toml)"
}
},
"required": [],
"additionalProperties": false
}),
},
Tool {
name: "get_buffer_metrics".to_string(),
description: "Get live buffer metrics for records matching a key string from a running AimDB instance.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"socket_path": {
"type": "string",
"description": "Unix socket path to the AimDB instance. Falls back to AIMDB_SOCKET env var if omitted."
},
"record_key": {
"type": "string",
"description": "Substring to match against record names (e.g., 'Temperature')"
}
},
"required": ["record_key"],
"additionalProperties": false
}),
},
Tool {
name: "save_memory".to_string(),
description: "Persist ideation context and design rationale to .aimdb/memory.md. \
Call this after every confirmed proposal with a narrative summary of what the user is building, \
the key question asked, the answer received, why the chosen buffer type fits, \
alternatives that were considered and rejected, and any future considerations noted. \
On session start, read aimdb://architecture/memory to restore this context.".to_string(),
input_schema: json!({
"type": "object",
"properties": {
"entry": {
"type": "string",
"description": "Markdown text to write. For append mode, structure as a '## RecordName' section with sub-headings: Context, Key question, Answer, Buffer choice & rationale, Alternatives considered, Future considerations."
},
"mode": {
"type": "string",
"enum": ["append", "overwrite"],
"description": "append (default): add a timestamped section to memory.md. overwrite: replace the entire file (use only to correct the whole document)."
},
"memory_path": {
"type": "string",
"description": "Override path (default: .aimdb/memory.md)"
}
},
"required": ["entry"],
"additionalProperties": false
}),
},
Tool {
name: "reset_session".to_string(),
description: "Reset the architecture agent session, discarding any pending proposals. Use when the user wants to start over or abandon the current ideation cycle.".to_string(),
input_schema: json!({
"type": "object",
"properties": {},
"additionalProperties": false
}),
},
];
Ok(ToolsListResult { tools })
}
pub async fn handle_tools_call(&self, params: ToolCallParams) -> McpResult<ToolCallResult> {
if !self.is_ready().await {
return Err(McpError::NotInitialized);
}
debug!("🛠️ Calling tool: {}", params.name);
let result = match params.name.as_str() {
"discover_instances" => tools::discover_instances(params.arguments).await?,
"list_records" => tools::list_records(params.arguments).await?,
"get_record" => tools::get_record(params.arguments).await?,
"set_record" => tools::set_record(params.arguments).await?,
"get_instance_info" => tools::get_instance_info(params.arguments).await?,
"query_schema" => tools::query_schema(params.arguments).await?,
"drain_record" => tools::drain_record(params.arguments).await?,
"graph_nodes" => tools::graph_nodes(params.arguments).await?,
"graph_edges" => tools::graph_edges(params.arguments).await?,
"graph_topo_order" => tools::graph_topo_order(params.arguments).await?,
"get_architecture" => tools::get_architecture(params.arguments).await?,
"propose_add_record" => tools::propose_add_record(params.arguments).await?,
"propose_modify_buffer" => tools::propose_modify_buffer(params.arguments).await?,
"propose_add_connector" => tools::propose_add_connector(params.arguments).await?,
"propose_modify_fields" => tools::propose_modify_fields(params.arguments).await?,
"propose_modify_key_variants" => {
tools::propose_modify_key_variants(params.arguments).await?
}
"propose_add_task" => tools::propose_add_task(params.arguments).await?,
"propose_add_binary" => tools::propose_add_binary(params.arguments).await?,
"resolve_proposal" => tools::resolve_proposal(params.arguments).await?,
"remove_record" => tools::remove_record(params.arguments).await?,
"rename_record" => tools::rename_record(params.arguments).await?,
"remove_task" => tools::remove_task(params.arguments).await?,
"remove_binary" => tools::remove_binary(params.arguments).await?,
"validate_against_instance" => {
tools::validate_against_instance(params.arguments).await?
}
"get_buffer_metrics" => tools::get_buffer_metrics(params.arguments).await?,
"save_memory" => tools::save_memory(params.arguments).await?,
"reset_session" => tools::reset_session(params.arguments).await?,
_ => {
return Err(McpError::MethodNotFound(format!(
"Unknown tool: {}",
params.name
)));
}
};
let content = vec![ToolContent::Text {
text: serde_json::to_string_pretty(&result)?,
}];
Ok(ToolCallResult {
content,
is_error: Some(false),
})
}
pub async fn handle_resources_list(&self) -> McpResult<ResourcesListResult> {
if !self.is_ready().await {
return Err(McpError::NotInitialized);
}
debug!("📋 Handling resources/list");
resources::list_resources().await
}
pub async fn handle_resources_read(
&self,
params: ResourceReadParams,
) -> McpResult<ResourceReadResult> {
if !self.is_ready().await {
return Err(McpError::NotInitialized);
}
debug!("📖 Handling resources/read: {}", params.uri);
resources::read_resource(¶ms.uri).await
}
pub async fn handle_prompts_list(&self) -> McpResult<PromptsListResult> {
if !self.is_ready().await {
return Err(McpError::NotInitialized);
}
debug!("📋 Listing available prompts");
let prompts = prompts::list_prompts();
Ok(PromptsListResult { prompts })
}
pub async fn handle_prompts_get(
&self,
params: PromptsGetParams,
) -> McpResult<PromptsGetResult> {
if !self.is_ready().await {
return Err(McpError::NotInitialized);
}
debug!("📝 Getting prompt: {}", params.name);
let messages = prompts::get_prompt(¶ms.name)
.ok_or_else(|| McpError::InvalidParams(format!("Unknown prompt: {}", params.name)))?;
Ok(PromptsGetResult {
description: Some(format!("Prompt: {}", params.name)),
messages,
})
}
}
impl Default for McpServer {
fn default() -> Self {
Self::new()
}
}