From d6ffd593a134e110f81652c95bb37865925e02f1 Mon Sep 17 00:00:00 2001
From: Floens <floens@gmail.com>
Date: Fri, 26 Jan 2018 10:38:31 +0100
Subject: [PATCH] fix vichan thumbnail urls.

---
 .../org/floens/chan/core/model/PostImage.java     |  1 +
 .../chan/core/site/sites/vichan/ViChan.java       | 15 ++++++++++++++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/Clover/app/src/main/java/org/floens/chan/core/model/PostImage.java b/Clover/app/src/main/java/org/floens/chan/core/model/PostImage.java
index 1375b6be..28ced983 100644
--- a/Clover/app/src/main/java/org/floens/chan/core/model/PostImage.java
+++ b/Clover/app/src/main/java/org/floens/chan/core/model/PostImage.java
@@ -56,6 +56,7 @@ public class PostImage {
                 type = Type.GIF;
                 break;
             case "webm":
+            case "mp4":
                 type = Type.MOVIE;
                 break;
             default:
diff --git a/Clover/app/src/main/java/org/floens/chan/core/site/sites/vichan/ViChan.java b/Clover/app/src/main/java/org/floens/chan/core/site/sites/vichan/ViChan.java
index 862b1370..14f46f1c 100644
--- a/Clover/app/src/main/java/org/floens/chan/core/site/sites/vichan/ViChan.java
+++ b/Clover/app/src/main/java/org/floens/chan/core/site/sites/vichan/ViChan.java
@@ -112,10 +112,23 @@ public class ViChan extends SiteBase {
 
         @Override
         public HttpUrl thumbnailUrl(Post.Builder post, boolean spoiler, Map<String, String> arg) {
+            String ext;
+            switch (arg.get("ext")) {
+                case "jpeg":
+                case "jpg":
+                case "png":
+                case "gif":
+                    ext = arg.get("ext");
+                    break;
+                default:
+                    ext = "jpg";
+                    break;
+            }
+
             return root.newBuilder()
                     .addPathSegment("file_store")
                     .addPathSegment("thumb")
-                    .addPathSegment(arg.get("tim") + "." + arg.get("ext"))
+                    .addPathSegment(arg.get("tim") + "." + ext)
                     .build();
         }