feat: Add column count to tables comment (#6) (#8)

* added column count to tables command

Co-authored-by: innoflash <[email protected]>

src/Commands/ListTablesCommand.php

@@ -15,9 +15,14 @@ class ListTablesCommand extends Command
 
     public function handle(SchemaContract $schema)
     {
-        $headers = ['Tables'];
+        $headers = ['Tables', 'Columns'];
         $rows = $schema->getTables();
 
+        $rows = collect($rows)
+            ->map(function ($row) use ($schema) {
+                return array_merge($row, [count($schema->getColumns($row[0]))]);
+            })->toArray();
+
         $this->table($headers, $rows);
     }
 }

src/LaravelSchemaListServiceProvider.php

@@ -16,6 +16,11 @@
 
 class LaravelSchemaListServiceProvider extends ServiceProvider
 {
+    private $connections = [
+        MySqlConnection::class    => MySQLSchema::class,
+        PostgresConnection::class => PostgresSchema::class,
+    ];
+
     /**
      * Bootstrap the application services.
      */
@@ -29,14 +34,10 @@ public function boot()
 
             $this->app->bind(SchemaContract::class, function () {
                 $connection = resolve(ConnectionInterface::class);
-
-                if ($connection instanceof MySqlConnection) {
-                    return new MySQLSchema($connection);
-                } elseif ($connection instanceof PostgresConnection) {
-                    return new PostgresSchema($connection);
-                }
-
-                return new UnsupportedSchema($connection);
+                $connectionClass = get_class($connection);
+                $schema = $this->connections[$connectionClass] ?? UnsupportedSchema::class;
+
+                return new $schema($connection);
             });
         }
     }