SluggableBehavior.php 6.13 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 83
    /**
     * @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.
     */
84
    public $ensureUnique = false;
85
    /**
86 87 88
     * @var array configuration for slug uniqueness validator. Parameter 'class' may be omitted - by default
     * [[UniqueValidator]] will be used.
     * @see UniqueValidator
89
     */
90
    public $uniqueValidator = [];
91
    /**
92 93
     * @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:
94 95
     *
     * ```php
96
     * function ($baseSlug, $iteration, $model)
97 98 99 100 101
     * {
     *     // return uniqueSlug
     * }
     * ```
     *
102
     * If not set unique slug will be generated adding incrementing suffix to the base slug.
103
     */
104
    public $uniqueSlugGenerator;
105

106

107 108 109 110 111
    /**
     * @inheritdoc
     */
    public function init()
    {
112 113 114
        parent::init();

        if (empty($this->attributes)) {
115
            $this->attributes = [BaseActiveRecord::EVENT_BEFORE_VALIDATE => $this->slugAttribute];
116 117
        }

118
        if ($this->attribute === null && $this->value === null) {
Qiang Xue committed
119
            throw new InvalidConfigException('Either "attribute" or "value" property must be specified.');
120 121 122 123 124 125 126 127
        }
    }

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

130
        if ($this->attribute !== null) {
131
            $attributes = (array)$this->attribute;
132 133 134 135 136 137 138 139 140 141 142 143 144
            /* @var $owner BaseActiveRecord */
            $owner = $this->owner;
            if (!$owner->getIsNewRecord() && !empty($owner->{$this->slugAttribute})) {
                $isNewSlug = false;
                foreach ($attributes as $attribute) {
                    if ($owner->isAttributeChanged($attribute)) {
                        $isNewSlug = true;
                        break;
                    }
                }
            }

            if ($isNewSlug) {
145
                $slugParts = [];
146 147
                foreach ($attributes as $attribute) {
                    $slugParts[] = $owner->{$attribute};
148
                }
149
                $slug = Inflector::slug(implode('-', $slugParts));
150
            } else {
151
                $slug = $owner->{$this->slugAttribute};
152
            }
153 154
        } else {
            $slug = parent::getValue($event);
155 156
        }

157
        if ($this->ensureUnique && $isNewSlug) {
158 159
            $baseSlug = $slug;
            $iteration = 0;
160
            while (!$this->validateSlug($slug)) {
161 162 163
                $iteration++;
                $slug = $this->generateUniqueSlug($baseSlug, $iteration);
            }
164
        }
165 166 167 168 169 170 171 172
        return $slug;
    }

    /**
     * Checks if given slug value is unique.
     * @param string $slug slug value
     * @return boolean whether slug is unique.
     */
173
    private function validateSlug($slug)
174
    {
175 176 177
        /* @var $validator UniqueValidator */
        /* @var $model BaseActiveRecord */
        $validator = Yii::createObject(array_merge(
178
            [
179
                'class' => UniqueValidator::className()
180
            ],
181 182 183 184 185 186 187 188
            $this->uniqueValidator
        ));

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

        $validator->validateAttribute($model, $this->slugAttribute);
189 190
        return !$model->hasErrors();
    }
191

192
    /**
193
     * Generates slug using configured callback or increment of iteration.
194 195
     * @param string $baseSlug base slug value
     * @param integer $iteration iteration number
196
     * @return string new slug value
197 198 199 200
     * @throws \yii\base\InvalidConfigException
     */
    private function generateUniqueSlug($baseSlug, $iteration)
    {
201
        if (is_callable($this->uniqueSlugGenerator)) {
202
            return call_user_func($this->uniqueSlugGenerator, $baseSlug, $iteration, $this->owner);
203 204
        } else {
            return $baseSlug . '-' . ($iteration + 1);
205
        }
206 207
    }
}