libreddit/src/user.rs

79 lines
2.3 KiB
Rust
Raw Normal View History

2020-10-25 21:25:59 +01:00
// CRATES
2021-01-14 00:55:10 +01:00
use crate::utils::{error, fetch_posts, format_url, param, prefs, request, Post, Preferences, User};
2021-01-01 06:03:44 +01:00
use actix_web::{HttpRequest, HttpResponse, Result};
2020-10-25 21:25:59 +01:00
use askama::Template;
2021-01-06 05:01:21 +01:00
use time::OffsetDateTime;
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),
2021-01-09 02:35:04 +01:00
prefs: Preferences,
2020-10-25 21:25:59 +01:00
}
2021-01-03 05:50:23 +01:00
// FUNCTIONS
pub async fn profile(req: HttpRequest) -> HttpResponse {
2021-01-01 00:54:13 +01:00
// Build the Reddit JSON API path
let path = format!("{}.json?{}&raw_json=1", req.path(), req.query_string());
2020-12-27 21:36:10 +01:00
2021-01-01 00:54:13 +01:00
// Retrieve other variables from Libreddit request
2021-01-01 21:33:57 +01:00
let sort = param(&path, "sort");
2021-01-01 00:54:13 +01:00
let username = req.match_info().get("username").unwrap_or("").to_string();
2020-11-21 07:05:27 +01:00
2021-01-01 00:54:13 +01:00
// Request user profile data and user posts/comments from Reddit
let user = user(&username).await.unwrap_or_default();
2021-01-01 21:55:09 +01:00
let posts = fetch_posts(&path, "Comment".to_string()).await;
2021-01-02 07:21:43 +01:00
2021-01-02 00:28:13 +01:00
match posts {
2021-01-07 06:27:24 +01:00
Ok((posts, after)) => {
2021-01-02 00:28:13 +01:00
let s = UserTemplate {
user,
2021-01-07 06:27:24 +01:00
posts,
2021-01-02 00:28:13 +01:00
sort: (sort, param(&path, "t")),
2021-01-07 06:27:24 +01:00
ends: (param(&path, "after"), after),
2021-01-09 02:35:04 +01:00
prefs: prefs(req),
2021-01-02 00:28:13 +01:00
}
.render()
.unwrap();
2021-01-03 05:50:23 +01:00
HttpResponse::Ok().content_type("text/html").body(s)
2021-01-02 07:21:43 +01:00
}
2021-01-02 00:28:13 +01:00
// If there is an error show error page
2021-01-09 06:11:20 +01:00
Err(msg) => error(msg.to_string()).await,
2020-11-20 05:42:18 +01:00
}
2020-10-25 21:25:59 +01:00
}
// USER
2021-01-01 21:33:57 +01:00
async fn user(name: &str) -> Result<User, &'static str> {
// Build the Reddit JSON API path
2021-01-11 19:33:48 +01:00
let path: String = format!("/user/{}/about.json", name);
2021-01-01 06:03:44 +01:00
2021-01-02 00:28:13 +01:00
// Send a request to the url
match request(&path).await {
// If success, receive JSON in response
2021-01-09 02:35:04 +01:00
Ok(res) => {
// Grab creation date as unix timestamp
let created: i64 = res["data"]["created"].as_f64().unwrap_or(0.0).round() as i64;
2021-01-14 00:55:10 +01:00
// nested_val function 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
// Parse the JSON output into a User struct
Ok(User {
name: name.to_string(),
2021-01-14 00:55:10 +01:00
title: about("title"),
icon: format_url(about("icon_img").as_str()),
2021-01-09 02:35:04 +01:00
karma: res["data"]["total_karma"].as_i64().unwrap_or(0),
created: OffsetDateTime::from_unix_timestamp(created).format("%b %d '%y"),
2021-01-14 00:55:10 +01:00
banner: about("banner_img"),
description: about("public_description"),
2021-01-09 02:35:04 +01:00
})
2021-01-02 07:21:43 +01:00
}
2021-01-02 00:28:13 +01:00
// If the Reddit API returns an error, exit this function
2021-01-02 07:21:43 +01:00
Err(msg) => return Err(msg),
2020-11-20 05:42:18 +01:00
}
2020-11-30 03:50:29 +01:00
}