From 9e0c94f1a4d85bbde6806dc33248ccafc4d35f63 Mon Sep 17 00:00:00 2001
From: Tyler Gebhard <walnut.alligator@gmail.com>
Date: Sun, 26 Oct 2014 03:03:02 -0400
Subject: [PATCH] changes to the way favicons are retrieved for YouTube feeds.
 Now instead of fetching the YouTube logo, it fetches the YouTube user's
 custom thumbnail.

---
 .../backend/feed/FaviconFetcher.java          | 59 +++++++++++++++++++
 .../com/commafeed/backend/feed/FeedUtils.java | 14 +++++
 .../commafeed/frontend/resource/FeedREST.java |  2 +-
 3 files changed, 74 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/commafeed/backend/feed/FaviconFetcher.java b/src/main/java/com/commafeed/backend/feed/FaviconFetcher.java
index a7349422..aeeec33e 100644
--- a/src/main/java/com/commafeed/backend/feed/FaviconFetcher.java
+++ b/src/main/java/com/commafeed/backend/feed/FaviconFetcher.java
@@ -39,6 +39,12 @@ public class FaviconFetcher {
 			log.debug("url is null");
 			return null;
 		}
+		
+		// Get YouTube Icon here
+		if (url.toLowerCase().contains("://gdata.youtube.com/")) {
+			byte[] icon = getYouTubeIcon(url);
+			return icon;
+		}
 
 		int doubleSlash = url.indexOf("//");
 		if (doubleSlash == -1) {
@@ -154,4 +160,57 @@ public class FaviconFetcher {
 		return bytes;
 	}
 
+	/*
+	 * Instead of grabbing the actual favicon, grab the user's icon
+	 * This prevents a whole bunch of repeated YouTube icons, replacing
+	 * each with identifiable user icons.
+	 */
+	private byte[] getYouTubeIcon(String url) {
+		byte[] bytes = null;
+		String contentType = null;
+		String username = null;
+		String imageUrl = null;
+		String thumbnailUrl = null;
+		try {
+			int apiOrBase = url.indexOf("/users/");
+			int userEndSlash = url.indexOf('/', apiOrBase + "/users/".length());
+			if (userEndSlash != -1) {
+				username = url.substring(apiOrBase + "/users/".length(), userEndSlash);
+			}
+			imageUrl = "https://gdata.youtube.com/feeds/users/" + username;
+			log.debug("Getting YouTube user's icon, {}", url);
+			
+			//initial get to translate username to obscure user thumbnail URL
+			HttpResult result = getter.getBinary(imageUrl, TIMEOUT);
+			bytes = result.getContent();
+			contentType = result.getContentType();
+			thumbnailUrl = FeedUtils.parseForImageUrl(bytes);
+
+			int thumbnailStart = thumbnailUrl.indexOf("<media:thumbnail url='");
+			int thumbnailEnd = thumbnailUrl.indexOf("'/>", thumbnailStart);
+			if (thumbnailStart != -1) {
+				thumbnailUrl = thumbnailUrl.substring(thumbnailStart+"<media:thumbnail url='".length(), thumbnailEnd);
+			}
+			
+			//final get to actually retrieve the thumbnail
+			result = getter.getBinary(thumbnailUrl, TIMEOUT);
+			bytes = result.getContent();
+			contentType = result.getContentType();
+		} catch (Exception e) {
+			log.debug("Failed to retrieve YouTubeIcon, instead retrieving default YouTube favicon: " + e.getMessage(), e);
+			return fetch("http://www.youtube.com/");
+		}
+
+		if (!isValidIconResponse(bytes, contentType)) {
+			bytes = null;
+		}
+		return bytes;
+	}
+	
+	public boolean exceptionUrl(String url) {
+		if (url.toLowerCase().contains("://gdata.youtube.com/")) {
+			return true;
+		}
+		return false;
+	}
 }
diff --git a/src/main/java/com/commafeed/backend/feed/FeedUtils.java b/src/main/java/com/commafeed/backend/feed/FeedUtils.java
index 88c056d1..6c6593ae 100644
--- a/src/main/java/com/commafeed/backend/feed/FeedUtils.java
+++ b/src/main/java/com/commafeed/backend/feed/FeedUtils.java
@@ -518,4 +518,18 @@ public class FeedUtils {
 		}
 	}
 
+	public static String parseForImageUrl(byte[] xml) {
+		String xmlString = null;
+		try {
+			String encoding = FeedUtils.guessEncoding(xml);
+			xmlString = FeedUtils.trimInvalidXmlCharacters(new String(xml, encoding));
+			/*if (xmlString == null) {
+				throw new FeedException("Input string is null for url " + feedUrl);
+			}*/
+			xmlString = FeedUtils.replaceHtmlEntitiesWithNumericEntities(xmlString);
+		} catch (Exception e) {
+		}
+		return xmlString;
+	}
+	
 }
diff --git a/src/main/java/com/commafeed/frontend/resource/FeedREST.java b/src/main/java/com/commafeed/frontend/resource/FeedREST.java
index ce0d72ea..c022c959 100644
--- a/src/main/java/com/commafeed/frontend/resource/FeedREST.java
+++ b/src/main/java/com/commafeed/frontend/resource/FeedREST.java
@@ -322,7 +322,7 @@ public class FeedREST {
 			return Response.status(Status.NOT_FOUND).build();
 		}
 		Feed feed = subscription.getFeed();
-		String url = feed.getLink() != null ? feed.getLink() : feed.getUrl();
+		String url = faviconFetcher.exceptionUrl(feed.getUrl()) ? feed.getUrl() : (feed.getLink() != null ? feed.getLink() : feed.getUrl());
 		byte[] icon = faviconFetcher.fetch(url);
 
 		ResponseBuilder builder = null;