X-Git-Url: https://git.wh0rd.org/?a=blobdiff_plain;f=plugins%2Fcache_starred_images%2Finit.php;h=ae0eaf56d679eb3834eab78d50ee3b403192cb74;hb=758752684c68efd179071cd77c92f78879e68f6d;hp=f721b38927605612c91292959fd2c89e02180c1e;hpb=f5967cf825bbcc26ac3c8ac1ba5ce86ef3be9855;p=tt-rss.git

diff --git a/plugins/cache_starred_images/init.php b/plugins/cache_starred_images/init.php
old mode 100644
new mode 100755
index f721b389..ae0eaf56
--- a/plugins/cache_starred_images/init.php
+++ b/plugins/cache_starred_images/init.php
@@ -1,16 +1,36 @@
 <?php
-class Cache_Starred_Images extends Plugin {
+class Cache_Starred_Images extends Plugin implements IHandler {
 
+	/* @var PluginHost $host */
 	private $host;
 	private $cache_dir;
+    private $max_cache_attempts = 5; // per-article
 
 	function about() {
 		return array(1.0,
-			"Automatically cache images in Starred articles",
+			"Automatically cache Starred articles' images and HTML5 video files",
 			"fox",
 			true);
 	}
 
+	/**
+	 * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+	 */
+	function csrf_ignore($method) {
+		return false;
+	}
+
+	/**
+	 * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+	 */
+	function before($method) {
+		return true;
+	}
+
+	function after() {
+		return true;
+	}
+
 	function init($host) {
 		$this->host = $host;
 
@@ -29,6 +49,8 @@ class Cache_Starred_Images extends Plugin {
 				$host->add_hook($host::HOOK_UPDATE_TASK, $this);
 				$host->add_hook($host::HOOK_HOUSE_KEEPING, $this);
 				$host->add_hook($host::HOOK_SANITIZE, $this);
+				$host->add_handler("public", "cache_starred_images_getimage", $this);
+
 			} else {
 				user_error("Starred cache directory is not writable.", E_USER_WARNING);
 			}
@@ -38,32 +60,19 @@ class Cache_Starred_Images extends Plugin {
 		}
 	}
 
-	function image() {
+	function cache_starred_images_getimage() {
+		ob_end_clean();
+
 		$hash = basename($_REQUEST["hash"]);
 
 		if ($hash) {
 
-			$filename = $this->cache_dir . "/" . $hash . '.png';
+			$filename = $this->cache_dir . "/" . basename($hash);
 
 			if (file_exists($filename)) {
-				/* See if we can use X-Sendfile */
-				$xsendfile = false;
-				if (function_exists('apache_get_modules') &&
-				    array_search('mod_xsendfile', apache_get_modules()))
-					$xsendfile = true;
-
-				if ($xsendfile) {
-					header("X-Sendfile: $filename");
-					header("Content-type: application/octet-stream");
-					header('Content-Disposition: attachment; filename="' . basename($filename) . '"');
-				} else {
-					header("Content-type: image/png");
-					$stamp = gmdate("D, d M Y H:i:s", filemtime($filename)). " GMT";
-					header("Last-Modified: $stamp", true);
-					ob_clean();   // discard any data in the output buffer (if possible)
-					flush();      // flush headers (if possible)
-					readfile($filename);
-				}
+				header("Content-Disposition: attachment; filename=\"$hash\"");
+
+				send_local_file($filename);
 			} else {
 				header($_SERVER["SERVER_PROTOCOL"]." 404 Not Found");
 				echo "File not found.";
@@ -71,8 +80,11 @@ class Cache_Starred_Images extends Plugin {
 		}
 	}
 
+	/**
+	 * @SuppressWarnings(PHPMD.UnusedLocalVariable)
+	 */
 	function hook_house_keeping() {
-		$files = glob($this->cache_dir . "/*.png");
+		$files = glob($this->cache_dir . "/*.{png,mp4,status}", GLOB_BRACE);
 
 		$last_article_id = 0;
 		$article_exists = 1;
@@ -82,11 +94,11 @@ class Cache_Starred_Images extends Plugin {
 
 			if ($article_id != $last_article_id) {
 				$last_article_id = $article_id;
-				$article_id = db_escape_string($article_id);
 
-				$result = db_query("SELECT id FROM ttrss_entries WHERE id = " . $article_id);
+				$sth = $this->pdo->prepare("SELECT id FROM ttrss_entries WHERE id = ?");
+				$sth->execute([$article_id]);
 
-				$article_exists = db_num_rows($result) > 0;
+				$article_exists = $sth->fetch();
 			}
 
 			if (!$article_exists) {
@@ -95,22 +107,26 @@ class Cache_Starred_Images extends Plugin {
 		}
 	}
 
+	/**
+	 * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+	 */
 	function hook_sanitize($doc, $site_url, $allowed_elements, $disallowed_attributes, $article_id) {
 		$xpath = new DOMXpath($doc);
 
 		if ($article_id) {
-			$entries = $xpath->query('(//img[@src])');
+			$entries = $xpath->query('(//img[@src])|(//video/source[@src])');
 
 			foreach ($entries as $entry) {
 				if ($entry->hasAttribute('src')) {
 					$src = rewrite_relative_url($site_url, $entry->getAttribute('src'));
 
-					$local_filename = $this->cache_dir . $article_id . "-" . sha1($src) . ".png";
+					$extension = $entry->tagName == 'source' ? '.mp4' : '.png';
+					$local_filename = $this->cache_dir . $article_id . "-" . sha1($src) . $extension;
 
 					if (file_exists($local_filename)) {
 						$entry->setAttribute("src", get_self_url_prefix() .
-							"/backend.php?op=pluginhandler&plugin=cache_starred_images&method=image&hash=" .
-							$article_id . "-" . sha1($src));
+							"/public.php?op=cache_starred_images_getimage&method=image&hash=" .
+							$article_id . "-" . sha1($src) . $extension);
 					}
 
 				}
@@ -121,33 +137,59 @@ class Cache_Starred_Images extends Plugin {
 	}
 
 	function hook_update_task() {
-		$result = db_query("SELECT content, ttrss_user_entries.owner_uid, link, site_url, ttrss_entries.id, plugin_data
+		$res = $this->pdo->query("SELECT content, ttrss_user_entries.owner_uid, link, site_url, ttrss_entries.id, plugin_data
 			FROM ttrss_entries, ttrss_user_entries LEFT JOIN ttrss_feeds ON
 				(ttrss_user_entries.feed_id = ttrss_feeds.id)
 			WHERE ref_id = ttrss_entries.id AND
 				marked = true AND
-				UPPER(content) LIKE '%<IMG%' AND
+				(UPPER(content) LIKE '%<IMG%' OR UPPER(content) LIKE '%<VIDEO%') AND
 				site_url != '' AND
 				plugin_data NOT LIKE '%starred_cache_images%'
 			ORDER BY ".sql_random_function()." LIMIT 100");
 
+		$usth = $this->pdo->prepare("UPDATE ttrss_entries SET plugin_data = ? WHERE id = ?");
 
-		while ($line = db_fetch_assoc($result)) {
+		while ($line = $res->fetch()) {
 			if ($line["site_url"]) {
 				$success = $this->cache_article_images($line["content"], $line["site_url"], $line["owner_uid"], $line["id"]);
 
 				if ($success) {
-					$plugin_data = db_escape_string("starred_cache_images,${line['owner_uid']}:" . $line["plugin_data"]);
+					$plugin_data = "starred_cache_images,${line['owner_uid']}:" . $line["plugin_data"];
 
-					db_query("UPDATE ttrss_entries SET plugin_data = '$plugin_data' WHERE id = " . $line["id"]);
+					$usth->execute([$plugin_data, $line['id']]);
 				}
 			}
 		}
 	}
 
+	/**
+	 * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+	 */
 	function cache_article_images($content, $site_url, $owner_uid, $article_id) {
 		libxml_use_internal_errors(true);
 
+		$status_filename = $this->cache_dir . $article_id . "-" . sha1($site_url) . ".status";
+
+		//_debug("status: $status_filename"); return;
+
+        if (file_exists($status_filename))
+            $status = json_decode(file_get_contents($status_filename), true);
+        else
+            $status = [];
+
+        $status["attempt"] += 1;
+
+        // only allow several download attempts for article
+        if ($status["attempt"] > $this->max_cache_attempts) {
+            _debug("too many attempts for $site_url");
+            return;
+        }
+
+        if (!file_put_contents($status_filename, json_encode($status))) {
+            user_error("unable to write status file: $status_filename", E_USER_WARNING);
+            return;
+        }
+
 		$charset_hack = '<head>
 			<meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
 		</head>';
@@ -156,25 +198,32 @@ class Cache_Starred_Images extends Plugin {
 		$doc->loadHTML($charset_hack . $content);
 		$xpath = new DOMXPath($doc);
 
-		$entries = $xpath->query('(//img[@src])');
+		$entries = $xpath->query('(//img[@src])|(//video/source[@src])');
 
 		$success = false;
 		$has_images = false;
 
 		foreach ($entries as $entry) {
-			if ($entry->hasAttribute('src')) {
+
+			if ($entry->hasAttribute('src') && strpos($entry->getAttribute('src'), "data:") !== 0) {
+
 				$has_images = true;
 				$src = rewrite_relative_url($site_url, $entry->getAttribute('src'));
 
-				$local_filename = $this->cache_dir . $article_id . "-" . sha1($src) . ".png";
+				$extension = $entry->tagName == 'source' ? '.mp4' : '.png';
+
+				$local_filename = $this->cache_dir . $article_id . "-" . sha1($src) . $extension;
 
 				//_debug("cache_images: downloading: $src to $local_filename");
 
 				if (!file_exists($local_filename)) {
-					$file_content = fetch_file_contents($src);
+					$file_content = fetch_file_contents(["url" => $src, "max_size" => MAX_CACHE_FILE_SIZE]);
+
+					if ($file_content) {
+                        if (strlen($file_content) > MIN_CACHE_FILE_SIZE) {
+                            file_put_contents($local_filename, $file_content);
+                        }
 
-					if ($file_content && strlen($file_content) > 0) {
-						file_put_contents($local_filename, $file_content);
 						$success = true;
 					}
 				} else {
@@ -190,4 +239,3 @@ class Cache_Starred_Images extends Plugin {
 		return 2;
 	}
 }
-?>