Add ability to run onSave method within Entity classes.

Author: lonnieezell

system/Model.php

@@ -546,6 +546,8 @@ protected function getHash($str, $len)
 	 */
 	public function save($data)
 	{
+	    $saveData = $data;
+
 	    // If $data is using a custom class with public or protected
         // properties representing the table elements, we need to grab
         // them as an array.
@@ -556,14 +558,27 @@ public function save($data)
 
 		if (is_object($data) && isset($data->{$this->primaryKey}))
 		{
-			return $this->update($data->{$this->primaryKey}, $data);
+			$response = $this->update($data->{$this->primaryKey}, $data);
 		}
 		elseif (is_array($data) && ! empty($data[$this->primaryKey]))
 		{
-			return $this->update($data[$this->primaryKey], $data);
+			$response = $this->update($data[$this->primaryKey], $data);
 		}
+        else
+        {
+            $response = $this->insert($data);
+        }
+
+        // If it was an Entity class, check it for an onSave method.
+        if (is_object($saveData) && ! $saveData instanceof \stdClass)
+        {
+            if (method_exists($saveData, 'onSave'))
+            {
+                $saveData->onSave();
+            }
+        }
 
-		return $this->insert($data);
+        return $response;
 	}
 
 	//--------------------------------------------------------------------
@@ -607,6 +622,14 @@ protected function classToArray($data): array
 	 */
 	public function insert($data)
 	{
+        // If $data is using a custom class with public or protected
+        // properties representing the table elements, we need to grab
+        // them as an array.
+        if (is_object($data) && ! $data instanceof \stdClass)
+        {
+            $data = $this->classToArray($data);
+        }
+
 	    // Validate data before saving.
 	    if ($this->skipValidation === false)
         {
@@ -653,6 +676,14 @@ public function insert($data)
 	 */
 	public function update($id, $data)
 	{
+        // If $data is using a custom class with public or protected
+        // properties representing the table elements, we need to grab
+        // them as an array.
+        if (is_object($data) && ! $data instanceof \stdClass)
+        {
+            $data = $this->classToArray($data);
+        }
+
 	    // Validate data before saving.
         if ($this->skipValidation === false)
         {