cast extended models when loading from database

src/Model.php

@@ -12,20 +12,27 @@ class Model implements JsonSerializable
     protected string $connection = "";
     protected string $pk;
     private bool $new;
+    private bool $ro;
 
-    public function __construct()
+    public function __construct($ro = false)
     {
         $this->new = true;
         $this->values = array();
         $this->updated = array();
         $this->pk = "id";
+        $this->ro = $ro;
         $this->initTable();
     }
 
     private function initTable(): void
     {
-        if ($this->table == "") {
-            $this->table =  $this->getDefaultTableName();
+        if (!$this->ro) {
+            if ($this->table == "") {
+                $this->table =  $this->getDefaultTableName();
+            }
+        } else {
+            $this->table = "ReadOnlyModel";
+            $this->pk = "__idpk";
         }
     }
 
@@ -77,7 +84,7 @@ class Model implements JsonSerializable
 
     public function find(mixed $key): mixed
     {
-        $qb = $this->newQueryBuilder();
+        $qb = $this->newQueryBuilder(get_class($this));
         $find = $qb->find($key);
         if ($find !== null) {
             $this->fillData($find->getValues());
@@ -89,27 +96,29 @@ class Model implements JsonSerializable
     public function save(): bool
     {
         $ret = false;
-        $qb = $this->newQueryBuilder();
-        if ($this->new) {
-            $query = $qb->getPreparedQuery($this->values);
-            $id = Connection::getInstance($this->connection)->insertQuery($query);
-            if ($id !== false) {
-                $this->values[$this->pk] = $id;
-                $this->new = false;
-                $ret = true;
-            }
-        } else {
-            if (isset($this->values[$this->pk])) {
-                if (count($this->updated) > 0) {
-                    $data = [];
-                    $data[$this->pk] = $this->values[$this->pk];
-                    foreach ($this->updated as $updatecol) {
-                        $data[$updatecol] = $this->values[$updatecol];
-                    }
-                    $query = $qb->getPreparedQuery($data, false);
-                    $ret = Connection::getInstance($this->connection)->updateQuery($query);
-                    if ($ret) {
-                        $this->updated = [];
+        if (!$this->ro) {
+            $qb = $this->newQueryBuilder("");
+            if ($this->new) {
+                $query = $qb->getPreparedQuery($this->values);
+                $id = Connection::getInstance($this->connection)->insertQuery($query);
+                if ($id !== false) {
+                    $this->values[$this->pk] = $id;
+                    $this->new = false;
+                    $ret = true;
+                }
+            } else {
+                if (isset($this->values[$this->pk])) {
+                    if (count($this->updated) > 0) {
+                        $data = [];
+                        $data[$this->pk] = $this->values[$this->pk];
+                        foreach ($this->updated as $updatecol) {
+                            $data[$updatecol] = $this->values[$updatecol];
+                        }
+                        $query = $qb->getPreparedQuery($data, false);
+                        $ret = Connection::getInstance($this->connection)->updateQuery($query);
+                        if ($ret) {
+                            $this->updated = [];
+                        }
                     }
                 }
             }
@@ -117,9 +126,9 @@ class Model implements JsonSerializable
         return $ret;
     }
 
-    public function newQueryBuilder(): QueryBuilder
+    public function newQueryBuilder($className): QueryBuilder
     {
-        return new QueryBuilder($this->table, $this->pk, $this->connection);
+        return new QueryBuilder($this->table, $this->pk, $this->connection, $className);
     }
 
     public function __set($name, $value): void
@@ -139,10 +148,16 @@ class Model implements JsonSerializable
         }
     }
 
+    public static function getClassName()
+    {
+        return static::class;
+    }
+
     public static function q(): QueryBuilder
     {
+        $cls = self::getClassName();
         $ist = new static();
-        return $ist->newQueryBuilder();
+        return $ist->newQueryBuilder($cls);
     }
 
     public function __get($name): mixed

src/QueryBuilder.php

@@ -10,6 +10,7 @@ class QueryBuilder
     protected array $orders;
     protected array $groupby;
     protected string $table;
+    protected string $classname;
     protected string $connection;
     protected string $pk;
     protected ?int $take;
@@ -20,8 +21,9 @@ class QueryBuilder
     public const MODE_AND = 0;
     public const MODE_OR = 1;
 
-    public function __construct($table, $pk = "id", $connection = "")
+    public function __construct($table, $pk = "id", $connection = "", $classname = "")
     {
+        $this->classname = $classname;
         $this->table = $table;
         $this->pk = $pk;
         $this->connection = $connection;
@@ -208,7 +210,12 @@ class QueryBuilder
     {
         $ret = array();
         foreach ($data as $row) {
-            $model = new Model();
+            if ($this->classname != "") {
+                $model = new $this->classname();
+            } else {
+                $model = new Model();
+            }
+
             $ret[] = $model->setPk($this->pk)->setTable($this->table)->fillData($row);
         }
         return new Collection($ret);
@@ -238,8 +245,7 @@ class QueryBuilder
 
         $this->fillData($ret);
 
-        if($where!="" && $Orwhere!="")
-        {
+        if ($where != "" && $Orwhere != "") {
             $Orwhere = " OR " . $Orwhere;
         }
 
@@ -257,7 +263,7 @@ class QueryBuilder
 
     private function fillData(Query &$query): void
     {
-        foreach (array_merge($this->wheres,$this->orwheres) as $expression) {
+        foreach (array_merge($this->wheres, $this->orwheres) as $expression) {
             if ($expression instanceof Expression) {
                 if ($expression->hasData()) {
                     $query->addData($expression->getValue());

tests/DbTest.php

@@ -116,6 +116,8 @@ test("extending model",function()
     $row = $qbs->find(1);
     expect($row->id)->toBe(1);
     expect($row->col1)->toBe("test");
+
+    expect($row)->toBeInstanceOf(Test::class);
 });
 
 test("order by", function(){