dynasty 1.4.1

Dynasty Reader's CLI downloader
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

mod methods;

use std::{fs, path::PathBuf};

use anyhow::Result;
use clap::ValueEnum;
use indicatif::ProgressBar;

#[derive(Debug, Clone, Copy)]
pub struct Archiver {
    pub kind: ArchiveKind,
    pub keep_archived: bool,
}

impl Archiver {
    pub fn archive(
        &self,
        chapter: &ArchivableChapter,
        progress_bar: &ProgressBar,
    ) -> Result<PathBuf> {
        let archive_function = match self.kind {
            #[cfg(feature = "archive-tar")]
            ArchiveKind::Cbt => methods::archive_cbt,
            #[cfg(feature = "archive-zip")]
            ArchiveKind::Cbz => methods::archive_cbz,
            #[cfg(feature = "archive-pdf")]
            ArchiveKind::Pdf => methods::archive_pdf,
            #[cfg(feature = "archive-tar")]
            ArchiveKind::Tar => methods::archive_tar,
            #[cfg(feature = "archive-zip")]
            ArchiveKind::Zip => methods::archive_zip,
        };

        let archive_path = archive_function(chapter, progress_bar)?;

        if !self.keep_archived {
            fs::remove_dir_all(&chapter.path)?;
        }

        Ok(archive_path)
    }
}

#[derive(Debug, Clone, Copy, ValueEnum)]
pub enum ArchiveKind {
    #[cfg(feature = "archive-tar")]
    Cbt,
    #[cfg(feature = "archive-zip")]
    Cbz,
    #[cfg(feature = "archive-pdf")]
    Pdf,
    #[cfg(feature = "archive-tar")]
    Tar,
    #[cfg(feature = "archive-zip")]
    Zip,
}

#[derive(Debug, Clone)]
pub struct ArchivableChapter {
    pub path: PathBuf,
    pub pages: Vec<PathBuf>,
    pub title: String,
}

impl ArchivableChapter {
    fn filename(&self) -> &str {
        self.path
            .file_name()
            .and_then(|os_str| os_str.to_str())
            .unwrap()
    }

    fn path_with_extension(&self, extension: &str) -> PathBuf {
        self.path
            .with_file_name(format!("{}.{}", self.filename(), extension))
    }
}