- Refactor CLI commands

Author: DannyBen

lib/completely/cli.rb

@@ -1,10 +1,20 @@
 require 'mister_bin'
-require 'completely/command'
+require 'completely/commands/init'
+require 'completely/commands/preview'
+require 'completely/commands/generate'
 
 module Completely
   class CLI
     def self.runner
-      MisterBin::Runner.new handler: Command
+      runner = MisterBin::Runner.new version: Completely::VERSION,
+        header: "Completely - Bash Completions Generator",
+        footer: "Run !txtpur!completely COMMAND --help!txtrst! for more information"
+
+      runner.route 'init',      to: Commands::Init
+      runner.route 'preview',   to: Commands::Preview
+      runner.route 'generate',  to: Commands::Generate
+
+      runner
     end
   end
 end

lib/completely/command.rb

@@ -0,0 +1,37 @@
+require 'mister_bin'
+
+module Completely
+  module Commands
+    class Base < MisterBin::Command
+
+      class << self
+        def config_path_usage
+          param "CONFIG_PATH", "Path to the YAML configuration file [default: completely.yaml]"
+        end
+
+        def function_usage
+          option "-f --function NAME", "Modify the name of the function in the generated script"
+        end
+      end
+
+    protected
+
+      def script
+        @script ||= completions.script
+      end
+
+      def completions
+        @completions ||= Completions.load(config_path, function_name: args['--function'])
+      end
+
+      def config_path
+        args['CONFIG_PATH'] || 'completely.yaml'
+      end
+
+      def syntax_warning
+        say! "\n!txtred!WARNING:\nYour configuration is invalid."
+        say! "!txtred!All patterns must start with the same word."
+      end
+    end
+  end
+end

lib/completely/commands/base.rb

@@ -0,0 +1,37 @@
+require 'completely/commands/base'
+
+module Completely
+  module Commands
+    class Generate < Base
+      help "Generate the bash completion script to a file"
+
+      usage "completely generate [CONFIG_PATH OUTPUT_PATH --function NAME --wrap NAME]"
+      usage "completely generate (-h|--help)"
+
+      function_usage
+      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"
+
+      config_path_usage
+      param "OUTPUT_PATH", "Path to the output bash script [default: completely.bash]"
+
+      def run
+        wrap = args['--wrap']
+        output = wrap ? wrapper_function(wrap) : script
+        File.write output_path, output
+        say "Saved !txtpur!#{output_path}"
+        syntax_warning unless completions.valid?
+      end
+
+    private
+
+      def wrapper_function(wrapper_name)
+        completions.wrapper_function wrapper_name
+      end
+
+      def output_path
+        @output_path ||= args['OUTPUT_PATH'] || "completely.bash"
+      end
+
+    end
+  end
+end

lib/completely/commands/generate.rb

@@ -0,0 +1,34 @@
+require 'completely/commands/base'
+
+module Completely
+  module Commands
+    class Init < Base
+      help "Create a new sample YAML configuration file"
+
+      usage "completely init [CONFIG_PATH]"
+      usage "completely init (-h|--help)"
+
+      config_path_usage
+
+      def run
+        if File.exist? config_path
+          raise "File already exists: #{config_path}"
+        else
+          File.write config_path, sample
+          say "Saved !txtpur!#{config_path}"
+        end
+      end
+
+    private
+
+      def sample
+        @sample ||= File.read sample_path
+      end
+
+      def sample_path
+        @sample_path ||= File.expand_path "../sample.yaml", __dir__
+      end
+
+    end
+  end
+end

lib/completely/commands/init.rb

@@ -0,0 +1,20 @@
+require 'completely/commands/base'
+
+module Completely
+  module Commands
+    class Preview < Base
+      help "Generate the bash completion script to STDOUT"
+
+      usage "completely preview [CONFIG_PATH --function NAME]"
+      usage "completely preview (-h|--help)"
+
+      function_usage      
+      config_path_usage
+
+      def run
+        puts script
+        syntax_warning unless completions.valid?
+      end
+    end
+  end
+end

lib/completely/commands/preview.rb

@@ -0,0 +1,8 @@
+Completely - Bash Completions Generator
+
+Commands:
+  init      Create a new sample YAML configuration file
+  preview   Generate the bash completion script to STDOUT
+  generate  Generate the bash completion script to a file
+
+Run completely COMMAND --help for more information