add withCount to transformer

knuckleswtf · Nov 19, 2024 · 6107ff9 · 6107ff9
1 parent da82b5b
commit 6107ff9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/Attributes/ResponseFromTransformer.php b/src/Attributes/ResponseFromTransformer.php
@@ -21,6 +21,7 @@ public function __construct(
 
         /* Format: [adapter, numberPerPage]. Example: [SomePaginator::class, 10] */
         public array $paginate = [],
+        public array $withCount = [],
     ) {
     }
 }
diff --git a/src/Extracting/InstantiatesExampleModels.php b/src/Extracting/InstantiatesExampleModels.php
@@ -93,12 +93,13 @@ protected function getExampleModelFromFactoryMake(string $type, array $factorySt
     /**
      * @param class-string $type
      * @param string[] $relations
+     * @param string[] $withCount
      *
      * @return \Illuminate\Database\Eloquent\Model|null
      */
-    protected function getExampleModelFromDatabaseFirst(string $type, array $relations = [])
+    protected function getExampleModelFromDatabaseFirst(string $type, array $relations = [], array $withCount = [])
     {
-        return $type::with($relations)->first();
+        return $type::with($relations)->withCount($withCount)->first();
     }
 
 }
diff --git a/src/Extracting/Strategies/Responses/UseResponseAttributes.php b/src/Extracting/Strategies/Responses/UseResponseAttributes.php
@@ -89,7 +89,7 @@ protected function getTransformerResponse(ResponseFromTransformer $attributeInst
     {
         $modelInstantiator = fn() => $this->instantiateExampleModel(
             $attributeInstance->model, $attributeInstance->factoryStates, $attributeInstance->with,
-            (new ReflectionClass($attributeInstance->name))->getMethod('transform')
+            (new ReflectionClass($attributeInstance->name))->getMethod('transform'), $attributeInstance->withCount,
         );
 
         $pagination = $attributeInstance->paginate ? [