[Command] Added better for custom default command

lyrixx · lyrixx · commit b342502a0284 · 2015-12-11T16:01:14.000+01:00
diff --git a/src/Symfony/Component/Console/Application.php b/src/Symfony/Component/Console/Application.php
@@ -66,6 +66,7 @@ class Application
     private $dispatcher;
     private $terminalDimensions;
     private $defaultCommand;
+    private $singleCommand;
 
     /**
      * Constructor.
@@ -167,8 +168,7 @@ public function doRun(InputInterface $input, OutputInterface $output)
         $name = $this->getCommandName($input);
         if (true === $input->hasParameterOption(array('--help', '-h'), true)) {
             if (!$name) {
-                $name = 'help';
-                $input = new ArrayInput(array('command' => 'help'));
+                $input = new ArrayInput(array('command_name' => $this->defaultCommand));
             } else {
                 $this->wantHelps = true;
             }
@@ -226,6 +226,13 @@ public function setDefinition(InputDefinition $definition)
      */
     public function getDefinition()
     {
+        if ($this->singleCommand) {
+            $inputDefinition = $this->definition;
+            $inputDefinition->setArguments();
+
+            return $inputDefinition;
+        }
+
         return $this->definition;
     }
 
@@ -832,7 +839,7 @@ protected function doRunCommand(Command $command, InputInterface $input, OutputI
      */
     protected function getCommandName(InputInterface $input)
     {
-        return $input->getFirstArgument();
+        return $this->singleCommand ?: $input->getFirstArgument();
     }
 
     /**
@@ -1017,6 +1024,22 @@ private function findAlternatives($name, $collection)
     public function setDefaultCommand($commandName)
     {
         $this->defaultCommand = $commandName;
+
+        return $this;
+    }
+
+    public function setSingleCommand($commandName)
+    {
+        if (null !== $this->singleCommand) {
+            throw new \LogicException('A Single command is already defined.');
+        }
+
+        // Ensure the command exist
+        $this->find($commandName);
+
+        $this->singleCommand = $commandName;
+
+        return $this;
     }
 
     private function stringWidth($string)
