-
-
Notifications
You must be signed in to change notification settings - Fork 25
/
request.rs
55 lines (47 loc) · 1.46 KB
/
request.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
43
44
45
46
47
48
49
50
51
52
53
54
55
use mime::Mime;
use reqwest::{header::CONTENT_TYPE, Client, Response};
use scraper::Html;
use super::result::Error;
lazy_static! {
static ref CLIENT: Client = reqwest::Client::builder()
.user_agent(
"Mozilla/5.0 (compatible; January/1.0; +https://gitlab.insrt.uk/revolt/january)"
)
.build()
.unwrap();
}
pub async fn fetch(url: &str) -> Result<(Response, Mime), Error> {
let resp = CLIENT
.get(url)
.send()
.await
.map_err(|_| Error::ReqwestFailed)?;
if !resp.status().is_success() {
return Err(Error::RequestFailed);
}
let content_type = resp
.headers()
.get(CONTENT_TYPE)
.ok_or_else(|| Error::MissingContentType)?
.to_str()
.map_err(|_| Error::ConversionFailed)?;
let mime: mime::Mime = content_type
.parse()
.map_err(|_| Error::FailedToParseContentType)?;
Ok((resp, mime))
}
pub async fn consume_fragment(resp: Response) -> Result<Html, Error> {
let body = resp.text().await.map_err(|_| Error::FailedToConsumeText)?;
Ok(Html::parse_document(&body))
}
pub async fn consume_size(resp: Response) -> Result<(isize, isize), Error> {
let bytes = resp
.bytes()
.await
.map_err(|_| Error::FailedToConsumeBytes)?;
if let Ok(size) = imagesize::blob_size(&bytes) {
Ok((size.width as isize, size.height as isize))
} else {
Err(Error::CouldNotDetermineImageSize)
}
}