mirror-fastadminnet/framework
1
0
Fork 0
mirror of https://gitee.com/fastadminnet/framework.git synced 2026-07-06 15:02:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #213 from oldrind/input

Browse Source 
优化Input类
This commit is contained in:
Chino Chang
2016-01-17 18:59:53 +08:00
parent 39b08c6b89 90cb9b0baf
commit 7ea9f45f57
1 changed files with 114 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

217
library/think/Input.php
View File

@@ -14,7 +14,7 @@ namespace think;
class Input class Input
{ {
// 全局过滤规则 // 全局过滤规则
public static $filter = null; public static $filters;
/** /**
* 获取get变量 * 获取get变量
@@ -23,7 +23,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function get($name = '', $default = null, $filter = '') public static function get($name = '', $default = null, $filter = null)
{ {
return self::getData($name, $_GET, $filter, $default); return self::getData($name, $_GET, $filter, $default);
} }
@@ -35,7 +35,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function post($name = '', $default = null, $filter = '') public static function post($name = '', $default = null, $filter = null)
{ {
return self::getData($name, $_POST, $filter, $default); return self::getData($name, $_POST, $filter, $default);
} }
@@ -47,7 +47,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function put($name = '', $default = null, $filter = '') public static function put($name = '', $default = null, $filter = null)
{ {
static $_PUT = null; static $_PUT = null;
if (is_null($_PUT)) { if (is_null($_PUT)) {
@@ -63,7 +63,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function param($name = '', $default = null, $filter = '') public static function param($name = '', $default = null, $filter = null)
{ {
switch ($_SERVER['REQUEST_METHOD']) { switch ($_SERVER['REQUEST_METHOD']) {
case 'POST': case 'POST':
@@ -85,7 +85,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function request($name = '', $default = null, $filter = '') public static function request($name = '', $default = null, $filter = null)
{ {
return self::getData($name, $_REQUEST, $filter, $default); return self::getData($name, $_REQUEST, $filter, $default);
} }
@@ -97,7 +97,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function session($name = '', $default = null, $filter = '') public static function session($name = '', $default = null, $filter = null)
{ {
return self::getData($name, $_SESSION, $filter, $default); return self::getData($name, $_SESSION, $filter, $default);
} }
@@ -109,7 +109,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function cookie($name = '', $default = null, $filter = '') public static function cookie($name = '', $default = null, $filter = null)
{ {
return self::getData($name, $_COOKIE, $filter, $default); return self::getData($name, $_COOKIE, $filter, $default);
} }
@@ -121,7 +121,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function server($name = '', $default = null, $filter = '') public static function server($name = '', $default = null, $filter = null)
{ {
return self::getData(strtoupper($name), $_SERVER, $filter, $default); return self::getData(strtoupper($name), $_SERVER, $filter, $default);
} }
@@ -133,7 +133,7 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function globals($name = '', $default = null, $filter = '') public static function globals($name = '', $default = null, $filter = null)
{ {
return self::getData($name, $GLOBALS, $filter, $default); return self::getData($name, $GLOBALS, $filter, $default);
} }
@@ -145,86 +145,72 @@ class Input
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @return mixed * @return mixed
*/ */
public static function env($name = '', $default = null, $filter = '') public static function env($name = '', $default = null, $filter = null)
{ {
return self::getData(strtoupper($name), $_ENV, $filter, $default); return self::getData(strtoupper($name), $_ENV, $filter, $default);
} }
/** /**
* 获取系统变量 支持过滤和默认值 * 获取系统变量 支持过滤和默认值
* @param $name * @param string $name
* @param $input * @param array $input
* @param $filter * @param mixed $filter
* @param $default * @param mixed $default
* @return mixed * @return mixed
*/ */
public static function getData($name, $input, $filter = '', $default = null) public static function getData($name, $input = [], $filter = null, $default = null)
{ {
// 解析name
list($name, $type) = static::parseName($name);
// 解析过滤器 // 解析过滤器
$filters = static::parseFilters($filter); $filters = static::parseFilter($filter);
// 解析值 // 为方便传参把默认值附加在过滤器后面
if ('' === $name) { $filters[] = $default;
// 过滤所有输入 if (!is_array($input)) {
$data = $input;
// 对数组应用过滤器
foreach ($filters as $filter) {
$data = self::filter($filter, $data);
}
// 递归过滤表达式
array_walk_recursive($data, 'self::filterExp');
// 返回结果
return $data;
} elseif (isset($input[$name])) {
// 过滤name指定的输入
$data = $input[$name];
} else {
// 无输入数据, 下面直接返回默认值
return $default;
}
// 强制类型转换
$data = static::typeCast($data, $type);
// 正则过滤
$regex = static::regexFilter($data, $filter);
if (false === $regex) {
// 过滤器是正则表达式, 但数据无匹配
// 返回默认值
$data = $default; $data = $default;
} elseif (!is_null($regex)) { } elseif (empty($name)) {
// 数据合法，对结果进行强类型转换 $data = $input;
$data = static::typeCast($regex, $type); array_walk_recursive($data, 'self::filter', $filters);
} else { } else {
// 假如值为数组 // 解析name
if (is_array($data)) { list($name, $type) = static::parseName($name);
// 递归过滤表达式 if (isset($input[$name])) {
array_walk_recursive($data, 'self::filterExp'); // 过滤name指定的输入
} $data = $input[$name];
foreach ($filters as $filter) { if (is_array($data)) {
if (is_callable($filter)) { array_walk_recursive($data, 'self::filter', $filters);
$data = is_array($data) ? self::filter($filter, $data) : call_user_func($filter, $data); // 参数过滤
} else { } else {
// filter函数不存在时, 则使用filter_var进行过滤 self::filter($data, $name, $filters);
// filter为非整形值时, 调用filter_id取得过滤id
$data = filter_var($data, is_int($filter) ? $filter : filter_id($filter));
if (false === $data) {
// 不通过过滤器则返回默认值
return $default;
}
} }
if ($data !== $default) {
// 强制类型转换
static::typeCast($data, $type);
}
} else {
// 无输入数据
$data = $default;
} }
} }
return $data; return $data;
} }
/**
* 设置默认的过滤函数
* @param string|array $name
* @return array
*/
public static function setFilter($name)
{
if (is_string($name)) {
$name = explode(',', $name);
}
static::$filters = $name;
}
/** /**
* 过滤表单中的表达式 * 过滤表单中的表达式
* @param string &$value * @param string $value
* @return void * @return void
*/ */
public static function filterExp(&$value) protected static function filterExp(&$value)
{ {
// TODO 其他安全过滤 // TODO 其他安全过滤
@@ -236,17 +222,43 @@ class Input
/** /**
* 递归过滤给定的值 * 递归过滤给定的值
* @param string $filter * @param mixed $value 键值
* @param mixed $data * @param mixed $key 键名
* @param array $filters 过滤方法+默认值
* @return mixed * @return mixed
*/ */
public static function filter($filter, $data) private static function filter(&$value, $key, $filters)
{ {
$result = []; if (!empty($value)) {
foreach ($data as $key => $val) { // 分离出默认值
$result[$key] = is_array($val) ? self::filter($filter, $val) : call_user_func($filter, $val); $default = array_pop($filters);
foreach ($filters as $filter) {
if (is_callable($filter)) {
// 调用函数过滤
$value = call_user_func($filter, $value);
} else {
$begin = substr($filter, 0, 1);
if (in_array($begin, ['/','#','~']) && $begin == $end = substr($filter, -1)) {
// 正则过滤
if (!preg_match($filter, $value)) {
// 匹配不成功返回默认值
$value = $default;
break;
}
} else {
// filter函数不存在时, 则使用filter_var进行过滤
// filter为非整形值时, 调用filter_id取得过滤id
$value = filter_var($value, is_int($filter) ? $filter : filter_id($filter));
if (false === $value) {
// 不通过过滤器则返回默认值
$value = $default;
break;
}
}
}
}
self::filterExp($value);
} }
return $result;
} }
/** /**
@@ -261,55 +273,55 @@ class Input
/** /**
* 解析过滤器 * 解析过滤器
* @param mixed $filters * @param mixed $filter
* @return array * @return array
*/ */
private static function parseFilters($filters) private static function parseFilter($filter)
{ {
$filters = ('' === $filters) ? static::$filter : $filters; if (is_null($filter)) {
$result = self::getFilter();
if (empty($filters)) { } elseif (empty($filter)) {
$result = []; $result = [];
} elseif (is_string($filters)) {
$result = explode(',', $filters);
} elseif (is_array($filters)) {
$result = $filters;
} else { } else {
$result = [$filters]; if (is_array($filter)) {
$result = $filter;
} elseif (is_string($filter) && strpos($filter, ',')) {
$result = explode(',', $filter);
} else {
$result = [$filter];
}
// 如果最后一项为0或false，表示覆盖默认的过滤函数，否则为叠加
if (!end($result)) {
array_pop($result);
} else {
$result = array_merge(self::getFilter(), array_diff($result, self::getFilter()));
}
} }
return $result; return $result;
} }
/** /**
* 正则过滤 * 获取过滤方法
* @param string $input * @return array
* @param string $filter
* @return string|false
*/ */
private static function regexFilter($input, $filter) private static function getFilter()
{ {
if (empty($filter) || is_array($input)) { if (is_null(static::$filters)) {
return null; // 从配置项中读取
$filters = \think\Config::get('default_filter');
static::$filters = empty($filters) ? [] : (is_array($filters) ? $filters : explode(',', $filters));
} }
$begin = $filter[0]; return static::$filters;
$end = $filter[strlen($filter) - 1];
if (
('/' === $begin && '/' === $end) ||
('#' === $begin && '#' === $end) ||
('~' === $begin && '~' === $end)
) {
return !preg_match($filter, $input) ? false : $input;
}
return null;
} }
/** /**
* 强类型转换 * 强类型转换
* @param string $data * @param string $data
* @param string $type * @param string $type
* @return mixed * @return mixed
*/ */
private static function typeCast($data, $type) private static function typeCast(&$data, $type)
{ {
switch (strtolower($type)) { switch (strtolower($type)) {
// 数组 // 数组
@@ -333,6 +345,5 @@ class Input
default: default:
$data = (string) $data; $data = (string) $data;
} }
return $data;
} }
} }