scraps 5.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

use crate::ir::loader;
use crate::usecase::scrap_type::list::usecase::ListTypeUsecase;
use crate::usecase::tag::list::usecase::ListTagUsecase;
use rmcp::model::ErrorCode;
use rmcp::model::{CallToolResult, ContentBlock};
use rmcp::service::RequestContext;
use rmcp::{ErrorData, RoleServer};
use serde_json::json;
use std::path::Path;

// Head of the topic map only; the full list stays list_tags' job.
const TOP_TAGS_LIMIT: usize = 10;
// Same split for the kinds of scraps; the full list stays list_types' job.
const TOP_TYPES_LIMIT: usize = 10;

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

    let mut contexts: Vec<String> = scraps
        .iter()
        .filter_map(|s| s.ctx().as_ref().map(|c| c.to_string()))
        .collect();
    contexts.sort();
    contexts.dedup();

    // Create tag usecase
    let tag_usecase = ListTagUsecase::new();

    let (tags, backlinks_map) = tag_usecase
        .execute(&scraps)
        .map_err(|e| ErrorData::new(ErrorCode(-32004), format!("List tags failed: {e}"), None))?;

    let tag_count = tags.len();
    let mut ranked: Vec<(String, usize)> = tags
        .into_iter()
        .map(|tag| {
            let backlinks_count = backlinks_map.get_tag(&tag).len();
            (tag.to_string(), backlinks_count)
        })
        .collect();
    // Tie-break by name so the head of the topic map is stable.
    ranked.sort_by(|a, b| b.1.cmp(&a.1).then_with(|| a.0.cmp(&b.0)));
    let top_tags: Vec<_> = ranked
        .into_iter()
        .take(TOP_TAGS_LIMIT)
        .map(|(title, backlinks_count)| {
            json!({
                "title": title,
                "backlinks_count": backlinks_count
            })
        })
        .collect();

    let types = ListTypeUsecase::new()
        .execute(&scraps)
        .map_err(|e| ErrorData::new(ErrorCode(-32004), format!("List types failed: {e}"), None))?;

    let type_count = types.len();
    let top_types: Vec<_> = types
        .into_iter()
        .take(TOP_TYPES_LIMIT)
        .map(|result| {
            json!({
                "type": result.name,
                "scraps_count": result.scraps_count
            })
        })
        .collect();

    let scrap_count = scraps.len();
    // An untyped wiki keeps the tag-only hint, so types never clutter it.
    let next = if scrap_count == 0 {
        "The wiki has no scraps yet."
    } else if type_count == 0 {
        "Search content with search_scraps, or expand a top tag with lookup_tag_backlinks {tag}; the full tag list is list_tags."
    } else {
        "Search content with search_scraps, expand a top tag with lookup_tag_backlinks {tag}, or expand a top type with lookup_type_scraps {type}; the full lists are list_tags and list_types."
    };

    let response = json!({
        "scrap_count": scrap_count,
        "tag_count": tag_count,
        "type_count": type_count,
        "contexts": contexts,
        "top_tags": top_tags,
        "top_types": top_types,
        "next": next,
    });

    Ok(CallToolResult::success(vec![ContentBlock::text(
        serde_json::to_string(&response).unwrap(),
    )]))
}