Copy over the rest

Author: drwl

lib/parser.rb

@@ -1,7 +1,7 @@
 require 'optparse'
 
 # Class for handling command line arguments
-class Parser
+class Parser # rubocop:disable Metrics/ClassLength
   def self.parse(args)
     new(args).parse
   end
@@ -10,6 +10,8 @@ def self.parse(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
+  EXCLUSION_LIST = %w[tests fixtures factories serializers].freeze
+  FORMAT_TYPES = %w[bare rdoc markdown].freeze
 
   def initialize(args)
     @args = args
@@ -25,7 +27,7 @@ def parse
 
   private
 
-  def parser(options)
+  def parser(options) # rubocop:disable Metrics/MethodLength
     has_set_position = {}
     positions = ANNOTATION_POSITIONS
 
@@ -93,6 +95,126 @@ def parser(options)
       opts.on('--wc', '--wrapper-close STR', 'Annotation wrapper closing') do |p|
         ENV['wrapper_close'] = p
       end
+
+      opts.on('-r', '--routes', "Annotate routes.rb with the output of 'rake routes'") do
+        ENV['routes'] = 'true'
+      end
+
+      opts.on('-a', '--active-admin', 'Annotate active_admin models') do
+        ENV['active_admin'] = 'true'
+      end
+
+      opts.on('-v', '--version', 'Show the current version of this gem') do
+        puts "annotate v#{Annotate.version}"
+      end
+
+      opts.on('-m', '--show-migration', 'Include the migration version number in the annotation') do
+        ENV['include_version'] = 'yes'
+      end
+
+      opts.on('-k', '--show-foreign-keys',
+              "List the table's foreign key constraints in the annotation") do
+        ENV['show_foreign_keys'] = 'yes'
+      end
+
+      opts.on('--ck',
+              '--complete-foreign-keys', 'Complete foreign key names in the annotation') do
+        ENV['show_foreign_keys'] = 'yes'
+        ENV['show_complete_foreign_keys'] = 'yes'
+      end
+
+      opts.on('-i', '--show-indexes',
+              "List the table's database indexes in the annotation") do
+        ENV['show_indexes'] = 'yes'
+      end
+
+      opts.on('-s', '--simple-indexes',
+              "Concat the column's related indexes in the annotation") do
+        ENV['simple_indexes'] = 'yes'
+      end
+
+      opts.on('--model-dir dir',
+              "Annotate model files stored in dir rather than app/models, separate multiple dirs with commas") do |dir|
+        ENV['model_dir'] = dir
+      end
+
+      opts.on('--root-dir dir',
+              "Annotate files stored within root dir projects, separate multiple dirs with commas") do |dir|
+        ENV['root_dir'] = dir
+      end
+
+      opts.on('--ignore-model-subdirects',
+              "Ignore subdirectories of the models directory") do |_dir|
+        ENV['ignore_model_sub_dir'] = 'yes'
+      end
+
+      opts.on('--sort',
+              "Sort columns alphabetically, rather than in creation order") do |_dir|
+        ENV['sort'] = 'yes'
+      end
+
+      opts.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|
+        ENV['classified_sort'] = 'yes'
+      end
+
+      opts.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}"
+                         else
+                           path
+                         end
+      end
+
+      opts.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
+
+      opts.on('-f', '--format [bare|rdoc|markdown]', FORMAT_TYPES, 'Render Schema Infomation as plain/RDoc/Markdown') do |fmt|
+        ENV["format_#{fmt}"] = 'yes'
+      end
+
+      opts.on('--force', 'Force new annotations even if there are no changes.') do |_force|
+        ENV['force'] = 'yes'
+      end
+
+      opts.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
+
+      opts.on('--timestamp', 'Include timestamp in (routes) annotation') do
+        ENV['timestamp'] = 'true'
+      end
+
+      opts.on('--trace', 'If unable to annotate a file, print the full stack trace, not just the exception message.') do |_value|
+        ENV['trace'] = 'yes'
+      end
+
+      opts.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
+
+      opts.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
+
+      opts.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
+
+      opts.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
+
+      opts.on('--ignore-unknown-models', "don't display warnings for bad model files") do |_values|
+        ENV['ignore_unknown_models'] = 'true'
+      end
+
+      opts.on('--with-comment', "include database comments in model annotations") do |_values|
+        ENV['with_comment'] = 'true'
+      end
     end
   end