From 002de23657f7c5b5f73a0cf41c4ce64f34a4cf92 Mon Sep 17 00:00:00 2001
From: FireMasterK <20838718+FireMasterK@users.noreply.github.com>
Date: Fri, 13 Nov 2020 13:15:51 +0530
Subject: [PATCH] Add support for webp.

---
 main.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index df1dd92..888625b 100644
--- a/main.go
+++ b/main.go
@@ -98,7 +98,7 @@ func getHost(path string) (host string) {
 
 	host = ""
 
-	if strings.HasPrefix(path, "/vi/") || strings.HasPrefix(path, "/sb/") {
+	if strings.HasPrefix(path, "/vi/") || strings.HasPrefix(path, "/vi_webp/") || strings.HasPrefix(path, "/sb/") {
 		host = "i.ytimg.com"
 	}
 
@@ -132,6 +132,7 @@ func getBestThumbnail(path string) (newpath string) {
 func main() {
 	http.HandleFunc("/videoplayback", genericHTTPProxy)
 	http.HandleFunc("/vi/", genericHTTPProxy)
+	http.HandleFunc("/vi_webp/", genericHTTPProxy)
 	http.HandleFunc("/a/", genericHTTPProxy)
 	http.HandleFunc("/ggpht/", genericHTTPProxy)
 	http.HandleFunc("/sb/", genericHTTPProxy)