Actix Rewrite

src/post.rs

@@ -0,0 +1,179 @@
+// 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")
+// }