gobby-wiki 0.2.0

Gobby wiki CLI shell
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

use std::path::PathBuf;

use serde_json::json;

use crate::graph::{LinkSuggestion, WikiBacklink};
use crate::support::graph::memory_graph_from_store;
use crate::support::scope::indexed_store_for_selection;
use crate::{CommandOutcome, ScopeIdentity, ScopeSelection, WikiError};

pub(crate) fn execute(
    page: String,
    selection: ScopeSelection,
) -> Result<CommandOutcome, WikiError> {
    let (_, output_scope, search_scope, store) = indexed_store_for_selection(&selection)?;
    let graph = memory_graph_from_store(&store, &search_scope);
    let backlinks = graph.backlinks(&search_scope, PathBuf::from(&page));
    Ok(render_backlinks(&page, output_scope, &backlinks))
}

pub(crate) fn execute_link_suggest(
    selection: ScopeSelection,
    limit: usize,
) -> Result<CommandOutcome, WikiError> {
    let (_, output_scope, search_scope, store) = indexed_store_for_selection(&selection)?;
    let graph = memory_graph_from_store(&store, &search_scope);
    let suggestions = graph.link_suggestions(&search_scope, limit);
    Ok(render_link_suggest(output_scope, limit, &suggestions))
}

fn render_backlinks(
    page: &str,
    scope: ScopeIdentity,
    backlinks: &[WikiBacklink],
) -> CommandOutcome {
    let backlink_payloads = backlinks
        .iter()
        .map(|backlink| {
            json!({
                "source_path": &backlink.source_path,
                "target_path": &backlink.target_path,
                "raw_target": &backlink.raw_target,
            })
        })
        .collect::<Vec<_>>();
    let payload = json!({
        "command": "backlinks",
        "scope": scope,
        "page": page,
        "backlinks": backlink_payloads,
    });
    let text = render_backlinks_text(page, &scope, backlinks);
    super::scoped_outcome("backlinks", &scope, payload, text)
}

fn render_link_suggest(
    scope: ScopeIdentity,
    limit: usize,
    suggestions: &[LinkSuggestion],
) -> CommandOutcome {
    let suggestion_payloads = suggestions
        .iter()
        .map(|suggestion| {
            json!({
                "target": &suggestion.target,
                "mention_count": suggestion.mention_count,
                "source_paths": &suggestion.source_paths,
            })
        })
        .collect::<Vec<_>>();
    let payload = json!({
        "command": "link-suggest",
        "scope": scope,
        "limit": limit,
        "suggestions": suggestion_payloads,
    });
    let text = render_link_suggest_text(&scope, suggestions);
    super::scoped_outcome("link-suggest", &scope, payload, text)
}

fn render_backlinks_text(page: &str, scope: &ScopeIdentity, backlinks: &[WikiBacklink]) -> String {
    let mut text = format!(
        "Backlinks for {page}
Scope: {scope}
"
    );
    if backlinks.is_empty() {
        text.push_str("No backlinks");
        return text;
    }

    for backlink in backlinks {
        text.push_str("- ");
        text.push_str(&backlink.source_path.display().to_string());
        text.push_str(" via ");
        text.push_str(&backlink.raw_target);
        text.push('\n');
    }
    text
}

fn render_link_suggest_text(scope: &ScopeIdentity, suggestions: &[LinkSuggestion]) -> String {
    let mut text = format!(
        "Link suggestions
Scope: {scope}
"
    );
    if suggestions.is_empty() {
        text.push_str("No suggestions");
        return text;
    }

    for suggestion in suggestions {
        text.push_str("- ");
        text.push_str(&suggestion.target);
        text.push_str(" (");
        text.push_str(&suggestion.mention_count.to_string());
        text.push(' ');
        text.push_str(if suggestion.mention_count == 1 {
            "mention"
        } else {
            "mentions"
        });
        text.push_str(")\n");
    }
    text
}