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

mod epub;

use std::{fs::File, io::BufReader, path::Path};

use chrono::Utc;
pub use epub::epub::get_metadata;
use epub::epub::{
    create_nav_file, create_opf_file, create_part_files, initialize_directory, rm_directory,
    zip_epub, Metadata,
};
use epub::images::{padding_image_file, sort_image_files};
use serde_json::from_reader;
use uuid::Uuid;

pub enum Direction {
    LTR,
    RTL,
}

pub fn img2epub(
    image_dir: &str,
    out: &str,
    title: Option<String>,
    creator: Option<String>,
    publisher: Option<String>,
    date: Option<String>,
    is_rtl: Option<bool>,
) -> Result<(), Box<dyn std::error::Error>> {
    // Create metadata
    let json_path = format!("{}/metadata.json", image_dir);
    let metadata: Metadata = if Path::new(&json_path).exists() {
        let buf = BufReader::new(File::open(&json_path)?);
        let mut data: Metadata = from_reader(buf).expect("metadata.json is invalid");
        data.override_with(title, creator, publisher, date, is_rtl);
        data
    } else {
        if let Some(title) = &title {
            Metadata {
                title: title.clone(),
                creator,
                publisher,
                date,
                is_rtl: is_rtl.unwrap_or(false),
            }
        } else {
            return Err("title is not specified".into());
        }
    };

    let epub_dir = format!("/tmp/epub-{}", Uuid::new_v4().to_string());
    initialize_directory(&epub_dir)?;

    // Sort image files by name
    let sorted_files = sort_image_files(image_dir);

    // Get the maximum width and height of the images
    let sizes = sorted_files.iter().map(|x| (x.width, x.height));
    let max_width: u32 = sizes.clone().map(|s| s.0).max().unwrap();
    let max_height = sizes.clone().map(|s| s.1).max().unwrap();

    // Copy image files to the epub directory
    for file in sorted_files.iter() {
        padding_image_file(
            file.clone(),
            max_width,
            max_height,
            format!("{}/OEBPS/images/{}.webp", epub_dir, file.file_name),
        );
    }
    padding_image_file(
        sorted_files[0].clone(),
        max_width,
        max_height,
        format!("{}/OEBPS/images/cover.webp", epub_dir),
    );

    // Create inner files of the epub
    create_nav_file(&epub_dir, max_width, max_height)?;
    create_opf_file(
        &epub_dir,
        &format!("urn:uuid:{}", Uuid::new_v4().to_string()),
        "ja-JP",
        Utc::now().format("%Y-%m-%dT%H:%M:%SZ").to_string().as_str(),
        &sorted_files,
        max_width,
        max_height,
        &metadata,
    )?;
    create_part_files(
        &epub_dir,
        &metadata.title,
        &sorted_files,
        max_width,
        max_height,
    )?;

    // Zip the directory
    zip_epub(&epub_dir, out)?;

    // Remove the directory
    rm_directory(&epub_dir)?;

    Ok(())
}