Add `CLI::Options` DSL.

1 files changed, 62 insertions, 0 deletions

diff --git a/Library/Homebrew/cask/lib/hbc/cli/options.rb b/Library/Homebrew/cask/lib/hbc/cli/options.rb
new file mode 100644
index 000000000..84126caa5
--- /dev/null
+++ b/Library/Homebrew/cask/lib/hbc/cli/options.rb
@@ -0,0 +1,62 @@
+module Hbc
+  class CLI
+    module Options
+      def self.included(klass)
+        klass.extend(ClassMethods)
+      end
+
+      module ClassMethods
+        def options
+          @options ||= {}
+          return @options unless superclass.respond_to?(:options)
+          superclass.options.merge(@options)
+        end
+
+        def option(name, method, default_value = nil)
+          @options ||= {}
+          @options[name] = method
+
+          return if method.respond_to?(:call)
+
+          define_method(:"#{method}=") do |value|
+            instance_variable_set(:"@#{method}", value)
+          end
+
+          if [true, false].include?(default_value)
+            define_method(:"#{method}?") do
+              instance_variable_get(:"@#{method}") == true
+            end
+          else
+            define_method(:"#{method}") do
+              instance_variable_get(:"@#{method}")
+            end
+          end
+        end
+      end
+
+      def process_arguments(*arguments)
+        parser = OptionParser.new do |opts|
+          next if self.class.options.nil?
+
+          self.class.options.each do |option_name, option_method|
+            option_type = case option_name.split(/(\ |\=)/).last
+            when "PATH"
+              Pathname
+            when /\w+(,\w+)+/
+              Array
+            end
+
+            opts.on(option_name, *option_type) do |value|
+              if option_method.respond_to?(:call)
+                option_method.call(value)
+              else
+                send(:"#{option_method}=", value)
+              end
+            end
+          end
+        end
+        parser.parse(*arguments)
+      end
+    end
+  end
+end