af_redditimgur: Rework retrieval of v.redd.it fallback URL.

[tt-rss.git] / plugins / af_redditimgur / init.php

diff --git a/plugins/af_redditimgur/init.php b/plugins/af_redditimgur/init.php
index c872b429a927af76792be100ccac9a2e5a527693..3a34b5d8cb89b725a73b3db86f6d78afc4dbf281 100755 (executable)
--- a/plugins/af_redditimgur/init.php
+++ b/plugins/af_redditimgur/init.php
@@ -159,7 +159,34 @@ class Af_RedditImgur extends Plugin {
                                                $poster_url = false;
                                        }
 
-                                       $source_stream = "https://v.redd.it/" . $matches[1] . "/DASH_600_K";
+                                       // Get original article URL from v.redd.it redirects
+                                       $source_article_url = $this->get_location($matches[0]);
+                                       _debug("Resolved ".$matches[0]." to ".$source_article_url, $debug);
+
+                                       $source_stream = false;
+
+                                       if ($source_article_url) {
+                                               $j = json_decode(fetch_file_contents($source_article_url.".json"), true);
+
+                                               if ($j) {
+                                                       foreach ($j as $listing) {
+                                                               foreach ($listing["data"]["children"] as $child) {
+                                                                       if ($child["data"]["url"] == $matches[0]) {
+                                                                               try {
+                                                                                       $source_stream = $child["data"]["media"]["reddit_video"]["fallback_url"];
+                                                                               }
+                                                                               catch (Exception $e) {
+                                                                               }
+                                                                               break 2;
+                                                                       }
+                                                               }
+                                                       }
+                                               }
+                                       }
+
+                                       if (!$source_stream) {
+                                               $source_stream = "https://v.redd.it/" . $matches[1] . "/DASH_600_K";
+                                       }
 
                                        $this->handle_as_video($doc, $entry, $source_stream, $poster_url);
                                        $found = 1;
@@ -471,6 +498,25 @@ class Af_RedditImgur extends Plugin {
                return $content_type;
        }
 
+       private function get_location($url, $useragent = SELF_USER_AGENT) {
+               $location = false;
+
+               if (function_exists("curl_init") && !defined("NO_CURL")) {
+                       $ch = curl_init($url);
+                       curl_setopt($ch, CURLOPT_TIMEOUT, 5);
+                       curl_setopt($ch, CURLOPT_RETURNTRANSFER, 1);
+                       curl_setopt($ch, CURLOPT_HEADER, true);
+                       curl_setopt($ch, CURLOPT_NOBODY, true);
+                       curl_setopt($ch, CURLOPT_FOLLOWLOCATION, !ini_get("open_basedir"));
+                       curl_setopt($ch, CURLOPT_USERAGENT, $useragent);
+
+                       @curl_exec($ch);
+                       $location = curl_getinfo($ch, CURLINFO_EFFECTIVE_URL);
+               }
+
+               return $location;
+       }
+
        /**
         * @SuppressWarnings(PHPMD.UnusedFormalParameter)
         */