genkit 0.3.1

A common generator kit for static site generator.
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
93
94
95
96

use std::{collections::HashMap, fmt::Write};

use anyhow::Result;

use crate::data::{self, PreviewEvent, UrlPreviewInfo};

use super::CodeBlock;

pub(crate) struct UrlPreviewBlock<'a> {
    url: &'a str,
    info: UrlPreviewInfo,
    // Whether show the preview image. default to true.
    show_image: bool,
}

impl<'a> UrlPreviewBlock<'a> {
    pub(crate) fn new(
        options: HashMap<String, &'a str>,
        url: &'a str,
        info: UrlPreviewInfo,
    ) -> Self {
        UrlPreviewBlock {
            url,
            info,
            show_image: options
                .get("image")
                .and_then(|v| str::parse::<bool>(v).ok())
                .unwrap_or(true),
        }
    }
}

impl<'a> CodeBlock for UrlPreviewBlock<'a> {
    fn render(&self) -> Result<String> {
        let mut html = String::new();
        writeln!(&mut html, r#"<div class="url-preview">"#)?;
        writeln!(&mut html, r#" <div>{}</div>"#, self.info.title)?;
        writeln!(&mut html, r#" <div>{}</div>"#, self.info.description)?;
        writeln!(&mut html, r#" <a href="{url}">{url}</a>"#, url = self.url)?;
        if self.show_image {
            if let Some(image) = self.info.image.as_ref().filter(|i| !i.is_empty()) {
                writeln!(&mut html, r#" <img src="{}" />"#, image)?;
            }
        }
        writeln!(&mut html, r#"</div>"#)?;
        Ok(html)
    }
}

pub(crate) struct UrlPreviewError<'a>(pub &'a str, pub &'a str);

impl<'a> CodeBlock for UrlPreviewError<'a> {
    fn render(&self) -> Result<String> {
        let mut html = String::new();
        writeln!(&mut html, r#"<div class="url-preview">"#)?;
        writeln!(&mut html, r#" <div></div>"#)?;
        writeln!(&mut html, r#" <div>Url preview error: {}</div>"#, self.1)?;
        writeln!(&mut html, r#" <a href="{url}">{url}</a>"#, url = self.0)?;
        writeln!(&mut html, r#"</div>"#)?;
        Ok(html)
    }
}

/// Render the preview url if success, otherwise return preview error
/// to remind user we have error.
#[tokio::main(flavor = "current_thread")]
pub(crate) async fn render(url: &str, options: HashMap<String, &str>) -> Option<String> {
    let (first_preview, mut rx) = {
        // parking_lot RwLock guard isn't async-aware,
        // we should keep this guard drop in this scope.
        let data = data::read();
        if let Some(info) = data.get_preview(url) {
            let html = UrlPreviewBlock::new(options, url, info).render().unwrap();
            return Some(html);
        }

        data.preview_url(url)
    };
    rx.changed()
        .await
        .expect("URL preview watch channel receive failed.");
    let event = rx.borrow();
    match event.to_owned().expect("Url preview didn't initialized.") {
        PreviewEvent::Finished(info) => {
            let html = UrlPreviewBlock::new(options, url, info).render().unwrap();
            if first_preview {
                println!("URL previewed: {url}");
            }
            Some(html)
        }
        PreviewEvent::Failed(err) => {
            // Return a preview error block.
            Some(UrlPreviewError(url, &err).render().unwrap())
        }
    }
}