Move cleanup code from controller to \service\files

Signed-off-by: Florian Pritz <bluewind@xinu.at>

1 files changed, 98 insertions, 0 deletions

diff --git a/application/service/files.php b/application/service/files.php
index 85efdfff8..062d17bf8 100644
--- a/application/service/files.php
+++ b/application/service/files.php
@@ -450,4 +450,102 @@ class files {
 		return $tooltip;
 	}
 
+	static public function clean_multipaste_tarballs()
+	{
+		$CI =& get_instance();
+
+		$tarball_dir = $CI->config->item("upload_path")."/special/multipaste-tarballs";
+		if (is_dir($tarball_dir)) {
+			$tarball_cache_time = $CI->config->item("tarball_cache_time");
+			$it = new RecursiveIteratorIterator(
+				new RecursiveDirectoryIterator($tarball_dir), RecursiveIteratorIterator::SELF_FIRST);
+
+			foreach ($it as $file) {
+				if ($file->isFile()) {
+					if ($file->getMTime() < time() - $tarball_cache_time) {
+						$lock = fopen($file, "r+");
+						flock($lock, LOCK_EX);
+						unlink($file);
+						flock($lock, LOCK_UN);
+					}
+				}
+			}
+		}
+	}
+
+	static public function remove_files_missing_in_db()
+	{
+		$CI =& get_instance();
+
+		$upload_path = $CI->config->item("upload_path");
+		$outer_dh = opendir($upload_path);
+
+		while (($dir = readdir($outer_dh)) !== false) {
+			if (!is_dir($upload_path."/".$dir) || $dir == ".." || $dir == "." || $dir == "special") {
+				continue;
+			}
+
+			$dh = opendir($upload_path."/".$dir);
+
+			$empty = true;
+
+			while (($file = readdir($dh)) !== false) {
+				if ($file == ".." || $file == ".") {
+					continue;
+				}
+
+				try {
+					list($hash, $storage_id) = explode("-", $file);
+				} catch (\ErrorException $e) {
+					unlink($upload_path."/".$dir."/".$file);
+					continue;
+				}
+
+				$query = $CI->db->select('hash, id')
+					->from('file_storage')
+					->where('hash', $hash)
+					->where('id', $storage_id)
+					->limit(1)
+					->get()->row_array();
+
+				if (empty($query)) {
+					$CI->mfile->delete_data_id($file);
+				} else {
+					$empty = false;
+				}
+			}
+
+			closedir($dh);
+
+			if ($empty && file_exists($upload_path."/".$dir)) {
+				rmdir($upload_path."/".$dir);
+			}
+		}
+		closedir($outer_dh);
+	}
+
+	static public function remove_files_missing_on_disk()
+	{
+		$CI =& get_instance();
+
+		$chunk = 500;
+		$total = $CI->db->count_all("file_storage");
+
+		for ($limit = 0; $limit < $total; $limit += $chunk) {
+			$query = $CI->db->select('hash, id')
+				->from('file_storage')
+				->limit($chunk, $limit)
+				->get()->result_array();
+
+			foreach ($query as $key => $item) {
+				$data_id = $item["hash"].'-'.$item['id'];
+				$file = $CI->mfile->file($data_id);
+
+				if (!$CI->mfile->file_exists($file)) {
+					$CI->mfile->delete_data_id($data_id);
+				}
+			}
+		}
+	}
+
 }