Start copying over CLI parsing

Author: drwl

lib/parser.rb

@@ -0,0 +1,55 @@
+require 'optparse'
+
+# Class for handling command line arguments
+class Parser
+  def self.parse(args)
+    new(args).parse
+  end
+
+  attr_reader :args
+
+  ANNOTATION_POSITIONS = %w[before top after bottom].freeze
+  FILE_TYPE_POSITIONS = %w[position_in_class position_in_factory position_in_fixture position_in_test position_in_routes position_in_serializer].freeze
+
+  def initialize(args)
+    @args = args
+  end
+
+  def parse
+    options = default_options
+
+    parser(options).parse!(args)
+
+    options
+  end
+
+  private
+
+  def parser(options)
+    has_set_position = {}
+    positions = ANNOTATION_POSITIONS
+
+    OptionParser.new do |opts|
+      opts.banner = 'Usage: annotate [options] [model_file]*'
+
+      opts.on('-d', '--delete', 'Remove annotations from all model files or the routes.rb file') do
+        options[:target_action] = :remove_annotations
+      end
+
+      opts.on('-p', '--position [before|top|after|bottom]', positions,
+              'Place the annotations at the top (before) or the bottom (after) of the model/test/fixture/factory/route/serializer file(s)') do |p|
+        ENV['position'] = p
+
+        FILE_TYPE_POSITIONS.each do |key|
+          ENV[key] = p unless has_set_position[key]
+        end
+      end
+    end
+  end
+
+  def default_options
+    {
+      target_action: :do_annotations
+    }
+  end
+end

spec/parser_spec.rb

@@ -0,0 +1,47 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+require 'pry'
+
+describe Parser do
+  context 'when given empty args' do
+    it 'returns an options hash with defaults' do
+      result = Parser.parse([])
+      expect(result).to be_a(Hash)
+      expect(result).to include(target_action: :do_annotations)
+    end
+  end
+
+  %w[-d --delete].each do |option|
+    describe option do
+      it 'sets target_action to :remove_annotations' do
+        result = Parser.parse([option])
+        expect(result).to include(target_action: :remove_annotations)
+      end
+    end
+  end
+
+  ['-p', '--position'].each do |option|
+    describe option do
+      Parser::ANNOTATION_POSITIONS.each do |position|
+        context "when specifying #{position}" do
+          it "#{position} position is an option" do
+            Parser.parse([option, position])
+            expect(Parser::ANNOTATION_POSITIONS).to include(position)
+          end
+
+          it "sets ENV['position'] to be position" do
+            Parser.parse([option, position])
+            expect(ENV['position']).to eq(position)
+          end
+
+          it 'sets the value in ENV for the different file types' do
+            Parser.parse([option, position])
+            Parser::FILE_TYPE_POSITIONS.each do |file_type|
+              expect(ENV[file_type]).to eq(position)
+            end
+          end
+        end
+      end
+    end
+  end
+end

spec/spec_helper.rb

@@ -28,6 +28,7 @@
 require 'active_support/core_ext/class/subclasses'
 require 'active_support/core_ext/string/inflections'
 require 'annotate'
+require 'parser'
 require 'byebug'
 
 module Annotate