From 322b7409d6a2ff4790435d4166e549db247417c5 Mon Sep 17 00:00:00 2001
From: Catur Septian <ccseptian@users.noreply.github.com>
Date: Thu, 13 Jan 2022 12:58:21 +0700
Subject: [PATCH] Update IseedCommand.php

separate files, so the resulting file size is not too large
---
 src/Orangehill/Iseed/IseedCommand.php | 104 +++++++++++++++-----------
 1 file changed, 59 insertions(+), 45 deletions(-)

diff --git a/src/Orangehill/Iseed/IseedCommand.php b/src/Orangehill/Iseed/IseedCommand.php
index 58517ec..8849af2 100755
--- a/src/Orangehill/Iseed/IseedCommand.php
+++ b/src/Orangehill/Iseed/IseedCommand.php
@@ -66,6 +66,7 @@ public function fire()
         $direction = $this->option('direction');
         $prefix = $this->option('classnameprefix');
         $suffix = $this->option('classnamesuffix');
+        $separate = intval($this->option('separate'));
 
         if ($max < 1) {
             $max = null;
@@ -74,6 +75,8 @@ public function fire()
         if ($chunkSize < 1) {
             $chunkSize = null;
         }
+        
+        $j = $separate > 1 ? $separate : 1;
 
         $tableIncrement = 0;
         foreach ($tables as $table) {
@@ -88,50 +91,60 @@ public function fire()
             }
             $tableIncrement++;
 
-            // generate file and class name based on name of the table
-            list($fileName, $className) = $this->generateFileName($table, $prefix, $suffix);
-
-            // if file does not exist or force option is turned on generate seeder
-            if (!\File::exists($fileName) || $this->option('force')) {
-                $this->printResult(
-                    app('iseed')->generateSeed(
+            for ($i = 0; $i < $j; $i++) {
+                // generate file and class name based on name of the table
+                list($fileName, $className) = $this->generateFileName($table, $prefix, $suffix, ($separate > 1 ? ($i + 1) : null));
+
+                // if file does not exist or force option is turned on generate seeder
+                if (!\File::exists($fileName) || $this->option('force')) {
+                    $this->printResult(
+                        app('iseed')->generateSeed(
+                            $table,
+                            $prefix,
+                            $suffix,
+                            $this->option('database'),
+                            $max,
+                            $chunkSize,
+                            $exclude,
+                            $prerunEvent,
+                            $postrunEvent,
+                            $dumpAuto,
+                            $indexed,
+                            $orderBy,
+                            $direction,
+                            $separate,
+                            ($separate > 1 ? $i : null)
+                        ),
                         $table,
-                        $prefix,
-                        $suffix,
-                        $this->option('database'),
-                        $max,
-                        $chunkSize,
-                        $exclude,
-                        $prerunEvent,
-                        $postrunEvent,
-                        $dumpAuto,
-                        $indexed,
-                        $orderBy,
-                        $direction
-                    ),
-                    $table
-                );
-                continue;
-            }
-
-            if ($this->confirm('File ' . $className . ' already exist. Do you wish to override it? [yes|no]')) {
-                // if user said yes overwrite old seeder
-                $this->printResult(
-                    app('iseed')->generateSeed(
+                        ($separate > 1 ? ($i + 1) : null)
+                    );
+                    continue;
+                }
+
+                if ($this->confirm('File ' . $className . ' already exist. Do you wish to override it? [yes|no]')) {
+                    // if user said yes overwrite old seeder
+                    $this->printResult(
+                        app('iseed')->generateSeed(
+                            $table,
+                            $prefix,
+                            $suffix,
+                            $this->option('database'),
+                            $max,
+                            $chunkSize,
+                            $exclude,
+                            $prerunEvent,
+                            $postrunEvent,
+                            $dumpAuto,
+                            $indexed,
+                            $orderBy,
+                            $direction,
+                            $separate,
+                            ($separate > 1 ? $i : null)
+                        ),
                         $table,
-                        $prefix,
-                        $suffix,
-                        $this->option('database'),
-                        $max,
-                        $chunkSize,
-                        $exclude,
-                        $prerunEvent,
-                        $postrunEvent,
-                        $dumpAuto,
-                        $indexed
-                    ),
-                    $table
-                );
+                        ($separate > 1 ? ($i + 1) : null)
+                    );
+                }
             }
         }
 
@@ -172,6 +185,7 @@ protected function getOptions()
             array('direction', null, InputOption::VALUE_OPTIONAL, 'orderby direction', null),
             array('classnameprefix', null, InputOption::VALUE_OPTIONAL, 'prefix for class and file name', null),
             array('classnamesuffix', null, InputOption::VALUE_OPTIONAL, 'suffix for class and file name', null),
+            array('separate', null, InputOption::VALUE_OPTIONAL, 'separate file', null),
         );
     }
 
@@ -185,11 +199,11 @@ protected function getOptions()
     protected function printResult($successful, $table)
     {
         if ($successful) {
-            $this->info("Created a seed file from table {$table}");
+            $this->info("Created a seed" . (' ' . $i . ' ' ?? ' ') . "file from table {$table}");
             return;
         }
 
-        $this->error("Could not create seed file from table {$table}");
+        $this->error("Could not create seed" . (' ' . $i . ' ' ?? ' ') . "file from table {$table}");
     }
 
     /**
@@ -205,7 +219,7 @@ protected function generateFileName($table, $prefix=null, $suffix=null)
         }
 
         // Generate class name and file name
-        $className = app('iseed')->generateClassName($table, $prefix, $suffix);
+        $className = app('iseed')->generateClassName($table, $prefix, $suffix, $i);
         $seedPath = base_path() . config('iseed::config.path');
         return [$seedPath . '/' . $className . '.php', $className . '.php'];
     }