[8.x] Allow to dynamically add Tests\Browsers\Page\Page or Laravel\Dusk\Page via dusk:page command (#1148)

Improvements based on #1145

Signed-off-by: Mior Muhammad Zaki <[email protected]>

Author: crynobone

src/Console/PageCommand.php

@@ -30,6 +30,41 @@ class PageCommand extends GeneratorCommand
      */
     protected $type = 'Page';
 
+    /**
+     * Build the class with the given name.
+     *
+     * @param  string  $name
+     * @return string
+     */
+    protected function buildClass($name)
+    {
+        $result = parent::buildClass($name);
+
+        $pageName = $this->argument('name');
+
+        $baseClass = 'Tests\Browser\Pages\Page';
+
+        if (! Str::contains($pageName, '/') && class_exists($baseClass)) {
+            return $result;
+        } elseif (! class_exists($baseClass)) {
+            $baseClass = 'Laravel\Dusk\Page';
+        }
+
+        $lineEndingCount = [
+            "\r\n" => substr_count($result, "\r\n"),
+            "\r" => substr_count($result, "\r"),
+            "\n" => substr_count($result, "\n"),
+        ];
+
+        $eol = array_keys($lineEndingCount, max($lineEndingCount))[0];
+
+        return str_replace(
+            'use Laravel\Dusk\Browser;'.$eol,
+            'use Laravel\Dusk\Browser;'.$eol."use {$baseClass};".$eol,
+            $result
+        );
+    }
+
     /**
      * Get the stub file for the generator.
      *