trnovel 0.10.2

Terminal reader for novel
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

use crate::errors::{Errors, Result};
use crate::novel::Novel;
use crate::novel::VolumeMarker;
use crate::novel::local_novel::LocalNovel;
use crate::utils::{get_path_md5, novel_catch_dir};

use serde::{Deserialize, Serialize};
use std::{
    fs::File,
    path::{Path, PathBuf},
};

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct LocalNovelCache {
    pub chapters: Vec<(String, usize)>,
    pub encoding: &'static encoding_rs::Encoding,
    pub current_chapter: usize,
    pub line_percent: f64,
    pub path: PathBuf,
    /// 分卷元数据。`#[serde(default)]` 保证旧缓存（无该字段）可正常读取。
    #[serde(default)]
    pub volumes: Vec<VolumeMarker>,
}

impl LocalNovelCache {
    pub fn save(&self) -> Result<()> {
        let cache_path = Self::cache_path(&self.path)?;
        let file = File::create(cache_path)?;
        serde_json::to_writer_pretty(file, self)?;
        Ok(())
    }

    pub fn cache_path<T: AsRef<Path>>(path: T) -> Result<PathBuf> {
        let novel_catch_dir = PathBuf::new().join(novel_catch_dir()?).join("local");

        if !novel_catch_dir.exists() {
            std::fs::create_dir_all(&novel_catch_dir)?;
        }

        Ok(novel_catch_dir
            .join(get_path_md5(path)?)
            .with_extension("json"))
    }
}

// 从本地小说创建缓存
impl TryFrom<&LocalNovel> for LocalNovelCache {
    type Error = Errors;
    fn try_from(value: &LocalNovel) -> Result<Self> {
        let novel_chapters = value.novel_chapters.clone();
        Ok(Self {
            chapters: value.get_chapters_result()?.to_vec(),
            encoding: value.encoding,
            current_chapter: novel_chapters.current_chapter,
            path: value.path.clone(),
            line_percent: novel_chapters.line_percent,
            volumes: novel_chapters.volumes.clone(),
        })
    }
}

// 从路径加载缓存
impl TryFrom<&Path> for LocalNovelCache {
    type Error = Errors;
    fn try_from(value: &Path) -> Result<Self> {
        let cache_path = Self::cache_path(value)?;
        let file = File::open(cache_path)?;
        Ok(serde_json::from_reader(file)?)
    }
}

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

    #[test]
    fn old_cache_without_volumes_deserializes() {
        // 旧缓存没有 `volumes` 字段，应能正常反序列化并退化为无卷。
        let json = r#"{
            "chapters": [["第一章 甲", 0], ["第二章 乙", 100]],
            "encoding": "UTF-8",
            "current_chapter": 1,
            "line_percent": 0.5,
            "path": "/tmp/x.txt"
        }"#;
        let cache: LocalNovelCache = serde_json::from_str(json).expect("应能反序列化旧缓存");
        assert!(cache.volumes.is_empty());
        assert_eq!(cache.chapters.len(), 2);
        assert_eq!(cache.current_chapter, 1);
    }
}