hydrus-api 0.10.2

A rust wrapper for the hydrus client api
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

use crate::api_core::common::OptionalStringNumber;
use crate::api_core::endpoints::managing_cookies_and_http_headers::CookieBuilder;
use crate::error::Result;
use crate::Client;
use std::time::{Duration, SystemTime, UNIX_EPOCH};

pub struct Address {
    client: Client,
    domain: String,
    path: String,
}

impl Address {
    pub(crate) fn from_str(client: Client, domain: &str) -> Self {
        let (domain, path) = domain.split_once("/").unwrap_or((domain, "/"));
        Self {
            client,
            domain: domain.to_string(),
            path: path.to_string(),
        }
    }

    /// Returns the path after the domain name
    pub fn path(&self) -> &str {
        &self.path
    }

    /// Sets the path of the domain that can be used for setting cookies
    pub fn set_path<S: ToString>(&mut self, path: S) {
        self.path = path.to_string();
    }

    /// Sets cookies for the domain
    pub async fn set_cookies(&self, cookies: Vec<DomainCookie>) -> Result<()> {
        let cookies = cookies
            .into_iter()
            .map(|cookie| {
                let mut builder = CookieBuilder::default()
                    .domain(&self.domain)
                    .path(&self.path)
                    .name(cookie.name)
                    .value(cookie.value);
                if let Some(expires) = cookie.expires {
                    builder =
                        builder.expires(expires.duration_since(UNIX_EPOCH).unwrap().as_secs());
                }
                builder.build()
            })
            .collect();

        self.client.set_cookies(cookies).await
    }

    /// Returns all cookies stored for this domain
    pub async fn get_cookies(&self) -> Result<Vec<DomainCookie>> {
        let response = self.client.get_cookies(&self.domain).await?;
        let cookies = response
            .cookies
            .into_iter()
            .map(DomainCookie::from)
            .collect();

        Ok(cookies)
    }
}

#[derive(Clone, Debug)]
pub struct DomainCookie {
    pub name: String,
    pub value: String,
    pub expires: Option<SystemTime>,
}

impl DomainCookie {
    /// Creates a new cookie that will be expire after the given instant or only last for the session
    pub fn new<S1: ToString, S2: ToString>(
        name: S1,
        value: S2,
        expires: Option<SystemTime>,
    ) -> Self {
        Self {
            name: name.to_string(),
            value: value.to_string(),
            expires,
        }
    }
}

impl From<[OptionalStringNumber; 5]> for DomainCookie {
    fn from(cookie_entry: [OptionalStringNumber; 5]) -> Self {
        let name = cookie_entry[0].string().unwrap_or("");
        let value = cookie_entry[1].string().unwrap_or("");
        let expires = cookie_entry[4]
            .number()
            .map(|n| UNIX_EPOCH + Duration::from_secs(n));

        Self::new(name, value, expires)
    }
}