179 lines
		
	
	
		
			6.5 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			179 lines
		
	
	
		
			6.5 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/**
 | 
						|
 * Name: Update
 | 
						|
 * Author: DakuTree <dakutree@codeanimu.net>
 | 
						|
 * Link: http://www.codeanimu.net
 | 
						|
 * License: GPLv2
 | 
						|
 * Description: Shimmie updater!
 | 
						|
 */
 | 
						|
class Update extends SimpleExtension {
 | 
						|
	public function onInitExt(Event $event) {
 | 
						|
		global $config;
 | 
						|
		$config->set_default_string("update_url", "http://nodeload.github.com/shish/shimmie2/zipball/master"); //best to avoid using https
 | 
						|
		$config->set_default_string("commit_hash", "");
 | 
						|
	}
 | 
						|
 | 
						|
	public function onSetupBuilding($event) {
 | 
						|
		global $config;
 | 
						|
		//Would prefer to use the admin panel for this.
 | 
						|
		//But since the admin panel is optional...kind of stuck to using this.
 | 
						|
		$sb = new SetupBlock("Update");
 | 
						|
		$sb->position = 75;
 | 
						|
		$sb->add_label("Current Commit: ".$config->get_string('commit_hash'));
 | 
						|
		$sb->add_text_option("update_url", "<br>Update URL: ");
 | 
						|
		$sb->add_label("<br><a href='".make_link('update')."'>Update</a>");
 | 
						|
		$event->panel->add_block($sb);
 | 
						|
	}
 | 
						|
 | 
						|
	public function onPageRequest(Event $event) {
 | 
						|
		global $config, $user;
 | 
						|
		if($event->page_matches("update") && $user->is_admin()) {
 | 
						|
			$ok = $this->update_shimmie();
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	private function update_shimmie() {
 | 
						|
		global $config, $page;
 | 
						|
		//This is a REALLY ugly function. (Damn my limited PHP knowledge >_<)
 | 
						|
		$html = "";
 | 
						|
		$url = $config->get_string("update_url");
 | 
						|
		$mfile = "master.zip";
 | 
						|
		if(glob("*-shimmie2*")){ //#3
 | 
						|
			$dir = glob("*-shimmie2*");
 | 
						|
			preg_match('@^([a-zA-Z0-9]+\-[0-9a-z]+\-)([^/]+)@i', $dir[0], $matches);
 | 
						|
			if(!empty($matches[2])){
 | 
						|
				$html .= "commit: ".$matches[2];
 | 
						|
				$commit = $matches[2];
 | 
						|
				mkdir("./backup");
 | 
						|
				$html .= "<br>backup folder created!";
 | 
						|
				$d_dir = "data/cache";
 | 
						|
				//This should empty the /data/cache/ folder.
 | 
						|
				if (is_dir($d_dir)) {
 | 
						|
					$objects = scandir($d_dir);
 | 
						|
					foreach ($objects as $object) {
 | 
						|
						if ($object != "." && $object != "..") {
 | 
						|
							if (filetype($d_dir."/".$object) == "dir") rmdir($d_dir."/".$object); else unlink($d_dir."/".$object);
 | 
						|
						}
 | 
						|
					}
 | 
						|
					reset($objects);
 | 
						|
					$html .= "<br>data folder emptied!";
 | 
						|
				}
 | 
						|
				copy ("./config.php", "./backup/config.php");//Although this stays the same, will keep backup just incase.
 | 
						|
				//FIXME: Somehow get rid of this massive rename list.
 | 
						|
				rename ("./core", "./backup/core");
 | 
						|
				rename ("./".$matches[0]."/core", "./core");
 | 
						|
				rename ("./lib", "./backup/lib");
 | 
						|
				rename ("./".$matches[0]."/lib", "./lib");
 | 
						|
				rename ("./themes", "./backup/themes");
 | 
						|
				rename ("./".$matches[0]."/themes", "./themes");
 | 
						|
				rename ("./.htaccess", "./backup/.htaccess");
 | 
						|
				rename ("./".$matches[0]."/.htaccess", "./.htaccess");
 | 
						|
				rename ("./doxygen.conf", "./backup/doxygen.conf");
 | 
						|
				rename ("./".$matches[0]."/doxygen.conf", "./doxygen.conf");
 | 
						|
				rename ("./index.php", "./backup/index.php");
 | 
						|
				rename ("./".$matches[0]."/index.php", "./index.php");
 | 
						|
				rename ("./install.php", "./backup/install.php");
 | 
						|
				rename ("./".$matches[0]."/install.php", "./install.php");
 | 
						|
				rename ("./ext", "./backup/ext");
 | 
						|
				rename ("./".$matches[0]."/ext", "./ext");
 | 
						|
				rename ("./contrib", "./backup/contrib");
 | 
						|
				rename ("./".$matches[0]."/contrib", "./contrib");
 | 
						|
				$html .= "<br>old shimmie setup has been moved to /backup/ (excluding images/thumbs)!";
 | 
						|
				if (is_dir($matches[0])) {
 | 
						|
					$objects = scandir($matches[0]);
 | 
						|
					foreach ($objects as $object) {
 | 
						|
						if ($object != "." && $object != "..") {
 | 
						|
							if (filetype($matches[0]."/".$object) == "dir") rmdir($matches[0]."/".$object); else unlink($matches[0]."/".$object);
 | 
						|
						}
 | 
						|
					}
 | 
						|
					reset($objects);
 | 
						|
					rmdir($matches[0]);
 | 
						|
					$html .= "<br>".$matches[0]." deleted!";
 | 
						|
				}
 | 
						|
				$html .= "<br>shimmie updated (although you may have gotten errors, it should have worked!";
 | 
						|
				$html .= "<br>due to the way shimmie loads extensions, all optional extensions have been disabled";
 | 
						|
				$config->set_string("commit_hash", $commit);
 | 
						|
				$html .= "<br>new commit_hash has been set!";
 | 
						|
			}else{
 | 
						|
				$html .= "Error! Folder does not exist!?"; //Although this should be impossible, shall have it anyway.
 | 
						|
			}
 | 
						|
		}elseif (file_exists($mfile)){ //#2
 | 
						|
			$zip = new ZipArchive;
 | 
						|
			if ($zip->open($mfile) === TRUE) {
 | 
						|
				$zip->extractTo('./');
 | 
						|
				$zip->close();
 | 
						|
				$html .= "extracted!";
 | 
						|
				$html .= "<br><a href='javascript:history.go(0)'>refresh</a> the page to continue!";
 | 
						|
				unlink($mfile); //Deletes master.zip
 | 
						|
			} else {
 | 
						|
				$html .= "failed!";
 | 
						|
			}
 | 
						|
		}else{ //#1
 | 
						|
			//Taken from the upload ext.
 | 
						|
			if($config->get_string("transload_engine") == "curl" && function_exists("curl_init")) {
 | 
						|
				$ch = curl_init($url);
 | 
						|
				$fp = fopen($mfile, "w");
 | 
						|
 | 
						|
				curl_setopt($ch, CURLOPT_FILE, $fp);
 | 
						|
				curl_setopt($ch, CURLOPT_HEADER, 0);
 | 
						|
				curl_setopt($ch, CURLOPT_REFERER, $url);
 | 
						|
				curl_setopt($ch, CURLOPT_USERAGENT, "Shimmie-".VERSION);
 | 
						|
 | 
						|
				curl_exec($ch);
 | 
						|
				curl_close($ch);
 | 
						|
				fclose($fp);
 | 
						|
				if(file_exists($mfile)){
 | 
						|
					$html .= "downloaded!";
 | 
						|
					$html .= "<br><a href='javascript:history.go(0)'>refresh</a> the page to continue!";
 | 
						|
				}else{
 | 
						|
					$html .= "download failed!";
 | 
						|
					$html .= "<br><a href='javascript:history.go(0)'>refresh</a> to try again!";
 | 
						|
					$html .= "<br>if you keep having this problem, you may have a problem with your transload engine!";
 | 
						|
				}
 | 
						|
			}elseif($config->get_string("transload_engine") == "wget") {
 | 
						|
				//this doesn't work?
 | 
						|
				$s_url = escapeshellarg($url);
 | 
						|
				system("wget $s_url --output-document=$mfile");
 | 
						|
				if(file_exists($mfile)){
 | 
						|
					$html .= "downloaded!";
 | 
						|
					$html .= "<br><a href='javascript:history.go(0)'>refresh</a> the page to continue!";
 | 
						|
				}else{
 | 
						|
					$html .= "download failed!";
 | 
						|
					$html .= "<br><a href='javascript:history.go(0)'>refresh</a> to try again!";
 | 
						|
					$html .= "<br>if you keep having this problem, you may have a problem with your transload engine!";
 | 
						|
				}
 | 
						|
			}elseif($config->get_string("transload_engine") == "fopen") {
 | 
						|
				$fp = @fopen($url, "r");
 | 
						|
				if(!$fp) {
 | 
						|
					return false;
 | 
						|
				}
 | 
						|
				$data = "";
 | 
						|
				$length = 0;
 | 
						|
				while(!feof($fp) && $length <= $config->get_int('upload_size')) {
 | 
						|
					$data .= fread($fp, 8192);
 | 
						|
					$length = strlen($data);
 | 
						|
				}
 | 
						|
				fclose($fp);
 | 
						|
 | 
						|
				$fp = fopen($mfile, "w");
 | 
						|
				fwrite($fp, $data);
 | 
						|
				fclose($fp);
 | 
						|
				if(file_exists($mfile)){
 | 
						|
					$html .= "downloaded!";
 | 
						|
					$html .= "<br><a href='javascript:history.go(0)'>refresh</a> the page to continue!";
 | 
						|
				}else{
 | 
						|
					$html .= "download failed!";
 | 
						|
					$html .= "<br><a href='javascript:history.go(0)'>refresh</a> to try again!";
 | 
						|
					$html .= "<br>if you keep having this problem, you may have a problem with your transload engine!";
 | 
						|
				}
 | 
						|
			}elseif($config->get_string("transload_engine") == "none"){
 | 
						|
				$html .= "no transload engine set!";
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		$page->add_block(new Block("Update", $html));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
?>
 |