mirror of
https://gitee.com/fastadminnet/framework.git
synced 2026-07-01 12:42:48 +08:00
401 lines
15 KiB
PHP
401 lines
15 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2006~2015 http://thinkphp.cn All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
|
// +----------------------------------------------------------------------
|
|
// | Author: liu21st <liu21st@gmail.com>
|
|
// +----------------------------------------------------------------------
|
|
|
|
namespace traits\think\model;
|
|
|
|
define('EXISTS_VALIDATE', 0);
|
|
define('MUST_VALIDATE', 1);
|
|
define('VALUE_VALIDATE', 2);
|
|
|
|
trait Auto
|
|
{
|
|
|
|
protected $validate = []; // 自动验证定义
|
|
protected $auto = []; // 自动完成定义
|
|
|
|
/**
|
|
* 创建数据对象 但不保存到数据库
|
|
* @access public
|
|
* @param mixed $data 创建数据
|
|
* @param string $type 状态
|
|
* @return mixed
|
|
*/
|
|
public function create($data = '', $type = '')
|
|
{
|
|
// 如果没有传值默认取POST数据
|
|
if (empty($data)) {
|
|
$data = $_POST;
|
|
} elseif (is_object($data)) {
|
|
$data = get_object_vars($data);
|
|
}
|
|
// 验证数据
|
|
if (empty($data) || !is_array($data)) {
|
|
$this->error = \think\Lang::get('_DATA_TYPE_INVALID_');
|
|
return false;
|
|
}
|
|
|
|
// 状态
|
|
$type = $type ? $type : (!empty($data[$this->getPk()]) ? self::MODEL_UPDATE : self::MODEL_INSERT);
|
|
|
|
// 检测提交字段的合法性
|
|
if (isset($this->options['field'])) {
|
|
// $this->field('field1,field2...')->create()
|
|
$fields = $this->options['field'];
|
|
unset($this->options['field']);
|
|
} elseif (self::MODEL_INSERT == $type && isset($this->insertFields)) {
|
|
$fields = $this->insertFields;
|
|
} elseif (self::MODEL_UPDATE == $type && isset($this->updateFields)) {
|
|
$fields = $this->updateFields;
|
|
}
|
|
if (isset($fields)) {
|
|
if (is_string($fields)) {
|
|
$fields = explode(',', $fields);
|
|
}
|
|
// 判断令牌验证字段
|
|
if (C('TOKEN_ON')) {
|
|
$fields[] = C('TOKEN_NAME');
|
|
}
|
|
|
|
foreach ($data as $key => $val) {
|
|
if (!in_array($key, $fields)) {
|
|
unset($data[$key]);
|
|
}
|
|
}
|
|
}
|
|
|
|
// 数据自动验证
|
|
if (!$this->autoValidation($data, $type)) {
|
|
return false;
|
|
}
|
|
|
|
// 验证完成生成数据对象
|
|
if ($this->autoCheckFields) {
|
|
// 开启字段检测 则过滤非法字段数据
|
|
$fields = $this->getDbFields();
|
|
foreach ($data as $key => $val) {
|
|
if (!in_array($key, $fields)) {
|
|
unset($data[$key]);
|
|
}
|
|
}
|
|
}
|
|
|
|
// 创建完成对数据进行自动处理
|
|
$this->autoOperation($data, $type);
|
|
// 赋值当前数据对象
|
|
$this->data = $data;
|
|
// 返回创建的数据以供其他调用
|
|
return $data;
|
|
}
|
|
|
|
/**
|
|
* 使用正则验证数据
|
|
* @access public
|
|
* @param string $value 要验证的数据
|
|
* @param string $rule 验证规则
|
|
* @return boolean
|
|
*/
|
|
public function regex($value, $rule)
|
|
{
|
|
static $validate = [
|
|
'require' => '/.+/',
|
|
'email' => '/^\w+([-+.]\w+)*@\w+([-.]\w+)*\.\w+([-.]\w+)*$/',
|
|
'url' => '/^http(s?):\/\/(?:[A-za-z0-9-]+\.)+[A-za-z]{2,4}(?:[\/\?#][\/=\?%\-&~`@[\]\':+!\.#\w]*)?$/',
|
|
'currency' => '/^\d+(\.\d+)?$/',
|
|
'number' => '/^\d+$/',
|
|
'zip' => '/^\d{6}$/',
|
|
'integer' => '/^[-\+]?\d+$/',
|
|
'double' => '/^[-\+]?\d+(\.\d+)?$/',
|
|
'english' => '/^[A-Za-z]+$/',
|
|
];
|
|
// 检查是否有内置的正则表达式
|
|
if (isset($validate[strtolower($rule)])) {
|
|
$rule = $validate[strtolower($rule)];
|
|
}
|
|
return preg_match($rule, $value) === 1;
|
|
}
|
|
|
|
/**
|
|
* 自动表单处理
|
|
* @access public
|
|
* @param array $data 创建数据
|
|
* @param string $type 创建类型
|
|
* @return mixed
|
|
*/
|
|
private function autoOperation(&$data, $type)
|
|
{
|
|
if (!empty($this->options['auto'])) {
|
|
$_auto = $this->options['auto'];
|
|
unset($this->options['auto']);
|
|
} elseif (!empty($this->auto)) {
|
|
$_auto = $this->auto;
|
|
}
|
|
// 自动填充
|
|
if (isset($_auto)) {
|
|
foreach ($_auto as $auto) {
|
|
// 填充因子定义格式
|
|
// ['field','填充内容','填充条件','附加规则',[额外参数]]
|
|
if (empty($auto[2])) {
|
|
$auto[2] = self::MODEL_INSERT; // 默认为新增的时候自动填充
|
|
}
|
|
if ($type == $auto[2] || self::MODEL_BOTH == $auto[2]) {
|
|
switch (trim($auto[3])) {
|
|
case 'function': // 使用函数进行填充 字段的值作为参数
|
|
case 'callback': // 使用回调方法
|
|
$args = isset($auto[4]) ? (array) $auto[4] : [];
|
|
if (isset($data[$auto[0]])) {
|
|
array_unshift($args, $data[$auto[0]]);
|
|
}
|
|
if ('function' == $auto[3]) {
|
|
$data[$auto[0]] = call_user_func_array($auto[1], $args);
|
|
} else {
|
|
$data[$auto[0]] = call_user_func_array([ & $this, $auto[1]], $args);
|
|
}
|
|
break;
|
|
case 'field': // 用其它字段的值进行填充
|
|
$data[$auto[0]] = $data[$auto[1]];
|
|
break;
|
|
case 'ignore': // 为空忽略
|
|
if ('' === $data[$auto[0]]) {
|
|
unset($data[$auto[0]]);
|
|
}
|
|
break;
|
|
case 'string':
|
|
default: // 默认作为字符串填充
|
|
$data[$auto[0]] = $auto[1];
|
|
}
|
|
if (false === $data[$auto[0]]) {
|
|
unset($data[$auto[0]]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return $data;
|
|
}
|
|
|
|
/**
|
|
* 自动表单验证
|
|
* @access protected
|
|
* @param array $data 创建数据
|
|
* @param string $type 创建类型
|
|
* @return boolean
|
|
*/
|
|
protected function autoValidation($data, $type)
|
|
{
|
|
if (!empty($this->options['validate'])) {
|
|
$_validate = $this->options['validate'];
|
|
unset($this->options['validate']);
|
|
} elseif (!empty($this->validate)) {
|
|
$_validate = $this->validate;
|
|
}
|
|
// 属性验证
|
|
if (isset($_validate)) {
|
|
// 如果设置了数据自动验证则进行数据验证
|
|
if ($this->patchValidate) {
|
|
// 重置验证错误信息
|
|
$this->error = [];
|
|
}
|
|
foreach ($_validate as $key => $val) {
|
|
// 验证因子定义格式
|
|
// [field,rule,message,condition,type,when,params]
|
|
// 判断是否需要执行验证
|
|
if (empty($val[5]) || self::MODEL_BOTH == $val[5] || $val[5] == $type) {
|
|
if (0 == strpos($val[2], '{%') && strpos($val[2], '}')) {
|
|
// 支持提示信息的多语言 使用 {%语言定义} 方式
|
|
$val[2] = \think\Lang::get(substr($val[2], 2, -1));
|
|
}
|
|
$val[3] = isset($val[3]) ? $val[3] : EXISTS_VALIDATE;
|
|
$val[4] = isset($val[4]) ? $val[4] : 'regex';
|
|
// 判断验证条件
|
|
switch ($val[3]) {
|
|
case MUST_VALIDATE: // 必须验证 不管表单是否有设置该字段
|
|
if (false === $this->_validationField($data, $val)) {
|
|
return false;
|
|
}
|
|
|
|
break;
|
|
case VALUE_VALIDATE: // 值不为空的时候才验证
|
|
if ('' != trim($data[$val[0]])) {
|
|
if (false === $this->_validationField($data, $val)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
break;
|
|
default: // 默认表单存在该字段就验证
|
|
if (isset($data[$val[0]]) && false === $this->_validationField($data, $val)) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// 批量验证的时候最后返回错误
|
|
if (!empty($this->error)) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* 验证表单字段 支持批量验证
|
|
* 如果批量验证返回错误的数组信息
|
|
* @access protected
|
|
* @param array $data 创建数据
|
|
* @param array $val 验证因子
|
|
* @return boolean
|
|
*/
|
|
protected function _validationField($data, $val)
|
|
{
|
|
if (false === $this->_validationFieldItem($data, $val)) {
|
|
if ($this->patchValidate) {
|
|
$this->error[$val[0]] = $val[2];
|
|
} else {
|
|
$this->error = $val[2];
|
|
return false;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
/**
|
|
* 根据验证因子验证字段
|
|
* @access protected
|
|
* @param array $data 创建数据
|
|
* @param array $val 验证因子
|
|
* @return boolean
|
|
*/
|
|
protected function _validationFieldItem($data, $val)
|
|
{
|
|
switch (strtolower(trim($val[4]))) {
|
|
case 'function': // 使用函数进行验证
|
|
case 'callback': // 调用方法进行验证
|
|
$args = isset($val[6]) ? (array) $val[6] : [];
|
|
if (is_string($val[0]) && strpos($val[0], ',')) {
|
|
$val[0] = explode(',', $val[0]);
|
|
}
|
|
if (is_array($val[0])) {
|
|
// 支持多个字段验证
|
|
foreach ($val[0] as $field) {
|
|
$_data[$field] = $data[$field];
|
|
}
|
|
array_unshift($args, $_data);
|
|
} else {
|
|
array_unshift($args, $data[$val[0]]);
|
|
}
|
|
return call_user_func_array('function' == $val[4] ? $val[1] : [ & $this, $val[1]], $args);
|
|
case 'confirm': // 验证两个字段是否相同
|
|
return $data[$val[0]] == $data[$val[1]];
|
|
case 'unique': // 验证某个值是否唯一
|
|
if (is_string($val[0]) && strpos($val[0], ',')) {
|
|
$val[0] = explode(',', $val[0]);
|
|
}
|
|
$map = [];
|
|
if (is_array($val[0])) {
|
|
// 支持多个字段验证
|
|
foreach ($val[0] as $field) {
|
|
$map[$field] = $data[$field];
|
|
}
|
|
} else {
|
|
$map[$val[0]] = $data[$val[0]];
|
|
}
|
|
if (!empty($data[$this->getPk()])) {
|
|
// 完善编辑的时候验证唯一
|
|
$map[$this->getPk()] = ['neq', $data[$this->getPk()]];
|
|
}
|
|
return $this->where($map)->find() ? false : true;
|
|
default: // 检查附加规则
|
|
return $this->check($data[$val[0]], $val[1], $val[4]);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 验证数据 支持 in between equal length regex expire ip_allow ip_deny
|
|
* @access public
|
|
* @param string $value 验证数据
|
|
* @param mixed $rule 验证表达式
|
|
* @param string $type 验证方式 默认为正则验证
|
|
* @return boolean
|
|
*/
|
|
public function check($value, $rule, $type = 'regex')
|
|
{
|
|
$type = strtolower(trim($type));
|
|
switch ($type) {
|
|
case 'in': // 验证是否在某个指定范围之内 逗号分隔字符串或者数组
|
|
case 'notin':
|
|
$range = is_array($rule) ? $rule : explode(',', $rule);
|
|
return 'in' == $type ? in_array($value, $range) : !in_array($value, $range);
|
|
case 'between': // 验证是否在某个范围
|
|
case 'notbetween': // 验证是否不在某个范围
|
|
if (is_array($rule)) {
|
|
$min = $rule[0];
|
|
$max = $rule[1];
|
|
} else {
|
|
list($min, $max) = explode(',', $rule);
|
|
}
|
|
return 'between' == $type ? $value >= $min && $value <= $max : $value < $min || $value > $max;
|
|
case 'equal': // 验证是否等于某个值
|
|
case 'notequal': // 验证是否等于某个值
|
|
return 'equal' == $type ? $value == $rule : $value != $rule;
|
|
case 'length': // 验证长度
|
|
$length = mb_strlen($value, 'utf-8'); // 当前数据长度
|
|
if (strpos($rule, ',')) {
|
|
// 长度区间
|
|
list($min, $max) = explode(',', $rule);
|
|
return $length >= $min && $length <= $max;
|
|
} else {
|
|
// 指定长度
|
|
return $length == $rule;
|
|
}
|
|
case 'expire':
|
|
list($start, $end) = explode(',', $rule);
|
|
if (!is_numeric($start)) {
|
|
$start = strtotime($start);
|
|
}
|
|
if (!is_numeric($end)) {
|
|
$end = strtotime($end);
|
|
}
|
|
return NOW_TIME >= $start && NOW_TIME <= $end;
|
|
case 'ip_allow': // IP 操作许可验证
|
|
return in_array(get_client_ip(), explode(',', $rule));
|
|
case 'ip_deny': // IP 操作禁止验证
|
|
return !in_array(get_client_ip(), explode(',', $rule));
|
|
case 'regex':
|
|
default: // 默认使用正则验证 可以使用验证类中定义的验证名称
|
|
// 检查附加规则
|
|
return $this->regex($value, $rule);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 指定自动完成
|
|
* @access public
|
|
* @param array $auto 自动完成设置
|
|
* @return Model
|
|
*/
|
|
public function auto($auto)
|
|
{
|
|
$this->options['auto'] = $auto;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 指定自动验证
|
|
* @access public
|
|
* @param array $validate 自动验证设置
|
|
* @return Model
|
|
*/
|
|
public function validate($validate)
|
|
{
|
|
$this->options['validate'] = $validate;
|
|
return $this;
|
|
}
|
|
}
|