redsunlib/src/post.rs

// CRATES
extern crate comrak;
use actix_web::{get, web, HttpResponse, Result};
use askama::Template;
use comrak::{markdown_to_html, ComrakOptions};
use chrono::{TimeZone, Utc};

// STRUCTS
#[derive(Template)]
#[template(path = "post.html", escape = "none")]
struct PostTemplate {
	comments: Vec<Comment>,
	post: Post,
	sort: String
}

pub struct Post {
	pub title: String,
	pub community: String,
	pub body: String,
	pub author: String,
	pub url: String,
	pub score: String,
	pub media: String,
	pub time: String
}

pub struct Comment {
	pub body: String,
	pub author: String,
	pub score: i64,
	pub time: String
}

async fn render(id: String, sort: String) -> Result<HttpResponse> {
	println!("id: {}", id);
	let post: Post = fetch_post(&id).await;
	let comments: Vec<Comment> = fetch_comments(id, &sort).await.unwrap();
	
	let s = PostTemplate {
		comments: comments,
		post: post,
		sort: sort
	}
	.render()
	.unwrap();
	Ok(HttpResponse::Ok().content_type("text/html").body(s))
}

// SERVICES
#[get("/{id}")]
async fn short(web::Path(id): web::Path<String>) -> Result<HttpResponse> {
	render(id.to_string(), "confidence".to_string()).await
}

#[get("/r/{sub}/comments/{id}/{title}")]
async fn page(web::Path((_sub, id)): web::Path<(String, String)>) -> Result<HttpResponse> {
	render(id.to_string(), "confidence".to_string()).await
}

#[get("/r/{sub}/comments/{id}/{title}/{sort}")]
async fn sorted(web::Path((_sub, id, _title, sort)): web::Path<(String, String, String, String)>) -> Result<HttpResponse> {
	render(id.to_string(), sort).await
}

// UTILITIES
async fn val (j: &serde_json::Value, k: &str) -> String { String::from(j["data"][k].as_str().unwrap_or("")) }

async fn media(data: &serde_json::Value) -> String {
	let post_hint: &str = data["data"]["post_hint"].as_str().unwrap_or("");
	let has_media: bool = data["data"]["media"].is_object();

	let media: String = if !has_media { format!(r#"<h4 class="post_body"><a href="{u}">{u}</a></h4>"#, u=data["data"]["url"].as_str().unwrap()) }
											else { format!(r#"<img class="post_image" src="{}.png"/>"#, data["data"]["url"].as_str().unwrap()) };

	match post_hint {
		"hosted:video" => format!(r#"<video class="post_image" src="{}" controls/>"#, data["data"]["media"]["reddit_video"]["fallback_url"].as_str().unwrap()),
		"image" => format!(r#"<img class="post_image" src="{}"/>"#, data["data"]["url"].as_str().unwrap()),
		"self" => String::from(""),
		_ => media
	}
}

// POSTS
// async fn post_html (post: Post) -> String {
// 	format!(r#"
// 		<div class="post" style="border: 2px solid #555;background: #222;">
// 			<div class="post_left" style="background: #333;">
// 				<button class="post_upvote">↑</button>
// 				<h3 class="post_score">{}</h3>
// 				<button class="post_upvote">↓</button>
// 			</div>
// 			<div class="post_right">
// 				<p>
// 					<b><a class="post_subreddit" href="/r/{sub}">r/{sub}</a></b>
// 					•
// 					Posted by 
// 					<a class="post_author" href="/u/{author}">u/{author}</a>
// 					<span style="float: right;">{time}</span>
// 				</p>
// 				<h3 class="post_title">{t}</h3>
// 				{media}
// 				<h4 class="post_body">{b}</h4>
// 			</div>
// 		</div><br>
// 	"#, if post.score>1000{format!("{}k", post.score/1000)} else {post.score.to_string()}, sub = post.community,
// 			author = post.author, t = post.title, media = post.media, b = post.body, time = post.time)
// }

async fn fetch_post (id: &String) -> Post {
	let url: String = format!("https://reddit.com/{}.json", id);
	let resp: String = reqwest::get(&url).await.unwrap().text().await.unwrap();
	
	let data: serde_json::Value = serde_json::from_str(resp.as_str()).expect("Failed to parse JSON");
	
	let post_data: &serde_json::Value = &data[0]["data"]["children"][0];

	let unix_time: i64 = post_data["data"]["created_utc"].as_f64().unwrap().round() as i64;
	let score = post_data["data"]["score"].as_i64().unwrap();

	Post {
		title: val(post_data, "title").await,
		community: val(post_data, "subreddit").await,
		body: markdown_to_html(post_data["data"]["selftext"].as_str().unwrap(), &ComrakOptions::default()),
		author: val(post_data, "author").await,
		url: val(post_data, "permalink").await,
		score: if score>1000 {format!("{}k",score/1000)} else {score.to_string()},
		media: media(post_data).await,
		time: Utc.timestamp(unix_time, 0).format("%b %e %Y %H:%M UTC").to_string()
	}
}

// COMMENTS
async fn fetch_comments (id: String, sort: &String) -> Result<Vec<Comment>, Box<dyn std::error::Error>> {
	let url: String = format!("https://reddit.com/{}.json?sort={}", id, sort);
	let resp: String = reqwest::get(&url).await?.text().await?;
	
	let data: serde_json::Value = serde_json::from_str(resp.as_str())?;
	
	let comment_data = data[1]["data"]["children"].as_array().unwrap();

	let mut comments: Vec<Comment> = Vec::new();
	
	for comment in comment_data.iter() {
		let unix_time: i64 = comment["data"]["created_utc"].as_f64().unwrap_or(0.0).round() as i64;
		comments.push(Comment {
			body: markdown_to_html(comment["data"]["body"].as_str().unwrap_or(""), &ComrakOptions::default()),
			author: val(comment, "author").await,
			score: comment["data"]["score"].as_i64().unwrap_or(0),
			time: Utc.timestamp(unix_time, 0).format("%b %e %Y %H:%M UTC").to_string()
		});
	}

	Ok(comments)
}

// async fn comments_html (comments: Vec<Comment>) -> String {
// 	let mut html: Vec<String> = Vec::new();
// 	for comment in comments.iter() {
// 		let hc: String = format!(r#"
// 			<div class="post">
// 				<div class="post_left">
// 					<button class="post_upvote">↑</button>
// 					<h3 class="post_score">{}</h3>
// 					<button class="post_upvote">↓</button>
// 				</div>
// 				<div class="post_right">
// 					<p>
// 						Posted by <a class="post_author" href="/u/{author}">u/{author}</a>
// 						<span style="float: right;">{time}</span>
// 					</p>
// 					<h4 class="post_body">{t}</h4>		
// 				</div>
// 			</div><br>
// 		"#, if comment.score>1000{format!("{}k", comment.score/1000)} else {comment.score.to_string()},
// 				author = comment.author, t = comment.body, time = comment.time);
// 		html.push(hc)
// 	}; html.join("\n")
// }