wty 0.8.0

Yomitan-compatible dictionaries from wikitionary data
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use std::{
    fs::{self, File},
    io::{BufWriter, Write},
    path::PathBuf,
};

use crate::{
    cli::Options, dict::writer::renderer::Renderer, models::yomitan::YomitanDict, path::PathManager,
};

use anyhow::Result;

mod renderer;
use renderer::HtmlRenderer;

const STYLES_CSS_HTML: &[u8] = include_bytes!("../../../../assets/styles_html.css");

#[allow(unused)]
pub fn write_html(opts: &Options, pm: &PathManager, ydict: YomitanDict) -> Result<PathBuf> {
    let dir_in_stage = pm.dir_in_stage("html");
    _ = fs::create_dir_all(&dir_in_stage);

    let dict_name = format!("{}.html", pm.dict_name_expanded());
    let path_dict = dir_in_stage.join(&dict_name);
    let path_css = dir_in_stage.join("styles.css");

    let file = File::create(&path_dict)?;
    let mut writer = BufWriter::new(file);

    writer.write_all(
        br#"<!DOCTYPE html>
            <html>
            <head>
            <meta charset="utf-8">
            <link rel="stylesheet" href="styles.css">
            </head>
            <body>
            "#,
    )?;

    for entry in ydict.into_iter_flat() {
        let html = HtmlRenderer::render_entry(&entry).into_string();
        // Ignore pretty since it can mess the result
        writer.write_all(html.as_bytes())?;
    }

    writer.write_all(br"</body></html>")?;

    // Copy assets/styles_html.css into dir_in_stage/styles.css
    fs::write(&path_css, STYLES_CSS_HTML)?;

    Ok(dir_in_stage)
}

pub fn write_test_html(opts: &Options, pm: &PathManager, ydict: YomitanDict) -> Result<PathBuf> {
    let dir_in_stage = pm.dir_in_stage("html");
    _ = fs::create_dir_all(&dir_in_stage);

    let dict_name = format!("{}.html", pm.dict_name_expanded());
    let path_dict = dir_in_stage.join(dict_name);

    let file = File::create(&path_dict)?;
    let mut writer = BufWriter::new(file);

    for entry in ydict.into_iter_flat() {
        let html = HtmlRenderer::render_entry(&entry).into_string();
        if opts.pretty {
            writer.write_all(prettify_html(&html).as_bytes())?;
        } else {
            writer.write_all(html.as_bytes())?;
        }
    }

    Ok(path_dict)
}

pub fn prettify_html(html: &str) -> String {
    let mut result = String::new();
    let mut indent: usize = 0;
    let mut in_tag = false;

    for ch in html.chars() {
        match ch {
            '<' => {
                if !in_tag {
                    result.push('\n');
                    // WARN: whitespace messes up browser rendering when coupled with yomitan
                    // css (white-space: pre-line).
                    result.push_str(&"  ".repeat(indent)); // main offender
                }
                in_tag = true;
                result.push(ch);
            }
            '>' => {
                result.push(ch);
                in_tag = false;

                // Adjust indent based on tag type
                let last_tag = result.split('<').next_back().unwrap_or("");
                if last_tag.starts_with('/') {
                    indent = indent.saturating_sub(1);
                } else if !last_tag.ends_with('/') && !last_tag.starts_with('!') {
                    indent += 1;
                }
            }
            _ => result.push(ch),
        }
    }

    result
}