libreddit/src/subreddit.rs

// CRATES
use crate::utils::{error, fetch_posts, format_num, format_url, param, request, 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<Post>,
	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<HttpResponse> {
	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<HttpResponse> {
	let sub = req.match_info().get("sub").unwrap();
	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: res["data"]["content_html"].as_str().unwrap().to_string(),
				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<Subreddit, &'static str> {
	// 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::<Vec<&str>>()[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: val(&res, "description_html").replace("\\", ""),
				icon: format_url(icon).await,
				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),
	}
}