From cf45d53fddd8862c097548e146f6551ecb58d34a Mon Sep 17 00:00:00 2001
From: robrobinbin <8597693+robrobinbin@users.noreply.github.com>
Date: Sat, 6 Feb 2021 21:05:11 +0100
Subject: [PATCH] Basic gallery support (#103)

---
 src/post.rs         |  3 ++-
 src/utils.rs        | 35 +++++++++++++++++++++++++++++++++--
 static/style.css    | 20 +++++++++++++++++++-
 templates/post.html | 14 ++++++++++++++
 4 files changed, 68 insertions(+), 4 deletions(-)

diff --git a/src/post.rs b/src/post.rs
index 1b67d03..3bf810d 100644
--- a/src/post.rs
+++ b/src/post.rs
@@ -72,7 +72,7 @@ async fn parse_post(json: &serde_json::Value) -> Post {
 	let ratio: f64 = post["data"]["upvote_ratio"].as_f64().unwrap_or(1.0) * 100.0;
 
 	// Determine the type of media along with the media URL
-	let (post_type, media) = media(&post["data"]).await;
+	let (post_type, media, gallery) = media(&post["data"]).await;
 
 	// Build a post using data parsed from Reddit post API
 	Post {
@@ -124,6 +124,7 @@ async fn parse_post(json: &serde_json::Value) -> Post {
 		rel_time,
 		created,
 		comments: format_num(post["data"]["num_comments"].as_i64().unwrap_or_default()),
+		gallery,
 	}
 }
 
diff --git a/src/utils.rs b/src/utils.rs
index f834055..512784c 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -46,6 +46,14 @@ pub struct Media {
 	pub height: i64,
 }
 
+pub struct GalleryMedia {
+	pub url: String,
+	pub width: i64,
+	pub height: i64,
+	pub caption: String,
+	pub outbound_url: String,	
+}
+
 // Post containing content, metadata and media
 pub struct Post {
 	pub id: String,
@@ -65,6 +73,7 @@ pub struct Post {
 	pub rel_time: String,
 	pub created: String,
 	pub comments: String,
+	pub gallery: Vec<GalleryMedia>,
 }
 
 // Comment with content, post, score and data/time that it was posted
@@ -188,8 +197,9 @@ pub fn format_num(num: i64) -> String {
 	}
 }
 
-pub async fn media(data: &Value) -> (String, Media) {
+pub async fn media(data: &Value) -> (String, Media, Vec<GalleryMedia>) {
 	let post_type: &str;
+	let mut gallery = Vec::new();
 	// If post is a video, return the video
 	let url = if data["preview"]["reddit_video_preview"]["fallback_url"].is_string() {
 		post_type = "video";
@@ -215,6 +225,25 @@ pub async fn media(data: &Value) -> (String, Media) {
 	} else if data["is_self"].as_bool().unwrap_or_default() {
 		post_type = "self";
 		data["permalink"].as_str().unwrap_or_default().to_string()
+	} else if data["is_gallery"].as_bool().unwrap_or_default() {
+		post_type = "gallery";		
+		gallery = data["gallery_data"]["items"]
+			.as_array()
+			.unwrap()
+			.iter()
+			.map(|item| {
+				let media_id = item["media_id"].as_str().unwrap_or_default();
+				let image = data["media_metadata"][media_id].as_object().unwrap();
+				GalleryMedia {
+					url: format_url(image["s"]["u"].as_str().unwrap_or_default()),
+					width: image["s"]["x"].as_i64().unwrap_or_default(),
+					height: image["s"]["y"].as_i64().unwrap_or_default(),
+					caption: item["caption"].as_str().unwrap_or_default().to_string(),
+					outbound_url: item["outbound_url"].as_str().unwrap_or_default().to_string(),
+				}
+			})
+			.collect::<Vec<GalleryMedia>>();
+		data["url"].as_str().unwrap_or_default().to_string()
 	} else {
 		post_type = "link";
 		data["url"].as_str().unwrap_or_default().to_string()
@@ -227,6 +256,7 @@ pub async fn media(data: &Value) -> (String, Media) {
 			width: data["preview"]["images"][0]["source"]["width"].as_i64().unwrap_or_default(),
 			height: data["preview"]["images"][0]["source"]["height"].as_i64().unwrap_or_default(),
 		},
+		gallery,
 	)
 }
 
@@ -323,7 +353,7 @@ pub async fn fetch_posts(path: &str, fallback_title: String) -> Result<(Vec<Post
 		let title = val(post, "title");
 
 		// Determine the type of media along with the media URL
-		let (post_type, media) = media(&post["data"]).await;
+		let (post_type, media, gallery) = media(&post["data"]).await;
 
 		posts.push(Post {
 			id: val(post, "id"),
@@ -378,6 +408,7 @@ pub async fn fetch_posts(path: &str, fallback_title: String) -> Result<(Vec<Post
 			rel_time,
 			created,
 			comments: format_num(post["data"]["num_comments"].as_i64().unwrap_or_default()),
+			gallery,
 		});
 	}
 
diff --git a/static/style.css b/static/style.css
index 9192936..c053703 100644
--- a/static/style.css
+++ b/static/style.css
@@ -637,7 +637,7 @@ a.search_subreddit:hover {
 	font-weight: bold;
 }
 
-.post_media_image, .post .__NoScript_PlaceHolder__, .post_media_video  {
+.post_media_image, .post .__NoScript_PlaceHolder__, .post_media_video, .gallery  {
 	max-width: calc(100% - 40px);
 	grid-area: post_media;
 	margin: 15px auto 5px auto;
@@ -664,6 +664,24 @@ a.search_subreddit:hover {
 	vertical-align: bottom;
 }
 
+.gallery img {
+	max-width: 100%;
+	vertical-align: bottom;
+}
+
+.gallery figcaption {
+	margin-top: 5px;
+}
+
+.gallery .outbound_url {
+	color: var(--accent);
+	text-overflow: ellipsis;
+	overflow: hidden;
+	white-space: nowrap;
+	display: block;
+	padding-top: 5px;
+}
+
 #post_url {
 	color: var(--accent);
 	margin: 5px 20px;
diff --git a/templates/post.html b/templates/post.html
index b9e78f3..6fac69e 100644
--- a/templates/post.html
+++ b/templates/post.html
@@ -78,6 +78,20 @@
 			</a>
 			{% else if post.post_type == "video" || post.post_type == "gif" %}
 			<video class="post_media_video" src="{{ post.media.url }}" controls autoplay loop></video>
+			{% else if post.post_type == "gallery" %}
+			<div class="gallery">
+			{% for image in post.gallery -%}
+				<figure>
+					<a href="{{ image.url }}" ><img alt="Gallery image" src="{{ image.url }}"/></a>
+					<figcaption>
+						<p>{{ image.caption }}</p>
+						{% if image.outbound_url.len() > 0 %}
+						<p><a class="outbound_url" href="{{ image.outbound_url }}">{{ image.outbound_url }}</a>
+						{% endif %}
+					</figcaption>
+				</figure>
+			{%- endfor %}
+			</div>			
 			{% else if post.post_type == "link" %}
 			<a id="post_url" href="{{ post.media.url }}">{{ post.media.url }}</a>
 			{% endif %}