scraps 1.0.0

Scraps is a portable CLI knowledge hub for managing interconnected Markdown documentation with Wiki-link notation.
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

use crate::input::file::read_scraps;
use crate::usecase::scrap::lookup_links::usecase::{LinkRefKind, LookupScrapLinksUsecase};
use rmcp::handler::server::wrapper::Parameters;
use rmcp::model::ErrorCode;
use rmcp::model::{CallToolResult, Content};
use rmcp::schemars::JsonSchema;
use rmcp::service::RequestContext;
use rmcp::{ErrorData, RoleServer};
use serde::{Deserialize, Serialize};
use std::path::Path;

#[derive(Debug, Deserialize, JsonSchema)]
#[schemars(deny_unknown_fields)]
pub struct LookupScrapLinksRequest {
    /// Title of the scrap to get links for
    pub title: String,
    /// Optional context if the scrap has one
    pub ctx: Option<String>,
}

#[derive(Debug, Serialize, PartialEq, Eq, Clone, Copy)]
#[serde(rename_all = "lowercase")]
pub enum LinkRefKindJson {
    Link,
    Embed,
}

impl From<LinkRefKind> for LinkRefKindJson {
    fn from(k: LinkRefKind) -> Self {
        match k {
            LinkRefKind::Link => LinkRefKindJson::Link,
            LinkRefKind::Embed => LinkRefKindJson::Embed,
        }
    }
}

#[derive(Debug, Serialize)]
pub struct LinkRefJson {
    pub kind: LinkRefKindJson,
    pub title: String,
    pub ctx: Option<String>,
    pub heading: Option<String>,
}

#[derive(Debug, Serialize)]
pub struct LookupScrapLinksResponse {
    pub results: Vec<LinkRefJson>,
    pub count: usize,
}

pub async fn lookup_scrap_links(
    scraps_dir: &Path,
    exclude_dirs: &[std::path::PathBuf],
    _context: RequestContext<RoleServer>,
    Parameters(request): Parameters<LookupScrapLinksRequest>,
) -> Result<CallToolResult, ErrorData> {
    // Load scraps from directory
    let scraps = read_scraps::to_all_scraps(scraps_dir, exclude_dirs).map_err(|e| {
        ErrorData::new(
            ErrorCode(-32003),
            format!("Failed to load scraps: {e}"),
            None,
        )
    })?;

    // Create get scrap links usecase
    let get_links_usecase = LookupScrapLinksUsecase::new();

    // Execute get links
    let title = scraps_libs::model::title::Title::from(request.title.as_str());
    let ctx = request
        .ctx
        .as_ref()
        .map(|c| scraps_libs::model::context::Ctx::from(c.as_str()));

    let results = get_links_usecase
        .execute(&scraps, &title, &ctx)
        .map_err(|e| {
            ErrorData::new(
                ErrorCode(-32004),
                format!("Get scrap links failed: {e}"),
                None,
            )
        })?;

    // Convert results to structured response
    let refs: Vec<LinkRefJson> = results
        .into_iter()
        .map(|result| LinkRefJson {
            kind: result.kind.into(),
            title: result.title.to_string(),
            ctx: result.ctx.map(|c| c.to_string()),
            heading: result.heading,
        })
        .collect();

    let count = refs.len();
    let response = LookupScrapLinksResponse {
        results: refs,
        count,
    };

    Ok(CallToolResult::success(vec![Content::text(
        serde_json::to_string(&response).map_err(|e| {
            ErrorData::new(
                ErrorCode(-32005),
                format!("JSON serialization failed: {e}"),
                None,
            )
        })?,
    )]))
}