aboutsummaryrefslogtreecommitdiffstats
path: root/lib/files/cache/scanner.php
blob: 526d4a2aab5799027f13b4c731f7ca7f524c9bee (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
124
125
126
127
128
129
<?php
/**
 * Copyright (c) 2012 Robin Appelman <icewind@owncloud.com>
 * This file is licensed under the Affero General Public License version 3 or
 * later.
 * See the COPYING-README file.
 */

namespace OC\Files\Cache;

class Scanner {
	/**
	 * @var \OC\Files\Storage\Storage $storage
	 */
	private $storage;

	/**
	 * @var string $storageId
	 */
	private $storageId;

	/**
	 * @var \OC\Files\Cache\Cache $cache
	 */
	private $cache;

	const SCAN_RECURSIVE = true;
	const SCAN_SHALLOW = false;

	public function __construct(\OC\Files\Storage\Storage $storage) {
		$this->storage = $storage;
		$this->storageId = $this->storage->getId();
		$this->cache = $storage->getCache();
	}

	/**
	 * get all the metadata of a file or folder
	 * *
	 *
	 * @param string $path
	 * @return array with metadata of the file
	 */
	public function getData($path) {
		$data = array();
		if (!$this->storage->isReadable($path)) return null; //cant read, nothing we can do
		$data['mimetype'] = $this->storage->getMimeType($path);
		$data['mtime'] = $this->storage->filemtime($path);
		if ($data['mimetype'] == 'httpd/unix-directory') {
			$data['size'] = -1; //unknown
		} else {
			$data['size'] = $this->storage->filesize($path);
		}
		return $data;
	}

	/**
	 * scan a single file and store it in the cache
	 *
	 * @param string $file
	 * @return array with metadata of the scanned file
	 */
	public function scanFile($file) {
		\OC_Hook::emit('\OC\Files\Cache\Scanner', 'scan_file', array('path' => $file, 'storage' => $this->storageId));
		$data = $this->getData($file);
		if ($data) {
			if ($file) {
				$parent = dirname($file);
				if ($parent === '.') {
					$parent = '';
				}
				if (!$this->cache->inCache($parent)) {
					$this->scanFile($parent);
				}
			}
			$id = $this->cache->put($file, $data);
		}
		return $data;
	}

	/**
	 * scan all the files in a folder and store them in the cache
	 *
	 * @param string $path
	 * @param SCAN_RECURSIVE/SCAN_SHALLOW $recursive
	 * @return int the size of the scanned folder or -1 if the size is unknown at this stage
	 */
	public function scan($path, $recursive = self::SCAN_RECURSIVE) {
		\OC_Hook::emit('\OC\Files\Cache\Scanner', 'scan_folder', array('path' => $path, 'storage' => $this->storageId));
		$this->scanFile($path);

		$size = 0;
		if ($dh = $this->storage->opendir($path)) {
			while ($file = readdir($dh)) {
				if ($file !== '.' and $file !== '..') {
					$child = ($path) ? $path . '/' . $file : $file;
					$data = $this->scanFile($child);
					if ($data) {
						if ($data['mimetype'] === 'httpd/unix-directory') {
							if ($recursive === self::SCAN_RECURSIVE) {
								$data['size'] = $this->scan($child, self::SCAN_RECURSIVE);
							} else {
								$data['size'] = -1;
							}
						}
						if ($data['size'] === -1) {
							$size = -1;
						} elseif ($size !== -1) {
							$size += $data['size'];
						}
					}
				}
			}
			if ($size !== -1) {
				$this->cache->put($path, array('size' => $size));
			}
		}
		return $size;
	}

	/**
	 * walk over any folders that are not fully scanned yet and scan them
	 */
	public function backgroundScan() {
		while ($path = $this->cache->getIncomplete()) {
			$this->scan($path);
			$this->cache->correctFolderSize($path);
		}
	}
}