libreddit/src/subreddit.rs

121 lines
3.4 KiB
Rust
Raw Normal View History

2020-10-25 21:25:59 +01:00
// CRATES
2021-01-17 00:13:34 +01:00
use crate::utils::*;
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;
2020-11-17 20:37:40 +01:00
2020-10-25 21:25:59 +01:00
// STRUCTS
#[derive(Template)]
#[template(path = "subreddit.html", escape = "none")]
struct SubredditTemplate {
sub: Subreddit,
posts: Vec<Post>,
2020-12-30 02:11:47 +01:00
sort: (String, String),
2020-11-30 03:50:29 +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-02 07:21:43 +01:00
#[derive(Template)]
#[template(path = "wiki.html", escape = "none")]
struct WikiTemplate {
sub: String,
wiki: String,
2021-01-02 19:58:21 +01:00
page: String,
2021-01-10 22:08:36 +01:00
prefs: Preferences,
2021-01-02 07:21:43 +01:00
}
2020-11-19 22:49:32 +01:00
// SERVICES
2021-01-03 05:50:23 +01:00
pub async fn page(req: HttpRequest) -> HttpResponse {
2021-01-01 00:54:13 +01:00
let path = format!("{}.json?{}", req.path(), req.query_string());
2021-01-09 05:55:40 +01:00
let default = cookie(&req, "front_page");
let sub_name = req
.match_info()
.get("sub")
.unwrap_or(if default.is_empty() { "popular" } else { default.as_str() })
.to_string();
2021-01-01 00:54:13 +01:00
let sort = req.match_info().get("sort").unwrap_or("hot").to_string();
2021-01-02 00:28:13 +01:00
match fetch_posts(&path, String::new()).await {
2021-01-07 17:38:05 +01:00
Ok((posts, after)) => {
2021-01-16 00:05:55 +01:00
// If you can get subreddit posts, also request subreddit metadata
let sub = if !sub_name.contains('+') && sub_name != "popular" && sub_name != "all" {
subreddit(&sub_name).await.unwrap_or_default()
} else if sub_name.contains('+') {
Subreddit {
name: sub_name,
..Subreddit::default()
}
} else {
Subreddit::default()
};
2021-01-02 00:28:13 +01:00
let s = SubredditTemplate {
2021-01-07 17:38:05 +01:00
sub,
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
}
Err(msg) => error(msg).await,
2021-01-02 07:21:43 +01:00
}
}
2021-01-03 05:50:23 +01:00
pub async fn wiki(req: HttpRequest) -> HttpResponse {
let sub = req.match_info().get("sub").unwrap_or("reddit.com").to_string();
let page = req.match_info().get("page").unwrap_or("index").to_string();
2021-01-11 19:33:42 +01:00
let path: String = format!("/r/{}/wiki/{}.json?raw_json=1", sub, page);
2021-01-02 07:21:43 +01:00
match request(&path).await {
Ok(res) => {
let s = WikiTemplate {
sub,
2021-01-04 06:31:21 +01:00
wiki: rewrite_url(res["data"]["content_html"].as_str().unwrap_or_default()),
page,
2021-01-10 22:08:36 +01:00
prefs: prefs(req),
2021-01-02 07:21:43 +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
}
Err(msg) => error(msg).await,
2020-11-18 01:03:28 +01:00
}
2020-10-25 21:25:59 +01:00
}
// SUBREDDIT
2021-01-14 18:53:54 +01:00
async fn subreddit(sub: &str) -> Result<Subreddit, String> {
2020-11-19 03:50:59 +01:00
// Build the Reddit JSON API url
2021-01-12 01:44:31 +01:00
let path: String = format!("/r/{}/about.json?raw_json=1", sub);
2020-10-25 21:25:59 +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-02 07:21:43 +01:00
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;
2020-11-20 05:42:18 +01:00
2021-01-02 07:21:43 +01:00
// Fetch subreddit icon either from the community_icon or icon_img value
2021-01-22 06:25:51 +01:00
let community_icon: &str = res["data"]["community_icon"].as_str().map_or("", |s| s.split('?').collect::<Vec<&str>>()[0]);
2021-01-02 07:21:43 +01:00
let icon = if community_icon.is_empty() { val(&res, "icon_img") } else { community_icon.to_string() };
2020-10-25 21:25:59 +01:00
2021-01-02 07:21:43 +01:00
let sub = Subreddit {
name: val(&res, "display_name"),
title: val(&res, "title"),
description: val(&res, "public_description"),
2021-01-02 20:09:26 +01:00
info: rewrite_url(&val(&res, "description_html").replace("\\", "")),
2021-01-12 02:47:14 +01:00
icon: format_url(icon.as_str()),
2021-01-02 07:21:43 +01:00
members: format_num(members),
active: format_num(active),
wiki: res["data"]["wiki_enabled"].as_bool().unwrap_or_default(),
};
2020-12-24 05:36:49 +01:00
2021-01-02 07:21:43 +01:00
Ok(sub)
}
// If the Reddit API returns an error, exit this function
Err(msg) => return Err(msg),
}
2020-11-30 03:50:29 +01:00
}