[Feature] Add with count validation to query stubs

Author: lindyhopchris

src/Console/MakeQuery.php

@@ -19,6 +19,8 @@
 
 namespace LaravelJsonApi\Laravel\Console;
 
+use Illuminate\Support\Str;
+use LaravelJsonApi\Core\Query\Custom\ExtendedQueryParameters;
 use Symfony\Component\Console\Input\InputOption;
 
 class MakeQuery extends GeneratorCommand
@@ -82,6 +84,36 @@ protected function getClassType(): string
         return 'Query';
     }
 
+    /**
+     * @inheritDoc
+     */
+    protected function buildClass($name)
+    {
+        $stub = parent::buildClass($name);
+
+        return $this->replaceAdditional($stub);
+    }
+
+    /**
+     * Replace additional placeholders.
+     *
+     * @param string $stub
+     * @return string
+     */
+    protected function replaceAdditional(string $stub): string
+    {
+        $withCount = ExtendedQueryParameters::withCount();
+
+        $replace = [
+            '{{ withCount }}' => $withCount,
+            '{{withCount}}' => $withCount,
+        ];
+
+        return str_replace(
+            array_keys($replace), array_values($replace), $stub
+        );
+    }
+
     /**
      * @inheritDoc
      */

stubs/query-collection.stub

@@ -41,6 +41,11 @@ class {{ class }} extends ResourceQuery
                 'string',
                 JsonApiRule::sort(),
             ],
+            '{{ withCount }}' => [
+                'nullable',
+                'string',
+                JsonApiRule::countable(),
+            ],
         ];
     }
 }

stubs/query.stub

@@ -33,6 +33,11 @@ class {{ class }} extends ResourceQuery
             ],
             'page' => JsonApiRule::notSupported(),
             'sort' => JsonApiRule::notSupported(),
+            '{{ withCount }}' => [
+                'nullable',
+                'string',
+                JsonApiRule::countable(),
+            ],
         ];
     }
 }

tests/lib/Integration/Console/MakeQueryTest.php

@@ -158,6 +158,7 @@ private function assertQueryCreated(): void
             'class PostQuery extends ResourceQuery',
             "'page' => JsonApiRule::notSupported(),",
             "'sort' => JsonApiRule::notSupported(),",
+            "'withCount' =>",
         ];
 
         foreach ($tests as $expected) {
@@ -180,6 +181,7 @@ private function assertQueryCollectionCreated(): void
             'class PostCollectionQuery extends ResourceQuery',
             'JsonApiRule::page(),',
             'JsonApiRule::sort(),',
+            "'withCount' =>",
         ];
 
         foreach ($tests as $expected) {