✨ add protected fields functionality

Signed-off-by: otengkwame <[email protected]>

Author: otengkwame

Core/core/Models/EasyModel.php

@@ -65,6 +65,13 @@ class EasyModel extends Model
      */
     public $returnAs = 'object';
 
+    /**
+     * Protected fields
+     *
+     * @var array
+     */
+    public $protected = [];
+
     /**
      * Construct the CI_Model
      */
@@ -416,6 +423,25 @@ public function getRowResult($query, $last = false)
         return $query->{$mode}('object');
     }
 
+    /**
+     * Protect fields by removing them from $row
+     *
+     * @param array|object $row
+     * @return mixed
+     */
+    public function protectFields($row)
+    {
+        foreach ($this->protected as $attr) {
+            if (is_object($row)) {
+                unset($row->$attr);
+            } else {
+                unset($row[$attr]);
+            }
+        }
+
+        return $row;
+    }
+
     /**
      * Grabs data from a table
      *       OR a single record by passing $id,
@@ -750,8 +776,17 @@ public function saveOnly($data, $show = false)
      */
     public function create($data)
     {
+        
+
         $this->db->set($data)->insert($this->table);
-        return $this->selectLast('*');
+        
+        $data = $this->selectLast('*');
+
+        if (!empty($this->protected)) {
+            return $this->protectFields($data);
+        }
+
+        return $data;
     }
 
     /**