appshots-mcp 0.2.0

MCP server for generating ASO-optimized App Store screenshots
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

use std::path::PathBuf;
use std::process::ExitCode;
use std::sync::Arc;

use clap::Parser;
use rmcp::ServiceExt;
use tracing_subscriber::EnvFilter;

#[derive(Parser)]
#[command(
    name = "appshots-mcp",
    about = "MCP server for ASO-optimized App Store screenshots",
    version
)]
struct Cli {
    /// Path to the target app project directory
    #[arg(long, default_value = ".")]
    project_dir: PathBuf,

    /// Path to the shared glossary file
    #[arg(long, default_value = "glossary.json")]
    glossary_path: PathBuf,

    /// Path to the appshots project config
    #[arg(long, default_value = "appshots.json")]
    config_path: PathBuf,
}

#[tokio::main(flavor = "current_thread")]
async fn main() -> ExitCode {
    let cli = Cli::parse();

    tracing_subscriber::fmt()
        .with_env_filter(EnvFilter::from_default_env())
        .with_writer(std::io::stderr)
        .init();

    let store = Arc::new(
        appshots_mcp::io::fs::FsFileStore::new().with_project_dir(cli.project_dir.clone()),
    );
    let server = appshots_mcp::server::AppShotsMcpServer::new(
        store,
        cli.project_dir,
        cli.glossary_path,
        cli.config_path,
    );

    let transport = rmcp::transport::io::stdio();

    match server.serve(transport).await {
        Ok(service) => {
            if let Err(e) = service.waiting().await {
                eprintln!("error: {e}");
                return ExitCode::FAILURE;
            }
            ExitCode::SUCCESS
        }
        Err(e) => {
            eprintln!("error: {e}");
            ExitCode::FAILURE
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn cli_default_values() {
        let cli = Cli::try_parse_from(["appshots-mcp"]).unwrap();
        assert_eq!(cli.project_dir, PathBuf::from("."));
        assert_eq!(cli.glossary_path, PathBuf::from("glossary.json"));
        assert_eq!(cli.config_path, PathBuf::from("appshots.json"));
    }

    #[test]
    fn cli_custom_values() {
        let cli = Cli::try_parse_from([
            "appshots-mcp",
            "--project-dir",
            "/my/project",
            "--glossary-path",
            "/my/glossary.json",
            "--config-path",
            "/my/config.json",
        ])
        .unwrap();
        assert_eq!(cli.project_dir, PathBuf::from("/my/project"));
        assert_eq!(cli.glossary_path, PathBuf::from("/my/glossary.json"));
        assert_eq!(cli.config_path, PathBuf::from("/my/config.json"));
    }

    #[test]
    fn cli_rejects_unknown_flag() {
        let result = Cli::try_parse_from(["appshots-mcp", "--unknown"]);
        assert!(result.is_err());
    }

    #[test]
    fn cli_version_flag() {
        let result = Cli::try_parse_from(["appshots-mcp", "--version"]);
        // --version causes clap to exit with an error containing the version
        assert!(result.is_err());
    }
}