Add more entity functionality.

Author: greg-md

src/EntityModel.php

@@ -4,8 +4,14 @@
 
 class EntityModel extends ModelAbstract
 {
+    const STATE_NEW = 1;
+
+    const STATE_MANAGED = 2;
+
     protected $entityClass;
 
+    private $entityStates = [];
+
     public function entityClass()
     {
         if (!$this->entityClass) {
@@ -31,6 +37,62 @@ public function create(array $record = [])
 
     public function save($entity)
     {
+        $entityClass = $this->entityClass();
+
+        if (!$entity instanceof $entityClass) {
+            throw new \Exception('Entity should be an instance of `' . $entityClass . '`.');
+        }
+
+        $record = [];
+
+        (function () use (&$record) {
+            $record = get_object_vars($this);
+        })->call($entity);
+
+        switch($this->getEntityState($entity)) {
+            case self::STATE_NEW:
+                $this->insert($record);
+
+                if ($autoIncrement = $this->autoIncrement()) {
+                    $id = (int) $this->connection()->lastInsertId();
+
+                    (function () use ($autoIncrement, $id) {
+                        $this->{$autoIncrement} = $id;
+                    })->call($entity);
+                }
+
+                $this->setEntityState($entity, self::STATE_MANAGED);
+
+                break;
+            case self::STATE_MANAGED:
+                $keys = [];
+
+                foreach ($this->firstUniqueKey() as $key) {
+                    $keys[$key] = $record[$key];
+                }
+
+                $query = $this->newUpdateQuery()
+                    ->setMultiple(array_diff_key($record, $keys))
+                    ->whereMultiple($keys);
+
+                $this->connection()->sqlExecute(...$query->toSql());
+
+                break;
+        }
+
+        return $this;
+    }
+
+    private function setEntityState($entity, $state)
+    {
+        $this->entityStates[spl_object_hash($entity)] = $state;
+
+        return $this;
+    }
+
+    private function getEntityState($entity)
+    {
+        return $this->entityStates[spl_object_hash($entity)] ?? self::STATE_NEW;
     }
 
     protected function prepareRowInstance(array $record)

src/Model.php

@@ -27,7 +27,6 @@ public function serialize()
             $this->autoIncrement,
             $this->uniqueKeys,
             $this->casts,
-            $this->customRecord,
 
             $this->fillable,
             $this->guarded,
@@ -46,7 +45,6 @@ public function unserialize($serialized)
             $this->autoIncrement,
             $this->uniqueKeys,
             $this->casts,
-            $this->customRecord,
 
             $this->fillable,
             $this->guarded,

src/ModelAbstract.php

@@ -81,8 +81,6 @@ public function serialize()
             $this->primaryKey,
             $this->autoIncrement,
             $this->uniqueKeys,
-            $this->casts,
-            $this->customRecord,
         ]);
     }
 
@@ -93,8 +91,6 @@ public function unserialize($serialized)
             $this->primaryKey,
             $this->autoIncrement,
             $this->uniqueKeys,
-            $this->casts,
-            $this->customRecord,
         ] = unserialize($serialized);
     }
 

src/RowsTrait.php

@@ -394,14 +394,13 @@ public function save(array $values = [])
 
         foreach ($this->rows as $key => &$row) {
             $data = $row;
+
             if ($this->rowStateIsCasted($key)) {
                 $data = $this->uncastRecord($data);
             }
 
             if ($this->rowStateIsNew($key)) {
-                $query = $this->newInsertQuery()->data($data);
-
-                $this->connection()->sqlExecute(...$query->toSql());
+                $this->insert($data);
 
                 if (!$this->getAutoIncrement() and $column = $this->autoIncrement()) {
                     $row[$column] = (int) $this->connection()->lastInsertId();
@@ -429,6 +428,7 @@ public function save(array $values = [])
 
     /**
      * @return $this
+     * @throws SqlException
      */
     public function destroy()
     {

tests/EntityModelTest.php

@@ -53,7 +53,7 @@ public function testCanFetchEntity()
 
         $model = $this->newEntityModel($connection);
 
-        $item = $model->create(['Id' => 1]);
+        $item = $model->new(['Id' => 1]);
 
         $this->assertInstanceOf(TableEntity::class, $item);