summaryrefslogtreecommitdiff
path: root/classes/diskcache.php
blob: 9ef62d5c413023c528682675d3416d5d71e9a78e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
<?php
class DiskCache
{
	private $dir;

	public function __construct($dir)
	{
		$this->dir = basename($dir);
	}

	public function getDir()
	{
		return $this->dir;
	}

	public function isWritable()
	{
		return is_dir($this->dir) && is_writable($this->dir);
	}

	public function exists($filename)
	{
		return file_exists($this->getFullPath($filename));
	}

	public function getSize($filename)
	{
		if ($this->exists($filename))
			return filesize($this->getFullPath($filename));
		else
			return -1;
	}

	public function getFullPath($filename)
	{
		$filename = basename($filename);

		return CACHE_DIR . "/" . $this->dir . "/" . $filename;
	}

	public function put($filename, $data)
	{
		return file_put_contents($this->getFullPath($filename), $data);
	}

	public function touch($filename)
	{
		return touch($this->getFullPath($filename));
	}

	public function get($filename)
	{
		if ($this->exists($filename))
			return file_get_contents($this->getFullPath($filename));
		else
			return null;
	}

	public function getMimeType($filename)
	{
		if ($this->exists($filename))
			return mime_content_type($this->getFullPath($filename));
		else
			return null;
	}

	public function send($filename)
	{
		return send_local_file($this->getFullPath($filename));
	}

	static public function getUrl($filename)
	{
		return get_self_url_prefix() . "/public.php?op=cached_url&file=" . $filename;
	}

	// check for locally cached (media) URLs and rewrite to local versions
	// this is called separately after sanitize() and plugin render article hooks to allow
	// plugins work on original source URLs used before caching
	static public function rewriteUrls($str)
	{
		$res = trim($str);
		if (!$res) return '';

		$doc = new DOMDocument();
		if ($doc->loadHTML('<?xml encoding="UTF-8">' . $res)) {
			$xpath = new DOMXPath($doc);
			$cache = new DiskCache("images");

			$entries = $xpath->query('(//img[@src]|//picture/source[@src]|//video[@poster]|//video/source[@src]|//audio/source[@src])');

			$need_saving = false;

			foreach ($entries as $entry) {

				if ($entry->hasAttribute('src') || $entry->hasAttribute('poster')) {

					// should be already absolutized because this is called after sanitize()
					$src = $entry->hasAttribute('poster') ? $entry->getAttribute('poster') : $entry->getAttribute('src');
					$cached_filename = sha1($src);

					if ($cache->getSize($cached_filename) > 0) {

						$src = DiskCache::getUrl(sha1($src));

						if ($entry->hasAttribute('poster'))
							$entry->setAttribute('poster', $src);
						else
							$entry->setAttribute('src', $src);

						$need_saving = true;
					}
				}
			}

			if ($need_saving) {
				$doc->removeChild($doc->firstChild); //remove doctype
				$res = $doc->saveHTML();
			}
		}
		return $res;
	}
}