Merge pull request #78 from dbpolito/patch-1

Making command environment more flexible
beyondcode · Aug 26, 2019 · 4ee28ca · 4ee28ca
2 parents ead1049 + 9e4522d
commit 4ee28ca
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/SelfDiagnosisCommand.php b/src/SelfDiagnosisCommand.php
@@ -12,7 +12,7 @@ class SelfDiagnosisCommand extends Command
      *
      * @var string
      */
-    protected $signature = 'self-diagnosis';
+    protected $signature = 'self-diagnosis {environment?}';
 
     /**
      * The console command description.
@@ -27,13 +27,15 @@ public function handle()
     {
         $this->runChecks(config('self-diagnosis.checks', []), trans('self-diagnosis::commands.self_diagnosis.common_checks'));
 
-        $environmentChecks = config('self-diagnosis.environment_checks.' . app()->environment(), []);
-        if (empty($environmentChecks) && array_key_exists(app()->environment(), config('self-diagnosis.environment_aliases'))) {
-            $environment = config('self-diagnosis.environment_aliases.' . app()->environment());
+        $environment = $this->argument('environment', app()->environment());
+        $environmentChecks = config('self-diagnosis.environment_checks.' . $environment, []);
+
+        if (empty($environmentChecks) && array_key_exists($environment, config('self-diagnosis.environment_aliases'))) {
+            $environment = config('self-diagnosis.environment_aliases.' . $environment);
             $environmentChecks = config('self-diagnosis.environment_checks.' . $environment, []);
         }
 
-        $this->runChecks($environmentChecks, trans('self-diagnosis::commands.self_diagnosis.environment_specific_checks', ['environment' => app()->environment()]));
+        $this->runChecks($environmentChecks, trans('self-diagnosis::commands.self_diagnosis.environment_specific_checks', ['environment' => $environment]));
 
         if (count($this->messages)) {
             $this->error(trans('self-diagnosis::commands.self_diagnosis.failed_checks'));