From 0579d6f61bcbc634b58ba0b0bfcb787ef2c4e375 Mon Sep 17 00:00:00 2001
From: Sonny <sonnyba871@gmail.com>
Date: Sun, 19 Jul 2020 13:06:26 +0200
Subject: [PATCH] Show gifv as mp4 video's

---
 src/newsreader/news/collection/reddit.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/newsreader/news/collection/reddit.py b/src/newsreader/news/collection/reddit.py
index 128131f..7af7eb1 100644
--- a/src/newsreader/news/collection/reddit.py
+++ b/src/newsreader/news/collection/reddit.py
@@ -43,6 +43,7 @@ RATE_LIMIT = 60
 RATE_LIMIT_DURATION = timedelta(seconds=60)
 
 REDDIT_IMAGE_EXTENSIONS = (".jpg", ".png", ".gif")
+REDDIT_VIDEO_EXTENSIONS = (".mp4", ".gifv", ".webm")
 
 
 def get_reddit_authorization_url(user):
@@ -149,6 +150,17 @@ class RedditBuilder(Builder):
                 video_info = post["data"]["secure_media"]["reddit_video"]
 
                 body = f"<div><video controls muted><source src='{video_info['fallback_url']}' type='video/mp4' /></video></div>"
+            elif not is_text_post and direct_url.endswith(REDDIT_VIDEO_EXTENSIONS):
+                extension = next(
+                    extension.replace(".", "")
+                    for extension in REDDIT_VIDEO_EXTENSIONS
+                    if direct_url.endswith(extension)
+                )
+
+                if extension == "gifv":
+                    body = f"<div><video controls muted><source src='{direct_url.replace(extension, 'mp4')}' type='video/mp4' /></video></div>"
+                else:
+                    body = f"<div><video controls muted><source src='{direct_url}' type='video/{extension}' /></video></div>"
 
             try:
                 parsed_date = datetime.fromtimestamp(post["data"]["created_utc"])