redsunlib/src/user.rs

85 lines
2.4 KiB
Rust
Raw Normal View History

2020-10-26 09:25:59 +13:00
// CRATES
2021-03-18 11:30:33 +13:00
use crate::client::json;
2021-03-12 17:15:26 +13:00
use crate::esc;
2021-03-18 11:30:33 +13:00
use crate::server::RequestExt;
use crate::utils::{error, format_url, param, template, Post, Preferences, User};
2020-10-26 09:25:59 +13:00
use askama::Template;
2021-03-18 11:30:33 +13:00
use hyper::{Body, Request, Response};
2021-01-06 17:01:21 +13:00
use time::OffsetDateTime;
2020-11-18 08:37:40 +13:00
2020-10-26 09:25:59 +13:00
// STRUCTS
#[derive(Template)]
#[template(path = "user.html", escape = "none")]
struct UserTemplate {
user: User,
posts: Vec<Post>,
2020-12-30 14:11:47 +13:00
sort: (String, String),
2020-12-28 09:36:10 +13:00
ends: (String, String),
2021-01-09 14:35:04 +13:00
prefs: Preferences,
2020-10-26 09:25:59 +13:00
}
2021-01-03 17:50:23 +13:00
// FUNCTIONS
2021-03-18 11:30:33 +13:00
pub async fn profile(req: Request<Body>) -> Result<Response<Body>, String> {
2021-01-01 12:54:13 +13:00
// Build the Reddit JSON API path
2021-03-18 17:40:55 +13:00
let path = format!(
"/user/{}.json?{}&raw_json=1",
req.param("name").unwrap_or("reddit".to_string()),
req.uri().query().unwrap_or_default()
);
2020-12-28 09:36:10 +13:00
2021-01-01 12:54:13 +13:00
// Retrieve other variables from Libreddit request
2021-01-02 09:33:57 +13:00
let sort = param(&path, "sort");
2021-03-18 11:30:33 +13:00
let username = req.param("name").unwrap_or_default();
2020-11-21 19:05:27 +13:00
2021-01-16 12:05:55 +13:00
// Request user posts/comments from Reddit
2021-02-25 18:29:23 +13:00
let posts = Post::fetch(&path, "Comment".to_string()).await;
2021-01-02 19:21:43 +13:00
2021-01-02 12:28:13 +13:00
match posts {
2021-01-07 18:27:24 +13:00
Ok((posts, after)) => {
2021-01-16 12:05:55 +13:00
// If you can get user posts, also request user data
let user = user(&username).await.unwrap_or_default();
template(UserTemplate {
user,
2021-01-07 18:27:24 +13:00
posts,
2021-01-02 12:28:13 +13:00
sort: (sort, param(&path, "t")),
2021-01-07 18:27:24 +13:00
ends: (param(&path, "after"), after),
2021-02-25 18:29:23 +13:00
prefs: Preferences::new(req),
})
2021-01-02 19:21:43 +13:00
}
2021-01-02 12:28:13 +13:00
// If there is an error show error page
2021-02-21 15:36:30 +13:00
Err(msg) => error(req, msg).await,
2020-11-20 17:42:18 +13:00
}
2020-10-26 09:25:59 +13:00
}
// USER
2021-01-15 06:53:54 +13:00
async fn user(name: &str) -> Result<User, String> {
// Build the Reddit JSON API path
let path: String = format!("/user/{}/about.json?raw_json=1", name);
2021-01-01 18:03:44 +13:00
2021-01-02 12:28:13 +13:00
// Send a request to the url
2021-03-18 11:30:33 +13:00
match json(path).await {
2021-01-02 12:28:13 +13:00
// If success, receive JSON in response
2021-01-09 14:35:04 +13:00
Ok(res) => {
// Grab creation date as unix timestamp
2021-03-10 04:22:17 +13:00
let created: i64 = res["data"]["created"].as_f64().unwrap_or(0.0).round() as i64;
2021-03-12 17:15:26 +13:00
// Closure used to parse JSON from Reddit APIs
2021-01-14 12:55:10 +13:00
let about = |item| res["data"]["subreddit"][item].as_str().unwrap_or_default().to_string();
2021-01-09 14:35:04 +13:00
// Parse the JSON output into a User struct
Ok(User {
name: name.to_string(),
2021-03-12 17:15:26 +13:00
title: esc!(about("title")),
2021-02-21 10:59:16 +13:00
icon: format_url(&about("icon_img")),
2021-01-09 14:35:04 +13:00
karma: res["data"]["total_karma"].as_i64().unwrap_or(0),
created: OffsetDateTime::from_unix_timestamp(created).format("%b %d '%y"),
2021-03-12 17:15:26 +13:00
banner: esc!(about("banner_img")),
2021-01-14 12:55:10 +13:00
description: about("public_description"),
2021-01-09 14:35:04 +13:00
})
2021-01-02 19:21:43 +13:00
}
2021-01-02 12:28:13 +13:00
// If the Reddit API returns an error, exit this function
2021-01-02 19:21:43 +13:00
Err(msg) => return Err(msg),
2020-11-20 17:42:18 +13:00
}
2020-11-30 15:50:29 +13:00
}