crates-docs 0.9.0

High-performance Rust crate documentation query MCP server, supports Stdio/HTTP/SSE transport and OAuth authentication
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

//! CLI module
//!
//! Command-line interface for the Crates Docs MCP Server.

mod api_key_cmd;
mod commands;
mod config_cmd;
mod health_cmd;
mod list_api_keys_cmd;
mod revoke_api_key_cmd;
mod serve_cmd;
mod test_cmd;
mod version_cmd;

use clap::Parser;
use std::path::PathBuf;

pub use api_key_cmd::run_generate_api_key_command;
pub use commands::Commands;
pub use config_cmd::run_config_command;
pub use health_cmd::run_health_command;
pub use list_api_keys_cmd::run_list_api_keys_command;
pub use revoke_api_key_cmd::run_revoke_api_key_command;
pub use serve_cmd::run_serve_command;
pub use test_cmd::run_test_command;
pub use version_cmd::run_version_command;

/// CLI configuration
#[derive(Parser)]
#[command(name = "crates-docs")]
#[command(version = env!("CARGO_PKG_VERSION"))]
#[command(about = "High-performance Rust crate documentation query MCP server", long_about = None)]
pub struct Cli {
    /// CLI command to execute
    #[command(subcommand)]
    pub command: Commands,

    /// Configuration file path
    #[arg(short, long, global = true, default_value = "config.toml")]
    pub config: PathBuf,

    /// Enable debug logging
    #[arg(short, long, global = true)]
    pub debug: bool,

    /// Enable verbose output
    #[arg(short, long, global = true)]
    pub verbose: bool,
}

/// Run the CLI application
pub async fn run(cli: Cli) -> Result<(), Box<dyn std::error::Error>> {
    match cli.command {
        Commands::Serve {
            mode,
            host,
            port,
            enable_oauth,
            oauth_client_id,
            oauth_client_secret,
            oauth_redirect_uri,
            enable_api_key,
            api_keys,
            api_key_header,
            api_key_query_param,
        } => {
            run_serve_command(
                &cli.config,
                cli.debug,
                mode,
                host,
                port,
                enable_oauth,
                oauth_client_id,
                oauth_client_secret,
                oauth_redirect_uri,
                enable_api_key,
                api_keys,
                api_key_header,
                api_key_query_param,
            )
            .await?;
        }
        Commands::GenerateApiKey { prefix } => {
            run_generate_api_key_command(&prefix)?;
        }
        Commands::ListApiKeys { config } => {
            run_list_api_keys_command(&config)?;
        }
        Commands::RevokeApiKey { config, key } => {
            run_revoke_api_key_command(&config, &key)?;
        }
        Commands::Config { output, force } => {
            run_config_command(&output, force)?;
        }
        Commands::Test {
            tool,
            crate_name,
            item_path,
            query,
            sort,
            version,
            limit,
            format,
        } => {
            run_test_command(
                &tool,
                crate_name.as_deref(),
                item_path.as_deref(),
                query.as_deref(),
                sort.as_deref(),
                version.as_deref(),
                limit,
                &format,
            )
            .await?;
        }
        Commands::Health {
            check_type,
            verbose,
        } => {
            run_health_command(&check_type, verbose).await?;
        }
        Commands::Version => {
            run_version_command();
        }
    }

    Ok(())
}