tatuin-providers 0.6.1

Providers lib of the Tatuin project
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

// SPDX-License-Identifier: MIT

use std::path::{Path, PathBuf};

use tatuin_core::RichStringTransformerTrait;

use crate::obsidian::{fs, markdown, md_file::TAG_RE};

#[derive(Debug, Clone, Default)]
pub struct InternalLinksTransformer {
    vault_path: PathBuf,
    remove_tags: bool,
}

impl InternalLinksTransformer {
    pub fn new(vault_path: &Path) -> Self {
        Self {
            vault_path: vault_path.to_path_buf(),
            remove_tags: false,
        }
    }

    pub fn with_remove_tags(mut self) -> Self {
        self.remove_tags = true;
        self
    }
}

impl RichStringTransformerTrait for InternalLinksTransformer {
    fn transform(&self, s: &str) -> String {
        let mut s = s.to_string();

        if self.remove_tags {
            s = clear_tags(&s);
        }
        s = fix_regular_links(&s, &self.vault_path);
        s = fix_wiki_links(&s, &self.vault_path);
        s
    }
}

fn clear_tags(name: &str) -> String {
    TAG_RE.replace_all(name, "").to_string()
}

fn fix_wiki_links(text: &str, vault_path: &Path) -> String {
    let mut result = text.to_string();

    for l in markdown::find_wiki_links(text).iter().rev() {
        let file_name = format!("{}.md", l.link);

        let link = if let Ok(file_name) = urlencoding::decode(&file_name)
            && let Ok(f) = fs::find_file(vault_path, &file_name)
        {
            fs::obsidian_url(vault_path, &f)
        } else {
            // there is no existed file, so render as-is
            continue;
        };

        let display = if l.display_text.is_empty() {
            l.link.to_string()
        } else {
            l.display_text.to_string()
        };

        result.replace_range(l.start..l.end + 1, format!("[{display}]({link})").as_str());
    }

    result
}

fn fix_regular_links(text: &str, vault_path: &Path) -> String {
    let mut result = text.to_string();

    for l in markdown::find_regular_links(text).iter().rev() {
        if let Ok(file_name) = urlencoding::decode(l.link)
            && let Ok(f) = fs::find_file(vault_path, &file_name)
        {
            result.replace_range(
                l.start..l.end,
                format!("[{}]({})", l.display_text, fs::obsidian_url(vault_path, &f)).as_str(),
            );
        }
    }

    result
}