Support for custom builder classes (#782)

barryvdh · Mar 26, 2019 · 39c148a · 39c148a
1 parent 2b3c9a2
commit 39c148a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/Console/ModelsCommand.php b/src/Console/ModelsCommand.php
@@ -446,7 +446,10 @@ protected function getPropertiesFromMethods($model)
                     }
                 } elseif (in_array($method, ['query', 'newQuery', 'newModelQuery'])) {
                     $reflection = new \ReflectionClass($model);
-                    $this->setMethod($method, '\Illuminate\Database\Eloquent\Builder|\\' . $reflection->getName());
+
+                    $builder = get_class($model->newModelQuery());
+
+                    $this->setMethod($method, "\\{$builder}|\\" . $reflection->getName());
                 } elseif (!method_exists('Illuminate\Database\Eloquent\Model', $method)
                     && !Str::startsWith($method, 'get')
                 ) {