attribute typecast

tecnocen-com · Oct 23, 2017 · b1d5795 · b1d5795
1 parent a1bc7bc
commit b1d5795
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 3 deletions.
diff --git a/src/Module.php b/src/Module.php
@@ -6,6 +6,7 @@
 use Yii;
 use yii\behaviors\BlameableBehavior;
 use yii\behaviors\TimestampBehavior;
+use yii\behaviors\TypecastBehavior;
 use yii\db\Expression as DbExpression;
 use yii\i18n\PhpMessageSource;
 use yii\web\Application as WebApplication;
@@ -25,6 +26,11 @@ class Module extends \yii\base\Module
      */
     public $timestampClass = TimestampBehavior::class;
 
+    /**
+     * @var string
+     */
+    public $typecastClass = AttributeTypecastBehavior::class;
+
     /**
      * @var string
      */

diff --git a/src/models/Entity.php b/src/models/Entity.php
@@ -18,6 +18,17 @@ abstract class Entity extends Pivot
      */
     protected $updatedAtAttribute = 'updated_at';
 
+    /**
+     * @inheritdoc
+     */
+    protected function attributeTypecast()
+    {
+        return parent::attributeTypecast() + [
+            $this->updatedByAttribute => 'integer',
+        ];
+    }
+
+
     /**
      * @inheritdoc
      */

diff --git a/src/models/PersistentEntity.php b/src/models/PersistentEntity.php
@@ -18,6 +18,16 @@ abstract class PersistentEntity extends Entity
      */
     protected $deletedAtAttribute = 'deleted_at';
 
+    /**
+     * @inheritdoc
+     */
+    protected function attributeTypecast()
+    {
+        return parent::attributeTypecast() + [
+            $this->deletedByAttribute => 'integer',
+        ];
+    }
+
     /**
      * @inheritdoc
      */

diff --git a/src/models/Pivot.php b/src/models/Pivot.php
@@ -53,17 +53,31 @@ public function behaviors()
                 'class' => $module->blameableClass,
                 'createdByAttribute' => $this->createdByAttribute,
                 'updatedByAttribute' => null,
-                'value' => $module->getUserId(),
+                'value' => $module->userId,
             ],
             'timestamp' => [
                 'class' => $module->timestampClass,
                 'createdAtAttribute' => $this->createdByAttribute,
                 'updatedAtAttribute' => null,
-                'value' => $module->getTimestampValue(),
-            ]
+                'value' => $module->timestampValue,
+            ],
+            'typecast' => [
+                'class' => $module->typecastClass,
+                'attributeTypes' => $this->attributeTypecast(),
+                'typecastAfterFind' => true,
+            ],
         ];
     }
 
+    /**
+     * @return ?array pairs of 'attribute' => 'typecast'. Return null if you
+     * want the typecast to be determined by `rules()` method.
+     */
+    protected function attributeTypecast()
+    {
+        return [$this->createdByAttribute => 'integer'];
+    }
+
     /**
      * @inheritdoc
      */