libreddit/src/user.rs

114 lines
3.6 KiB
Rust
Raw Normal View History

2020-10-25 21:25:59 +01:00
// CRATES
2021-03-17 23:30:33 +01:00
use crate::client::json;
2021-03-12 05:15:26 +01:00
use crate::esc;
2021-03-17 23:30:33 +01:00
use crate::server::RequestExt;
use crate::utils::{error, filter_posts, format_url, get_filters, param, template, Post, Preferences, User};
2020-10-25 21:25:59 +01:00
use askama::Template;
2021-03-17 23:30:33 +01:00
use hyper::{Body, Request, Response};
use time::{OffsetDateTime, macros::format_description};
2020-11-17 20:37:40 +01:00
2020-10-25 21:25:59 +01:00
// STRUCTS
#[derive(Template)]
#[template(path = "user.html", escape = "none")]
struct UserTemplate {
user: User,
posts: Vec<Post>,
2020-12-30 02:11:47 +01:00
sort: (String, String),
2020-12-27 21:36:10 +01:00
ends: (String, String),
/// "overview", "comments", or "submitted"
listing: String,
2021-01-09 02:35:04 +01:00
prefs: Preferences,
url: String,
redirect_url: String,
/// Whether the user themself is filtered.
is_filtered: bool,
/// Whether all fetched posts are filtered (to differentiate between no posts fetched in the first place,
/// and all fetched posts being filtered).
all_posts_filtered: bool,
2020-10-25 21:25:59 +01:00
}
2021-01-03 05:50:23 +01:00
// FUNCTIONS
2021-03-17 23:30:33 +01:00
pub async fn profile(req: Request<Body>) -> Result<Response<Body>, String> {
let listing = req.param("listing").unwrap_or_else(|| "overview".to_string());
2021-01-01 00:54:13 +01:00
// Build the Reddit JSON API path
2021-03-18 05:40:55 +01:00
let path = format!(
"/user/{}/{}.json?{}&raw_json=1",
2021-03-27 04:00:47 +01:00
req.param("name").unwrap_or_else(|| "reddit".to_string()),
listing,
req.uri().query().unwrap_or_default(),
2021-03-18 05:40:55 +01:00
);
let url = String::from(req.uri().path_and_query().map_or("", |val| val.as_str()));
let redirect_url = url[1..].replace('?', "%3F").replace('&', "%26");
2020-12-27 21:36:10 +01:00
2021-01-01 00:54:13 +01:00
// Retrieve other variables from Libreddit request
let sort = param(&path, "sort").unwrap_or_default();
2021-03-17 23:30:33 +01:00
let username = req.param("name").unwrap_or_default();
let user = user(&username).await.unwrap_or_default();
2020-11-21 07:05:27 +01:00
let filters = get_filters(&req);
if filters.contains(&["u_", &username].concat()) {
template(UserTemplate {
user,
posts: Vec::new(),
sort: (sort, param(&path, "t").unwrap_or_default()),
ends: (param(&path, "after").unwrap_or_default(), "".to_string()),
listing,
prefs: Preferences::new(req),
url,
redirect_url,
is_filtered: true,
all_posts_filtered: false,
})
} else {
// Request user posts/comments from Reddit
match Post::fetch(&path, false).await {
Ok((mut posts, after)) => {
let all_posts_filtered = filter_posts(&mut posts, &filters);
2021-01-16 00:05:55 +01:00
template(UserTemplate {
user,
posts,
sort: (sort, param(&path, "t").unwrap_or_default()),
ends: (param(&path, "after").unwrap_or_default(), after),
listing,
prefs: Preferences::new(req),
url,
redirect_url,
is_filtered: false,
all_posts_filtered,
})
}
// If there is an error show error page
Err(msg) => error(req, msg).await,
2021-01-02 07:21:43 +01:00
}
2020-11-20 05:42:18 +01:00
}
2020-10-25 21:25:59 +01:00
}
// USER
2021-01-14 18:53:54 +01:00
async fn user(name: &str) -> Result<User, String> {
// Build the Reddit JSON API path
let path: String = format!("/user/{}/about.json?raw_json=1", name);
2021-01-01 06:03:44 +01:00
2021-01-02 00:28:13 +01:00
// Send a request to the url
2021-05-20 21:24:06 +02:00
json(path, false).await.map(|res| {
// Grab creation date as unix timestamp
let created_unix = res["data"]["created"].as_f64().unwrap_or(0.0).round() as i64;
let created = OffsetDateTime::from_unix_timestamp(created_unix).unwrap_or(OffsetDateTime::UNIX_EPOCH);
2021-05-20 21:24:06 +02:00
// Closure used to parse JSON from Reddit APIs
let about = |item| res["data"]["subreddit"][item].as_str().unwrap_or_default().to_string();
2021-01-09 02:35:04 +01:00
2021-05-20 21:24:06 +02:00
// Parse the JSON output into a User struct
User {
name: res["data"]["name"].as_str().unwrap_or(name).to_owned(),
2021-05-20 21:24:06 +02:00
title: esc!(about("title")),
icon: format_url(&about("icon_img")),
karma: res["data"]["total_karma"].as_i64().unwrap_or(0),
2021-12-29 21:48:57 +01:00
created: created.format(format_description!("[month repr:short] [day] '[year repr:last_two]")).unwrap_or_default(),
2021-05-20 21:24:06 +02:00
banner: esc!(about("banner_img")),
description: about("public_description"),
2021-01-02 07:21:43 +01:00
}
2021-05-20 21:24:06 +02:00
})
2020-11-30 03:50:29 +01:00
}