af_redditimgur: gifv: fix iframe not being wide enough, fix indenting

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

<?php
class Af_RedditImgur extends Plugin {
	private $host;

	function about() {
		return array(1.0,
			"Inline image links in Reddit RSS feeds",
			"fox");
	}

	function init($host) {
		$this->host = $host;

		$host->add_hook($host::HOOK_ARTICLE_FILTER, $this);
	}

	function hook_article_filter($article) {

		if (strpos($article["link"], "reddit.com/r/") !== FALSE) {
				$doc = new DOMDocument();
				@$doc->loadHTML($article["content"]);

				if ($doc) {
					$xpath = new DOMXPath($doc);
					$entries = $xpath->query('(//a[@href]|//img[@src])');

					$found = false;

					foreach ($entries as $entry) {
						if ($entry->hasAttribute("href")) {

							if (preg_match("/\.(gifv)$/i", $entry->getAttribute("href"))) {

								$gifv_meta = fetch_file_contents($entry->getAttribute("href"),
									false, false, false, false, 10);

								if ($gifv_meta) {
									$adoc = new DOMDocument();
									@$adoc->loadHTML($gifv_meta);

									if ($adoc) {
										$axpath = new DOMXPath($adoc);
										$aentries = $axpath->query('(//meta)');

										$width = false;
										$height = false;

										foreach ($aentries as $aentry) {
											if (strpos($aentry->getAttribute("property"), "og:image:width") !== FALSE) {
												$width = $aentry->getAttribute("content");
											}

											if (strpos($aentry->getAttribute("property"), "og:image:height") !== FALSE) {
												$height = $aentry->getAttribute("content");
											}
										}
									}
								}

								if ($width && $height) {

									$iframe = $doc->createElement('iframe');
									$iframe->setAttribute("src", str_replace("http:", "", $entry->getAttribute("href")));
									$iframe->setAttribute("frameborder", "0");
									$iframe->setAttribute("width", $width + 64);
									$iframe->setAttribute("height", $height + 64);

									$br = $doc->createElement('br');
									$entry->parentNode->insertBefore($iframe, $entry);
									$entry->parentNode->insertBefore($br, $entry);

									// add empty img tag to disable display of attachment
									$img = $doc->createElement('img');
									$img->setAttribute("src", "data:image/gif;base64,R0lGODlhAQABAAD/ACwAAAAAAQABAAACADs%3D");
									$img->setAttribute("width", "0");
									$img->setAttribute("height", "0");
									$entry->parentNode->insertBefore($img, $entry);
									$found = true;
								}
							}

							if (preg_match("/\.(jpg|jpeg|gif|png)(\?[0-9])?$/i", $entry->getAttribute("href"))) {

							 	$img = $doc->createElement('img');
								$img->setAttribute("src", $entry->getAttribute("href"));

								$br = $doc->createElement('br');
								$entry->parentNode->insertBefore($img, $entry);
								$entry->parentNode->insertBefore($br, $entry);

								$found = true;
							}

							// links to imgur pages
							$matches = array();
							if (preg_match("/^https?:\/\/(m\.)?imgur.com\/([^\.\/]+$)/", $entry->getAttribute("href"), $matches)) {

								$token = $matches[2];

								$album_content = fetch_file_contents($entry->getAttribute("href"),
									false, false, false, false, 10);

								if ($album_content && $token) {
									$adoc = new DOMDocument();
									@$adoc->loadHTML($album_content);

									if ($adoc) {
										$axpath = new DOMXPath($adoc);
										$aentries = $axpath->query('(//img[@src])');

										foreach ($aentries as $aentry) {
											if (preg_match("/\/\/i.imgur.com\/$token\./", $aentry->getAttribute("src"))) {
												$img = $doc->createElement('img');
												$img->setAttribute("src", $aentry->getAttribute("src"));

												$br = $doc->createElement('br');

												$entry->parentNode->insertBefore($img, $entry);
												$entry->parentNode->insertBefore($br, $entry);

												$found = true;

												break;
											}
										}
									}
								}
							}

							// linked albums, ffs
							if (preg_match("/^https?:\/\/imgur.com\/(a|album)\/[^\.]+$/", $entry->getAttribute("href"), $matches)) {

								$album_content = fetch_file_contents($entry->getAttribute("href"),
									false, false, false, false, 10);

								if ($album_content) {
									$adoc = new DOMDocument();
									@$adoc->loadHTML($album_content);

									if ($adoc) {
										$axpath = new DOMXPath($adoc);
										$aentries = $axpath->query("//meta[@property='og:image']");

										foreach ($aentries as $aentry) {
											$img = $doc->createElement('img');
											$img->setAttribute("src", $aentry->getAttribute("content"));
											$entry->parentNode->insertBefore($doc->createElement('br'), $entry);

											$br = $doc->createElement('br');

											$entry->parentNode->insertBefore($img, $entry);
											$entry->parentNode->insertBefore($br, $entry);

											$found = true;
										}
									}
								}
							}
						}

						// remove tiny thumbnails
						if ($entry->hasAttribute("src")) {
							if ($entry->parentNode && $entry->parentNode->parentNode) {
								$entry->parentNode->parentNode->removeChild($entry->parentNode);
							}
						}
					}

					$node = $doc->getElementsByTagName('body')->item(0);

					if ($node && $found) {
						$article["content"] = $doc->saveXML($node);
					}
				}
		}

		return $article;
	}

	function api_version() {
		return 2;
	}

}
?>