annotate 3.0.3 → 3.1.0

data/lib/annotate/annotate_routes.rb

@@ -22,58 +22,54 @@
 module AnnotateRoutes
   PREFIX = '== Route Map'.freeze
   PREFIX_MD = '## Route Map'.freeze
-  HEADER_ROW = ['Prefix', 'Verb', 'URI Pattern', 'Controller#Action']
+  HEADER_ROW = ['Prefix', 'Verb', 'URI Pattern', 'Controller#Action'].freeze
+
+  MAGIC_COMMENT_MATCHER = Regexp.new(/(^#\s*encoding:.*)|(^# coding:.*)|(^# -\*- coding:.*)|(^# -\*- encoding\s?:.*)|(^#\s*frozen_string_literal:.+)|(^# -\*- frozen_string_literal\s*:.+-\*-)/).freeze
 
   class << self
     def do_annotations(options = {})
-      return unless routes_exists?
-      existing_text = File.read(routes_file)
-
-      if rewrite_contents_with_header(existing_text, header(options), options)
-        puts "#{routes_file} annotated."
+      if routes_file_exist?
+        existing_text = File.read(routes_file)
+        content, header_position = strip_annotations(existing_text)
+        new_content = annotate_routes(header(options), content, header_position, options)
+        new_text = new_content.join("\n")
+
+        if rewrite_contents(existing_text, new_text)
+          puts "#{routes_file} was annotated."
+        else
+          puts "#{routes_file} was not changed."
+        end
+      else
+        puts "#{routes_file} could not be found."
       end
     end
 
     def remove_annotations(_options={})
-      return unless routes_exists?
-      existing_text = File.read(routes_file)
-      content, where_header_found = strip_annotations(existing_text)
-      new_content = strip_on_removal(content, where_header_found)
-      if rewrite_contents(existing_text, new_content)
-        puts "Removed annotations from #{routes_file}."
+      if routes_file_exist?
+        existing_text = File.read(routes_file)
+        content, header_position = strip_annotations(existing_text)
+        new_content = strip_on_removal(content, header_position)
+        new_text = new_content.join("\n")
+        if rewrite_contents(existing_text, new_text)
+          puts "Annotations were removed from #{routes_file}."
+        else
+          puts "#{routes_file} was not changed (Annotation did not exist)."
+        end
+      else
+        puts "#{routes_file} could not be found."
       end
     end
 
     private
 
-    def routes_exists?
-      routes_exists = File.exists?(routes_file)
-      puts "Can't find routes.rb" unless routes_exists
-
-      routes_exists
+    def routes_file_exist?
+      File.exist?(routes_file)
     end
 
     def routes_file
       @routes_rb ||= File.join('config', 'routes.rb')
     end
 
-    def rewrite_contents_with_header(existing_text, header, options = {})
-      content, where_header_found = strip_annotations(existing_text)
-      new_content = annotate_routes(header, content, where_header_found, options)
-
-      # Make sure we end on a trailing newline.
-      new_content << '' unless new_content.last == ''
-      new_text = new_content.join("\n")
-
-      if existing_text == new_text
-        puts "#{routes_file} unchanged."
-        false
-      else
-        File.open(routes_file, 'wb') { |f| f.puts(new_text) }
-        true
-      end
-    end
-
     def header(options = {})
       routes_map = app_routes_map(options)
 
@@ -86,10 +82,10 @@ module AnnotateRoutes
       end
       out << '' if magic_comments_map.any?
 
-      out += ["# #{options[:wrapper_open]}"] if options[:wrapper_open]
+      out << comment(options[:wrapper_open]) if options[:wrapper_open]
 
-      out += ["# #{options[:format_markdown] ? PREFIX_MD : PREFIX}" + (options[:timestamp] ? " (Updated #{Time.now.strftime('%Y-%m-%d %H:%M')})" : '')]
-      out += ['#']
+      out << comment(options[:format_markdown] ? PREFIX_MD : PREFIX) + (options[:timestamp] ? " (Updated #{Time.now.strftime('%Y-%m-%d %H:%M')})" : '')
+      out << comment
       return out if routes_map.size.zero?
 
       maxs = [HEADER_ROW.map(&:size)] + routes_map[1..-1].map { |line| line.split.map(&:size) }
@@ -97,20 +93,29 @@ module AnnotateRoutes
       if options[:format_markdown]
         max = maxs.map(&:max).compact.max
 
-        out += ["# #{content(HEADER_ROW, maxs, options)}"]
-        out += ["# #{content(['-' * max, '-' * max, '-' * max, '-' * max], maxs, options)}"]
+        out << comment(content(HEADER_ROW, maxs, options))
+        out << comment(content(['-' * max, '-' * max, '-' * max, '-' * max], maxs, options))
       else
-        out += ["# #{content(routes_map[0], maxs, options)}"]
+        out << comment(content(routes_map[0], maxs, options))
       end
 
-      out += routes_map[1..-1].map { |line| "# #{content(options[:format_markdown] ? line.split(' ') : line, maxs, options)}" }
-      out += ["# #{options[:wrapper_close]}"] if options[:wrapper_close]
+      out += routes_map[1..-1].map { |line| comment(content(options[:format_markdown] ? line.split(' ') : line, maxs, options)) }
+      out << comment(options[:wrapper_close]) if options[:wrapper_close]
 
       out
     end
 
+    def comment(row = '')
+      if row == ''
+        '#'
+      else
+        "# #{row}"
+      end
+    end
+
     # TODO: write the method doc using ruby rdoc formats
-    # where_header_found => This will either be :before, :after, or
+    # This method returns an array of 'real_content' and 'header_position'.
+    # 'header_position' will either be :before, :after, or
     # a number.  If the number is > 0, the
     # annotation was found somewhere in the
     # middle of the file.  If the number is
@@ -118,7 +123,7 @@ module AnnotateRoutes
     def strip_annotations(content)
       real_content = []
       mode = :content
-      header_found_at = 0
+      header_position = 0
 
       content.split(/\n/, -1).each_with_index do |line, line_number|
         if mode == :header && line !~ /\s*#/
@@ -126,7 +131,7 @@ module AnnotateRoutes
           real_content << line unless line.blank?
         elsif mode == :content
           if line =~ /^\s*#\s*== Route.*$/
-            header_found_at = line_number + 1 # index start's at 0
+            header_position = line_number + 1 # index start's at 0
             mode = :header
           else
             real_content << line
@@ -134,30 +139,27 @@ module AnnotateRoutes
         end
       end
 
-      where_header_found(real_content, header_found_at)
+      real_content_and_header_position(real_content, header_position)
     end
 
-    def strip_on_removal(content, where_header_found)
-      if where_header_found == :before
+    def strip_on_removal(content, header_position)
+      if header_position == :before
         content.shift while content.first == ''
-      elsif where_header_found == :after
+      elsif header_position == :after
         content.pop while content.last == ''
       end
 
+      # Make sure we end on a trailing newline.
+      content << '' unless content.last == ''
+
       # TODO: If the user buried it in the middle, we should probably see about
       # TODO: preserving a single line of space between the content above and
       # TODO: below...
       content
     end
 
-    # @param [String, Array<String>]
-    def rewrite_contents(existing_text, new_content)
-      # Make sure we end on a trailing newline.
-      new_content << '' unless new_content.last == ''
-      new_text = new_content.join("\n")
-
+    def rewrite_contents(existing_text, new_text)
       if existing_text == new_text
-        puts "#{routes_file} unchanged."
         false
       else
         File.open(routes_file, 'wb') { |f| f.puts(new_text) }
@@ -165,7 +167,7 @@ module AnnotateRoutes
       end
     end
 
-    def annotate_routes(header, content, where_header_found, options = {})
+    def annotate_routes(header, content, header_position, options = {})
       magic_comments_map, content = extract_magic_comments_from_array(content)
       if %w(before top).include?(options[:position_in_routes])
         header = header << '' if content.first != ''
@@ -178,11 +180,14 @@ module AnnotateRoutes
 
         # We're moving something from the top of the file to the bottom, so ditch
         # the spacer we put in the first time around.
-        content.shift if where_header_found == :before && content.first == ''
+        content.shift if header_position == :before && content.first == ''
 
         new_content = magic_comments_map + content + header
       end
 
+      # Make sure we end on a trailing newline.
+      new_content << '' unless new_content.last == ''
+
       new_content
     end
 
@@ -210,8 +215,8 @@ module AnnotateRoutes
       magic_comments = []
       new_content = []
 
-      content_array.map do |row|
-        if row =~ magic_comment_matcher
+      content_array.each do |row|
+        if row =~ MAGIC_COMMENT_MATCHER
           magic_comments << row.strip
         else
           new_content << row
@@ -231,21 +236,17 @@ module AnnotateRoutes
       end.join(' | ')
     end
 
-    def where_header_found(real_content, header_found_at)
+    def real_content_and_header_position(real_content, header_position)
       # By default assume the annotation was found in the middle of the file
 
       # ... unless we have evidence it was at the beginning ...
-      return real_content, :before if header_found_at == 1
+      return real_content, :before if header_position == 1
 
       # ... or that it was at the end.
-      return real_content, :after if header_found_at >= real_content.count
+      return real_content, :after if header_position >= real_content.count
 
       # and the default
-      return real_content, header_found_at
-    end
-
-    def magic_comment_matcher
-      Regexp.new(/(^#\s*encoding:.*)|(^# coding:.*)|(^# -\*- coding:.*)|(^# -\*- encoding\s?:.*)|(^#\s*frozen_string_literal:.+)|(^# -\*- frozen_string_literal\s*:.+-\*-)/)
+      return real_content, header_position
     end
   end
 end

data/lib/annotate/constants.rb

@@ -1,5 +1,38 @@
 module Annotate
   module Constants
     TRUE_RE = /^(true|t|yes|y|1)$/i.freeze
+
+    ##
+    # The set of available options to customize the behavior of Annotate.
+    #
+    POSITION_OPTIONS = [
+      :position_in_routes, :position_in_class, :position_in_test,
+      :position_in_fixture, :position_in_factory, :position,
+      :position_in_serializer
+    ].freeze
+
+    FLAG_OPTIONS = [
+      :show_indexes, :simple_indexes, :include_version, :exclude_tests,
+      :exclude_fixtures, :exclude_factories, :ignore_model_sub_dir,
+      :format_bare, :format_rdoc, :format_yard, :format_markdown, :sort, :force, :frozen,
+      :trace, :timestamp, :exclude_serializers, :classified_sort,
+      :show_foreign_keys, :show_complete_foreign_keys,
+      :exclude_scaffolds, :exclude_controllers, :exclude_helpers,
+      :exclude_sti_subclasses, :ignore_unknown_models, :with_comment
+    ].freeze
+
+    OTHER_OPTIONS = [
+      :additional_file_patterns, :ignore_columns, :skip_on_db_migrate, :wrapper_open, :wrapper_close,
+      :wrapper, :routes, :models, :hide_limit_column_types, :hide_default_column_types,
+      :ignore_routes, :active_admin
+    ].freeze
+
+    PATH_OPTIONS = [
+      :require, :model_dir, :root_dir
+    ].freeze
+
+    ALL_ANNOTATE_OPTIONS = [
+      POSITION_OPTIONS, FLAG_OPTIONS, OTHER_OPTIONS, PATH_OPTIONS
+    ].freeze
   end
 end

data/lib/annotate/helpers.rb

@@ -0,0 +1,30 @@
+module Annotate
+  # Class for holding helper methods. Done to make lib/annotate.rb less bloated.
+  class Helpers
+    class << self
+      def skip_on_migration?
+        ENV['ANNOTATE_SKIP_ON_DB_MIGRATE'] =~ Constants::TRUE_RE || ENV['skip_on_db_migrate'] =~ Constants::TRUE_RE
+      end
+
+      def include_routes?
+        ENV['routes'] =~ Constants::TRUE_RE
+      end
+
+      def include_models?
+        ENV['models'] =~ Constants::TRUE_RE
+      end
+
+      def true?(val)
+        val.present? && Constants::TRUE_RE.match?(val)
+      end
+
+      def fallback(*args)
+        args.detect(&:present?)
+      end
+
+      def reset_options(options)
+        options.flatten.each { |key| ENV[key.to_s] = nil }
+      end
+    end
+  end
+end

data/lib/annotate/parser.rb

@@ -9,14 +9,19 @@ module Annotate
 
     attr_reader :args, :options, :env
 
+    DEFAULT_OPTIONS = {
+      target_action: :do_annotations,
+      exit: false
+    }.freeze
+
     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
     EXCLUSION_LIST = %w[tests fixtures factories serializers].freeze
-    FORMAT_TYPES = %w[bare rdoc markdown].freeze
+    FORMAT_TYPES = %w[bare rdoc yard markdown].freeze
 
     def initialize(args, env)
       @args = args
-      @options = default_options
+      @options = DEFAULT_OPTIONS.dup
       @env = env
     end
 
@@ -31,128 +36,164 @@ module Annotate
 
     private
 
-    def commit
-      env.each_pair do |key, value|
-        ENV[key] = value
-      end
-    end
-
     def parser
       OptionParser.new do |option_parser|
         add_options_to_parser(option_parser)
       end
     end
 
+    def commit
+      env.each_pair do |key, value|
+        ENV[key] = value
+      end
+    end
+
     def add_options_to_parser(option_parser) # rubocop:disable Metrics/MethodLength
       has_set_position = {}
-      positions = ANNOTATION_POSITIONS
 
       option_parser.banner = 'Usage: annotate [options] [model_file]*'
 
-      option_parser.on('--additional-file-patterns path1,path2,path3', Array, "Additional file paths or globs to annotate, separated by commas (e.g. `/foo/bar/%model_name%/*.rb,/baz/%model_name%.rb`)") do |additional_file_patterns|
+      option_parser.on('--additional-file-patterns path1,path2,path3',
+                       Array,
+                       "Additional file paths or globs to annotate, separated by commas (e.g. `/foo/bar/%model_name%/*.rb,/baz/%model_name%.rb`)") do |additional_file_patterns|
         ENV['additional_file_patterns'] = additional_file_patterns
       end
 
-      option_parser.on('-d', '--delete', 'Remove annotations from all model files or the routes.rb file') do
+      option_parser.on('-d',
+                       '--delete',
+                       'Remove annotations from all model files or the routes.rb file') do
         @options[:target_action] = :remove_annotations
       end
 
-      option_parser.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
+      option_parser.on('-p',
+                       '--position [before|top|after|bottom]',
+                       ANNOTATION_POSITIONS,
+                       'Place the annotations at the top (before) or the bottom (after) of the model/test/fixture/factory/route/serializer file(s)') do |position|
+        env['position'] = position
 
         FILE_TYPE_POSITIONS.each do |key|
-          env[key] = p unless has_set_position[key]
+          env[key] = position unless has_set_position[key]
         end
       end
 
-      option_parser.on('--pc', '--position-in-class [before|top|after|bottom]', positions,
-                       'Place the annotations at the top (before) or the bottom (after) of the model file') do |p|
-        env['position_in_class'] = p
+      option_parser.on('--pc',
+                       '--position-in-class [before|top|after|bottom]',
+                       ANNOTATION_POSITIONS,
+                       'Place the annotations at the top (before) or the bottom (after) of the model file') do |position_in_class|
+        env['position_in_class'] = position_in_class
         has_set_position['position_in_class'] = true
       end
 
-      option_parser.on('--pf', '--position-in-factory [before|top|after|bottom]', positions,
-                       'Place the annotations at the top (before) or the bottom (after) of any factory files') do |p|
-        env['position_in_factory'] = p
+      option_parser.on('--pf',
+                       '--position-in-factory [before|top|after|bottom]',
+                       ANNOTATION_POSITIONS,
+                       'Place the annotations at the top (before) or the bottom (after) of any factory files') do |position_in_factory|
+        env['position_in_factory'] = position_in_factory
         has_set_position['position_in_factory'] = true
       end
 
-      option_parser.on('--px', '--position-in-fixture [before|top|after|bottom]', positions,
-                       'Place the annotations at the top (before) or the bottom (after) of any fixture files') do |p|
-        env['position_in_fixture'] = p
+      option_parser.on('--px',
+                       '--position-in-fixture [before|top|after|bottom]',
+                       ANNOTATION_POSITIONS,
+                       'Place the annotations at the top (before) or the bottom (after) of any fixture files') do |position_in_fixture|
+        env['position_in_fixture'] = position_in_fixture
         has_set_position['position_in_fixture'] = true
       end
 
-      option_parser.on('--pt', '--position-in-test [before|top|after|bottom]', positions,
-                       'Place the annotations at the top (before) or the bottom (after) of any test files') do |p|
-        env['position_in_test'] = p
+      option_parser.on('--pt',
+                       '--position-in-test [before|top|after|bottom]',
+                       ANNOTATION_POSITIONS,
+                       'Place the annotations at the top (before) or the bottom (after) of any test files') do |position_in_test|
+        env['position_in_test'] = position_in_test
         has_set_position['position_in_test'] = true
       end
 
-      option_parser.on('--pr', '--position-in-routes [before|top|after|bottom]', positions,
-                       'Place the annotations at the top (before) or the bottom (after) of the routes.rb file') do |p|
-        env['position_in_routes'] = p
+      option_parser.on('--pr',
+                       '--position-in-routes [before|top|after|bottom]',
+                       ANNOTATION_POSITIONS,
+                       'Place the annotations at the top (before) or the bottom (after) of the routes.rb file') do |position_in_routes|
+        env['position_in_routes'] = position_in_routes
         has_set_position['position_in_routes'] = true
       end
 
-      option_parser.on('--ps', '--position-in-serializer [before|top|after|bottom]', positions,
-                       'Place the annotations at the top (before) or the bottom (after) of the serializer files') do |p|
-        env['position_in_serializer'] = p
+      option_parser.on('--ps',
+                       '--position-in-serializer [before|top|after|bottom]',
+                       ANNOTATION_POSITIONS,
+                       'Place the annotations at the top (before) or the bottom (after) of the serializer files') do |position_in_serializer|
+        env['position_in_serializer'] = position_in_serializer
         has_set_position['position_in_serializer'] = true
       end
 
-      option_parser.on('--w', '--wrapper STR', 'Wrap annotation with the text passed as parameter.',
-                       'If --w option is used, the same text will be used as opening and closing') do |p|
-        env['wrapper'] = p
+      option_parser.on('--w',
+                       '--wrapper STR',
+                       'Wrap annotation with the text passed as parameter.',
+                       'If --w option is used, the same text will be used as opening and closing') do |wrapper|
+        env['wrapper'] = wrapper
       end
 
-      option_parser.on('--wo', '--wrapper-open STR', 'Annotation wrapper opening.') do |p|
-        env['wrapper_open'] = p
+      option_parser.on('--wo',
+                       '--wrapper-open STR',
+                       'Annotation wrapper opening.') do |wrapper_open|
+        env['wrapper_open'] = wrapper_open
       end
 
-      option_parser.on('--wc', '--wrapper-close STR', 'Annotation wrapper closing') do |p|
-        env['wrapper_close'] = p
+      option_parser.on('--wc',
+                       '--wrapper-close STR',
+                       'Annotation wrapper closing') do |wrapper_close|
+        env['wrapper_close'] = wrapper_close
       end
 
-      option_parser.on('-r', '--routes', "Annotate routes.rb with the output of 'rake routes'") do
+      option_parser.on('-r',
+                       '--routes',
+                       "Annotate routes.rb with the output of 'rake routes'") do
         env['routes'] = 'true'
       end
 
-      option_parser.on('--models', "Annotate ActiveRecord models") do
+      option_parser.on('--models',
+                       "Annotate ActiveRecord models") do
         env['models'] = 'true'
       end
 
-      option_parser.on('-a', '--active-admin', 'Annotate active_admin models') do
+      option_parser.on('-a',
+                       '--active-admin',
+                       'Annotate active_admin models') do
         env['active_admin'] = 'true'
       end
 
-      option_parser.on('-v', '--version', 'Show the current version of this gem') do
+      option_parser.on('-v',
+                       '--version',
+                       'Show the current version of this gem') do
         puts "annotate v#{Annotate.version}"
         @options[:exit] = true
       end
 
-      option_parser.on('-m', '--show-migration', 'Include the migration version number in the annotation') do
+      option_parser.on('-m',
+                       '--show-migration',
+                       'Include the migration version number in the annotation') do
         env['include_version'] = 'yes'
       end
 
-      option_parser.on('-k', '--show-foreign-keys',
+      option_parser.on('-k',
+                       '--show-foreign-keys',
                        "List the table's foreign key constraints in the annotation") do
         env['show_foreign_keys'] = 'yes'
       end
 
       option_parser.on('--ck',
-                       '--complete-foreign-keys', 'Complete foreign key names in the annotation') do
+                       '--complete-foreign-keys',
+                       'Complete foreign key names in the annotation') do
         env['show_foreign_keys'] = 'yes'
         env['show_complete_foreign_keys'] = 'yes'
       end
 
-      option_parser.on('-i', '--show-indexes',
+      option_parser.on('-i',
+                       '--show-indexes',
                        "List the table's database indexes in the annotation") do
         env['show_indexes'] = 'yes'
       end
 
-      option_parser.on('-s', '--simple-indexes',
+      option_parser.on('-s',
+                       '--simple-indexes',
                        "Concat the column's related indexes in the annotation") do
         env['simple_indexes'] = 'yes'
       end
@@ -168,84 +209,95 @@ module Annotate
       end
 
       option_parser.on('--ignore-model-subdirects',
-                       "Ignore subdirectories of the models directory") do |_dir|
+                       "Ignore subdirectories of the models directory") do
         env['ignore_model_sub_dir'] = 'yes'
       end
 
       option_parser.on('--sort',
-                       "Sort columns alphabetically, rather than in creation order") do |_dir|
+                       "Sort columns alphabetically, rather than in creation order") do
         env['sort'] = 'yes'
       end
 
       option_parser.on('--classified-sort',
-                       "Sort columns alphabetically, but first goes id, then the rest columns, then the timestamp columns and then the association columns") do |_dir|
+                       "Sort columns alphabetically, but first goes id, then the rest columns, then the timestamp columns and then the association columns") do
         env['classified_sort'] = 'yes'
       end
 
-      option_parser.on('-R', '--require path',
+      option_parser.on('-R',
+                       '--require path',
                        "Additional file to require before loading models, may be used multiple times") do |path|
-        env['require'] = if !env['require'].blank?
-                           env['require'] + ",#{path}"
+        env['require'] = if env['require'].present?
+                           "#{env['require']},#{path}"
                          else
                            path
                          end
       end
 
-      option_parser.on('-e', '--exclude [tests,fixtures,factories,serializers]', Array, "Do not annotate fixtures, test files, factories, and/or serializers") do |exclusions|
+      option_parser.on('-e',
+                       '--exclude [tests,fixtures,factories,serializers]',
+                       Array,
+                       "Do not annotate fixtures, test files, factories, and/or serializers") do |exclusions|
         exclusions ||= EXCLUSION_LIST
         exclusions.each { |exclusion| env["exclude_#{exclusion}"] = 'yes' }
       end
 
-      option_parser.on('-f', '--format [bare|rdoc|markdown]', FORMAT_TYPES, 'Render Schema Infomation as plain/RDoc/Markdown') do |fmt|
-        env["format_#{fmt}"] = 'yes'
+      option_parser.on('-f',
+                       '--format [bare|rdoc|yard|markdown]',
+                       FORMAT_TYPES,
+                       'Render Schema Infomation as plain/RDoc/Yard/Markdown') do |format_type|
+        env["format_#{format_type}"] = 'yes'
       end
 
-      option_parser.on('--force', 'Force new annotations even if there are no changes.') do |_force|
+      option_parser.on('--force',
+                       'Force new annotations even if there are no changes.') do
         env['force'] = 'yes'
       end
 
-      option_parser.on('--frozen', 'Do not allow to change annotations. Exits non-zero if there are going to be changes to files.') do
+      option_parser.on('--frozen',
+                       'Do not allow to change annotations. Exits non-zero if there are going to be changes to files.') do
         env['frozen'] = 'yes'
       end
 
-      option_parser.on('--timestamp', 'Include timestamp in (routes) annotation') do
+      option_parser.on('--timestamp',
+                       'Include timestamp in (routes) annotation') do
         env['timestamp'] = 'true'
       end
 
-      option_parser.on('--trace', 'If unable to annotate a file, print the full stack trace, not just the exception message.') do |_value|
+      option_parser.on('--trace',
+                       'If unable to annotate a file, print the full stack trace, not just the exception message.') do
         env['trace'] = 'yes'
       end
 
-      option_parser.on('-I', '--ignore-columns REGEX', "don't annotate columns that match a given REGEX (i.e., `annotate -I '^(id|updated_at|created_at)'`") do |regex|
+      option_parser.on('-I',
+                       '--ignore-columns REGEX',
+                       "don't annotate columns that match a given REGEX (i.e., `annotate -I '^(id|updated_at|created_at)'`") do |regex|
         env['ignore_columns'] = regex
       end
 
-      option_parser.on('--ignore-routes REGEX', "don't annotate routes that match a given REGEX (i.e., `annotate -I '(mobile|resque|pghero)'`") do |regex|
+      option_parser.on('--ignore-routes REGEX',
+                       "don't annotate routes that match a given REGEX (i.e., `annotate -I '(mobile|resque|pghero)'`") do |regex|
         env['ignore_routes'] = regex
       end
 
-      option_parser.on('--hide-limit-column-types VALUES', "don't show limit for given column types, separated by commas (i.e., `integer,boolean,text`)") do |values|
+      option_parser.on('--hide-limit-column-types VALUES',
+                       "don't show limit for given column types, separated by commas (i.e., `integer,boolean,text`)") do |values|
         env['hide_limit_column_types'] = values.to_s
       end
 
-      option_parser.on('--hide-default-column-types VALUES', "don't show default for given column types, separated by commas (i.e., `json,jsonb,hstore`)") do |values|
+      option_parser.on('--hide-default-column-types VALUES',
+                       "don't show default for given column types, separated by commas (i.e., `json,jsonb,hstore`)") do |values|
         env['hide_default_column_types'] = values.to_s
       end
 
-      option_parser.on('--ignore-unknown-models', "don't display warnings for bad model files") do |_values|
+      option_parser.on('--ignore-unknown-models',
+                       "don't display warnings for bad model files") do
         env['ignore_unknown_models'] = 'true'
       end
 
-      option_parser.on('--with-comment', "include database comments in model annotations") do |_values|
+      option_parser.on('--with-comment',
+                       "include database comments in model annotations") do
         env['with_comment'] = 'true'
       end
     end
-
-    def default_options
-      {
-        target_action: :do_annotations,
-        exit: false
-      }
-    end
   end
 end