Handle alternative status codes

This commit is contained in:
spikecodes 2021-03-09 22:23:26 -08:00
parent e59b2b1346
commit 2e89a85858
No known key found for this signature in database
GPG Key ID: 004CECFF9B463BCB
1 changed files with 23 additions and 29 deletions

View File

@ -514,7 +514,7 @@ pub async fn error(req: Request<()>, msg: String) -> tide::Result {
}
#[async_recursion]
async fn connect(path: String) -> io::Result<(i16, String)> {
async fn connect(path: String) -> io::Result<String> {
// Build reddit-compliant user agent for Libreddit
let user_agent = format!("web:libreddit:{}", env!("CARGO_PKG_VERSION"));
@ -557,7 +557,7 @@ async fn connect(path: String) -> io::Result<(i16, String)> {
.collect::<Vec<&str>>()[1];
connect(location.replace("https://www.reddit.com", "")).await
} else {
Ok((status, body))
Ok(body)
}
}
@ -572,35 +572,29 @@ pub async fn request(path: String) -> Result<Value, String> {
};
match connect(path).await {
Ok((status, body)) => {
match status {
// If response is success
200 => {
// Parse the response from Reddit as JSON
let parsed: Result<Value, Error> = from_str(&body);
match parsed {
Ok(json) => {
// If Reddit returned an error
if json["error"].is_i64() {
Err(
json["reason"]
.as_str()
.unwrap_or_else(|| {
json["message"].as_str().unwrap_or_else(|| {
eprintln!("{} - Error parsing reddit error", url);
"Error parsing reddit error"
})
})
.to_string(),
)
} else {
Ok(json)
}
}
Err(e) => err("Failed to parse page JSON data", e.to_string()),
Ok(body) => {
// Parse the response from Reddit as JSON
let parsed: Result<Value, Error> = from_str(&body);
match parsed {
Ok(json) => {
// If Reddit returned an error
if json["error"].is_i64() {
Err(
json["reason"]
.as_str()
.unwrap_or_else(|| {
json["message"].as_str().unwrap_or_else(|| {
eprintln!("{} - Error parsing reddit error", url);
"Error parsing reddit error"
})
})
.to_string(),
)
} else {
Ok(json)
}
}
_ => err("Couldn't send request to Reddit", status.to_string()),
Err(e) => err("Failed to parse page JSON data", e.to_string()),
}
}
Err(e) => err("Couldn't send request to Reddit", e.to_string()),