inkhaven 1.6.10

Inkhaven — TUI literary work editor for Typst books
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

//! TDOC-4 — image assets. Rewrite `<img src="…">` for local images to
//! `assets/<basename>` and record the sources so the orchestrator can copy them
//! into the site. External URLs and already-rewritten paths are left alone.

use std::collections::BTreeSet;
use std::path::Path;

/// Rewrite local `<img src>` to `assets/<basename>`, collecting the original
/// (project-relative) sources into `collected`.
pub fn rewrite_images(html: &str, collected: &mut BTreeSet<String>) -> String {
    let mut out = String::new();
    let mut rest = html;
    while let Some(pos) = rest.find("<img ") {
        out.push_str(&rest[..pos]);
        let tag = &rest[pos..];
        if let Some(src_rel) = tag.find("src=\"") {
            let abs = src_rel + 5;
            if let Some(end) = tag[abs..].find('"') {
                let src = &tag[abs..abs + end];
                if is_local(src) {
                    if let Some(base) = Path::new(src).file_name().and_then(|s| s.to_str()) {
                        collected.insert(src.to_string());
                        out.push_str(&tag[..abs]);
                        out.push_str("assets/");
                        out.push_str(base);
                        rest = &tag[abs + end..];
                        continue;
                    }
                }
            }
        }
        // No rewrite — emit `<img ` and keep scanning after it.
        out.push_str("<img ");
        rest = &tag[5..];
    }
    out.push_str(rest);
    out
}

fn is_local(src: &str) -> bool {
    !src.starts_with("http://")
        && !src.starts_with("https://")
        && !src.starts_with("data:")
        && !src.starts_with("assets/")
}

/// Copy every collected image into `<out_dir>/assets/`. Missing files warn, never fail.
pub fn copy_assets(project_root: &Path, out_dir: &Path, collected: &BTreeSet<String>) -> std::io::Result<()> {
    if collected.is_empty() {
        return Ok(());
    }
    let assets = out_dir.join("assets");
    std::fs::create_dir_all(&assets)?;
    for src in collected {
        let from = project_root.join(src);
        let Some(base) = Path::new(src).file_name() else { continue };
        if from.exists() {
            let _ = std::fs::copy(&from, assets.join(base));
        } else {
            tracing::warn!(target: "inkhaven::html", "html: image `{}` not found — skipped", from.display());
        }
    }
    Ok(())
}

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

    #[test]
    fn rewrites_local_leaves_remote() {
        let mut c = BTreeSet::new();
        let html = rewrite_images(
            "<p><img src=\"img/map.png\" alt=\"m\"> and <img src=\"https://x/y.png\" alt=\"r\"></p>",
            &mut c,
        );
        assert!(html.contains("src=\"assets/map.png\""));
        assert!(html.contains("src=\"https://x/y.png\""));
        assert!(c.contains("img/map.png"));
        assert_eq!(c.len(), 1);
    }
}