1 files changed, 76 insertions, 3 deletions
diff --git a/tests/FunctionalTests/protected/pages/ViewSource.php b/tests/FunctionalTests/protected/pages/ViewSource.php
index eeaf0f80..84f39161 100644
--- a/tests/FunctionalTests/protected/pages/ViewSource.php
+++ b/tests/FunctionalTests/protected/pages/ViewSource.php
@@ -2,10 +2,83 @@
 
 class ViewSource extends TPage
 {
-	protected function onLoad($param)
+	private $_path=null;
+	private $_fullPath=null;
+	private $_fileType=null;
+
+	protected function isFileTypeAllowed($extension)
+	{
+		return in_array($extension,array('tpl','page','php'));
+	}
+
+	protected function getFileExtension($fileName)
+	{
+		if(($pos=strrpos($fileName,'.'))===false)
+			return '';
+		else
+			return substr($fileName,$pos+1);
+	}
+
+	public function onLoad($param)
 	{
-		$pageName = $this->Request->getParameter("source");
-		var_dump($pageName);
+		parent::onLoad($param);
+		$path=$this->Request->Items['path'];
+		$fullPath=realpath($this->Service->BasePath.'/'.$path);
+		$fileExt=$this->getFileExtension($fullPath);
+		if($fullPath!==false && is_file($fullPath) && strpos($fullPath,$this->Service->BasePath)!==false)
+		{
+ 			if($this->isFileTypeAllowed($fileExt))
+ 			{
+				$this->_fullPath=strtr($fullPath,'\\','/');
+				$this->_path=strtr(substr($fullPath,strlen($this->Service->BasePath)),'\\','/');
+ 			}
+		}
+		if($this->_fullPath===null)
+			throw new THttpException(500,'File Not Found: %s',$path);
+		$basePath=dirname($this->_fullPath);
+		if($dh=opendir($basePath))
+		{
+			$str="<h2>{$this->_path}</h2>\n";
+			while(($file=readdir($dh))!==false)
+			{
+				if(is_file($basePath.'/'.$file))
+				{
+					$fileType=$this->getFileExtension($basePath.'/'.$file);
+					if($this->isFileTypeAllowed($fileType))
+					{
+						if($fileType==='tpl' || $fileType==='page')
+							$type='Template file';
+						else
+							$type='Class file';
+						$path='/'.ltrim(strtr(dirname($this->_path),'\\','/').'/'.$file,'/');
+						$str.="$type: <a href=\"?page=ViewSource&amp;path=$path\">$file</a><br/>";
+					}
+				}
+
+			}
+			closedir($dh);
+			$this->SourceList->Text=$str;
+		}
+
+		switch($fileExt)
+		{
+			case 'page' :
+			case 'tpl' :
+				$this->Highlighter->Language='prado';
+				break;
+			case 'php' :
+				$this->Highlighter->Language='php';
+				break;
+			case 'xml' :
+				$this->Highlighter->Language='xml';
+				break;
+			default :
+				$this->Highlighter->Language='html';
+				break;
+		}
+
+		$this->SourceView->Text=file_get_contents($this->_fullPath);
+		$this->dataBind();
 	}
 }