refactor: support string-class model in hasOne and hasMany

SonyPradana · SonyPradana · commit 5a61a0ff9a06 · 2024-03-03T21:09:59.000+07:00
diff --git a/src/System/Database/MyModel/Model.php b/src/System/Database/MyModel/Model.php
@@ -4,7 +4,6 @@
 
 namespace System\Database\MyModel;
 
-use System\Collection\CollectionImmutable;
 use System\Database\MyPDO;
 use System\Database\MyQuery;
 use System\Database\MyQuery\Bind;
@@ -112,6 +111,17 @@ public function setUp(
      */
     public function __get(string $name)
     {
+        if (method_exists($this, $name)) {
+            $highorder = $this->{$name}();
+            if (is_a($highorder, Model::class)) {
+                return $highorder->first();
+            }
+
+            if (is_a($highorder, ModelCollection::class)) {
+                return $highorder->toArrayArray();
+            }
+        }
+
         return $this->getter($name);
     }
 
@@ -330,31 +340,59 @@ public function isExist(): bool
     }
 
     /**
-     * @return CollectionImmutable<string, mixed>
+     * Get get model relation.
+     *
+     * @param class-string|string $model
+     *
+     * @return Model
      */
-    public function hasOne(string $table, string $ref = 'id')
+    public function hasOne($model, ?string $ref = null)
     {
-        $ref = MyQuery::from($this->table_name, $this->pdo)
-            ->select([$table . '.*'])
-            ->join(InnerJoin::ref($table, $this->primery_key, $ref))
+        if (class_exists($model)) {
+            $model      = new $model($this->pdo, []);
+            $table_name = $model->table_name;
+            $join_ref   = $ref ?? $model->primery_key;
+        } else {
+            $table_name = $model;
+            $join_ref   = $ref ?? $this->primery_key;
+            $model      = new Model($this->pdo, []);
+        }
+        $result   = MyQuery::from($this->table_name, $this->pdo)
+            ->select([$table_name . '.*'])
+            ->join(InnerJoin::ref($table_name, $this->primery_key, $join_ref))
             ->whereRef($this->where)
             ->single();
+        $model->columns = $model->fresh = [$result];
 
-        return new CollectionImmutable($ref);
+        return $model;
     }
 
     /**
-     * @return CollectionImmutable<string|int, mixed>
+     * Get get model relation.
+     *
+     * @param class-string|string $model
+     *
+     * @return ModelCollection<array-key, Model>
      */
-    public function hasMany(string $table, string $ref = 'id')
+    public function hasMany($model, ?string $ref = null)
     {
-        $ref = MyQuery::from($this->table_name, $this->pdo)
-            ->select([$table . '.*'])
-            ->join(InnerJoin::ref($table, $this->primery_key, $ref))
-            ->whereRef($this->where)
-            ->get();
+        if (class_exists($model)) {
+            $model      = new $model($this->pdo, []);
+            $table_name = $model->table_name;
+            $join_ref   = $ref ?? $model->primery_key;
+        } else {
+            $table_name = $model;
+            $join_ref   = $ref ?? $this->primery_key;
+            $model      = new Model($this->pdo, []);
+        }
+        $result = MyQuery::from($this->table_name, $this->pdo)
+             ->select([$table_name . '.*'])
+             ->join(InnerJoin::ref($table_name, $this->primery_key, $join_ref))
+             ->whereRef($this->where)
+             ->get();
+        $model->columns = $model->fresh = $result->toArray();
 
-        return new CollectionImmutable($ref->immutable());
+        return $model->get();
     }
 
     /**
diff --git a/src/System/Database/MyModel/ModelCollection.php b/src/System/Database/MyModel/ModelCollection.php
@@ -43,18 +43,24 @@ public function getPrimeryKey()
         return $primeryKeys;
     }
 
+    /**
+     *  Check every Model has clean column.
+     */
     public function isClean(?string $column = null): bool
     {
         return $this->every(fn ($model) => $model->isClean($column));
     }
 
+    /**
+     * Check every Model has dirty column.
+     */
     public function isDirty(?string $column = null): bool
     {
         return !$this->isClean($column);
     }
 
     /**
-     * Global update (base on primerykey).
+     * Update using query using model primery key.
      *
      * @param array<array-key, mixed> $values
      */
@@ -70,6 +76,9 @@ public function update(array $values): bool
         return $update->execute();
     }
 
+    /**
+     * Delete using query using model primery key.
+     */
     public function delete(): bool
     {
         $table_name  = (fn () => $this->{'table_name'})->call($this->model);
@@ -81,4 +90,20 @@ public function delete(): bool
 
         return $delete->execute();
     }
+
+    /**
+     * Convert array of model to pure array;.
+     *
+     * @return array<array-key, mixed>
+     */
+    public function toArrayArray(): array
+    {
+        /** @var array<array-key, mixed> */
+        $arr = [];
+        foreach ($this->collection as $model) {
+            $arr = array_merge($arr, $model->toArray());
+        }
+
+        return $arr;
+    }
 }
diff --git a/tests/DataBase/Model/BaseModelTest.php b/tests/DataBase/Model/BaseModelTest.php
@@ -166,9 +166,50 @@ public function itCanGetHasOne()
         $this->createProfileSchema();
         $this->createProfiles([$profile]);
 
+        $user   = $this->user();
+        $result = $user->hasOne(Profile::class, 'user');
+        $this->assertEquals($profile, $result->first());
+    }
+
+    /**
+     * @test
+     *
+     * @group database
+     */
+    public function itCanGetHasOneUsingMagicGetter()
+    {
+        // profile
+        $profile = [
+            'user'   => 'taylor',
+            'name'   => 'taylor otwell',
+            'gender' => 'male',
+        ];
+        $this->createProfileSchema();
+        $this->createProfiles([$profile]);
+
+        $user   = $this->user();
+        $this->assertEquals($profile, $user->profile);
+    }
+
+    /**
+     * @test
+     *
+     * @group database
+     */
+    public function itCanGetHasOneWithTableName()
+    {
+        // profile
+        $profile = [
+            'user'   => 'taylor',
+            'name'   => 'taylor otwell',
+            'gender' => 'male',
+        ];
+        $this->createProfileSchema();
+        $this->createProfiles([$profile]);
+
         $user   = $this->user();
         $result = $user->hasOne('profiles', 'user');
-        $this->assertEquals($profile, $result->toArray());
+        $this->assertEquals($profile, $result->first());
     }
 
     /**
@@ -196,16 +237,73 @@ public function itCanGetHasMany()
         $this->createOrders($order);
 
         $user   = $this->user();
-        $result = $user->hasMany('orders', 'user');
-        $this->assertEquals($order, $result->toArray());
+        $result = $user->hasMany(Order::class, 'user');
+        $this->assertEquals($order, $result->toArrayArray());
     }
 
     /**
      * @test
      *
      * @group database
      */
-    public function itCanCheckisClean()
+    public function itCanGetHasManyWithMagicGetter()
+    {
+        // order
+        $order = [
+            [
+                'id'     => '1',
+                'user'   => 'taylor',
+                'name'   => 'order 1',
+                'type'   => 'gadget',
+            ], [
+                'id'     => '3',
+                'user'   => 'taylor',
+                'name'   => 'order 2',
+                'type'   => 'gadget',
+            ],
+        ];
+        $this->createOrderSchema();
+        $this->createOrders($order);
+
+        $user   = $this->user();
+        $this->assertEquals($order, $user->orders);
+    }
+
+    /**
+     * @test
+     *
+     * @group database
+     */
+    public function itCanGetHasManyWithTableName()
+    {
+        // order
+        $order = [
+            [
+                'id'     => '1',
+                'user'   => 'taylor',
+                'name'   => 'order 1',
+                'type'   => 'gadget',
+            ], [
+                'id'     => '3',
+                'user'   => 'taylor',
+                'name'   => 'order 2',
+                'type'   => 'gadget',
+            ],
+        ];
+        $this->createOrderSchema();
+        $this->createOrders($order);
+
+        $user   = $this->user();
+        $result = $user->hasMany(Order::class, 'user');
+        $this->assertEquals($order, $result->toArrayArray());
+    }
+
+    /**
+     * @test
+     *
+     * @group database
+     */
+    public function itCanCheckisCleanWith()
     {
         $user = $this->user();
         $this->assertTrue($user->isClean(), 'Check all column');
@@ -540,4 +638,19 @@ class User extends Model
     protected string $primery_key = 'user';
     /** @var string[] Hide from shoing column */
     protected $stash = ['password'];
+
+    public function profile()
+    {
+        return $this->hasOne(Profile::class, 'user');
+    }
+
+    public function orders()
+    {
+        return $this->hasMany(Order::class, 'user');
+    }
+}
+
+class Order extends Model
+{
+    protected string $table_name = 'orders';
 }
diff --git a/tests/DataBase/Model/BaseMultyModelTest.php b/tests/DataBase/Model/BaseMultyModelTest.php
@@ -161,8 +161,27 @@ public function itCanGetHasOne()
         $this->createProfiles([$profile]);
 
         $user   = $this->users();
-        $result = $user->hasOne('profiles', 'user');
-        $this->assertEquals($profile, $result->toArray());
+        $this->assertEquals($profile, $user->profile()->first());
+    }
+
+    /**
+     * @test
+     *
+     * @group database
+     */
+    public function itCanGetHasOneUsingMagicGetter()
+    {
+        // profile
+        $profile = [
+            'user'   => 'taylor',
+            'name'   => 'taylor otwell',
+            'gender' => 'male',
+        ];
+        $this->createProfileSchema();
+        $this->createProfiles([$profile]);
+
+        $user   = $this->users();
+        $this->assertEquals($profile, $user->profile);
     }
 
     /**
@@ -190,8 +209,8 @@ public function itCanGetHasMany()
         $this->createOrders($order);
 
         $user   = $this->users();
-        $result = $user->hasMany('orders', 'user');
-        $this->assertEquals($order, $result->toArray());
+        $result = $user->hasMany(Order::class, 'user');
+        $this->assertEquals($order, $result->toArrayArray());
     }
 
     /**

Original file line number	Diff line number	Diff line change
`@@ -43,18 +43,24 @@ public function getPrimeryKey()`
`43`	`43`	`return $primeryKeys;`
`44`	`44`	`}`
`45`	`45`
	`46`	`+ /**`
	`47`	`+ * Check every Model has clean column.`
	`48`	`+ */`
`46`	`49`	`public function isClean(?string $column = null): bool`
`47`	`50`	`{`
`48`	`51`	`return $this->every(fn ($model) => $model->isClean($column));`
`49`	`52`	`}`
`50`	`53`
	`54`	`+ /**`
	`55`	`+ * Check every Model has dirty column.`
	`56`	`+ */`
`51`	`57`	`public function isDirty(?string $column = null): bool`
`52`	`58`	`{`
`53`	`59`	`return !$this->isClean($column);`
`54`	`60`	`}`
`55`	`61`
`56`	`62`	`/**`
`57`		`- * Global update (base on primerykey).`
	`63`	`+ * Update using query using model primery key.`
`58`	`64`	`*`
`59`	`65`	`* @param array<array-key, mixed> $values`
`60`	`66`	`*/`
`@@ -70,6 +76,9 @@ public function update(array $values): bool`
`70`	`76`	`return $update->execute();`
`71`	`77`	`}`
`72`	`78`
	`79`	`+ /**`
	`80`	`+ * Delete using query using model primery key.`
	`81`	`+ */`
`73`	`82`	`public function delete(): bool`
`74`	`83`	`{`
`75`	`84`	`$table_name = (fn () => $this->{'table_name'})->call($this->model);`
`@@ -81,4 +90,20 @@ public function delete(): bool`
`81`	`90`
`82`	`91`	`return $delete->execute();`
`83`	`92`	`}`
	`93`	`+`
	`94`	`+ /**`
	`95`	`+ * Convert array of model to pure array;.`
	`96`	`+ *`
	`97`	`+ * @return array<array-key, mixed>`
	`98`	`+ */`
	`99`	`+ public function toArrayArray(): array`
	`100`	`+ {`
	`101`	`+ /** @var array<array-key, mixed> */`
	`102`	`+ $arr = [];`
	`103`	`+ foreach ($this->collection as $model) {`
	`104`	`+ $arr = array_merge($arr, $model->toArray());`
	`105`	`+ }`
	`106`	`+`
	`107`	`+ return $arr;`
	`108`	`+ }`
`84`	`109`	`}`