105 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			105 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/**
 | 
						|
 * Name: Archive File Handler
 | 
						|
 * Author: Shish <webmaster@shishnet.org>
 | 
						|
 * Description: Allow users to upload archives (zip, etc)
 | 
						|
 * Documentation:
 | 
						|
 *  Note: requires exec() access and an external unzip command
 | 
						|
 *  <p>Any command line unzipper should work, some examples:
 | 
						|
 *  <p>unzip: <code>unzip -d "%d" "%f"</code>
 | 
						|
 *  <br>7-zip: <code>7zr x -o"%d" "%f"</code>
 | 
						|
 */
 | 
						|
 | 
						|
class ArchiveFileHandler extends SimpleExtension {
 | 
						|
	public function onInitExt($event) {
 | 
						|
		global $config;
 | 
						|
		$config->set_default_string('archive_extract_command', 'unzip -d "%d" "%f"');
 | 
						|
	}
 | 
						|
 | 
						|
	public function onSetupBuilding($event) {
 | 
						|
		$sb = new SetupBlock("Archive Handler Options");
 | 
						|
		$sb->add_text_option("archive_tmp_dir", "Temporary folder: ");
 | 
						|
		$sb->add_text_option("archive_extract_command", "<br>Extraction command: ");
 | 
						|
		$sb->add_label("<br>%f for archive, %d for temporary directory");
 | 
						|
		$event->panel->add_block($sb);
 | 
						|
	}
 | 
						|
 | 
						|
	public function onDataUpload($event) {
 | 
						|
		if($this->supported_ext($event->type)) {
 | 
						|
			global $config;
 | 
						|
			$tmp = sys_get_temp_dir();
 | 
						|
			$tmpdir = "$tmp/shimmie-archive-{$event->hash}";
 | 
						|
			$cmd = $config->get_string('archive_extract_command');
 | 
						|
			$cmd = str_replace('%f', $event->tmpfile);
 | 
						|
			$cmd = str_replace('%d', $tmpdir);
 | 
						|
			system($cmd);
 | 
						|
			$this->add_dir($tmpdir);
 | 
						|
			unlink($tmpdir);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
 | 
						|
	private function supported_ext($ext) {
 | 
						|
		$exts = array("zip");
 | 
						|
		return in_array(strtolower($ext), $exts);
 | 
						|
	}
 | 
						|
 | 
						|
	// copied from bulk add extension
 | 
						|
	private function add_image($tmpname, $filename, $tags) {
 | 
						|
		if(file_exists($tmpname)) {
 | 
						|
			global $user;
 | 
						|
			$pathinfo = pathinfo($filename);
 | 
						|
			$metadata['filename'] = $pathinfo['basename'];
 | 
						|
			$metadata['extension'] = $pathinfo['extension'];
 | 
						|
			$metadata['tags'] = $tags;
 | 
						|
			$metadata['source'] = null;
 | 
						|
			try {
 | 
						|
				$event = new DataUploadEvent($user, $tmpname, $metadata);
 | 
						|
				send_event($event);
 | 
						|
			}
 | 
						|
			catch(UploadException $ex) {
 | 
						|
				return $ex->getMessage();
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// copied from bulk add extension
 | 
						|
	private function add_dir($base, $subdir="") {
 | 
						|
		global $page;
 | 
						|
 | 
						|
		$list = "";
 | 
						|
 | 
						|
		$dir = opendir("$base/$subdir");
 | 
						|
		while($filename = readdir($dir)) {
 | 
						|
			$fullpath = "$base/$subdir/$filename";
 | 
						|
 | 
						|
			if(is_link($fullpath)) {
 | 
						|
				// ignore
 | 
						|
			}
 | 
						|
			else if(is_dir($fullpath)) {
 | 
						|
				if($filename[0] != ".") {
 | 
						|
					$this->add_dir($base, "$subdir/$filename");
 | 
						|
				}
 | 
						|
			}
 | 
						|
			else {
 | 
						|
				$tmpfile = $fullpath;
 | 
						|
				$tags = $subdir;
 | 
						|
				$tags = str_replace("/", " ", $tags);
 | 
						|
				$tags = str_replace("__", " ", $tags);
 | 
						|
				$list .= "<br>".html_escape("$subdir/$filename (".str_replace(" ", ",", $tags).")...");
 | 
						|
				$error = $this->add_image($tmpfile, $filename, $tags);
 | 
						|
				if(is_null($error)) {
 | 
						|
					$list .= "ok\n";
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					$list .= "failed: $error\n";
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		closedir($dir);
 | 
						|
 | 
						|
		// $this->theme->add_status("Adding $subdir", $list);
 | 
						|
	}
 | 
						|
}
 | 
						|
?>
 |