From f140f8fe5a3d27d522da409dfd3de980a6628e6c Mon Sep 17 00:00:00 2001
From: Kavin <20838718+FireMasterK@users.noreply.github.com>
Date: Sun, 27 Nov 2022 16:44:42 +0000
Subject: [PATCH] Localize image URLs to current instance.

---
 .../piped/server/handlers/StreamHandlers.java    | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/main/java/me/kavin/piped/server/handlers/StreamHandlers.java b/src/main/java/me/kavin/piped/server/handlers/StreamHandlers.java
index 23703bf..db47d3b 100644
--- a/src/main/java/me/kavin/piped/server/handlers/StreamHandlers.java
+++ b/src/main/java/me/kavin/piped/server/handlers/StreamHandlers.java
@@ -145,6 +145,22 @@ public class StreamHandlers {
 
                 Streams streams = federatedGeoBypassResponse.getData();
 
+                // re-rewrite image URLs
+                streams.chapters.forEach(chapter -> chapter.image = rewriteURL(chapter.image));
+                streams.relatedStreams.forEach(contentItem -> {
+                    if (contentItem instanceof StreamItem streamItem) {
+                        streamItem.thumbnail = rewriteURL(streamItem.thumbnail);
+                        streamItem.uploaderAvatar = rewriteURL(streamItem.uploaderAvatar);
+                    } else if (contentItem instanceof ChannelItem channelItem) {
+                        channelItem.thumbnail = rewriteURL(channelItem.thumbnail);
+                    } else if (contentItem instanceof PlaylistItem playlistItem) {
+                        playlistItem.thumbnail = rewriteURL(playlistItem.thumbnail);
+                    }
+                });
+                streams.subtitles.forEach(subtitle -> subtitle.url = rewriteURL(subtitle.url));
+                streams.thumbnailUrl = rewriteURL(streams.thumbnailUrl);
+                streams.uploaderAvatar = rewriteURL(streams.uploaderAvatar);
+
                 String lbryId;
 
                 try {