wenda-php/protected/plugin/xsearch/util/XSUtil.class.php

<?php
/**
 * XSUtil 类定义文件
 *
 * @author hightman
 * @link http://www.xunsearch.com/
 * @copyright Copyright &copy; 2011 HangZhou YunSheng Network Technology Co., Ltd.
 * @license http://www.xunsearch.com/license/
 * @version $Id$
 */

/**
 * XSUtil 工具程序通用代码
 * 
 * @author hightman <hightman@twomice.net>
 * @version 1.0.0
 * @package XS.util
 */
class XSUtil
{
	private static $optind, $options = null;
	private static $charset = null;

	/**
	 * 修正字符串至固定宽度
	 * 其中一个全角符号、汉字的宽度为半角字符的 2 倍。
	 * @param string $text 要修正的字符串
	 * @param int $size 修正的目标宽度
	 * @param string $pad 用于填充补足的字符
	 * @return type 
	 */
	public static function fixWidth($text, $size, $pad = ' ')
	{
		for ($i = $j = 0; $i < strlen($text) && $j < $size; $i++, $j++)
		{
			if ((ord($text[$i]) & 0xe0) == 0xe0)
			{
				if (($size - $j) == 1)
					break;
				$j++;
				$i += 2;
			}
		}
		return substr($text, 0, $i) . str_repeat($pad, $size - $j);
	}

	/**
	 * 设置输出、输入编码
	 * 默认输出的中文编码均为 UTF-8
	 * @param string $charset 期望得到的字符集
	 */
	public static function setCharset($charset)
	{
		if ($charset !== null && strcasecmp($charset, 'utf8') && strcasecmp($charset, 'utf-8'))
		{
			self::$charset = $charset;
			ob_start(array(__CLASS__, 'convertOut'));
		}
	}

	/**
	 * 把 UTF-8 字符串转换为用户编码
	 * @param string $buf 要转换字符串
	 * @return string 转换后的字符串
	 */
	public static function convertOut($buf)
	{
		if (self::$charset !== null)
			return XS::convert($buf, self::$charset, 'UTF-8');
		return $buf;
	}

	/**
	 * 把用户输入的字符串转换为 UTF-8 编码
	 * @param string $buf 要转换字符串
	 * @return string 转换后的字符串
	 */
	public static function convertIn($buf)
	{
		if (self::$charset !== null)
			return XS::convert($buf, 'UTF-8', self::$charset);
		return $buf;
	}

	/**
	 * 解析命令行参数
	 * @param array $valued 需要附加值的参数列表
	 * @return array 解析完的参数数组，未指定 - 开头的选项统一放入 '-' 的子数组
	 */
	public static function parseOpt($valued = array())
	{
		$result = array('-' => array());
		$params = isset($_SERVER['argv']) ? $_SERVER['argv'] : array();
		for ($i = 0; $i < count($params); $i++)
		{
			if ($params[$i] === '--')
			{
				for ($i = $i + 1; $i < count($params); $i++)
					$result['-'][] = $params[$i];
				break;
			}
			else if ($params[$i][0] === '-')
			{
				$value = true;
				$pname = substr($params[$i], 1);
				if ($pname[0] === '-')
				{
					$pname = substr($pname, 1);
					if (($pos = strpos($pname, '=')) !== false)
					{
						$value = substr($pname, $pos + 1);
						$pname = substr($pname, 0, $pos);
					}
				}
				else if (strlen($pname) > 1)
				{
					for ($j = 1; $j < strlen($params[$i]); $j++)
					{
						$pname = substr($params[$i], $j, 1);
						if (in_array($pname, $valued))
						{
							$value = substr($params[$i], $j + 1);
							break;
						}
						else if (($j + 1) != strlen($params[$i]))
						{
							$result[$pname] = true;
						}
					}
				}
				if ($value === true && in_array($pname, $valued) && isset($params[$i + 1]))
				{
					$value = $params[$i + 1];
					$i++;
				}
				$result[$pname] = $value;
			}
			else
			{
				$result['-'][] = $params[$i];
			}
		}
		self::$options = $result;
		self::$optind = 1;
		return $result;
	}

	/**
	 * 取得命令行参数
	 * 要求事先调用 parseOpt, 否则会自动以默认参数调用它。
	 * @param string $short 短参数名
	 * @param string $long 长参数名
	 * @param bool $extra 是否补用默认顺序的参数
	 * @return string 返回可用的参数值，若不存在则返回 null
	 * @see parseOpt
	 */
	public static function getOpt($short, $long = null, $extra = false)
	{
		if (self::$options === null)
			self::parseOpt();

		$value = null;
		$options = self::$options;
		if ($long !== null && isset($options[$long]))
			$value = $options[$long];
		else if ($short !== null && isset($options[$short]))
			$value = $options[$short];
		else if ($extra === true && isset($options['-'][self::$optind]))
		{
			$value = $options['-'][self::$optind];
			self::$optind++;
		}
		return $value;
	}

	/**
	 * 刷新标准输出缓冲区
	 */
	public static function flush()
	{
		flush();
		if (ob_get_level() > 0)
			ob_flush();
	}
}