civitai 0.1.3

Load models from civit ai.
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

use trauma::download::Download;

use crate::CivitResult;

pub use self::defines::ModelInfo;

mod defines;

/// Get the information from model id
///
/// <https://github.com/civitai/civitai/wiki/REST-API-Reference#get-apiv1modelsmodelid>
#[derive(Clone, Debug)]
pub struct RequestModel {
    pub id: usize,
}

impl RequestModel {
    /// Create a new request by model id
    pub fn new(id: usize) -> Self {
        Self { id }
    }
    /// Send the request
    pub async fn send(&self) -> CivitResult<ModelInfo> {
        Ok(reqwest::get(self.url()).await?.json().await?)
    }
    /// Get the url for the request
    pub fn url(&self) -> String {
        format!("https://civitai.com/api/v1/models/{}", self.id)
    }
}

impl ModelInfo {
    /// Get the primary download link for this model
    ///
    /// Primary model is the latest model in the list
    pub fn download_link(&self) -> &str {
        match self.model_versions.first() {
            Some(s) => s.download_url.as_str(),
            None => "",
        }
    }
    /// Download the primary model, it creates a new download task
    ///
    /// Primary model is the latest model in the list
    pub fn download(&self, local: &str) -> CivitResult<Download> {
        let mut file_name = local;
        let download_link = match self.model_versions.first() {
            Some(s) => {
                if file_name.is_empty() {
                    file_name = s.name.as_str()
                }
                s.download_url.as_str()
            }
            None => Err(trauma::Error::InvalidUrl("Missing download link".to_string()))?,
        };
        let mut task = Download::try_from(download_link)?;
        task.filename = file_name.to_string();
        Ok(task)
    }
}