Fix: upgrade hotlinked http URLs on HTTPS

1 files changed, 12 insertions, 6 deletions

diff --git a/static/js/pichat.butt.js b/static/js/pichat.butt.js
index 3f16519..220e873 100755
--- a/static/js/pichat.butt.js
+++ b/static/js/pichat.butt.js
@@ -52,11 +52,16 @@ function escapeHtml(txt) {
 
 function normalizeUrl(url) {
     if (!url) { return url; }
-    var lowerurl = url.toLowerCase();
-    if (lowerurl.indexOf('http://') == 0 || lowerurl.indexOf('https://') == 0 || lowerurl.indexOf('ftp://') == 0 || lowerurl.indexOf('//') == 0)
-        return url;
+    var trimmed = url.trim();
+    var lowerurl = trimmed.toLowerCase();
+    if (lowerurl.indexOf('//') == 0 || lowerurl.indexOf('https://') == 0 || lowerurl.indexOf('ftp://') == 0)
+        return trimmed;
+    if (lowerurl.indexOf('http://') == 0) {
+        if (location && location.protocol == 'https:') return 'https://' + trimmed.substr('http://'.length);
+        return trimmed;
+    }
     var scheme = (location && location.protocol == 'https:') ? 'https://' : 'http://';
-    return scheme + url;
+    return scheme + trimmed;
 }
 
 URLRegex = /((\b(http\:\/\/|https\:\/\/|ftp\:\/\/)|(www\.))+(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(\/|\/([\w#!:.?+=&%@!\-\/]))?)/gi;
@@ -68,9 +73,10 @@ function getImagesAsArray(text) {
   if (urls === null) return imgs
   for (var i = 0; i<urls.length; i++){
     var url = urls[i]
-    var urlWithoutParams = url.replace(/\?.*$/i, "");
+    var normalized = normalizeUrl(url);
+    var urlWithoutParams = normalized.replace(/[?#].*$/i, "");
     if (PicRegex.test(urlWithoutParams))
-      imgs.push(url)
+      imgs.push(normalized)
   }
   return imgs
 }