<?php
// Copyright (c) 2013 Datenstrom, http://datenstrom.se
// This file may be used and distributed under the terms of the public license.

// Command line core plugin
class YellowCommandline
{
	const Version = "0.2.1";
	var $yellow;			//access to API

	// Initialise plugin
	function onLoad($yellow)
	{
		$this->yellow = $yellow;
		$this->yellow->config->setDefault("commandBuildDefaultFile", "index.html");
		$this->yellow->config->setDefault("commandBuildCustomMediaExtension", ".txt");
		$this->yellow->config->setDefault("commandBuildCustomErrorFile", "error404.html");
	}
	
	// Handle command
	function onCommand($args)
	{
		list($name, $command) = $args;
		switch($command)
		{
			case "build":	$statusCode = $this->build($args); break;
			case "version":	$statusCode = $this->version(); break;
			default:		$statusCode = $this->help();
		}
		return $statusCode;
	}
	
	// Show available commands
	function help()
	{
		echo "Yellow command line ".YellowCommandline::Version."\n";
		echo "Syntax: yellow.php build DIRECTORY [LOCATION]\n";
		echo "        yellow.php version\n";
		return 0;
	}
	
	// Show software version
	function version()
	{
		echo "Yellow ".Yellow::Version."\n";
		foreach($this->yellow->plugins->plugins as $key=>$value) echo "$value[class] $value[version]\n";
		return 0;
	}
	
	// Build website
	function build($args)
	{
		$statusCode = 0;
		list($name, $command, $path, $location) = $args;
		if(!empty($path) && $path!="/")
		{
			if($this->yellow->config->isExisting("serverName"))
			{
				$serverName = $this->yellow->config->get("serverName");
				$serverBase = $this->yellow->config->get("serverBase");
				list($statusCode, $content, $media, $system, $error) = $this->buildStatic($serverName, $serverBase, $location, $path);
			} else {
				list($statusCode, $content, $media, $system, $error) = array(500, 0, 0, 0, 1);
				$fileName = $this->yellow->config->get("configDir").$this->yellow->config->get("configFile");
				echo "ERROR bulding website: Please configure serverName and serverBase in file '$fileName'!\n";
			}
			echo "Yellow build: $content content, $media media, $system system";
			echo ", $error error".($error!=1 ? 's' : '');
			echo ", status $statusCode\n";
		} else {
			echo "Yellow build: Invalid arguments\n";
		}
		return $statusCode;
	}
	
	// Build static files
	function buildStatic($serverName, $serverBase, $location, $path)
	{
		$this->yellow->toolbox->timerStart($time);
		$statusCodeMax = $error = 0;
		if(empty($location))
		{
			$pages = $this->yellow->pages->index(true);
			$fileNamesMedia = $this->yellow->toolbox->getDirectoryEntriesrecursive(
				$this->yellow->config->get("mediaDir"), "/.*/", false, false);
			$fileNamesMedia = array_merge($fileNamesMedia, $this->yellow->toolbox->getDirectoryEntries(
				".", "/.*\\".$this->yellow->config->get("commandBuildCustomMediaExtension")."/", false, false));
			$fileNamesSystem = array($this->yellow->config->get("commandBuildCustomErrorFile"));
		} else {
			$pages = new YellowPageCollection($this->yellow, $location);
			$pages->append(new YellowPage($this->yellow, $location));
			$fileNamesMedia = array();
			$fileNamesSystem = array();
		}
		foreach($pages as $page)
		{
			$statusCode = $this->buildStaticLocation($serverName, $serverBase, $page->location, $path);
			$statusCodeMax = max($statusCodeMax, $statusCode);
			if($statusCode >= 400)
			{
				++$error;
				echo "ERROR building location '".$page->location."', ".$this->yellow->page->getStatusCode(true)."\n";
			}
			if(defined("DEBUG") && DEBUG>=1) echo "YellowCommandline::buildStatic status:$statusCode location:".$page->location."\n";
		}
		foreach($fileNamesMedia as $fileName)
		{
			$statusCode = $this->copyStaticFile($fileName, "$path/$fileName") ? 200 : 500;
			$statusCodeMax = max($statusCodeMax, $statusCode);
			if($statusCode >= 400)
			{
				++$error;
				echo "ERROR building media file '$path/$fileName', ".$this->yellow->toolbox->getHttpStatusFormatted($statusCode)."\n";
			}
			if(defined("DEBUG") && DEBUG>=1) echo "YellowCommandline::buildStatic status:$statusCode file:$fileName\n";
		}
		foreach($fileNamesSystem as $fileName)
		{
			$statusCode = $this->buildStaticError($serverName, $serverBase, "$path/$fileName", 404) ? 200 : 500;
			$statusCodeMax = max($statusCodeMax, $statusCode);
			if($statusCode >= 400)
			{
				++$error;
				echo "ERROR building system file '$path/$fileName', ".$this->yellow->toolbox->getHttpStatusFormatted($statusCode)."\n";
			}
			if(defined("DEBUG") && DEBUG>=1) echo "YellowCommandline::buildStatic status:$statusCode file:$fileName\n";
		}
		$this->yellow->toolbox->timerStop($time);
		if(defined("DEBUG") && DEBUG>=1) echo "YellowCommandline::buildStatic time:$time ms\n";
		return array($statusCodeMax, count($pages), count($fileNamesMedia), count($fileNamesSystem), $error);
	}
	
	// Build static location as file
	function buildStaticLocation($serverName, $serverBase, $location, $path)
	{		
		ob_start();
		$_SERVER["SERVER_PROTOCOL"] = "HTTP/1.1";
		$_SERVER["SERVER_NAME"] = $serverName;
		$_SERVER["REQUEST_URI"] = $serverBase.$location;
		$_SERVER["SCRIPT_NAME"] = $serverBase."yellow.php";
		$statusCode = $this->yellow->request();
		if($statusCode != 404)
		{
			$fileOk = true;
			$modified = strtotime($this->yellow->page->getHeader("Last-Modified"));
			list($contentType, $contentEncoding) = explode(';', $this->yellow->page->getHeader("Content-Type"), 2);
			$staticLocation = $this->getStaticLocation($location, $contentType);
			if($location == $staticLocation)
			{
				$fileName = $this->getStaticFileName($location, $path);
				$fileData = ob_get_contents();
				if($statusCode>=301 && $statusCode<=303) $fileData = $this->getStaticRedirect($this->yellow->page->getHeader("Location"));
				$fileOk = $this->makeStaticFile($fileName, $fileData, $modified);
			} else {
				if(!$this->yellow->toolbox->isFileLocation($location))
				{
					$fileName = $this->getStaticFileName($location, $path);
					$fileData = $this->getStaticRedirect("http://$serverName$serverBase$staticLocation");
					$fileOk = $this->makeStaticFile($fileName, $fileData, $modified);
					if($fileOk)
					{
						$fileName = $this->getStaticFileName($staticLocation, $path);
						$fileData = ob_get_contents();
						$fileOk = $this->makeStaticFile($fileName, $fileData, $modified);
					}
				} else {
					$statusCode = 409;
					$this->yellow->page->error($statusCode, "Type '$contentType' does not match file name!");
				}
			}
			if(!$fileOk)
			{
				$statusCode = 500;
				$this->yellow->page->error($statusCode, "Can't write file '$fileName'!");
			}
		}
		ob_end_clean();
		return $statusCode;
	}
	
	// Build static error as file
	function buildStaticError($serverName, $serverBase, $fileName, $statusCodeRequest)
	{
		ob_start();
		$_SERVER["SERVER_PROTOCOL"] = "HTTP/1.1";
		$_SERVER["SERVER_NAME"] = $serverName;
		$_SERVER["REQUEST_URI"] = $serverBase."/";
		$_SERVER["SCRIPT_NAME"] = $serverBase."yellow.php";
		$statusCode = $this->yellow->request($statusCodeRequest);
		if($statusCode == $statusCodeRequest)
		{
			$modified = strtotime($this->yellow->page->getHeader("Last-Modified"));			
			if(!$this->makeStaticFile($fileName, ob_get_contents(), $modified))
			{
				$statusCode = 500;
				$this->yellow->page->error($statusCode, "Can't write file '$fileName'!");
			}
		}
		ob_end_clean();
		return $statusCode == $statusCodeRequest;
	}
	
	// Create static file
	function makeStaticFile($fileName, $fileData, $modified)
	{
		return $this->yellow->toolbox->makeFile($fileName, $fileData, true) &&
			$this->yellow->toolbox->modifyFile($fileName, $modified);
	}
	
	// Copy static file
	function copyStaticFile($fileNameSource, $fileNameDest)
	{
		return $this->yellow->toolbox->copyFile($fileNameSource, $fileNameDest, true) &&
			$this->yellow->toolbox->modifyFile($fileNameDest, filemtime($fileNameSource));
	}
	
	// Return static location corresponding to content type
	function getStaticLocation($location, $contentType)
	{
		if(!empty($contentType))
		{
			$extension = ($pos = strrposu($location, '.')) ? substru($location, $pos) : "";
			if($contentType == "text/html")
			{
				if($this->yellow->toolbox->isFileLocation($location))
				{
					if(!empty($extension) && $extension!=".html") $location .= ".html";
				}
			} else {
				if($this->yellow->toolbox->isFileLocation($location))
				{
					if(empty($extension)) $location .= ".unknown";
				} else {
					if(preg_match("/^(\w+)\/(\w+)/", $contentType, $matches)) $extension = ".$matches[2]";
					$location .= "index$extension";
				}
			}
		}
		return $location;
	}
	
	// Return static file name from location
	function getStaticFileName($location, $path)
	{
		$fileName = $path.$location;
		if(!$this->yellow->toolbox->isFileLocation($location))
		{
			$fileName .= $this->yellow->config->get("commandBuildDefaultFile");
		}
		return $fileName;
	}
	
	// Return static redirect data
	function getStaticRedirect($url)
	{
		$data  = "<!DOCTYPE html><html>\n";
		$data .= "<head>\n";
		$data .= "<meta http-equiv=\"content-type\" content=\"text/html; charset=utf-8\" />\n";
		$data .= "<meta http-equiv=\"refresh\" content=\"0;url=$url\" />\n";
		$data .= "</head>\n";
		$data .= "</html>\n";
		return $data;
	}
}
	
$yellow->registerPlugin("commandline", "YellowCommandline", YellowCommandline::Version);
?>