Add autocomplete for console command options.

KnpLabs · Jan 21, 2011 · f1a1eee · f1a1eee
1 parent 3137a6f
commit f1a1eee
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 7 deletions.
diff --git a/Command/AutocompleteCommand.php b/Command/AutocompleteCommand.php
@@ -4,6 +4,7 @@
 
 use Symfony\Bundle\FrameworkBundle\Command;
 use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Output\OutputInterface;
 
 /**
@@ -22,6 +23,7 @@ protected function configure()
     {
         $this
             ->setDefinition(array(
+                new InputArgument('command_name', InputArgument::OPTIONAL, 'A command name to generate autocomplete options for'),
             ))
             ->setName('console:autocomplete')
             ->setHelp(<<<EOT
@@ -30,18 +32,28 @@ protected function configure()
 
   <info>php app/console console:autocomplete</info>
 EOT
-        );
-        ;
+            );
     }
 
     /**
      * @see Command
      */
     protected function execute(InputInterface $input, OutputInterface $output)
     {
-        $commands = $this->application->all();
-        $commands = array_keys($commands);
-
-        $output->write(join(" ", $commands), false); 
+        $commandName = $input->getArgument('command_name');
+        if ($commandName !== null && $this->application->has($commandName)) {
+            $options = array_merge(
+                $this->application->get($commandName)->getDefinition()->getOptions(),
+                $this->application->getDefinition()->getOptions()
+            );
+            $options = array_map(function($option) {
+                return '--' . $option->getName();
+            }, $options);
+            $output->write(join(" ", $options), false);
+        } else {
+            $commands = $this->application->all();
+            $commands = array_keys($commands);
+            $output->write(join(" ", $commands), false);
+        }
     }
 }
diff --git a/Resources/Shells/symfony2-completion.bash b/Resources/Shells/symfony2-completion.bash
@@ -26,7 +26,17 @@ _console()
                 return 0
                 ;;
             *)
-            ;;
+                ;;
+        esac
+    fi
+    if [[ ${COMP_CWORD} > 1 ]] ; then
+        case "${cur}" in
+            --*)
+                COMPREPLY=( $(compgen -W "$(${cmd} console:autocomplete ${COMP_WORDS[1]})" -- ${cur}) )
+                return 0
+                ;;
+            *)
+                ;;
         esac
     fi