From 26b92c751a8589ec6953ea226e84eaa279d70ec0 Mon Sep 17 00:00:00 2001
From: FireMasterK <20838718+FireMasterK@users.noreply.github.com>
Date: Wed, 9 Dec 2020 18:59:12 +0530
Subject: [PATCH] Better proxying of images.

---
 .../java/me/kavin/piped/utils/ResponseHelper.java    | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/main/java/me/kavin/piped/utils/ResponseHelper.java b/src/main/java/me/kavin/piped/utils/ResponseHelper.java
index ea6a05d..4aacb65 100644
--- a/src/main/java/me/kavin/piped/utils/ResponseHelper.java
+++ b/src/main/java/me/kavin/piped/utils/ResponseHelper.java
@@ -234,6 +234,9 @@ public class ResponseHelper {
 	if (Constants.debug)
 	    return old;
 
+	if (old == null || old.isEmpty())
+	    return null;
+
 	URL url = null;
 
 	try {
@@ -246,14 +249,17 @@ public class ResponseHelper {
 
 	String query = url.getQuery();
 
-	final boolean hasQuery = query != null;
+	boolean hasQuery = query != null;
 
 	String path = url.getPath();
 
 	path = path.replace("-rj", "-rw");
 
-	if (!hasQuery && path.startsWith("/vi/"))
-	    path = path.replace("/vi/", "/vi_webp/").replace(".jpg", ".webp");
+	if (path.startsWith("/vi/") && !path.contains("_live")) {
+	    path = path.replace("/vi/", "/vi_webp/").replace(".jpg", ".webp").replace("hq720", "mqdefault")
+		    .replace("hqdefault", "mqdefault");
+	    hasQuery = false;
+	}
 
 	return Constants.PROXY_PART + path + (hasQuery ? "?" + query + "&host=" : "?host=")
 		+ URLUtils.silentEncode(host);