// CRATES use crate::esc; use crate::utils::{catch_random, error, format_num, format_url, param, redirect, rewrite_urls, setting, template, val, Post, Preferences, Subreddit}; use crate::{client::json, server::ResponseExt, RequestExt}; use askama::Template; use cookie::Cookie; use hyper::{Body, Request, Response}; use time::{Duration, OffsetDateTime}; // STRUCTS #[derive(Template)] #[template(path = "subreddit.html", escape = "none")] struct SubredditTemplate { sub: Subreddit, posts: Vec, sort: (String, String), ends: (String, String), prefs: Preferences, url: String, } #[derive(Template)] #[template(path = "wiki.html", escape = "none")] struct WikiTemplate { sub: String, wiki: String, page: String, prefs: Preferences, } #[derive(Template)] #[template(path = "wall.html", escape = "none")] struct WallTemplate { title: String, sub: String, msg: String, prefs: Preferences, url: String, } // SERVICES pub async fn community(req: Request) -> Result, String> { // Build Reddit API path let root = req.uri().path() == "/"; let subscribed = setting(&req, "subscriptions"); let front_page = setting(&req, "front_page"); let post_sort = req.cookie("post_sort").map_or_else(|| "hot".to_string(), |c| c.value().to_string()); let sort = req.param("sort").unwrap_or_else(|| req.param("id").unwrap_or(post_sort)); let sub = req.param("sub").unwrap_or(if front_page == "default" || front_page.is_empty() { if subscribed.is_empty() { "popular".to_string() } else { subscribed.to_owned() } } else { front_page.to_owned() }); let quarantined = can_access_quarantine(&req, &sub) || root; // Handle random subreddits if let Ok(random) = catch_random(&sub, "").await { return Ok(random); } if req.param("sub").is_some() && sub.starts_with("u_") { return Ok(redirect(["/user/", &sub[2..]].concat())); } let path = format!("/r/{}/{}.json?{}&raw_json=1", sub, sort, req.uri().query().unwrap_or_default()); match Post::fetch(&path, String::new(), quarantined).await { Ok((posts, after)) => { // If you can get subreddit posts, also request subreddit metadata let sub = if !sub.contains('+') && sub != subscribed && sub != "popular" && sub != "all" { // Regular subreddit subreddit(&sub, quarantined).await.unwrap_or_default() } else if sub == subscribed { // Subscription feed if req.uri().path().starts_with("/r/") { subreddit(&sub, quarantined).await.unwrap_or_default() } else { Subreddit::default() } } else if sub.contains('+') { // Multireddit Subreddit { name: sub, ..Subreddit::default() } } else { Subreddit::default() }; let url = String::from(req.uri().path_and_query().map_or("", |val| val.as_str())); template(SubredditTemplate { sub, posts, sort: (sort, param(&path, "t").unwrap_or_default()), ends: (param(&path, "after").unwrap_or_default(), after), prefs: Preferences::new(req), url, }) } Err(msg) => match msg.as_str() { "quarantined" => quarantine(req, sub), "private" => error(req, format!("r/{} is a private community", sub)).await, "banned" => error(req, format!("r/{} has been banned from Reddit", sub)).await, _ => error(req, msg).await, }, } } pub fn quarantine(req: Request, sub: String) -> Result, String> { let wall = WallTemplate { title: format!("r/{} is quarantined", sub), msg: "Please click the button below to continue to this subreddit.".to_string(), url: req.uri().to_string(), sub, prefs: Preferences::new(req), }; Ok( Response::builder() .status(403) .header("content-type", "text/html") .body(wall.render().unwrap_or_default().into()) .unwrap_or_default(), ) } pub async fn add_quarantine_exception(req: Request) -> Result, String> { let subreddit = req.param("sub").ok_or("Invalid URL")?; let redir = param(&format!("?{}", req.uri().query().unwrap_or_default()), "redir").ok_or("Invalid URL")?; let mut res = redirect(redir); res.insert_cookie( Cookie::build(&format!("allow_quaran_{}", subreddit.to_lowercase()), "true") .path("/") .http_only(true) .expires(cookie::Expiration::Session) .finish(), ); Ok(res) } pub fn can_access_quarantine(req: &Request, sub: &str) -> bool { // Determine if the subreddit can be accessed setting(&req, &format!("allow_quaran_{}", sub.to_lowercase())).parse().unwrap_or_default() } // Sub or unsub by setting subscription cookie using response "Set-Cookie" header pub async fn subscriptions(req: Request) -> Result, String> { let sub = req.param("sub").unwrap_or_default(); // Handle random subreddits if sub == "random" || sub == "randnsfw" { return Err("Can't subscribe to random subreddit!".to_string()); } let query = req.uri().query().unwrap_or_default().to_string(); let action: Vec = req.uri().path().split('/').map(String::from).collect(); let mut sub_list = Preferences::new(req).subscriptions; // Retrieve list of posts for these subreddits to extract display names let posts = json(format!("/r/{}/hot.json?raw_json=1", sub), true).await?; let display_lookup: Vec<(String, &str)> = posts["data"]["children"] .as_array() .map(|list| { list .iter() .map(|post| { let display_name = post["data"]["subreddit"].as_str().unwrap_or_default(); (display_name.to_lowercase(), display_name) }) .collect::>() }) .unwrap_or_default(); // Find each subreddit name (separated by '+') in sub parameter for part in sub.split('+') { // Retrieve display name for the subreddit let display; let part = if let Some(&(_, display)) = display_lookup.iter().find(|x| x.0 == part.to_lowercase()) { // This is already known, doesn't require seperate request display } else { // This subreddit display name isn't known, retrieve it let path: String = format!("/r/{}/about.json?raw_json=1", part); display = json(path, true).await?; display["data"]["display_name"].as_str().ok_or_else(|| "Failed to query subreddit name".to_string())? }; // Modify sub list based on action if action.contains(&"subscribe".to_string()) && !sub_list.contains(&part.to_owned()) { // Add each sub name to the subscribed list sub_list.push(part.to_owned()); // Reorder sub names alphabettically sub_list.sort_by_key(|a| a.to_lowercase()) } else if action.contains(&"unsubscribe".to_string()) { // Remove sub name from subscribed list sub_list.retain(|s| s.to_lowercase() != part.to_lowercase()); } } // Redirect back to subreddit // check for redirect parameter if unsubscribing from outside sidebar let path = if let Some(redirect_path) = param(&format!("?{}", query), "redirect") { format!("/{}/", redirect_path) } else { format!("/r/{}", sub) }; let mut res = redirect(path); // Delete cookie if empty, else set if sub_list.is_empty() { res.remove_cookie("subscriptions".to_string()); } else { res.insert_cookie( Cookie::build("subscriptions", sub_list.join("+")) .path("/") .http_only(true) .expires(OffsetDateTime::now_utc() + Duration::weeks(52)) .finish(), ); } Ok(res) } pub async fn wiki(req: Request) -> Result, String> { let sub = req.param("sub").unwrap_or_else(|| "reddit.com".to_string()); let quarantined = can_access_quarantine(&req, &sub); // Handle random subreddits if let Ok(random) = catch_random(&sub, "/wiki").await { return Ok(random); } let page = req.param("page").unwrap_or_else(|| "index".to_string()); let path: String = format!("/r/{}/wiki/{}.json?raw_json=1", sub, page); match json(path, quarantined).await { Ok(response) => template(WikiTemplate { sub, wiki: rewrite_urls(response["data"]["content_html"].as_str().unwrap_or("

Wiki not found

")), page, prefs: Preferences::new(req), }), Err(msg) => { if msg == "quarantined" { quarantine(req, sub) } else { error(req, msg).await } } } } pub async fn sidebar(req: Request) -> Result, String> { let sub = req.param("sub").unwrap_or_else(|| "reddit.com".to_string()); let quarantined = can_access_quarantine(&req, &sub); // Handle random subreddits if let Ok(random) = catch_random(&sub, "/about/sidebar").await { return Ok(random); } // Build the Reddit JSON API url let path: String = format!("/r/{}/about.json?raw_json=1", sub); // Send a request to the url match json(path, quarantined).await { // If success, receive JSON in response Ok(response) => template(WikiTemplate { wiki: format!( "{}

Moderators


    {}
", rewrite_urls(&val(&response, "description_html").replace("\\", "")), moderators(&sub, quarantined).await?.join(""), ), sub, page: "Sidebar".to_string(), prefs: Preferences::new(req), }), Err(msg) => { if msg == "quarantined" { quarantine(req, sub) } else { error(req, msg).await } } } } pub async fn moderators(sub: &str, quarantined: bool) -> Result, String> { // Retrieve and format the html for the moderators list Ok( moderators_list(sub, quarantined) .await? .iter() .map(|m| format!("
  • {name}
    • ", name = m)) .collect(), ) } async fn moderators_list(sub: &str, quarantined: bool) -> Result, String> { // Build the moderator list URL let path: String = format!("/r/{}/about/moderators.json?raw_json=1", sub); // Retrieve response let response = json(path, quarantined).await?["data"]["children"].clone(); Ok( // Traverse json tree and format into list of strings response .as_array() .unwrap_or(&Vec::new()) .iter() .filter_map(|moderator| { let name = moderator["name"].as_str().unwrap_or_default(); if name.is_empty() { None } else { Some(name.to_string()) } }) .collect::>(), ) } // SUBREDDIT async fn subreddit(sub: &str, quarantined: bool) -> Result { // Build the Reddit JSON API url let path: String = format!("/r/{}/about.json?raw_json=1", sub); // Send a request to the url match json(path, quarantined).await { // If success, receive JSON in response Ok(res) => { // Metadata regarding the subreddit let members: i64 = res["data"]["subscribers"].as_u64().unwrap_or_default() as i64; let active: i64 = res["data"]["accounts_active"].as_u64().unwrap_or_default() as i64; // Fetch subreddit icon either from the community_icon or icon_img value let community_icon: &str = res["data"]["community_icon"].as_str().unwrap_or_default(); let icon = if community_icon.is_empty() { val(&res, "icon_img") } else { community_icon.to_string() }; let sub = Subreddit { name: esc!(&res, "display_name"), title: esc!(&res, "title"), description: esc!(&res, "public_description"), info: rewrite_urls(&val(&res, "description_html").replace("\\", "")), moderators: moderators_list(sub, quarantined).await?, icon: format_url(&icon), members: format_num(members), active: format_num(active), wiki: res["data"]["wiki_enabled"].as_bool().unwrap_or_default(), }; Ok(sub) } // If the Reddit API returns an error, exit this function Err(msg) => return Err(msg), } }