From 5edd605ae1a9269b8b6b91b587da3d54a35424e9 Mon Sep 17 00:00:00 2001
From: Andrew Dolgov <noreply@fakecake.org>
Date: Sat, 4 Feb 2017 11:50:01 +0300
Subject: image cache: do not try to cache data: schema urls; add caching of
 html5 video content (similar to cache_starred_images plugin)

---
 include/functions2.php | 33 +++++++++++++++++----------------
 1 file changed, 17 insertions(+), 16 deletions(-)

(limited to 'include/functions2.php')

diff --git a/include/functions2.php b/include/functions2.php
index 2b34f287c..6017a78f8 100644
--- a/include/functions2.php
+++ b/include/functions2.php
@@ -888,7 +888,7 @@
 		$doc->loadHTML($charset_hack . $res);
 		$xpath = new DOMXPath($doc);
 
-		$entries = $xpath->query('(//a[@href]|//img[@src])');
+		$entries = $xpath->query('(//a[@href]|//img[@src]|//video/source[@src])');
 
 		$ttrss_uses_https = parse_url(get_self_url_prefix(), PHP_URL_SCHEME) === 'https';
 
@@ -903,28 +903,29 @@
 					$entry->setAttribute('rel', 'noopener noreferrer');
 				}
 
-				if ($entry->nodeName == 'img') {
-
-					if ($entry->hasAttribute('src')) {
-						$src = rewrite_relative_url($site_url, $entry->getAttribute('src'));
-
-						$cached_filename = CACHE_DIR . '/images/' . sha1($src) . '.png';
+				if ($entry->hasAttribute('src')) {
+					$src = rewrite_relative_url($site_url, $entry->getAttribute('src'));
 
-						if (file_exists($cached_filename)) {
-							$src = SELF_URL_PATH . '/public.php?op=cached_image&hash=' . sha1($src) . '.png';
+					$extension = $entry->tagName == 'source' ? '.mp4' : '.png';
+					$cached_filename = CACHE_DIR . '/images/' . sha1($src) . $extension;
 
-							if ($entry->hasAttribute('srcset')) {
-								$entry->removeAttribute('srcset');
-							}
+					if (file_exists($cached_filename)) {
+						$src = SELF_URL_PATH . '/public.php?op=cached_image&hash=' . sha1($src) . $extension;
 
-							if ($entry->hasAttribute('sizes')) {
-								$entry->removeAttribute('sizes');
-							}
+						if ($entry->hasAttribute('srcset')) {
+							$entry->removeAttribute('srcset');
 						}
 
-						$entry->setAttribute('src', $src);
+						if ($entry->hasAttribute('sizes')) {
+							$entry->removeAttribute('sizes');
+						}
 					}
 
+					$entry->setAttribute('src', $src);
+				}
+				
+				if ($entry->nodeName == 'img') {
+
 					if ($entry->hasAttribute('src')) {
 						$is_https_url = parse_url($entry->getAttribute('src'), PHP_URL_SCHEME) === 'https';
 
-- 
cgit v1.2.3