view/Smarty.php

<?php
/**
 * Smarty
 *
 * @package ko\view
 * @author zhangchu
 */

if (!defined('KO_SMARTY_INC'))
{
	/**
	 * Smarty 类文件定义
	 */
	define('KO_SMARTY_INC', KO_DIR.DS.'vendor'.DS.'Smarty-3.1.21'.DS.'Smarty.class.php');
}
if (!defined('KO_SMARTY_PLUGINS'))
{
	/**
	 * Smarty 自定义插件目录定义
	 */
	define('KO_SMARTY_PLUGINS', dirname(KO_SMARTY_INC).DS.'plugins'.DS);
}
if (!defined('KO_TEMPLATE_DIR'))
{
	/**
	 * 模板路径
	 */
	define('KO_TEMPLATE_DIR', dirname(KO_INCLUDE_DIR).DS.'templates'.DS);
}
if (!defined('KO_TEMPLATE_C_DIR'))
{
	/**
	 * 模板编译路径
	 */
	define('KO_TEMPLATE_C_DIR', dirname(KO_INCLUDE_DIR).DS.'templates_c'.DS);
}

include_once(KO_SMARTY_INC);

/**
 * Smarty操作封装
 */
class Ko_View_Smarty
{
	private $_oSmarty;
	private $_aAutoInfo = array();

	public function __construct($aParam = array())
	{
		$this->_oSmarty = new Smarty();
		$this->_oSmarty->setTemplateDir(KO_TEMPLATE_DIR);
		$this->_oSmarty->setCompileDir(KO_TEMPLATE_C_DIR);
		$this->_oSmarty->addPluginsDir(KO_SMARTY_PLUGINS);
		foreach ($aParam as $key => $val)
		{
			$this->_oSmarty->$key = $val;
		}
	}

	/**
	 * 单行文本 input编辑/显示 或 多行文本 textarea编辑
	 */
	public function vAssignHtml($vTplVar, $vValue=null, $aExclude=array())
	{
		$this->_vAssignEscape($vTplVar, $vValue, array('Ko_View_Escape', 'VEscapeHtml'), $aExclude);
	}

	/**
	 * assign raw，不推荐使用 或 html文本 显示
	 */
	public function vAssignRaw($vTplVar, $vValue=null)
	{
		$this->_oSmarty->assign($vTplVar, $vValue);
	}

	public function vClearAssign($vVar)
	{
		$this->_oSmarty->clearAssign($vVar);
	}
	
	public function vClearAllAssign()
	{
		$this->_oSmarty->clearAllAssign();
	}

	/**
	 * 输出模板
	 */
	public function vDisplay($sFilePath, $sCacheId = null)
	{
		echo $this->sFetch($sFilePath, $sCacheId);
	}

	/**
	 * 返回模板替换后内容
	 *
	 * @return string
	 */
	public function sFetch($sFilePath, $sCacheId = null)
	{
		$this->_vPreAutoFetch($sFilePath);
		return $this->_oSmarty->fetch($sFilePath, $sCacheId);
	}

	/**
	 * @return string|array
	 */
	public function vGetTemplateDir($sKey = null)
	{
		return $this->_oSmarty->getTemplateDir($sKey);
	}

	public function vAddTemplateDir($vDir, $sKey = null)
	{
		$this->_oSmarty->addTemplateDir($vDir, $sKey);
	}

	public function vSetTemplateDir($vDir)
	{
		$this->_oSmarty->setTemplateDir($vDir);
	}
	
	/**
	 * @return string
	 */
	public function sGetCompileDir()
	{
		return $this->_oSmarty->getCompileDir();
	}
	
	public function vSetCompileDir($sDir)
	{
		$this->_oSmarty->setCompileDir($sDir);
	}
	
	/**
	 * @return array
	 */
	public function aGetPluginsDir()
	{
		return $this->_oSmarty->getPluginsDir();
	}

	public function vAddPluginsDir($vDir)
	{
		$this->_oSmarty->addPluginsDir($vDir);
	}
	
	public function vSetPluginsDir($vDir)
	{
		$this->_oSmarty->setPluginsDir($vDir);
	}

	/**
	 * @return boolean
	 */
	public function bIsCached($sFilePath, $sCacheId = null)
	{
		return $this->_oSmarty->isCached($sFilePath, $sCacheId);
	}
	
	public function vClearCache($sFilePath, $sCacheId = null)
	{
		$this->_oSmarty->clearCache($sFilePath, $sCacheId);
	}
	
	public function vClearAllCache()
	{
		$this->_oSmarty->clearAllCache();
	}
	
	public function vSetCaching($iCaching)
	{
		$this->_oSmarty->caching = $iCaching;
	}
	
	public function vSetCachingType($sType)
	{
		$this->_oSmarty->caching_type = $sType;
	}
	
	public function vSetCacheLifeTime($iLifeTime)
	{
		$this->_oSmarty->cache_lifetime = $iLifeTime;
	}

	/**
	 * @return boolean
	 */
	public function bTemplateExists($sFilePath)
	{
		return $this->_oSmarty->templateExists($sFilePath);
	}

	public function vGetTemplateVars($sName)
	{
		return $this->_oSmarty->getTemplateVars($sName);
	}

	public function vRegisterPlugin($sType, $sName, $vCallback)
	{
		return $this->_oSmarty->registerPlugin($sType, $sName, $vCallback);
	}

	/**
	 * 注册模版自动分析处理类
	 */
	public function vRegAutoInfoClass($sName, $sClass)
	{
		$this->_aAutoInfo[$sName]['class'] = $sClass;
	}

	/**
	 * 设置模版自动分析函数的参数
	 */
	public function vRegAutoInfoFunc($sName, $sFunc, $aPara)
	{
		$this->_aAutoInfo[$sName]['func'][$sFunc] = $aPara;
	}

	private function _vPreAutoFetch($sFilePath)
	{
		if (empty($this->_aAutoInfo))
		{
			return;
		}
		
		$aArr = $this->_aGetAutoArr($sFilePath);
		foreach ($aArr as $k => $v)
		{
			if ('L' === substr($k, -1))
			{
				continue;
			}
			list($sRegName, $sFuncName, $aViewPara) = Ko_View_Str::AParseAutoStr($k);
			if (!class_exists($this->_aAutoInfo[$sRegName]['class']))
			{
				continue;
			}
			$viewobj = Ko_Tool_Singleton::OInstance($this->_aAutoInfo[$sRegName]['class']);
			$funcname = 'vAuto_'.$sFuncName;
			if (!method_exists($viewobj, $funcname))
			{
				continue;
			}
			$realk = Ko_View_Str::SAssembleAutoStr($sRegName, $sFuncName, $aViewPara);
			$aRegPara = isset($this->_aAutoInfo[$sRegName]['func'][$sFuncName]) ? $this->_aAutoInfo[$sRegName]['func'][$sFuncName] : null;
			$viewobj->$funcname($this, $k, isset($aArr[$realk]) ? $aArr[$realk] : array(), $aRegPara, $aViewPara);
		}
	}
	
	private function _aGetAutoArr($sFilePath)
	{
		$templateDir = $this->_sGetTemplateDir();
		$sFullname = Ko_View_Str::SGetAbsoluteFile($sFilePath, $templateDir);
		if (!is_file($sFullname))
		{
			return array();
		}
		$stime = filemtime($sFullname);

		$sHashFile = $this->_oSmarty->getCompileDir().'/'.KO_VIEW_AUTOTAG.'_'.md5($sFullname);
		$htime = is_file($sHashFile) ? filemtime($sHashFile) : 0;
		if ($htime >= $stime)
		{
			$arr = Ko_Tool_Enc::ADecode(file_get_contents($sHashFile));
			if (false !== $arr)
			{
				$rebuild = false;
				foreach ($arr[0] as $file)
				{
					$itime = is_file($file) ? filemtime($file) : time();
					if ($htime < $itime)
					{
						$rebuild = true;
						break;
					}
				}
				if (!$rebuild)
				{
					return $arr[1];
				}
			}
		}

		$aFilelist = array();
		$sContent = file_get_contents($sFullname);
		$str = new Ko_View_Str($sContent);
		$aArr = $str->aParseArr($this->_oSmarty->left_delimiter, $this->_oSmarty->right_delimiter, $templateDir, $aFilelist);
		file_put_contents($sHashFile, Ko_Tool_Enc::SEncode(array($aFilelist, $aArr)));
		return $aArr;
	}

	private function _vAssignEscape($vTplVar, $vValue, $fnEscape, $aExclude)
	{
		if (is_array($vTplVar))
		{
			$vTplVar = call_user_func($fnEscape, $vTplVar, $aExclude);
		}
		else
		{
			$vValue = call_user_func($fnEscape, $vValue, $aExclude);
		}
		$this->_oSmarty->assign($vTplVar, $vValue);
	}
	
	private function _sGetTemplateDir()
	{
		$dirs = $this->_oSmarty->getTemplateDir();
		return is_array($dirs) ? $dirs[0] : $dirs;
	}
}