Checks

Author: Casmo

src/Commands/ComponentsCommand.php

@@ -13,18 +13,18 @@ class ComponentsCommand extends Command
 
     public function handle(): int
     {
-        $component = $this->argument('component');
+        $components = $this->getComponents();
 
-        if (! $component) {
+        if (empty($components)) {
+            $this->warn('No components found');
+            return Command::FAILURE;
+        }
 
-            $publishableGroups = ServiceProvider::$publishGroups;
-            $keys = array_keys($publishableGroups);
-            foreach ($keys as $key) {
-                if (str_starts_with($key, 'backstage-components-')) {
-                    $components[] = str_replace('backstage-components-', '', $key);
-                }
-            }
-            $component = $this->choice('Which component do you want to publish?', $components);
+        $component = $this->argument('component') ?? $this->choice('Which component do you want to publish?', $components);
+
+        if (! in_array($component, $components)) {
+            $this->error("Component {$component} not found");
+            return Command::FAILURE;
         }
 
         $this->info("Publishing component: {$component}");
@@ -34,4 +34,18 @@ public function handle(): int
 
         return Command::SUCCESS;
     }
+
+    private function getComponents(): array
+    {
+        $components = [];
+        $publishableGroups = ServiceProvider::$publishGroups;
+        $keys = array_keys($publishableGroups);
+        foreach ($keys as $key) {
+            if (str_starts_with($key, 'backstage-components-')) {
+                $components[] = str_replace('backstage-components-', '', $key);
+            }
+        }
+        
+        return $components;
+    }
 }