This commit is contained in:
thinkphp
2016-01-21 17:37:40 +08:00
3 changed files with 183 additions and 118 deletions

View File

@@ -32,9 +32,14 @@ function C($name = '', $value = null, $range = '')
// 获取输入数据 支持默认值和过滤 // 获取输入数据 支持默认值和过滤
function I($key, $default = null, $filter = '') function I($key, $default = null, $filter = '')
{ {
if (strpos($key, '.')) { if ($pos = strpos($key, '.')) {
// 指定参数来源 // 指定参数来源
list($method, $key) = explode('.', $key, 2); $method = substr($key, 0, $pos);
if (in_array($method, ['get', 'post', 'put', 'param', 'request', 'session', 'cookie', 'server', 'globals', 'env', 'path', 'file', 'data'])) {
$key = substr($key, $pos + 1);
} else {
$method = 'param';
}
} else { } else {
// 默认为自动判断 // 默认为自动判断
$method = 'param'; $method = 'param';

View File

@@ -21,11 +21,12 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function get($name = '', $default = null, $filter = null) public static function get($name = '', $default = null, $filter = null, $merge = false)
{ {
return self::getData($name, $_GET, $filter, $default); return self::data($name, $default, $filter, $merge, $_GET);
} }
/** /**
@@ -33,11 +34,12 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function post($name = '', $default = null, $filter = null) public static function post($name = '', $default = null, $filter = null, $merge = false)
{ {
return self::getData($name, $_POST, $filter, $default); return self::data($name, $default, $filter, $merge, $_POST);
} }
/** /**
@@ -45,15 +47,16 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function put($name = '', $default = null, $filter = null) public static function put($name = '', $default = null, $filter = null, $merge = false)
{ {
static $_PUT = null; static $_PUT = null;
if (is_null($_PUT)) { if (is_null($_PUT)) {
parse_str(file_get_contents('php://input'), $_PUT); parse_str(file_get_contents('php://input'), $_PUT);
} }
return self::getData($name, $_PUT, $filter, $default); return self::data($name, $default, $filter, $merge, $_PUT);
} }
/** /**
@@ -61,21 +64,22 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function param($name = '', $default = null, $filter = null) public static function param($name = '', $default = null, $filter = null, $merge = false)
{ {
switch ($_SERVER['REQUEST_METHOD']) { switch ($_SERVER['REQUEST_METHOD']) {
case 'POST': case 'POST':
$result = self::post($name, $default, $filter); $method = 'post';
break; break;
case 'PUT': case 'PUT':
$result = self::put($name, $default, $filter); $method = 'put';
break; break;
default: default:
$result = self::get($name, $default, $filter); $method = 'get';
} }
return $result; return self::$method($name, $default, $filter, $merge);
} }
/** /**
@@ -83,11 +87,12 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function request($name = '', $default = null, $filter = null) public static function request($name = '', $default = null, $filter = null, $merge = false)
{ {
return self::getData($name, $_REQUEST, $filter, $default); return self::data($name, $default, $filter, $merge, $_REQUEST);
} }
/** /**
@@ -95,11 +100,12 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function session($name = '', $default = null, $filter = null) public static function session($name = '', $default = null, $filter = null, $merge = false)
{ {
return self::getData($name, $_SESSION, $filter, $default); return self::data($name, $default, $filter, $merge, $_SESSION);
} }
/** /**
@@ -107,11 +113,12 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function cookie($name = '', $default = null, $filter = null) public static function cookie($name = '', $default = null, $filter = null, $merge = false)
{ {
return self::getData($name, $_COOKIE, $filter, $default); return self::data($name, $default, $filter, $merge, $_COOKIE);
} }
/** /**
@@ -119,11 +126,12 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function server($name = '', $default = null, $filter = null) public static function server($name = '', $default = null, $filter = null, $merge = false)
{ {
return self::getData(strtoupper($name), $_SERVER, $filter, $default); return self::data(strtoupper($name), $default, $filter, $merge, $_SERVER);
} }
/** /**
@@ -131,11 +139,12 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function globals($name = '', $default = null, $filter = null) public static function globals($name = '', $default = null, $filter = null, $merge = false)
{ {
return self::getData($name, $GLOBALS, $filter, $default); return self::data($name, $default, $filter, $merge, $GLOBALS);
} }
/** /**
@@ -143,51 +152,93 @@ class Input
* @param string $name 数据名称 * @param string $name 数据名称
* @param string $default 默认值 * @param string $default 默认值
* @param string $filter 过滤方法 * @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed * @return mixed
*/ */
public static function env($name = '', $default = null, $filter = null) public static function env($name = '', $default = null, $filter = null, $merge = false)
{ {
return self::getData(strtoupper($name), $_ENV, $filter, $default); return self::data(strtoupper($name), $default, $filter, $merge, $_ENV);
}
/**
* 获取PATH_INFO
* @param string $name 数据名称
* @param string $default 默认值
* @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed
*/
public static function path($name = '', $default = null, $filter = null, $merge = false)
{
if (!empty($_SERVER['PATH_INFO'])) {
$depr = \think\Config::get('pathinfo_depr');
$input = explode($depr, trim($_SERVER['PATH_INFO'], $depr));
return self::data($name, $default, $filter, $merge, $input);
} else {
return $default;
}
}
/**
* 获取$_FILES
* @param string $name 数据名称
* @param string $default 默认值
* @param string $filter 过滤方法
* @param boolean $merge 是否与默认的过虑方法合并
* @return mixed
*/
public static function file($name = '', $default = null, $filter = null, $merge = false)
{
return self::data($name, $default, $filter, $merge, $_FILES);
} }
/** /**
* 获取系统变量 支持过滤和默认值 * 获取系统变量 支持过滤和默认值
* @param string $name * @param string $name 字段名
* @param array $input * @param mixed $default 默认值
* @param mixed $filter * @param mixed $filter 过滤函数
* @param mixed $default * @param boolean $merge 是否与默认的过虑方法合并
* @param array $input 数据源
* @return mixed * @return mixed
*/ */
public static function getData($name, $input = [], $filter = null, $default = null) public static function data($name, $default = null, $filter = null, $merge = false, $input = null)
{ {
// 解析过滤器 if (is_null($input) && !empty($name)) {
$filters = static::parseFilter($filter); $input = $name;
// 为方便传参把默认值附加在过滤器后面 $name = '';
$filters[] = $default; }
if (!is_array($input)) { if (!empty($input)) {
$data = $default;
} elseif (empty($name)) {
$data = $input; $data = $input;
array_walk_recursive($data, 'self::filter', $filters); $name = (string) $name;
} else { if ('' != $name) {
// 解析name // 解析name
list($name, $type) = static::parseName($name); list($name, $type) = static::parseName($name);
if (isset($input[$name])) { // 按.拆分成多维数组进行判断
// 过滤name指定的输入 foreach (explode('.', $name) as $val) {
$data = $input[$name]; if (isset($data[$val])) {
if (is_array($data)) { $data = $data[$val];
array_walk_recursive($data, 'self::filter', $filters); } else {
} else { // 无输入数据,返回默认值
self::filter($data, $name, $filters); return $default;
}
} }
if ($data !== $default) {
// 强制类型转换
static::typeCast($data, $type);
}
} else {
// 无输入数据
$data = $default;
} }
// 解析过滤器
$filters = static::parseFilter($filter, $merge);
// 为方便传参把默认值附加在过滤器后面
$filters[] = $default;
if (is_array($data)) {
array_walk_recursive($data, 'self::filter', $filters);
} else {
self::filter($data, $name?:0, $filters);
}
if (isset($type) && $data !== $default) {
// 强制类型转换
static::typeCast($data, $type);
}
} else {
$data = $default;
} }
return $data; return $data;
} }
@@ -210,7 +261,7 @@ class Input
* @param string $value * @param string $value
* @return void * @return void
*/ */
protected static function filterExp(&$value) public static function filterExp(&$value)
{ {
// TODO 其他安全过滤 // TODO 其他安全过滤
@@ -229,36 +280,34 @@ class Input
*/ */
private static function filter(&$value, $key, $filters) private static function filter(&$value, $key, $filters)
{ {
if (!empty($value)) { // 分离出默认值
// 分离出默认值 $default = array_pop($filters);
$default = array_pop($filters); foreach ($filters as $filter) {
foreach ($filters as $filter) { if (is_callable($filter)) {
if (is_callable($filter)) { // 调用函数过滤
// 调用函数过滤 $value = call_user_func($filter, $value);
$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 { } else {
$begin = substr($filter, 0, 1); // filter函数不存在时, 则使用filter_var进行过滤
if (in_array($begin, ['/','#','~']) && $begin == $end = substr($filter, -1)) { // filter为非整形值时, 调用filter_id取得过滤id
// 正则过滤 $value = filter_var($value, is_int($filter) ? $filter : filter_id($filter));
if (!preg_match($filter, $value)) { if (false === $value) {
// 匹配不成功返回默认值 // 不通过过滤器则返回默认值
$value = $default; $value = $default;
break; 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);
} }
self::filterExp($value);
} }
/** /**
@@ -276,7 +325,7 @@ class Input
* @param mixed $filter * @param mixed $filter
* @return array * @return array
*/ */
private static function parseFilter($filter) private static function parseFilter($filter, $merge = false)
{ {
if (is_null($filter)) { if (is_null($filter)) {
$result = self::getFilter(); $result = self::getFilter();
@@ -290,10 +339,8 @@ class Input
} else { } else {
$result = [$filter]; $result = [$filter];
} }
// 如果最后一项为0或false表示覆盖默认的过滤函数否则为叠加 if ($merge) {
if (!end($result)) { // 与默认的过滤函数合并
array_pop($result);
} else {
$result = array_merge(self::getFilter(), array_diff($result, self::getFilter())); $result = array_merge(self::getFilter(), array_diff($result, self::getFilter()));
} }
} }

View File

@@ -20,48 +20,47 @@ use think\Input;
class inputTest extends \PHPUnit_Framework_TestCase class inputTest extends \PHPUnit_Framework_TestCase
{ {
public function testEmptyStringName()
{
$input = ['a' => 'test'];
$this->assertEquals($input, Input::getData('', $input, 'trim'));
}
public function testInputName() public function testInputName()
{ {
$input = ['a' => 'test']; $input = ['a' => 'a', 'b' => ['c' => [' one ', 'two']]];
$this->assertEquals($input['a'], Input::getData('a', $input)); $this->assertEquals($input, Input::data($input));
$this->assertEquals($input['a'], Input::data($input['a']));
$this->assertEquals('one', Input::data('b.c.0/s', 'default', 'trim', false, $input));
} }
public function testDefaultValue() public function testDefaultValue()
{ {
$input = ['a' => 'test']; $input = ['a' => 'test'];
$default = 'default'; $default = 'default';
$this->assertEquals($default, Input::getData('foo', $input, null, $default)); $this->assertEquals($default, Input::data($input['b'], $default));
$this->assertEquals($default, Input::data($input, $default, '', false, $input));
$this->assertEquals($default, Input::get('a', $default));
} }
public function testStringFilter() public function testStringFilter()
{ {
$input = ['a' => ' test ', 'b' => ' test<> ']; $input = ['a' => ' test ', 'b' => ' test<> '];
$filters = 'trim'; $filters = 'trim';
$this->assertEquals('test', Input::getData('a', $input, $filters)); $this->assertEquals('test', Input::data('a', '', $filters, false, $input));
$filters = 'trim,htmlspecialchars'; $filters = 'trim,htmlspecialchars';
$this->assertEquals('test&lt;&gt;', Input::getData('b', $input, $filters)); $this->assertEquals('test&lt;&gt;', Input::data('b', '', $filters, false, $input));
} }
public function testArrayFilter() public function testArrayFilter()
{ {
$input = ['a' => ' test ', 'b' => ' test<> ']; $input = ['a' => ' test ', 'b' => ' test<> '];
$filters = ['trim']; $filters = ['trim'];
$this->assertEquals('test', Input::getData('a', $input, $filters)); $this->assertEquals('test', Input::data('a', '', $filters, false, $input));
$filters = ['trim', 'htmlspecialchars']; $filters = ['trim', 'htmlspecialchars'];
$this->assertEquals('test&lt;&gt;', Input::getData('b', $input, $filters)); $this->assertEquals('test&lt;&gt;', Input::data('b', '', $filters, false, $input));
} }
public function testFilterExp() public function testFilterExp()
{ {
$src = 'EXP|NEQ|GT|EGT|LT|ELT|OR|XOR|LIKE|NOTLIKE|NOT BETWEEN|NOTBETWEEN|BETWEEN|NOTIN|NOT IN|IN'; $src = 'EXP|NEQ|GT|EGT|LT|ELT|OR|XOR|LIKE|NOTLIKE|NOT BETWEEN|NOTBETWEEN|BETWEEN|NOTIN|NOT IN|IN';
$regexs = explode('|', $src); $regexs = explode('|', $src);
$data = Input::getData('', $regexs); $data = Input::data($regexs);
foreach ($regexs as $key => $value) { foreach ($regexs as $key => $value) {
$expected = $value . ' '; $expected = $value . ' ';
$this->assertEquals($expected, $data[$key]); $this->assertEquals($expected, $data[$key]);
@@ -70,11 +69,13 @@ class inputTest extends \PHPUnit_Framework_TestCase
public function testFiltrateWithRegex() public function testFiltrateWithRegex()
{ {
$input = ['a' => 'test1', 'b' => '_test2']; $input = ['a' => 'test1', 'b' => '_test2', 'c' => ''];
$filters = '/^test/'; $filters = '/^test/';
$this->assertEquals('test1', Input::getData('a', $input, $filters)); $this->assertEquals('test1', Input::data('a', '', $filters, false, $input));
$default = 'default value'; $default = 'default value';
$this->assertEquals($default, Input::getData('b', $input, $filters, $default)); $this->assertEquals($default, Input::data('b', $default, $filters, false, $input));
$filters = '/.+/';
$this->assertEquals('default value', Input::data('c', $default, $filters, false, $input));
} }
public function testFiltrateWithFilterVar() public function testFiltrateWithFilterVar()
@@ -84,10 +85,10 @@ class inputTest extends \PHPUnit_Framework_TestCase
$default = false; $default = false;
$input = ['a' => $email, 'b' => $error]; $input = ['a' => $email, 'b' => $error];
$filters = FILTER_VALIDATE_EMAIL; $filters = FILTER_VALIDATE_EMAIL;
$this->assertEquals($email, Input::getData('a', $input, $filters)); $this->assertEquals($email, Input::data('a', '', $filters, false, $input));
$this->assertFalse(Input::getData('b', $input, $filters, $default)); $this->assertFalse(Input::data('b', $default, $filters, false, $input));
$filters = 'validate_email'; $filters = 'validate_email';
$this->assertFalse(Input::getData('b', $input, $filters, $default)); $this->assertFalse(Input::data('b', $default, $filters, false, $input));
} }
public function testAllInput() public function testAllInput()
@@ -109,21 +110,21 @@ class inputTest extends \PHPUnit_Framework_TestCase
'e' => 'NEQ ', 'e' => 'NEQ ',
'f' => 'gt ', 'f' => 'gt ',
]; ];
$this->assertEquals($excepted, Input::getData('', $input, $filters)); $this->assertEquals($excepted, Input::data($input, '', $filters));
} }
public function testTypeCast() public function testTypeCast()
{ {
$input = [ $_POST = [
'a' => [1, 2, 3], 'a' => [1, 2, 3],
'b' => '1000', 'b' => '1000',
'c' => '3.14', 'c' => '3.14',
'd' => 'test boolean', 'd' => 'test boolean',
]; ];
$this->assertEquals([1, 2, 3], Input::getData('a/a', $input)); $this->assertEquals([1, 2, 3], Input::post('a/a'));
$this->assertEquals(1000, Input::getData('b/d', $input)); $this->assertEquals(1000, Input::post('b/d'));
$this->assertEquals(3.14, Input::getData('c/f', $input)); $this->assertEquals(3.14, Input::post('c/f'));
$this->assertEquals(true, Input::getData('d/b', $input)); $this->assertEquals(true, Input::post('d/b'));
} }
public function testSuperglobals() public function testSuperglobals()
@@ -140,6 +141,9 @@ class inputTest extends \PHPUnit_Framework_TestCase
$_SERVER['REQUEST_METHOD'] = 'GET'; $_SERVER['REQUEST_METHOD'] = 'GET';
$this->assertEquals('get value', Input::param('get')); $this->assertEquals('get value', Input::param('get'));
$this->assertEquals(null, Input::param('post')); $this->assertEquals(null, Input::param('post'));
$this->assertEquals(null, Input::param('put'));
$_REQUEST = array_merge($_GET, $_POST);
$this->assertEquals('get value', Input::request('get'));
session_start(); session_start();
$_SESSION['test'] = 'session value '; $_SESSION['test'] = 'session value ';
@@ -152,21 +156,30 @@ class inputTest extends \PHPUnit_Framework_TestCase
$_SERVER['REQUEST_METHOD'] = 'GET '; $_SERVER['REQUEST_METHOD'] = 'GET ';
$this->assertEquals('GET', Input::server('REQUEST_METHOD')); $this->assertEquals('GET', Input::server('REQUEST_METHOD'));
$GLOBALS['total'] = 1000;
$this->assertEquals(1000, Input::globals('total'));
$this->assertEquals('testing', Input::env('APP_ENV')); $this->assertEquals('testing', Input::env('APP_ENV'));
//$_SERVER['PATH_INFO'] = 'path/info';
//$path = $_SERVER['PATH_INFO'] ? explode('/', $_SERVER['PATH_INFO'])[0] : '';
//$this->assertEquals($path, Input::path('0', ''));
$_FILES = ['file'=>['name'=>'test.png', 'type'=>'image/png', 'tmp_name'=>'/tmp/php5Wx0aJ', 'error'=>0, size=>15726]];
$this->assertEquals('image/png', Input::file('file.type'));
} }
public function testFilterCover() public function testFilterMerge()
{ {
Input::setFilter('htmlspecialchars'); Input::setFilter('htmlspecialchars');
$input = ['a' => ' test<> ', 'b' => '<b\\ar />']; $input = ['a' => ' test<> ', 'b' => '<b\\ar />'];
$this->assertEquals(' test<> ', Input::data('a', '', '', false, $input));
$filters = ['trim']; $filters = ['trim'];
$this->assertEquals('test&lt;&gt;', Input::getData('a', $input, $filters)); $this->assertEquals('test<>', Input::data('a', '', $filters, false, $input));
$filters = ['trim', false]; $this->assertEquals('test&lt;&gt;', Input::data('a', '', $filters, true, $input));
$this->assertEquals('test<>', Input::getData('a', $input, $filters));
$filters = 'stripslashes'; $filters = 'stripslashes';
$this->assertEquals("&lt;bar /&gt;", Input::getData('b', $input, $filters)); $this->assertEquals("&lt;bar /&gt;", Input::data('b', '', $filters, true, $input));
$filters = 'stripslashes,0';
$this->assertEquals("<bar />", Input::getData('b', $input, $filters));
} }
} }