systemprompt-mcp 0.1.19

Core MCP (Model Context Protocol) functionality for systemprompt.io OS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use anyhow::Result;
use std::sync::Arc;
use tracing::Instrument;

use super::event_bus::EventBus;
use super::events::McpEvent;
use crate::services::database::DatabaseManager;
use crate::services::lifecycle::LifecycleManager;
use crate::services::registry::RegistryManager;

pub async fn run_daemon(
    event_bus: &Arc<EventBus>,
    lifecycle: &LifecycleManager,
    database: &DatabaseManager,
) -> Result<()> {
    let span: tracing::Span = systemprompt_logging::SystemSpan::new("mcp_orchestrator").into();
    async move {
        tracing::info!("Starting MCP daemon mode");

        database.cleanup_stale_services().await?;
        let servers = RegistryManager::get_enabled_servers()?;
        database.sync_state(&servers).await?;
        let server_count = servers.len();

        tracing::info!(
            mode = "daemon",
            enabled_services = server_count,
            services = ?servers.iter().map(|s| s.name.clone()).collect::<Vec<_>>(),
            "MCP daemon started"
        );

        for server in &servers {
            event_bus
                .publish(McpEvent::ServiceStartRequested {
                    service_name: server.name.clone(),
                })
                .await?;
        }

        tokio::time::sleep(tokio::time::Duration::from_secs(2)).await;

        tracing::info!("All MCP servers started with proper OAuth enforcement");
        tracing::info!("MCP manager will keep servers running. Press Ctrl+C to stop.");

        tokio::signal::ctrl_c().await?;
        tracing::info!("Shutting down MCP servers");

        let running_servers = database.get_running_servers().await?;
        let running_count = running_servers.len();

        tracing::info!(
            running_services = running_count,
            shutdown_reason = "user_interrupt",
            "MCP daemon shutdown initiated"
        );

        for server in running_servers {
            lifecycle.stop_server(&server).await?;

            event_bus
                .publish(McpEvent::ServiceStopped {
                    service_name: server.name,
                    exit_code: None,
                })
                .await?;
        }

        tracing::info!("MCP daemon shutdown completed");

        Ok(())
    }
    .instrument(span)
    .await
}