<?php
// Upgrade the installation
function upgradeInstall($branch = 'v2-master', $stage)
{
	$notWritable = array_search(false, pathsWritable($GLOBALS['paths']));
	if ($notWritable == false) {
		ini_set('max_execution_time', 0);
		set_time_limit(0);
		$url = 'https://github.com/causefx/Organizr/archive/' . $branch . '.zip';
		$file = "upgrade.zip";
		$source = dirname(__DIR__, 2) . DIRECTORY_SEPARATOR . 'upgrade' . DIRECTORY_SEPARATOR . 'Organizr-' . str_replace('v2', '2', $branch) . DIRECTORY_SEPARATOR;
		$cleanup = dirname(__DIR__, 2) . DIRECTORY_SEPARATOR . "upgrade" . DIRECTORY_SEPARATOR;
		$destination = dirname(__DIR__, 2) . DIRECTORY_SEPARATOR;
		switch ($stage) {
			case '1':
				writeLog('success', 'Update Function -  Started Upgrade Process', $GLOBALS['organizrUser']['username']);
				if (downloadFile($url, $file)) {
					writeLog('success', 'Update Function -  Downloaded Update File for Branch: ' . $branch, $GLOBALS['organizrUser']['username']);
					return true;
				} else {
					writeLog('error', 'Update Function -  Downloaded Update File Failed  for Branch: ' . $branch, $GLOBALS['organizrUser']['username']);
					return false;
				}
				break;
			case '2':
				if (unzipFile($file)) {
					writeLog('success', 'Update Function -  Unzipped Update File for Branch: ' . $branch, $GLOBALS['organizrUser']['username']);
					return true;
				} else {
					writeLog('error', 'Update Function -  Unzip Failed for Branch: ' . $branch, $GLOBALS['organizrUser']['username']);
					return false;
				}
				break;
			case '3':
				if (rcopy($source, $destination)) {
					writeLog('success', 'Update Function -  Overwrited Files using Updated Files from Branch: ' . $branch, $GLOBALS['organizrUser']['username']);
					$updateComplete = $GLOBALS['dbLocation'] . 'completed.txt';
					if (!file_exists($updateComplete)) {
						touch($updateComplete);
					}
					return true;
				} else {
					writeLog('error', 'Update Function -  Overwrite Failed for Branch: ' . $branch, $GLOBALS['organizrUser']['username']);
					return false;
				}
				break;
			case '4':
				if (rrmdir($cleanup)) {
					writeLog('success', 'Update Function -  Deleted Update Files from Branch: ' . $branch, $GLOBALS['organizrUser']['username']);
					writeLog('success', 'Update Function -  Update Completed', $GLOBALS['organizrUser']['username']);
					return true;
				} else {
					writeLog('error', 'Update Function -  Removal of Update Files Failed for Branch: ' . $branch, $GLOBALS['organizrUser']['username']);
					return false;
				}
				break;
			default:
				return false;
				break;
		}
	} else {
		return 'permissions';
	}
	
}

function downloadFile($url, $path)
{
	ini_set('max_execution_time', 0);
	set_time_limit(0);
	$folderPath = dirname(__DIR__, 2) . DIRECTORY_SEPARATOR . "upgrade" . DIRECTORY_SEPARATOR;
	if (!file_exists($folderPath)) {
		if (@!mkdir($folderPath)) {
			writeLog('error', 'Update Function -  Folder Creation failed', $GLOBALS['organizrUser']['username']);
			return false;
		}
	}
	$newfname = $folderPath . $path;
	$context = stream_context_create(
		array(
			'ssl' => array(
				'verify_peer' => true,
				'cafile' => getCert()
			)
		)
	);
	$file = fopen($url, 'rb', false, $context);
	if ($file) {
		$newf = fopen($newfname, 'wb');
		if ($newf) {
			while (!feof($file)) {
				fwrite($newf, fread($file, 1024 * 8), 1024 * 8);
			}
		}
	} else {
		writeLog("error", "organizr could not download $url");
		return false;
	}
	if ($file) {
		fclose($file);
		writeLog("success", "organizr finished downloading the github zip file");
	} else {
		writeLog("error", "organizr could not download the github zip file");
		return false;
	}
	if ($newf) {
		fclose($newf);
		writeLog("success", "organizr created upgrade zip file from github zip file");
	} else {
		writeLog("error", "organizr could not create upgrade zip file from github zip file");
		return false;
	}
	return true;
}

function downloadFileToPath($from, $to, $path)
{
	ini_set('max_execution_time', 0);
	set_time_limit(0);
	if (@!mkdir($path, 0777, true)) {
		writeLog("error", "organizr could not create upgrade folder");
	}
	$file = fopen($from, 'rb');
	if ($file) {
		$newf = fopen($to, 'wb');
		if ($newf) {
			while (!feof($file)) {
				fwrite($newf, fread($file, 1024 * 8), 1024 * 8);
			}
		}
	} else {
		writeLog("error", "organizr could not download $url");
	}
	if ($file) {
		fclose($file);
		writeLog("success", "organizr finished downloading the github zip file");
	} else {
		writeLog("error", "organizr could not download the github zip file");
	}
	if ($newf) {
		fclose($newf);
		writeLog("success", "organizr created upgrade zip file from github zip file");
	} else {
		writeLog("error", "organizr could not create upgrade zip file from github zip file");
	}
	return true;
}

function unzipFile($zipFile)
{
	ini_set('max_execution_time', 0);
	set_time_limit(0);
	$zip = new ZipArchive;
	$extractPath = dirname(__DIR__, 2) . DIRECTORY_SEPARATOR . "upgrade/";
	if ($zip->open($extractPath . $zipFile) != "true") {
		writeLog("error", "organizr could not unzip upgrade.zip");
	} else {
		writeLog("success", "organizr unzipped upgrade.zip");
	}
	/* Extract Zip File */
	$zip->extractTo($extractPath);
	$zip->close();
	return true;
}

// Function to remove folders and files
function rrmdir($dir)
{
	ini_set('max_execution_time', 0);
	set_time_limit(0);
	if (is_dir($dir)) {
		$files = scandir($dir);
		foreach ($files as $file) {
			if ($file != "." && $file != "..") {
				rrmdir("$dir/$file");
			}
		}
		rmdir($dir);
	} elseif (file_exists($dir)) {
		unlink($dir);
	}
	return true;
}

// Function to Copy folders and files
function rcopy($src, $dst)
{
	ini_set('max_execution_time', 0);
	set_time_limit(0);
	$src = cleanPath($src);
	$dst = cleanPath($dst);
	if (is_dir($src)) {
		if (!file_exists($dst)) : mkdir($dst);
		endif;
		$files = scandir($src);
		foreach ($files as $file) {
			if ($file != "." && $file != "..") {
				rcopy("$src/$file", "$dst/$file");
			}
		}
	} elseif (file_exists($src)) {
		copy($src, $dst);
	}
	return true;
}