SluggableBehavior.php 6.42 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
<?php
/**
 * @link http://www.yiiframework.com/
 * @copyright Copyright (c) 2008 Yii Software LLC
 * @license http://www.yiiframework.com/license/
 */

namespace yii\behaviors;

use yii\base\InvalidConfigException;
use yii\db\BaseActiveRecord;
use yii\helpers\Inflector;
13 14
use yii\validators\UniqueValidator;
use Yii;
15 16

/**
Qiang Xue committed
17
 * SluggableBehavior automatically fills the specified attribute with a value that can be used a slug in a URL.
18
 *
19
 * To use SluggableBehavior, insert the following code to your ActiveRecord class:
20 21 22 23 24 25 26 27
 *
 * ```php
 * use yii\behaviors\SluggableBehavior;
 *
 * public function behaviors()
 * {
 *     return [
 *         [
28
 *             'class' => SluggableBehavior::className(),
29
 *             'attribute' => 'title',
Qiang Xue committed
30
 *             // 'slugAttribute' => 'slug',
31 32 33 34
 *         ],
 *     ];
 * }
 * ```
35
 *
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
 * By default, SluggableBehavior will fill the `slug` attribute with a value that can be used a slug in a URL
 * when the associated AR object is being validated. If your attribute name is different, you may configure
 * the [[slugAttribute]] property like the following:
 *
 * ```php
 * public function behaviors()
 * {
 *     return [
 *         [
 *             'class' => SluggableBehavior::className(),
 *             'slugAttribute' => 'alias',
 *         ],
 *     ];
 * }
 * ```
51
 *
52
 * @author Alexander Kochetov <creocoder@gmail.com>
53
 * @author Paul Klimov <klimov.paul@gmail.com>
54 55 56 57 58
 * @since 2.0
 */
class SluggableBehavior extends AttributeBehavior
{
    /**
Qiang Xue committed
59
     * @var string the attribute that will receive the slug value
60
     */
61
    public $slugAttribute = 'slug';
62
    /**
63
     * @var string|array the attribute or list of attributes whose value will be converted into a slug
64 65
     */
    public $attribute;
Qiang Xue committed
66
    /**
67
     * @var string|callable the value that will be used as a slug. This can be an anonymous function
Qiang Xue committed
68 69 70 71 72 73 74 75 76 77 78
     * or an arbitrary value. If the former, the return value of the function will be used as a slug.
     * The signature of the function should be as follows,
     *
     * ```php
     * function ($event)
     * {
     *     // return slug
     * }
     * ```
     */
    public $value;
79
    /**
80 81 82
     * @var boolean whether to generate a new slug if it has already been generated before.
     * If true, the behavior will not generate a new slug even if [[attribute]] is changed.
     * @since 2.0.2
83 84
     */
    public $immutable = false;
85 86 87 88 89
    /**
     * @var boolean whether to ensure generated slug value to be unique among owner class records.
     * If enabled behavior will validate slug uniqueness automatically. If validation fails it will attempt
     * generating unique slug value from based one until success.
     */
90
    public $ensureUnique = false;
91
    /**
92 93 94
     * @var array configuration for slug uniqueness validator. Parameter 'class' may be omitted - by default
     * [[UniqueValidator]] will be used.
     * @see UniqueValidator
95
     */
96
    public $uniqueValidator = [];
97
    /**
98 99
     * @var callable slug unique value generator. It is used in case [[ensureUnique]] enabled and generated
     * slug is not unique. This should be a PHP callable with following signature:
100 101
     *
     * ```php
102
     * function ($baseSlug, $iteration, $model)
103 104 105 106 107
     * {
     *     // return uniqueSlug
     * }
     * ```
     *
108
     * If not set unique slug will be generated adding incrementing suffix to the base slug.
109
     */
110
    public $uniqueSlugGenerator;
111

112

113 114 115 116 117
    /**
     * @inheritdoc
     */
    public function init()
    {
118 119 120
        parent::init();

        if (empty($this->attributes)) {
121
            $this->attributes = [BaseActiveRecord::EVENT_BEFORE_VALIDATE => $this->slugAttribute];
122 123
        }

124
        if ($this->attribute === null && $this->value === null) {
Qiang Xue committed
125
            throw new InvalidConfigException('Either "attribute" or "value" property must be specified.');
126 127 128 129 130 131 132 133
        }
    }

    /**
     * @inheritdoc
     */
    protected function getValue($event)
    {
134 135
        $isNewSlug = true;

136
        if ($this->attribute !== null) {
Alexander Mohorev committed
137
            $attributes = (array) $this->attribute;
138 139
            /* @var $owner BaseActiveRecord */
            $owner = $this->owner;
140
            if (!empty($owner->{$this->slugAttribute})) {
141
                $isNewSlug = false;
142
                if (!$this->immutable) {
143 144 145 146 147
                    foreach ($attributes as $attribute) {
                        if ($owner->isAttributeChanged($attribute)) {
                            $isNewSlug = true;
                            break;
                        }
148 149 150 151 152
                    }
                }
            }

            if ($isNewSlug) {
153
                $slugParts = [];
154 155
                foreach ($attributes as $attribute) {
                    $slugParts[] = $owner->{$attribute};
156
                }
157
                $slug = Inflector::slug(implode('-', $slugParts));
158
            } else {
159
                $slug = $owner->{$this->slugAttribute};
160
            }
161 162
        } else {
            $slug = parent::getValue($event);
163 164
        }

165
        if ($this->ensureUnique && $isNewSlug) {
166 167
            $baseSlug = $slug;
            $iteration = 0;
168
            while (!$this->validateSlug($slug)) {
169 170 171
                $iteration++;
                $slug = $this->generateUniqueSlug($baseSlug, $iteration);
            }
172
        }
173 174 175 176 177 178 179 180
        return $slug;
    }

    /**
     * Checks if given slug value is unique.
     * @param string $slug slug value
     * @return boolean whether slug is unique.
     */
181
    private function validateSlug($slug)
182
    {
183 184 185
        /* @var $validator UniqueValidator */
        /* @var $model BaseActiveRecord */
        $validator = Yii::createObject(array_merge(
186
            [
187
                'class' => UniqueValidator::className()
188
            ],
189 190 191 192 193 194 195 196
            $this->uniqueValidator
        ));

        $model = clone $this->owner;
        $model->clearErrors();
        $model->{$this->slugAttribute} = $slug;

        $validator->validateAttribute($model, $this->slugAttribute);
197 198
        return !$model->hasErrors();
    }
199

200
    /**
201
     * Generates slug using configured callback or increment of iteration.
202 203
     * @param string $baseSlug base slug value
     * @param integer $iteration iteration number
204
     * @return string new slug value
205 206 207 208
     * @throws \yii\base\InvalidConfigException
     */
    private function generateUniqueSlug($baseSlug, $iteration)
    {
209
        if (is_callable($this->uniqueSlugGenerator)) {
210
            return call_user_func($this->uniqueSlugGenerator, $baseSlug, $iteration, $this->owner);
211 212
        } else {
            return $baseSlug . '-' . ($iteration + 1);
213
        }
214 215
    }
}