// CRATES use actix_web::{get, web, HttpResponse, Result}; use askama::Template; use chrono::{TimeZone, Utc}; #[path = "utils.rs"] mod utils; use utils::{nested_val, request, val, ErrorTemplate, Flair, Params, Post, User}; // STRUCTS #[derive(Template)] #[template(path = "user.html", escape = "none")] struct UserTemplate { user: User, posts: Vec, sort: String, } async fn render(username: String, sort: String) -> Result { let user = user(&username).await; let posts = posts(username, &sort).await; if user.is_err() || posts.is_err() { let s = ErrorTemplate { message: user.err().unwrap().to_string(), } .render() .unwrap(); Ok(HttpResponse::Ok().status(actix_web::http::StatusCode::NOT_FOUND).content_type("text/html").body(s)) } else { let s = UserTemplate { user: user.unwrap(), posts: posts.unwrap(), sort: sort, } .render() .unwrap(); Ok(HttpResponse::Ok().content_type("text/html").body(s)) } } // SERVICES #[get("/u/{username}")] async fn page(web::Path(username): web::Path, params: web::Query) -> Result { match ¶ms.sort { Some(sort) => render(username, sort.to_string()).await, None => render(username, "hot".to_string()).await, } } // USER async fn user(name: &String) -> Result { // Build the Reddit JSON API url let url: String = format!("https://www.reddit.com/user/{}/about.json", name); // Send a request to the url, receive JSON in response let req = request(url).await; // If the Reddit API returns an error, exit this function if req.is_err() { return Err(req.err().unwrap()); } // Otherwise, grab the JSON output from the request let res = req.unwrap(); // Parse the JSON output into a User struct Ok(User { name: name.to_string(), icon: nested_val(&res, "subreddit", "icon_img").await, karma: res["data"]["total_karma"].as_i64().unwrap(), banner: nested_val(&res, "subreddit", "banner_img").await, description: nested_val(&res, "subreddit", "public_description").await, }) } // POSTS async fn posts(sub: String, sort: &String) -> Result, &'static str> { // Build the Reddit JSON API url let url: String = format!("https://www.reddit.com/u/{}/.json?sort={}", sub, sort); // Send a request to the url, receive JSON in response let req = request(url).await; // If the Reddit API returns an error, exit this function if req.is_err() { return Err(req.err().unwrap()); } // Otherwise, grab the JSON output from the request let res = req.unwrap(); let post_list = res["data"]["children"].as_array().unwrap(); let mut posts: Vec = Vec::new(); for post in post_list.iter() { let img = if val(post, "thumbnail").await.starts_with("https:/") { val(post, "thumbnail").await } else { String::new() }; let unix_time: i64 = post["data"]["created_utc"].as_f64().unwrap().round() as i64; let score = post["data"]["score"].as_i64().unwrap(); let title = val(post, "title").await; posts.push(Post { title: if title.is_empty() { "Comment".to_string() } else { title }, community: val(post, "subreddit").await, body: String::new(), author: val(post, "author").await, score: if score > 1000 { format!("{}k", score / 1000) } else { score.to_string() }, media: img, url: val(post, "permalink").await, time: Utc.timestamp(unix_time, 0).format("%b %e '%y").to_string(), flair: Flair( val(post, "link_flair_text").await, val(post, "link_flair_background_color").await, if val(post, "link_flair_text_color").await == "dark" { "black".to_string() } else { "white".to_string() }, ), }); } Ok(posts) }