123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228 |
- <?php
- class Cache_Starred_Images extends Plugin implements IHandler {
- private $host;
- private $cache_dir;
- function about() {
- return array(1.0,
- "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;
- $this->cache_dir = CACHE_DIR . "/starred-images/";
- if (!is_dir($this->cache_dir)) {
- mkdir($this->cache_dir);
- }
- if (is_dir($this->cache_dir)) {
- if (!is_writable($this->cache_dir))
- chmod($this->cache_dir, 0777);
- if (is_writable($this->cache_dir)) {
- $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);
- }
- } else {
- user_error("Unable to create starred cache directory.", E_USER_WARNING);
- }
- }
- function cache_starred_images_getimage() {
- ob_end_clean();
- $hash = basename($_REQUEST["hash"]);
- if ($hash) {
- $filename = $this->cache_dir . "/" . basename($hash);
- $is_video = strpos($filename, ".mp4") !== FALSE;
- if (file_exists($filename)) {
- header("Content-Disposition: attachment; filename=\"$hash\"");
- /* 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: " . ($is_video ? "video/mp4" : "image/png"));
- $stamp = gmdate("D, d M Y H:i:s", filemtime($filename)). " GMT";
- header("Last-Modified: $stamp", true);
- readfile($filename);
- }
- } else {
- header($_SERVER["SERVER_PROTOCOL"]." 404 Not Found");
- echo "File not found.";
- }
- }
- }
- /**
- * @SuppressWarnings(PHPMD.UnusedLocalVariable)
- */
- function hook_house_keeping() {
- $files = glob($this->cache_dir . "/*.{png,mp4}", GLOB_BRACE);
- $last_article_id = 0;
- $article_exists = 1;
- foreach ($files as $file) {
- list ($article_id, $hash) = explode("-", basename($file));
- 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);
- $article_exists = db_num_rows($result) > 0;
- }
- if (!$article_exists) {
- unlink($file);
- }
- }
- }
- /**
- * @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])|(//video/source[@src])');
- foreach ($entries as $entry) {
- if ($entry->hasAttribute('src')) {
- $src = rewrite_relative_url($site_url, $entry->getAttribute('src'));
- $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() .
- "/public.php?op=cache_starred_images_getimage&method=image&hash=" .
- $article_id . "-" . sha1($src) . $extension);
- }
- }
- }
- }
- return $doc;
- }
- function hook_update_task() {
- $result = db_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%' OR UPPER(content) LIKE '%<VIDEO%') AND
- site_url != '' AND
- plugin_data NOT LIKE '%starred_cache_images%'
- ORDER BY ".sql_random_function()." LIMIT 100");
- while ($line = db_fetch_assoc($result)) {
- 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"]);
- db_query("UPDATE ttrss_entries SET plugin_data = '$plugin_data' WHERE id = " . $line["id"]);
- }
- }
- }
- }
- /**
- * @SuppressWarnings(PHPMD.UnusedFormalParameter)
- */
- function cache_article_images($content, $site_url, $owner_uid, $article_id) {
- libxml_use_internal_errors(true);
- $charset_hack = '<head>
- <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
- </head>';
- $doc = new DOMDocument();
- $doc->loadHTML($charset_hack . $content);
- $xpath = new DOMXPath($doc);
- $entries = $xpath->query('(//img[@src])|(//video/source[@src])');
- $success = false;
- $has_images = false;
- foreach ($entries as $entry) {
- if ($entry->hasAttribute('src') && strpos($entry->getAttribute('src'), "data:") !== 0) {
- $has_images = true;
- $src = rewrite_relative_url($site_url, $entry->getAttribute('src'));
- $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);
- if ($file_content && strlen($file_content) > MIN_CACHE_FILE_SIZE) {
- file_put_contents($local_filename, $file_content);
- $success = true;
- }
- } else {
- $success = true;
- }
- }
- }
- return $success || !$has_images;
- }
- function api_version() {
- return 2;
- }
- }
|