ExistValidator.php 5.18 KB
Newer Older
w  
Qiang Xue committed
1 2 3
<?php
/**
 * @link http://www.yiiframework.com/
Qiang Xue committed
4
 * @copyright Copyright (c) 2008 Yii Software LLC
w  
Qiang Xue committed
5 6 7
 * @license http://www.yiiframework.com/license/
 */

w  
Qiang Xue committed
8
namespace yii\validators;
Qiang Xue committed
9 10

use Yii;
Qiang Xue committed
11
use yii\base\InvalidConfigException;
w  
Qiang Xue committed
12

w  
Qiang Xue committed
13
/**
Alexander Makarov committed
14
 * ExistValidator validates that the attribute value exists in a table.
w  
Qiang Xue committed
15
 *
16 17 18
 * ExistValidator checks if the value being validated can be found in the table column specified by
 * the ActiveRecord class [[targetClass]] and the attribute [[targetAttribute]].
 *
w  
Qiang Xue committed
19 20 21
 * This validator is often used to verify that a foreign key contains a value
 * that can be found in the foreign table.
 *
22 23 24 25 26 27 28 29
 * The followings are examples of validation rules using this validator:
 *
 * ```php
 * // a1 needs to exist
 * ['a1', 'exist']
 * // a1 needs to exist, but its value will use a2 to check for the existence
 * ['a1', 'exist', 'targetAttribute' => 'a2']
 * // a1 and a2 need to exist together, and they both will receive error message
Qiang Xue committed
30
 * [['a1', 'a2'], 'exist', 'targetAttribute' => ['a1', 'a2']]
31 32
 * // a1 and a2 need to exist together, only a1 will receive error message
 * ['a1', 'exist', 'targetAttribute' => ['a1', 'a2']]
Qiang Xue committed
33 34
 * // a1 needs to exist by checking the existence of both a2 and a3 (using a1 value)
 * ['a1', 'exist', 'targetAttribute' => ['a2', 'a1' => 'a3']]
35 36
 * ```
 *
w  
Qiang Xue committed
37
 * @author Qiang Xue <qiang.xue@gmail.com>
Alexander Makarov committed
38
 * @since 2.0
w  
Qiang Xue committed
39
 */
Alexander Makarov committed
40
class ExistValidator extends Validator
w  
Qiang Xue committed
41
{
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
    /**
     * @var string the name of the ActiveRecord class that should be used to validate the existence
     * of the current attribute value. It not set, it will use the ActiveRecord class of the attribute being validated.
     * @see targetAttribute
     */
    public $targetClass;
    /**
     * @var string|array the name of the ActiveRecord attribute that should be used to
     * validate the existence of the current attribute value. If not set, it will use the name
     * of the attribute currently being validated. You may use an array to validate the existence
     * of multiple columns at the same time. The array values are the attributes that will be
     * used to validate the existence, while the array keys are the attributes whose values are to be validated.
     * If the key and the value are the same, you can just specify the value.
     */
    public $targetAttribute;
    /**
     * @var string|array|\Closure additional filter to be applied to the DB query used to check the existence of the attribute value.
     * This can be a string or an array representing the additional query condition (refer to [[\yii\db\Query::where()]]
     * on the format of query condition), or an anonymous function with the signature `function ($query)`, where `$query`
     * is the [[\yii\db\Query|Query]] object that you can modify in the function.
     */
    public $filter;
w  
Qiang Xue committed
64

65 66 67 68 69 70 71 72 73 74
    /**
     * @inheritdoc
     */
    public function init()
    {
        parent::init();
        if ($this->message === null) {
            $this->message = Yii::t('yii', '{attribute} is invalid.');
        }
    }
Qiang Xue committed
75

76 77 78 79 80 81
    /**
     * @inheritdoc
     */
    public function validateAttribute($object, $attribute)
    {
        $targetAttribute = $this->targetAttribute === null ? $attribute : $this->targetAttribute;
Qiang Xue committed
82

83 84 85 86 87 88 89 90
        if (is_array($targetAttribute)) {
            $params = [];
            foreach ($targetAttribute as $k => $v) {
                $params[$v] = is_integer($k) ? $object->$v : $object->$k;
            }
        } else {
            $params = [$targetAttribute => $object->$attribute];
        }
Alexander Makarov committed
91

92 93 94
        foreach ($params as $value) {
            if (is_array($value)) {
                $this->addError($object, $attribute, Yii::t('yii', '{attribute} is invalid.'));
95

96 97 98
                return;
            }
        }
99

100 101
        $targetClass = $this->targetClass === null ? get_class($object) : $this->targetClass;
        $query = $this->createQuery($targetClass, $params);
102

103 104 105 106
        if (!$query->exists()) {
            $this->addError($object, $attribute, $this->message);
        }
    }
Qiang Xue committed
107

108 109 110 111 112 113 114 115 116 117 118 119 120 121
    /**
     * @inheritdoc
     */
    protected function validateValue($value)
    {
        if (is_array($value)) {
            return [$this->message, []];
        }
        if ($this->targetClass === null) {
            throw new InvalidConfigException('The "targetClass" property must be set.');
        }
        if (!is_string($this->targetAttribute)) {
            throw new InvalidConfigException('The "targetAttribute" property must be configured as a string.');
        }
122

123
        $query = $this->createQuery($this->targetClass, [$this->targetAttribute => $value]);
124

125 126
        return $query->exists() ? null : [$this->message, []];
    }
127

128 129
    /**
     * Creates a query instance with the given condition.
130 131
     * @param string $targetClass the target AR class
     * @param mixed $condition query condition
132 133 134 135 136 137 138 139 140 141 142 143 144 145
     * @return \yii\db\ActiveQueryInterface the query instance
     */
    protected function createQuery($targetClass, $condition)
    {
        /** @var \yii\db\ActiveRecordInterface $targetClass */
        $query = $targetClass::find()->where($condition);
        if ($this->filter instanceof \Closure) {
            call_user_func($this->filter, $query);
        } elseif ($this->filter !== null) {
            $query->andWhere($this->filter);
        }

        return $query;
    }
w  
Qiang Xue committed
146
}