poet 0.7.0

Static site generator with optional MCP server and LLM SEO optimizations
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

use std::path::PathBuf;

use anyhow::Result;
use anyhow::anyhow;
use blake3::Hash;
use blake3::hash;
use rhai::CustomType;
use rhai::TypeBuilder;

use crate::filesystem::file_entry_kind::FileEntryKind;
use crate::filesystem::file_entry_stub::FileEntryStub;

#[derive(Clone, Debug)]
pub struct FileEntry {
    pub contents: String,
    pub contents_hash: Hash,
    pub kind: FileEntryKind,
    pub relative_path: PathBuf,
}

impl FileEntry {
    pub fn get_stem_path_relative_to(&self, base: &PathBuf) -> PathBuf {
        self.relative_path
            .strip_prefix(base)
            .unwrap_or(&self.relative_path)
            .with_extension("")
    }

    pub fn get_stem_relative_to(&self, base: &PathBuf) -> String {
        self.get_stem_path_relative_to(base)
            .to_string_lossy()
            .to_string()
    }

    fn rhai_relative_path(&mut self) -> String {
        self.relative_path.to_string_lossy().to_string()
    }
}

impl CustomType for FileEntry {
    fn build(mut builder: TypeBuilder<Self>) {
        builder
            .with_name("FileEntry")
            .with_get("relative_path", Self::rhai_relative_path);
    }
}

impl TryFrom<FileEntryStub> for FileEntry {
    type Error = anyhow::Error;

    fn try_from(file_entry_stub: FileEntryStub) -> Result<Self> {
        let top_directory: String = file_entry_stub
            .top_directory()
            .ok_or_else(|| anyhow!("Unable to find file's top directory"))?;
        let extension: String = file_entry_stub
            .extension()
            .ok_or_else(|| anyhow!("Unable to find file's extension"))?;

        Ok(Self {
            contents_hash: hash(file_entry_stub.contents.as_bytes()),
            contents: file_entry_stub.contents,
            kind: match (top_directory.as_str(), extension.as_str()) {
                ("authors", "toml") => FileEntryKind::Author,
                ("content", "md") => FileEntryKind::Content,
                ("prompts", "md") => FileEntryKind::Prompt,
                ("shortcodes", "rhai") => FileEntryKind::Shortcode,
                _ => FileEntryKind::Other,
            },
            relative_path: file_entry_stub.relative_path,
        })
    }
}

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

    #[test]
    fn test_get_stem_relative_to() {
        let file_entry: FileEntry = FileEntryStub {
            contents: String::new(),
            relative_path: PathBuf::from("shortcodes/example/foo.rhai"),
        }
        .try_into()
        .unwrap();

        let base = PathBuf::from("shortcodes");
        let stem = file_entry.get_stem_relative_to(&base);

        assert_eq!(stem, "example/foo");
    }
}