fluffyf 0.2.0

An e621 api wrapper and 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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

// Copyright 2024 Ezra Alvarion
//
// Licensed under the Apache License, Version 2.0 <LICENSE-Apache or
// https://www.apache.org/licenses/LICENSE-2.0> or the BSD 2-Clause 
// License <LICENSE-BSD or https://opensource.org/license/bsd-2-clause/>,
// at Your option. This file may not be copied, modified, or distributed
// except according to those terms.

use {
    std::{
        collections::HashMap,
        error::Error,
        fs
    },
    
    bytes::Bytes,
    log::debug,
    reqwest::Client,
    serde::Deserialize,
    
    crate::RqResult,
};

#[derive(Debug, PartialEq, Eq, Deserialize)]
pub enum Extension {
    #[serde(rename = "jpg")]
    Jpeg,
    #[serde(rename = "png")]
    Png,
    #[serde(rename = "gif")]
    Gif,
    #[serde(rename = "swf")]
    Swf,
    #[serde(rename = "webm")]
    WebM
}

#[derive(Debug, Deserialize)]
pub struct Alternates {
    pub r#type: String,
    pub width: u32,
    pub height: u32,
    pub urls: Vec<String>
}

#[derive(Debug, Deserialize)]
pub struct PostFile {
    /// **Funfact**: The widest image on E621 \[112770\] is 21616 pixels long.
    pub width: u16,

    /// And the tallest \[37061\] is 17700 pixels tall.
    pub height: u16, 

    pub ext: Extension,
    pub size: u64,   // Size is not as clear-cut as dimension because of videos
    pub md5: String,

    /// The URL where the file is hosted on E6.
    pub url: String,
}

#[derive(Debug, Deserialize)]
pub struct PostPreview {
    pub width: u16,
    pub height: u16,

    /// The URL where the preview file is hosted on E6.
    pub url: String,
}

#[derive(Debug, Deserialize)]
pub struct PostSample {
    /// If the post has a sample/thumbnail or not.
    pub has: bool,
    pub height: u16,
    pub width: u16,

    /// The URL where the sample file is hosted on E6.
    pub url: String,

    /// Alternate samples to use, usually only relevant for
    /// videos and specifies different format per sample.
    /// 
    /// The form is Key-Paired Objects where the key is something like `480p`.
    pub alternates: HashMap<String, Alternates>
}

impl PostFile {
    pub async fn get_image_data(&self, client: Client)
        -> RqResult<Bytes> {
        let res = client.get(&self.url).send().await?.bytes().await?;
        debug!("got the bytes");
        Ok(res)
    }
    
    pub async fn write(&self, client: Client, filepath: &str) -> Result<(), Box<dyn Error>> {
        debug!("writing {filepath} using {:#?}", client);
        fs::write(filepath, self.get_image_data(client).await?.as_ref())?;
        Ok(())
    }
}