agd 0.1.0

Local-first CLI and stdio MCP server for org-mode files
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
75
76
77
78
79

mod resources;
mod tools;

use std::path::Path;

use anyhow::{Context, Result};
use clap::Args;
use org_core::{Settings, load_settings};
use rmcp::{ServiceExt, handler::server::tool::ToolRouter, transport::stdio};
use tokio::runtime::Builder;
use tracing::{error, info};
use tracing_subscriber::EnvFilter;

#[derive(Args, Debug)]
pub(crate) struct McpCommand {
    /// Log level (trace, debug, info, warn, error)
    #[arg(long)]
    pub(crate) log_level: Option<String>,
}

pub(crate) struct OrgModeRouter {
    pub(crate) settings: Settings,
    pub(crate) project: String,
}

impl McpCommand {
    pub(crate) fn execute(
        self,
        config: Option<String>,
        root_directory: Option<String>,
    ) -> Result<()> {
        let project = root_directory.clone().unwrap_or_else(|| {
            std::env::current_dir()
                .map(|path| path.to_string_lossy().into_owned())
                .unwrap_or_else(|_| "unknown".to_string())
        });
        let settings = load_settings(
            config.as_deref().map(Path::new),
            root_directory.as_deref().map(Path::new),
        )?
        .settings;
        let log_level = self
            .log_level
            .as_deref()
            .unwrap_or("info")
            .parse::<tracing::Level>()
            .unwrap_or(tracing::Level::INFO);
        tracing_subscriber::fmt()
            .with_env_filter(EnvFilter::from_default_env().add_directive(log_level.into()))
            .with_writer(std::io::stderr)
            .try_init()
            .map_err(|error| anyhow::anyhow!("failed to initialize MCP logging: {error}"))?;
        info!(roots = ?settings.roots, "starting stdio MCP server");
        Builder::new_multi_thread()
            .enable_all()
            .build()
            .context("failed to create MCP runtime")?
            .block_on(serve(settings, project))
    }
}

async fn serve(settings: Settings, project: String) -> Result<()> {
    let service = OrgModeRouter { settings, project }
        .serve(stdio())
        .await
        .inspect_err(|error| error!("error starting MCP server: {error}"))
        .context("failed to start MCP server")?;
    service
        .waiting()
        .await
        .context("MCP server stopped with an error")?;
    Ok(())
}

impl OrgModeRouter {
    pub(crate) fn tool_router() -> ToolRouter<Self> {
        Self::tool_router_native()
    }
}