- Add support for generate --install

Author: DannyBen

lib/completely/commands/generate.rb

@@ -6,12 +6,15 @@ class Generate < Base
       help 'Generate the bash completion script to file or stdout'
 
       usage 'completely generate [CONFIG_PATH OUTPUT_PATH --function NAME --wrap NAME]'
+      usage 'completely generate [CONFIG_PATH --install PROGRAM --function NAME]'
       usage 'completely generate (-h|--help)'
 
       option_function
       option '-w --wrap NAME', 'Wrap the completion script inside a function that echos the ' \
         'script. This is useful if you wish to embed it directly in your script.'
 
+      option '-i --install PROGRAM', 'Install the generated script as completions for PROGRAM.'
+
       param 'CONFIG_PATH', <<~USAGE
         Path to the YAML configuration file [default: completely.yaml].
         Use '-' to read from stdin.
@@ -35,17 +38,34 @@ class Generate < Base
       def run
         wrap = args['--wrap']
         output = wrap ? wrapper_function(wrap) : script
+
+        if args['--install']
+          install output
+        else
+          show_or_save output
+        end
+      end
+
+    private
+
+      def install(content)
+        installer = Installer.from_string program: args['--install'], string: content
+        success = installer.install force: true
+        raise InstallError, "Failed running command:\nnb`#{installer.install_command_string}`" unless success
+        say "Saved m`#{installer.target_path}`"
+        say 'You may need to restart your session to test it'
+      end
+
+      def show_or_save(content)
         if output_path == '-'
-          puts output
+          puts content
         else
-          File.write output_path, output
+          File.write output_path, content
           say "Saved m`#{output_path}`"
         end
         syntax_warning unless completions.valid?
       end
 
-    private
-
       def wrapper_function(wrapper_name)
         completions.wrapper_function wrapper_name
       end

spec/approvals/cli/generate/help

@@ -2,6 +2,7 @@ Generate the bash completion script to file or stdout
 
 Usage:
   completely generate [CONFIG_PATH OUTPUT_PATH --function NAME --wrap NAME]
+  completely generate [CONFIG_PATH --install PROGRAM --function NAME]
   completely generate (-h|--help)
 
 Options:
@@ -12,6 +13,9 @@ Options:
     Wrap the completion script inside a function that echos the script. This is
     useful if you wish to embed it directly in your script.
 
+  -i --install PROGRAM
+    Install the generated script as completions for PROGRAM.
+
   -h --help
     Show this help