// CRATES use crate::utils::{error, fetch_posts, format_num, format_url, param, request, rewrite_url, val, Post, Subreddit}; use actix_web::{HttpRequest, HttpResponse, Result}; use askama::Template; // STRUCTS #[derive(Template)] #[template(path = "subreddit.html", escape = "none")] struct SubredditTemplate { sub: Subreddit, posts: Vec, sort: (String, String), ends: (String, String), } #[derive(Template)] #[template(path = "wiki.html", escape = "none")] struct WikiTemplate { sub: String, wiki: String, page: String, } // SERVICES pub async fn page(req: HttpRequest) -> Result { let path = format!("{}.json?{}", req.path(), req.query_string()); let sub = req.match_info().get("sub").unwrap_or("popular").to_string(); let sort = req.match_info().get("sort").unwrap_or("hot").to_string(); let sub_result = if !&sub.contains('+') && sub != "popular" { subreddit(&sub).await.unwrap_or_default() } else { Subreddit::default() }; match fetch_posts(&path, String::new()).await { Ok(items) => { let s = SubredditTemplate { sub: sub_result, posts: items.0, sort: (sort, param(&path, "t")), ends: (param(&path, "after"), items.1), } .render() .unwrap(); Ok(HttpResponse::Ok().content_type("text/html").body(s)) } Err(msg) => error(msg.to_string()).await, } } pub async fn wiki(req: HttpRequest) -> Result { let sub = req.match_info().get("sub").unwrap_or("reddit.com"); let page = req.match_info().get("page").unwrap_or("index"); let path: String = format!("r/{}/wiki/{}.json?raw_json=1", sub, page); match request(&path).await { Ok(res) => { let s = WikiTemplate { sub: sub.to_string(), wiki: rewrite_url(res["data"]["content_html"].as_str().unwrap()), page: page.to_string(), } .render() .unwrap(); Ok(HttpResponse::Ok().content_type("text/html").body(s)) } Err(msg) => error(msg.to_string()).await, } } // SUBREDDIT async fn subreddit(sub: &str) -> 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 request(&path).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("").split('?').collect::>()[0]; let icon = if community_icon.is_empty() { val(&res, "icon_img") } else { community_icon.to_string() }; let sub = Subreddit { name: val(&res, "display_name"), title: val(&res, "title"), description: val(&res, "public_description"), info: rewrite_url(&val(&res, "description_html").replace("\\", "")), 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), } }