138 lines
4.6 KiB
Rust
138 lines
4.6 KiB
Rust
// CRATES
|
|
use crate::client::json;
|
|
use crate::server::RequestExt;
|
|
use crate::utils::{error, filter_posts, format_url, get_filters, nsfw_landing, param, setting, template, Post, Preferences, User};
|
|
use askama::Template;
|
|
use hyper::{Body, Request, Response};
|
|
use time::{macros::format_description, OffsetDateTime};
|
|
|
|
// STRUCTS
|
|
#[derive(Template)]
|
|
#[template(path = "user.html")]
|
|
struct UserTemplate {
|
|
user: User,
|
|
posts: Vec<Post>,
|
|
sort: (String, String),
|
|
ends: (String, String),
|
|
/// "overview", "comments", or "submitted"
|
|
listing: String,
|
|
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,
|
|
/// Whether all posts were hidden because they are NSFW (and user has disabled show NSFW)
|
|
all_posts_hidden_nsfw: bool,
|
|
no_posts: bool,
|
|
}
|
|
|
|
// FUNCTIONS
|
|
pub async fn profile(req: Request<Body>) -> Result<Response<Body>, String> {
|
|
let listing = req.param("listing").unwrap_or_else(|| "overview".to_string());
|
|
|
|
// Build the Reddit JSON API path
|
|
let path = format!(
|
|
"/user/{}/{listing}.json?{}&raw_json=1",
|
|
req.param("name").unwrap_or_else(|| "reddit".to_string()),
|
|
req.uri().query().unwrap_or_default(),
|
|
);
|
|
let url = String::from(req.uri().path_and_query().map_or("", |val| val.as_str()));
|
|
let redirect_url = url[1..].replace('?', "%3F").replace('&', "%26");
|
|
|
|
// Retrieve other variables from Redlib request
|
|
let sort = param(&path, "sort").unwrap_or_default();
|
|
let username = req.param("name").unwrap_or_default();
|
|
|
|
// Retrieve info from user about page.
|
|
let user = user(&username).await.unwrap_or_default();
|
|
|
|
let req_url = req.uri().to_string();
|
|
// Return landing page if this post if this Reddit deems this user NSFW,
|
|
// but we have also disabled the display of NSFW content or if the instance
|
|
// is SFW-only.
|
|
if user.nsfw && crate::utils::should_be_nsfw_gated(&req, &req_url) {
|
|
return Ok(nsfw_landing(req, req_url).await.unwrap_or_default());
|
|
}
|
|
|
|
let filters = get_filters(&req);
|
|
if filters.contains(&["u_", &username].concat()) {
|
|
Ok(template(&UserTemplate {
|
|
user,
|
|
posts: Vec::new(),
|
|
sort: (sort, param(&path, "t").unwrap_or_default()),
|
|
ends: (param(&path, "after").unwrap_or_default(), String::new()),
|
|
listing,
|
|
prefs: Preferences::new(&req),
|
|
url,
|
|
redirect_url,
|
|
is_filtered: true,
|
|
all_posts_filtered: false,
|
|
all_posts_hidden_nsfw: false,
|
|
no_posts: 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);
|
|
let no_posts = posts.is_empty();
|
|
let all_posts_hidden_nsfw = !no_posts && (posts.iter().all(|p| p.flags.nsfw) && setting(&req, "show_nsfw") != "on");
|
|
Ok(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,
|
|
all_posts_hidden_nsfw,
|
|
no_posts,
|
|
}))
|
|
}
|
|
// If there is an error show error page
|
|
Err(msg) => error(req, &msg).await,
|
|
}
|
|
}
|
|
}
|
|
|
|
// USER
|
|
async fn user(name: &str) -> Result<User, String> {
|
|
// Build the Reddit JSON API path
|
|
let path: String = format!("/user/{name}/about.json?raw_json=1");
|
|
|
|
// Send a request to the url
|
|
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);
|
|
|
|
// Closure used to parse JSON from Reddit APIs
|
|
let about = |item| res["data"]["subreddit"][item].as_str().unwrap_or_default().to_string();
|
|
|
|
// Parse the JSON output into a User struct
|
|
User {
|
|
name: res["data"]["name"].as_str().unwrap_or(name).to_owned(),
|
|
title: about("title"),
|
|
icon: format_url(&about("icon_img")),
|
|
karma: res["data"]["total_karma"].as_i64().unwrap_or(0),
|
|
created: created.format(format_description!("[month repr:short] [day] '[year repr:last_two]")).unwrap_or_default(),
|
|
banner: about("banner_img"),
|
|
description: about("public_description"),
|
|
nsfw: res["data"]["subreddit"]["over_18"].as_bool().unwrap_or_default(),
|
|
}
|
|
})
|
|
}
|
|
|
|
#[tokio::test(flavor = "multi_thread")]
|
|
async fn test_fetching_user() {
|
|
let user = user("spez").await;
|
|
assert!(user.is_ok());
|
|
assert!(user.unwrap().karma > 100);
|
|
}
|