novel_cli/utils/
convert.rs

use std::sync::OnceLock;

use color_eyre::eyre::Result;
use novel_api::Timing;
use opencc_rs::{Config, OpenCC};

use super::{Content, Novel};
use crate::cmd::Convert;

pub fn convert<T>(novel: &mut Novel, converts: T) -> Result<()>
where
    T: AsRef<[Convert]>,
{
    if converts.as_ref().is_empty() {
        return Ok(());
    }

    let mut timing = Timing::new();

    novel.name = convert_str(&novel.name, &converts, false)?;

    novel.author_name = convert_str(&novel.author_name, &converts, false)?;

    if novel.introduction.is_some() {
        #[allow(clippy::needless_borrows_for_generic_args)]
        for line in novel.introduction.as_mut().unwrap() {
            *line = convert_str(&line, &converts, false)?;
        }
    }

    for volume in &mut novel.volumes {
        volume.title = convert_str(&volume.title, &converts, true)?;

        for chapter in &mut volume.chapters {
            chapter.title = convert_str(&chapter.title, &converts, true)?;

            if chapter.contents.is_some() {
                let mut contents = chapter.contents.take().unwrap();

                for content in &mut contents {
                    #[allow(clippy::needless_borrows_for_generic_args)]
                    if let Content::Text(line) = content {
                        *line = convert_str(&line, &converts, false)?;
                    }
                }

                chapter.contents = Some(contents);
            }
        }
    }

    tracing::debug!("Time spent on `convert`: {}", timing.elapsed()?);

    Ok(())
}

pub fn convert_str<T, E>(str: T, converts: E, in_heading: bool) -> Result<String>
where
    T: AsRef<str>,
    E: AsRef<[Convert]>,
{
    let converts = converts.as_ref();

    if converts.is_empty() {
        Ok(str.as_ref().to_string())
    } else {
        let mut result = String::new();

        static OPENCC_S2T: OnceLock<OpenCC> = OnceLock::new();
        static OPENCC_T2S: OnceLock<OpenCC> = OnceLock::new();
        static OPENCC_JP2T2S: OnceLock<OpenCC> = OnceLock::new();

        if converts.contains(&Convert::JP2T2S) {
            result = OPENCC_JP2T2S
                .get_or_init(|| {
                    OpenCC::new(vec![Config::JP2T, Config::TW2S])
                        .expect("OpenCC initialization failure")
                })
                .convert(&str)?;
        } else if converts.contains(&Convert::T2S) {
            result = OPENCC_T2S
                .get_or_init(|| {
                    OpenCC::new(vec![Config::TW2S]).expect("OpenCC initialization failure")
                })
                .convert(&str)?;
        } else if converts.contains(&Convert::S2T) {
            result = OPENCC_S2T
                .get_or_init(|| {
                    OpenCC::new(vec![Config::S2T]).expect("OpenCC initialization failure")
                })
                .convert(&str)?;
        }

        if converts.contains(&Convert::CUSTOM) {
            if result.is_empty() {
                result = custom_convert(str, in_heading);
            } else {
                result = custom_convert(result, in_heading);
            }

            if converts.contains(&Convert::JP2T2S) || converts.contains(&Convert::T2S) {
                let mut new_result = String::with_capacity(result.len());
                for c in result.chars() {
                    match super::CONVERT_T2S_MAP.get(&c) {
                        Some(new) => {
                            new_result.push(*new);
                        }
                        None => new_result.push(c),
                    }
                }

                result = new_result;
            }
        }

        Ok(result.trim().to_string())
    }
}

#[must_use]
fn custom_convert<T>(str: T, in_heading: bool) -> String
where
    T: AsRef<str>,
{
    if str.as_ref().is_empty() {
        return String::default();
    }

    let mut s = String::new();
    for c in html_escape::decode_html_entities(str.as_ref())
        .to_string()
        .chars()
    {
        match super::CONVERT_MAP.get(&c) {
            Some(new) => {
                s.push(*new);
            }
            None => s.push(c),
        }
    }

    let mut result = String::new();
    for (c, next_c) in s.chars().zip(s.chars().skip(1)) {
        do_custom_convert(c, Some(next_c), &mut result, in_heading);
    }
    do_custom_convert(s.chars().last().unwrap(), None, &mut result, in_heading);

    result
}

fn do_custom_convert(c: char, next_c: Option<char>, result: &mut String, in_heading: bool) {
    let space = ' ';
    let last = result.chars().last();

    if
    // https://en.wikipedia.org/wiki/Zero-width_space
    c == '\u{200B}'
        // https://en.wikipedia.org/wiki/Zero-width_non-joiner
        || c == '\u{200C}'
        // https://en.wikipedia.org/wiki/Zero-width_joiner
        || c == '\u{200D}'
        // https://en.wikipedia.org/wiki/Word_joiner
        || c == '\u{2060}'
        // https://en.wikipedia.org/wiki/Byte_order_mark
        || c == '\u{FEFF}'
        // https://en.wikipedia.org/wiki/Unicode_control_characters
        || c.is_control()
    {
        // do nothing
    } else if c.is_whitespace() {
        if last.is_some_and(|c| !super::is_punctuation(c)) {
            result.push(space)
        }
    } else if super::is_punctuation(c) {
        if !in_heading && last.is_some_and(|c| c.is_whitespace()) {
            result.pop();
        }

        if c == ':' {
            // e.g. 08:00
            if last.is_some_and(|c| c.is_ascii_digit())
                && next_c.is_some_and(|c| c.is_ascii_digit())
            {
                result.push(':');
            } else {
                result.push('：');
            }
        } else {
            result.push(c);
        }
    } else {
        result.push(c);
    }
}

#[cfg(test)]
mod tests {
    use pretty_assertions::assert_eq;
    use testresult::TestResult;

    use super::*;

    #[test]
    fn convert() -> TestResult {
        let config = vec![Convert::JP2T2S, Convert::CUSTOM];

        assert_eq!(convert_str("幺", &config, false)?, "幺");
        assert_eq!(convert_str("妳", &config, false)?, "你");
        assert_eq!(convert_str("Ｑ０", &config, false)?, "Q0");
        assert_eq!(convert_str("“安装后”", &config, false)?, "“安装后”");
        assert_eq!(convert_str("&amp;", &config, false)?, "&");
        assert_eq!(convert_str("安裝後?", &config, false)?, "安装后？");
        assert_eq!(convert_str("安　装", &config, false)?, "安 装");
        assert_eq!(convert_str("你\n好", &config, false)?, "你好");
        assert_eq!(convert_str("08:00", &config, false)?, "08:00");
        assert_eq!(convert_str("接著", &config, false)?, "接着");
        assert_eq!(
            convert_str("第一章 “你好”", &config, false)?,
            "第一章“你好”"
        );
        assert_eq!(
            convert_str("第一章 “你好”", &config, true)?,
            "第一章 “你好”"
        );

        Ok(())
    }
}
novel_cli/utils/convert.rs

novel_cli/utils/
convert.rs
