src/epub.rs


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

use crate::models::FileEntry;
use anyhow::Result;
use reqwest::Client;
use std::path::Path;
use tokio::{
    fs::{self, File},
    io::AsyncWriteExt,
};

/// Creates and writes the mimetype. Assumes dest_root already exists.
pub async fn write_mimetype(dest_root: &Path) -> Result<()> {
    let dest_path = dest_root.join("mimetype");
    let mut file = File::create(dest_path).await?;
    file.write_all(b"application/epub+zip").await?;
    Ok(())
}

pub async fn download_all_files(
    client: &Client,
    file_entries: &[FileEntry],
    dest_root: &Path,
) -> Result<()> {
    for entry in file_entries {
        let dest_path = dest_root.join(&entry.full_path);

        if let Some(parent_dir) = dest_path.parent() {
            fs::create_dir_all(parent_dir).await?;
        }

        let mut file = File::create(dest_path).await?;
        let bytes = client
            .get(&entry.url)
            .send()
            .await?
            .error_for_status()?
            .bytes()
            .await?;

        file.write_all(&bytes).await?;
    }
    Ok(())
}