Add all option to generate the factories of all tables in database.

Author: aiiro

README.md

@@ -43,12 +43,23 @@ After installing and Configuration, you can generate the factory file by running
 
 Please pass the table name to `generate:factory` command as the argument.
 
-**NOTE**
-This command connects to the database to retrieve the columns from table, so make sure that the database is configured.
 ``` shell
 php artisan generate:factory some_samples
 ```
 
+**NOTE**
+This command connects to the database to retrieve the columns from table, so make sure that the database is configured.
+
+#### To generate factories of all tables in database.
+
+Use `--all` option without table name, to generate factories of all tables in database.
+
+If a factory of table exists, it will be skipped and continue to generate factories of other tables.
+
+```bash
+php artisan generate:factory  --all
+```
+
 ## License
 This project is released under MIT License. See [MIT License](LICENSE)
  for the detail.

src/Commands/GenerateFactory.php

@@ -2,6 +2,9 @@
 
 namespace Aiiro\Factory\Commands;
 
+use Aiiro\Factory\Connections\DatabaseContract;
+use Aiiro\Factory\Connections\MySql;
+use Aiiro\Factory\Connections\Sqlite;
 use Aiiro\Factory\Exceptions\UnknownConnectionException;
 use Illuminate\Config\Repository;
 use Illuminate\Console\GeneratorCommand;
@@ -22,11 +25,18 @@ class GenerateFactory extends GeneratorCommand
     /**
      * @var string
      */
-    protected $name = 'generate:factory';
+    protected $signature = 'generate:factory {name?} {--all}';
 
-    /** @var \Illuminate\Config\Repository */
+    /**
+     * @var \Illuminate\Config\Repository
+     */
     protected $config;
 
+    /**
+     * @var DatabaseContract
+     */
+    protected $database;
+
     /**
      * GenerateFactory constructor.
      *
@@ -40,47 +50,53 @@ public function __construct(Filesystem $files, Repository $config)
     }
 
     /**
-     * @throws \Illuminate\Contracts\Filesystem\FileNotFoundException
+     * @return bool|null|void
      * @throws \Aiiro\Factory\Exceptions\UnknownConnectionException
      */
     public function handle()
     {
+        /** @var Connection $conn */
+        $conn = \DB::connection();
+        /** @var \PDO $pdo */
+        $pdo = \DB::connection()->getPdo();
+
+        if ($conn instanceof MySqlConnection) {
+            $this->database = new MySql($pdo);
+        } elseif ($conn instanceof SQLiteConnection) {
+            $this->database = new Sqlite($pdo);
+        } else {
+            throw new UnknownConnectionException('Unknown connection is set.');
+        }
+
+        if ($this->hasOption('all') && $this->option('all')) {
+            $tables = $this->database->fetchTables();
+            foreach ($tables as $tableRecord) {
+                $this->runFactoryBuilder($tableRecord[0]);
+            }
+        } else {
+            $table = $this->getNameInput();
+            $this->runFactoryBuilder($table);
+        }
+    }
 
-        $table = $this->getNameInput();
+    /**
+     * @param $table
+     * @return bool
+     * @throws \Illuminate\Contracts\Filesystem\FileNotFoundException
+     */
+    protected function runFactoryBuilder($table)
+    {
         $model = Str::ucfirst(Str::camel($table));
         $factory = Str::singular($model) . 'Factory';
 
         $path = $this->getPath($factory);
 
         if ($this->alreadyExists($path)) {
             $this->error($factory . ' already exists!');
-
             return false;
         }
 
-        /** @var Connection $conn */
-        $conn = \DB::connection();
-        /** @var \PDO $pdo */
-        $pdo = \DB::connection()->getPdo();
-
-        $stmt = null;
-        $columns = [];
-
-        if ($conn instanceof MySqlConnection) {
-            $stmt = $pdo->prepare("DESCRIBE $table");
-            $stmt->execute();
-            foreach ($stmt->fetchAll() as $column) {
-                $columns[] = $column['Field'];
-            }
-        } elseif ($conn instanceof SQLiteConnection) {
-            $stmt = $pdo->prepare("PRAGMA table_info($table)");
-            $stmt->execute();
-            foreach ($stmt->fetchAll() as $column) {
-                $columns[] = $column['name'];
-            }
-        } else {
-            throw new UnknownConnectionException('Unknown connection is set.');
-        }
+        $columns = $this->database->fetchColumns($table);
 
         $factoryContent = $this->buildFactory(
             $this->config->get('factory-generator.namespace.model'),
@@ -91,7 +107,6 @@ public function handle()
         $this->createFile($path, $factoryContent);
 
         $this->info($factory . ' created successfully.');
-
     }
 
     /**

src/Connections/DatabaseContract.php

@@ -0,0 +1,8 @@
+<?php
+
+namespace Aiiro\Factory\Connections;
+
+interface DatabaseContract
+{
+    public function fetchColumns($table);
+}

src/Connections/MySql.php

@@ -0,0 +1,45 @@
+<?php
+
+namespace Aiiro\Factory\Connections;
+
+class MySql implements DatabaseContract
+{
+
+    /**
+     * @var \PDO
+     */
+    protected $pdo;
+
+    /**
+     * MySql constructor.
+     *
+     * @param $pdo
+     */
+    public function __construct($pdo)
+    {
+        $this->pdo = $pdo;
+    }
+
+    public function fetchTables()
+    {
+        $stmt = $this->pdo->prepare("SHOW TABLES");
+        $stmt->execute();
+
+        return $stmt->fetchAll();
+    }
+
+
+    public function fetchColumns($table)
+    {
+        $columns = [];
+
+        $stmt = $this->pdo->prepare("DESCRIBE $table");
+        $stmt->execute();
+
+        foreach ($stmt->fetchAll() as $column) {
+            $columns[] = $column['Field'];
+        }
+
+        return $columns;
+    }
+}

src/Connections/Sqlite.php

@@ -0,0 +1,43 @@
+<?php
+
+namespace Aiiro\Factory\Connections;
+
+class Sqlite implements DatabaseContract
+{
+
+    /**
+     * @var \PDO
+     */
+    protected $pdo;
+
+    /**
+     * Sqlite constructor.
+     *
+     * @param $pdo
+     */
+    public function __construct($pdo)
+    {
+        $this->pdo = $pdo;
+    }
+
+    public function fetchTables()
+    {
+        $stmt = $this->pdo->prepare("SELECT name FROM sqlite_master WHERE type = 'table' AND name NOT IN ('migrations', 'sqlite_sequence');");
+        $stmt->execute();
+
+        return $stmt->fetchAll();
+    }
+
+    public function fetchColumns($table)
+    {
+        $columns = [];
+
+        $stmt = $this->pdo->prepare("PRAGMA table_info($table)");
+        $stmt->execute();
+        foreach ($stmt->fetchAll() as $column) {
+            $columns[] = $column['name'];
+        }
+
+        return $columns;
+    }
+}