Module: AnnotateModels
- Defined in:
- lib/annotate/annotate_models.rb
Defined Under Namespace
Classes: BadModelFileError
Constant Summary collapse
- COMPAT_PREFIX =
Annotate Models plugin use this header
'== Schema Info'.freeze
- COMPAT_PREFIX_MD =
'## Schema Info'.freeze
- PREFIX =
'== Schema Information'.freeze
- PREFIX_MD =
'## Schema Information'.freeze
- END_MARK =
'== Schema Information End'.freeze
- SKIP_ANNOTATION_PREFIX =
'# -\*- SkipSchemaAnnotations'.freeze
- MATCHED_TYPES =
%w(test fixture factory serializer scaffold controller helper).freeze
- UNIT_TEST_DIR =
File.join for windows reverse bar compat? I dont use windows, can`t test
File.join('test', "unit")
- MODEL_TEST_DIR =
since rails 4.0
File.join('test', "models")
- SPEC_MODEL_DIR =
File.join('spec', "models")
- FIXTURE_TEST_DIR =
File.join('test', "fixtures")
- FIXTURE_SPEC_DIR =
File.join('spec', "fixtures")
- CONTROLLER_TEST_DIR =
Other test files
File.join('test', "controllers")
- CONTROLLER_SPEC_DIR =
File.join('spec', "controllers")
- REQUEST_SPEC_DIR =
File.join('spec', "requests")
- ROUTING_SPEC_DIR =
File.join('spec', "routing")
- EXEMPLARS_TEST_DIR =
Object Daddy github.com/flogic/object_daddy/tree/master
File.join('test', "exemplars")
- EXEMPLARS_SPEC_DIR =
File.join('spec', "exemplars")
- BLUEPRINTS_TEST_DIR =
Machinist github.com/notahat/machinist
File.join('test', "blueprints")
- BLUEPRINTS_SPEC_DIR =
File.join('spec', "blueprints")
- FACTORY_BOT_TEST_DIR =
Factory Bot github.com/thoughtbot/factory_bot
File.join('test', "factories")
- FACTORY_BOT_SPEC_DIR =
File.join('spec', "factories")
- FABRICATORS_TEST_DIR =
Fabrication github.com/paulelliott/fabrication.git
File.join('test', "fabricators")
- FABRICATORS_SPEC_DIR =
File.join('spec', "fabricators")
- SERIALIZERS_DIR =
Serializers github.com/rails-api/active_model_serializers
File.join('app', "serializers")
- SERIALIZERS_TEST_DIR =
File.join('test', "serializers")
- SERIALIZERS_SPEC_DIR =
File.join('spec', "serializers")
- CONTROLLER_DIR =
Controller files
File.join('app', "controllers")
- ACTIVEADMIN_DIR =
Active admin registry files
File.join('app', "admin")
- HELPER_DIR =
Helper files
File.join('app', "helpers")
- NO_LIMIT_COL_TYPES =
Don’t show limit (#) on these column types Example: show “integer” instead of “integer(4)”
%w(integer bigint boolean).freeze
- NO_DEFAULT_COL_TYPES =
Don’t show default value for these column types
%w(json jsonb hstore).freeze
- INDEX_CLAUSES =
{ unique: { default: 'UNIQUE', markdown: '_unique_' }, where: { default: 'WHERE', markdown: '_where_' }, using: { default: 'USING', markdown: '_using_' } }.freeze
- MAGIC_COMMENT_MATCHER =
Regexp.new(/(^#\s*encoding:.*(?:\n|r\n))|(^# coding:.*(?:\n|\r\n))|(^# -\*- coding:.*(?:\n|\r\n))|(^# -\*- encoding\s?:.*(?:\n|\r\n))|(^#\s*frozen_string_literal:.+(?:\n|\r\n))|(^# -\*- frozen_string_literal\s*:.+-\*-(?:\n|\r\n))/).freeze
Class Attribute Summary collapse
Class Method Summary collapse
-
.annotate(klass, file, header, options = {}) ⇒ Object
Given the name of an ActiveRecord class, create a schema info block (basically a comment containing information on the columns and their types) and put it at the front of the model and fixture source files.
- .annotate_model_file(annotated, file, header, options) ⇒ Object
-
.annotate_one_file(file_name, info_block, position, options = {}) ⇒ Object
Add a schema block to a file.
- .annotate_pattern(options = {}) ⇒ Object
- .classified_sort(cols) ⇒ Object
-
.do_annotations(options = {}) ⇒ Object
We’re passed a name of things that might be ActiveRecord models.
- .expand_glob_into_files(glob) ⇒ Object
- .factory_files(root_directory) ⇒ Object
- .files_by_pattern(root_directory, pattern_type, options) ⇒ Object
- .final_index_string(index, max_size) ⇒ Object
- .final_index_string_in_markdown(index) ⇒ Object
- .fixture_files(root_directory) ⇒ Object
- .get_col_type(col) ⇒ Object
- .get_foreign_key_info(klass, options = {}) ⇒ Object
- .get_index_info(klass, options = {}) ⇒ Object
-
.get_loaded_model(model_path, file) ⇒ Object
Retrieve loaded model class.
-
.get_loaded_model_by_path(model_path) ⇒ Object
Retrieve loaded model class by path to the file where it’s supposed to be defined.
-
.get_model_class(file) ⇒ Object
Retrieve the classes belonging to the model names we’re asked to process Check for namespaced models in subdirectories as well as models in subdirectories without namespacing.
-
.get_model_files(options) ⇒ Object
Return a list of the model files to annotate.
- .get_patterns(options, pattern_types = []) ⇒ Object
- .get_schema_footer_text(_klass, options = {}) ⇒ Object
- .get_schema_header_text(klass, options = {}) ⇒ Object
-
.get_schema_info(klass, header, options = {}) ⇒ Object
Use the column information in an ActiveRecord class to create a comment block containing a line for each column.
- .hide_default?(col_type, options) ⇒ Boolean
- .hide_limit?(col_type, options) ⇒ Boolean
- .index_columns_info(index) ⇒ Object
- .index_unique_info(index, format = :default) ⇒ Object
- .index_using_info(index, format = :default) ⇒ Object
- .index_where_info(index, format = :default) ⇒ Object
- .list_model_files_from_argument ⇒ Object
- .magic_comments_as_string(content) ⇒ Object
- .matched_types(options) ⇒ Object
-
.options_with_position(options, position_in) ⇒ Object
position = :position_in_fixture or :position_in_class.
- .parse_options(options = {}) ⇒ Object
-
.quote(value) ⇒ Object
Simple quoting for the default column value.
- .remove_annotation_of_file(file_name, options = {}) ⇒ Object
- .remove_annotations(options = {}) ⇒ Object
- .resolve_filename(filename_template, model_name, table_name) ⇒ Object
- .retrieve_indexes_from_table(klass) ⇒ Object
- .scaffold_files(root_directory) ⇒ Object
- .schema_default(klass, column) ⇒ Object
- .serialize_files(root_directory) ⇒ Object
- .split_model_dir(option_value) ⇒ Object
- .test_files(root_directory) ⇒ Object
Class Attribute Details
.model_dir ⇒ Object
95 96 97 |
# File 'lib/annotate/annotate_models.rb', line 95 def model_dir @model_dir.is_a?(Array) ? @model_dir : [@model_dir || 'app/models'] end |
.root_dir ⇒ Object
101 102 103 104 105 106 107 108 109 |
# File 'lib/annotate/annotate_models.rb', line 101 def root_dir if @root_dir.blank? [''] elsif @root_dir.is_a?(String) @root_dir.split(',') else @root_dir end end |
Class Method Details
.annotate(klass, file, header, options = {}) ⇒ Object
Given the name of an ActiveRecord class, create a schema info block (basically a comment containing information on the columns and their types) and put it at the front of the model and fixture source files.
Options (opts)
:position_in_class<Symbol>:: where to place the annotated section in model file
:position_in_test<Symbol>:: where to place the annotated section in test/spec file(s)
:position_in_fixture<Symbol>:: where to place the annotated section in fixture file
:position_in_factory<Symbol>:: where to place the annotated section in factory file
:position_in_serializer<Symbol>:: where to place the annotated section in serializer file
:exclude_tests<Symbol>:: whether to skip modification of test/spec files
:exclude_fixtures<Symbol>:: whether to skip modification of fixture files
:exclude_factories<Symbol>:: whether to skip modification of factory files
:exclude_serializers<Symbol>:: whether to skip modification of serializer files
:exclude_scaffolds<Symbol>:: whether to skip modification of scaffold files
:exclude_controllers<Symbol>:: whether to skip modification of controller files
:exclude_helpers<Symbol>:: whether to skip modification of helper files
:exclude_sti_subclasses<Symbol>:: whether to skip modification of files for STI subclasses
Returns:
an array of file names that were annotated.
618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 |
# File 'lib/annotate/annotate_models.rb', line 618 def annotate(klass, file, header, = {}) begin klass.reset_column_information info = get_schema_info(klass, header, ) model_name = klass.name.underscore table_name = klass.table_name model_file_name = File.join(file) annotated = [] if annotate_one_file(model_file_name, info, :position_in_class, (, :position_in_class)) annotated << model_file_name end matched_types().each do |key| exclusion_key = "exclude_#{key.pluralize}".to_sym position_key = "position_in_#{key}".to_sym # Same options for active_admin models if key == 'admin' exclusion_key = 'exclude_class'.to_sym position_key = 'position_in_class'.to_sym end next if [exclusion_key] get_patterns(, key) .map { |f| resolve_filename(f, model_name, table_name) } .map { |f| (f) } .flatten .each do |f| if annotate_one_file(f, info, position_key, (, position_key)) annotated << f end end end rescue StandardError => e $stderr.puts "Unable to annotate #{file}: #{e.}" $stderr.puts "\t" + e.backtrace.join("\n\t") if [:trace] end annotated end |
.annotate_model_file(annotated, file, header, options) ⇒ Object
808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 |
# File 'lib/annotate/annotate_models.rb', line 808 def annotate_model_file(annotated, file, header, ) begin return false if /#{SKIP_ANNOTATION_PREFIX}.*/ =~ (File.exist?(file) ? File.read(file) : '') klass = get_model_class(file) do_annotate = klass && klass < ActiveRecord::Base && (![:exclude_sti_subclasses] || !(klass.superclass < ActiveRecord::Base && klass.table_name == klass.superclass.table_name)) && !klass.abstract_class? && klass.table_exists? annotated.concat(annotate(klass, file, header, )) if do_annotate rescue BadModelFileError => e unless [:ignore_unknown_models] $stderr.puts "Unable to annotate #{file}: #{e.}" $stderr.puts "\t" + e.backtrace.join("\n\t") if [:trace] end rescue StandardError => e $stderr.puts "Unable to annotate #{file}: #{e.}" $stderr.puts "\t" + e.backtrace.join("\n\t") if [:trace] end end |
.annotate_one_file(file_name, info_block, position, options = {}) ⇒ Object
Add a schema block to a file. If the file already contains a schema info block (a comment starting with “== Schema Information”), check if it matches the block that is already there. If so, leave it be. If not, remove the old info block and write a new one.
Returns:
true or false depending on whether the file was modified.
Options (opts)
:force<Symbol>:: whether to update the file even if it doesn't seem to need it.
:position_in_*<Symbol>:: where to place the annotated section in fixture or model file,
:before, :top, :after or :bottom. Default is :before.
508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 |
# File 'lib/annotate/annotate_models.rb', line 508 def annotate_one_file(file_name, info_block, position, = {}) return false unless File.exist?(file_name) old_content = File.read(file_name) return false if old_content =~ /#{SKIP_ANNOTATION_PREFIX}.*\n/ # Ignore the Schema version line because it changes with each migration header_pattern = /(^# Table name:.*?\n(#.*[\r]?\n)*[\r]?)/ old_header = old_content.match(header_pattern).to_s new_header = info_block.match(header_pattern).to_s column_pattern = /^#[\t ]+[\w\*\.`]+[\t ]+.+$/ old_columns = old_header && old_header.scan(column_pattern).sort new_columns = new_header && new_header.scan(column_pattern).sort return false if old_columns == new_columns && ![:force] abort "annotate error. #{file_name} needs to be updated, but annotate was run with `--frozen`." if [:frozen] # Replace inline the old schema info with the new schema info wrapper_open = [:wrapper_open] ? "# #{[:wrapper_open]}\n" : "" wrapper_close = [:wrapper_close] ? "# #{[:wrapper_close]}\n" : "" wrapped_info_block = "#{wrapper_open}#{info_block}#{wrapper_close}" old_annotation = old_content.match(annotate_pattern()).to_s # if there *was* no old schema info or :force was passed, we simply # need to insert it in correct position if old_annotation.empty? || [:force] magic_comments_block = magic_comments_as_string(old_content) old_content.gsub!(MAGIC_COMMENT_MATCHER, '') old_content.sub!(annotate_pattern(), '') new_content = if %w(after bottom).include?([position].to_s) magic_comments_block + (old_content.rstrip + "\n\n" + wrapped_info_block) elsif magic_comments_block.empty? magic_comments_block + wrapped_info_block + old_content.lstrip else magic_comments_block + "\n" + wrapped_info_block + old_content.lstrip end else # replace the old annotation with the new one # keep the surrounding whitespace the same space_match = old_annotation.match(/\A(?<start>\s*).*?\n(?<end>\s*)\z/m) new_annotation = space_match[:start] + wrapped_info_block + space_match[:end] new_content = old_content.sub(annotate_pattern(), new_annotation) end File.open(file_name, 'wb') { |f| f.puts new_content } true end |
.annotate_pattern(options = {}) ⇒ Object
88 89 90 91 92 93 |
# File 'lib/annotate/annotate_models.rb', line 88 def annotate_pattern( = {}) if [:wrapper_open] return /(?:^(\n|\r\n)?# (?:#{[:wrapper_open]}).*(\n|\r\n)?# (?:#{COMPAT_PREFIX}|#{COMPAT_PREFIX_MD}).*?(\n|\r\n)(#.*(\n|\r\n))*(\n|\r\n)*)|^(\n|\r\n)?# (?:#{COMPAT_PREFIX}|#{COMPAT_PREFIX_MD}).*?(\n|\r\n)(#.*(\n|\r\n))*(\n|\r\n)*/ end /^(\n|\r\n)?# (?:#{COMPAT_PREFIX}|#{COMPAT_PREFIX_MD}).*?(\n|\r\n)(#.*(\n|\r\n))*(\n|\r\n)*/ end |
.classified_sort(cols) ⇒ Object
870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 |
# File 'lib/annotate/annotate_models.rb', line 870 def classified_sort(cols) rest_cols = [] = [] associations = [] id = nil cols.each do |c| if c.name.eql?('id') id = c elsif c.name.eql?('created_at') || c.name.eql?('updated_at') << c elsif c.name[-3, 3].eql?('_id') associations << c else rest_cols << c end end [rest_cols, , associations].each { |a| a.sort_by!(&:name) } ([id] << rest_cols << << associations).flatten.compact end |
.do_annotations(options = {}) ⇒ Object
We’re passed a name of things that might be ActiveRecord models. If we can find the class, and if its a subclass of ActiveRecord::Base, then pass it to the associated block
783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 |
# File 'lib/annotate/annotate_models.rb', line 783 def do_annotations( = {}) () header = [:format_markdown] ? PREFIX_MD.dup : PREFIX.dup version = ActiveRecord::Migrator.current_version rescue 0 if [:include_version] && version > 0 header << "\n# Schema version: #{version}" end annotated = [] get_model_files().each do |path, filename| annotate_model_file(annotated, File.join(path, filename), header, ) end if annotated.empty? puts 'Model files unchanged.' else puts "Annotated (#{annotated.length}): #{annotated.join(', ')}" end end |
.expand_glob_into_files(glob) ⇒ Object
804 805 806 |
# File 'lib/annotate/annotate_models.rb', line 804 def (glob) Dir.glob(glob) end |
.factory_files(root_directory) ⇒ Object
139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 |
# File 'lib/annotate/annotate_models.rb', line 139 def factory_files(root_directory) [ File.join(root_directory, EXEMPLARS_TEST_DIR, "%MODEL_NAME%_exemplar.rb"), File.join(root_directory, EXEMPLARS_SPEC_DIR, "%MODEL_NAME%_exemplar.rb"), File.join(root_directory, BLUEPRINTS_TEST_DIR, "%MODEL_NAME%_blueprint.rb"), File.join(root_directory, BLUEPRINTS_SPEC_DIR, "%MODEL_NAME%_blueprint.rb"), File.join(root_directory, FACTORY_BOT_TEST_DIR, "%MODEL_NAME%_factory.rb"), # (old style) File.join(root_directory, FACTORY_BOT_SPEC_DIR, "%MODEL_NAME%_factory.rb"), # (old style) File.join(root_directory, FACTORY_BOT_TEST_DIR, "%TABLE_NAME%.rb"), # (new style) File.join(root_directory, FACTORY_BOT_SPEC_DIR, "%TABLE_NAME%.rb"), # (new style) File.join(root_directory, FACTORY_BOT_TEST_DIR, "%PLURALIZED_MODEL_NAME%.rb"), # (new style) File.join(root_directory, FACTORY_BOT_SPEC_DIR, "%PLURALIZED_MODEL_NAME%.rb"), # (new style) File.join(root_directory, FABRICATORS_TEST_DIR, "%MODEL_NAME%_fabricator.rb"), File.join(root_directory, FABRICATORS_SPEC_DIR, "%MODEL_NAME%_fabricator.rb") ] end |
.files_by_pattern(root_directory, pattern_type, options) ⇒ Object
164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 |
# File 'lib/annotate/annotate_models.rb', line 164 def files_by_pattern(root_directory, pattern_type, ) case pattern_type when 'test' then test_files(root_directory) when 'fixture' then fixture_files(root_directory) when 'scaffold' then scaffold_files(root_directory) when 'factory' then factory_files(root_directory) when 'serializer' then serialize_files(root_directory) when 'additional_file_patterns' [[:additional_file_patterns] || []].flatten when 'controller' [File.join(root_directory, CONTROLLER_DIR, "%PLURALIZED_MODEL_NAME%_controller.rb")] when 'admin' [File.join(root_directory, ACTIVEADMIN_DIR, "%MODEL_NAME%.rb")] when 'helper' [File.join(root_directory, HELPER_DIR, "%PLURALIZED_MODEL_NAME%_helper.rb")] else [] end end |
.final_index_string(index, max_size) ⇒ Object
426 427 428 429 430 431 432 433 434 435 |
# File 'lib/annotate/annotate_models.rb', line 426 def final_index_string(index, max_size) sprintf( "# %-#{max_size}.#{max_size}s %s%s%s%s", index.name, "(#{index_columns_info(index).join(',')})", index_unique_info(index), index_where_info(index), index_using_info(index) ).rstrip + "\n" end |
.final_index_string_in_markdown(index) ⇒ Object
409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 |
# File 'lib/annotate/annotate_models.rb', line 409 def final_index_string_in_markdown(index) details = sprintf( "%s%s%s", index_unique_info(index, :markdown), index_where_info(index, :markdown), index_using_info(index, :markdown) ).strip details = " (#{details})" unless details.blank? sprintf( "# * `%s`%s:\n# * **`%s`**\n", index.name, details, index_columns_info(index).join("`**\n# * **`") ) end |
.fixture_files(root_directory) ⇒ Object
121 122 123 124 125 126 127 128 |
# File 'lib/annotate/annotate_models.rb', line 121 def fixture_files(root_directory) [ File.join(root_directory, FIXTURE_TEST_DIR, "%TABLE_NAME%.yml"), File.join(root_directory, FIXTURE_SPEC_DIR, "%TABLE_NAME%.yml"), File.join(root_directory, FIXTURE_TEST_DIR, "%PLURALIZED_MODEL_NAME%.yml"), File.join(root_directory, FIXTURE_SPEC_DIR, "%PLURALIZED_MODEL_NAME%.yml") ] end |
.get_col_type(col) ⇒ Object
369 370 371 372 373 374 375 |
# File 'lib/annotate/annotate_models.rb', line 369 def get_col_type(col) if (col.respond_to?(:bigint?) && col.bigint?) || /\Abigint\b/ =~ col.sql_type 'bigint' else (col.type || col.sql_type).to_s end end |
.get_foreign_key_info(klass, options = {}) ⇒ Object
459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 |
# File 'lib/annotate/annotate_models.rb', line 459 def get_foreign_key_info(klass, = {}) fk_info = if [:format_markdown] "#\n# ### Foreign Keys\n#\n" else "#\n# Foreign Keys\n#\n" end return '' unless klass.connection.respond_to?(:supports_foreign_keys?) && klass.connection.supports_foreign_keys? && klass.connection.respond_to?(:foreign_keys) foreign_keys = klass.connection.foreign_keys(klass.table_name) return '' if foreign_keys.empty? format_name = lambda do |fk| return fk.[:column] if fk.name.blank? [:show_complete_foreign_keys] ? fk.name : fk.name.gsub(/(?<=^fk_rails_)[0-9a-f]{10}$/, '...') end max_size = foreign_keys.map(&format_name).map(&:size).max + 1 foreign_keys.sort_by {|fk| [format_name.call(fk), fk.column]}.each do |fk| ref_info = "#{fk.column} => #{fk.to_table}.#{fk.primary_key}" constraints_info = '' constraints_info += "ON DELETE => #{fk.on_delete} " if fk.on_delete constraints_info += "ON UPDATE => #{fk.on_update} " if fk.on_update constraints_info.strip! fk_info << if [:format_markdown] sprintf("# * `%s`%s:\n# * **`%s`**\n", format_name.call(fk), constraints_info.blank? ? '' : " (_#{constraints_info}_)", ref_info) else sprintf("# %-#{max_size}.#{max_size}s %s %s", format_name.call(fk), "(#{ref_info})", constraints_info).rstrip + "\n" end end fk_info end |
.get_index_info(klass, options = {}) ⇒ Object
347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 |
# File 'lib/annotate/annotate_models.rb', line 347 def get_index_info(klass, = {}) index_info = if [:format_markdown] "#\n# ### Indexes\n#\n" else "#\n# Indexes\n#\n" end indexes = retrieve_indexes_from_table(klass) return '' if indexes.empty? max_size = indexes.collect{|index| index.name.size}.max + 1 indexes.sort_by(&:name).each do |index| index_info << if [:format_markdown] final_index_string_in_markdown(index) else final_index_string(index, max_size) end end index_info end |
.get_loaded_model(model_path, file) ⇒ Object
Retrieve loaded model class
741 742 743 744 745 746 747 748 749 750 751 752 753 754 |
# File 'lib/annotate/annotate_models.rb', line 741 def get_loaded_model(model_path, file) loaded_model_class = get_loaded_model_by_path(model_path) return loaded_model_class if loaded_model_class # We cannot get loaded model when `model_path` is loaded by Rails # auto_load/eager_load paths. Try all possible model paths one by one. absolute_file = File.(file) model_paths = $LOAD_PATH.select { |path| absolute_file.include?(path) } .map { |path| absolute_file.sub(path, '').sub(/\.rb$/, '').sub(/^\//, '') } model_paths .map { |path| get_loaded_model_by_path(path) } .find { |loaded_model| !loaded_model.nil? } end |
.get_loaded_model_by_path(model_path) ⇒ Object
Retrieve loaded model class by path to the file where it’s supposed to be defined.
757 758 759 760 761 762 763 764 765 766 767 |
# File 'lib/annotate/annotate_models.rb', line 757 def get_loaded_model_by_path(model_path) ActiveSupport::Inflector.constantize(ActiveSupport::Inflector.camelize(model_path)) rescue StandardError, LoadError # Revert to the old way but it is not really robust ObjectSpace.each_object(::Class) .select do |c| Class === c && # note: we use === to avoid a bug in activesupport 2.3.14 OptionMerger vs. is_a? c.ancestors.respond_to?(:include?) && # to fix FactoryGirl bug, see https://github.com/ctran/annotate_models/pull/82 c.ancestors.include?(ActiveRecord::Base) end.detect { |c| ActiveSupport::Inflector.underscore(c.to_s) == model_path } end |
.get_model_class(file) ⇒ Object
Retrieve the classes belonging to the model names we’re asked to process Check for namespaced models in subdirectories as well as models in subdirectories without namespacing.
721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 |
# File 'lib/annotate/annotate_models.rb', line 721 def get_model_class(file) model_path = file.gsub(/\.rb$/, '') model_dir.each { |dir| model_path = model_path.gsub(/^#{dir}/, '').gsub(/^\//, '') } begin get_loaded_model(model_path, file) || raise(BadModelFileError.new) rescue LoadError # this is for non-rails projects, which don't get Rails auto-require magic file_path = File.(file) if File.file?(file_path) && Kernel.require(file_path) retry elsif model_path =~ /\// model_path = model_path.split('/')[1..-1].join('/').to_s retry else raise end end end |
.get_model_files(options) ⇒ Object
Return a list of the model files to annotate. If we have command line arguments, they’re assumed to the path of model files from root dir. Otherwise we take all the model files in the model_dir directory.
670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 |
# File 'lib/annotate/annotate_models.rb', line 670 def get_model_files() model_files = [] model_files = list_model_files_from_argument unless [:is_rake] return model_files unless model_files.empty? model_dir.each do |dir| Dir.chdir(dir) do list = if [:ignore_model_sub_dir] Dir["*.rb"].map { |f| [dir, f] } else Dir["**/*.rb"].reject { |f| f["concerns/"] }.map { |f| [dir, f] } end model_files.concat(list) end end model_files rescue SystemCallError $stderr.puts "No models found in directory '#{model_dir.join("', '")}'." $stderr.puts "Either specify models on the command line, or use the --model-dir option." $stderr.puts "Call 'annotate --help' for more info." exit 1 end |
.get_patterns(options, pattern_types = []) ⇒ Object
184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 |
# File 'lib/annotate/annotate_models.rb', line 184 def get_patterns(, pattern_types = []) current_patterns = [] root_dir.each do |root_directory| Array(pattern_types).each do |pattern_type| patterns = files_by_pattern(root_directory, pattern_type, ) current_patterns += if pattern_type.to_sym == :additional_file_patterns patterns else patterns.map { |p| p.sub(/^[\/]*/, '') } end end end current_patterns end |
.get_schema_footer_text(_klass, options = {}) ⇒ Object
336 337 338 339 340 341 342 343 344 345 |
# File 'lib/annotate/annotate_models.rb', line 336 def (_klass, = {}) info = '' if [:format_rdoc] info << "#--\n" info << "# #{END_MARK}\n" info << "#++\n" else info << "#\n" end end |
.get_schema_header_text(klass, options = {}) ⇒ Object
324 325 326 327 328 329 330 331 332 333 334 |
# File 'lib/annotate/annotate_models.rb', line 324 def get_schema_header_text(klass, = {}) info = "#\n" if [:format_markdown] info << "# Table name: `#{klass.table_name}`\n" info << "#\n" info << "# ### Columns\n" else info << "# Table name: #{klass.table_name}\n" end info << "#\n" end |
.get_schema_info(klass, header, options = {}) ⇒ Object
Use the column information in an ActiveRecord class to create a comment block containing a line for each column. The line contains the column name, the type (and length), and any optional attributes
235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 |
# File 'lib/annotate/annotate_models.rb', line 235 def get_schema_info(klass, header, = {}) info = "# #{header}\n" info << get_schema_header_text(klass, ) max_size = max_schema_info_width(klass, ) md_names_overhead = 6 md_type_allowance = 18 = 16 if [:format_markdown] info << sprintf( "# %-#{max_size + md_names_overhead}.#{max_size + md_names_overhead}s | %-#{md_type_allowance}.#{md_type_allowance}s | %s\n", 'Name', 'Type', 'Attributes' ) info << "# #{ '-' * ( max_size + md_names_overhead ) } | #{'-' * md_type_allowance} | #{ '-' * 27 }\n" end cols = columns(klass, ) cols.each do |col| col_type = get_col_type(col) attrs = [] attrs << "default(#{schema_default(klass, col)})" unless col.default.nil? || hide_default?(col_type, ) attrs << 'unsigned' if col.respond_to?(:unsigned?) && col.unsigned? attrs << 'not null' unless col.null attrs << 'primary key' if klass.primary_key && (klass.primary_key.is_a?(Array) ? klass.primary_key.collect(&:to_sym).include?(col.name.to_sym) : col.name.to_sym == klass.primary_key.to_sym) if col_type == 'decimal' col_type << "(#{col.precision}, #{col.scale})" elsif !%w[spatial geometry geography].include?(col_type) if col.limit && ![:format_yard] if col.limit.is_a? Array attrs << "(#{col.limit.join(', ')})" else col_type << "(#{col.limit})" unless hide_limit?(col_type, ) end end end # Check out if we got an array column attrs << 'is an Array' if col.respond_to?(:array) && col.array # Check out if we got a geometric column # and print the type and SRID if col.respond_to?(:geometry_type) attrs << "#{col.geometry_type}, #{col.srid}" elsif col.respond_to?(:geometric_type) && col.geometric_type.present? attrs << "#{col.geometric_type.to_s.downcase}, #{col.srid}" end # Check if the column has indices and print "indexed" if true # If the index includes another column, print it too. if [:simple_indexes] && klass.table_exists?# Check out if this column is indexed indices = retrieve_indexes_from_table(klass) if indices = indices.select { |ind| ind.columns.include? col.name } indices.sort_by(&:name).each do |ind| next if ind.columns.is_a?(String) ind = ind.columns.reject! { |i| i == col.name } attrs << (ind.empty? ? "indexed" : "indexed => [#{ind.join(", ")}]") end end end col_name = if with_comments?(klass, ) && col.comment "#{col.name}(#{col.comment})" else col.name end if [:format_rdoc] info << sprintf("# %-#{max_size}.#{max_size}s<tt>%s</tt>", "*#{col_name}*::", attrs.unshift(col_type).join(", ")).rstrip + "\n" elsif [:format_yard] info << sprintf("# @!attribute #{col_name}") + "\n" ruby_class = col.respond_to?(:array) && col.array ? "Array<#{map_col_type_to_ruby_classes(col_type)}>": map_col_type_to_ruby_classes(col_type) info << sprintf("# @return [#{ruby_class}]") + "\n" elsif [:format_markdown] name_remainder = max_size - col_name.length - non_ascii_length(col_name) type_remainder = (md_type_allowance - 2) - col_type.length info << (sprintf("# **`%s`**%#{name_remainder}s | `%s`%#{type_remainder}s | `%s`", col_name, " ", col_type, " ", attrs.join(", ").rstrip)).gsub('``', ' ').rstrip + "\n" else info << format_default(col_name, max_size, col_type, , attrs) end end if [:show_indexes] && klass.table_exists? info << get_index_info(klass, ) end if [:show_foreign_keys] && klass.table_exists? info << get_foreign_key_info(klass, ) end info << (klass, ) end |
.hide_default?(col_type, options) ⇒ Boolean
448 449 450 451 452 453 454 455 456 457 |
# File 'lib/annotate/annotate_models.rb', line 448 def hide_default?(col_type, ) excludes = if [:hide_default_column_types].blank? NO_DEFAULT_COL_TYPES else [:hide_default_column_types].split(',') end excludes.include?(col_type) end |
.hide_limit?(col_type, options) ⇒ Boolean
437 438 439 440 441 442 443 444 445 446 |
# File 'lib/annotate/annotate_models.rb', line 437 def hide_limit?(col_type, ) excludes = if [:hide_limit_column_types].blank? NO_LIMIT_COL_TYPES else [:hide_limit_column_types].split(',') end excludes.include?(col_type) end |
.index_columns_info(index) ⇒ Object
377 378 379 380 381 382 383 384 385 |
# File 'lib/annotate/annotate_models.rb', line 377 def index_columns_info(index) Array(index.columns).map do |col| if index.try(:orders) && index.orders[col.to_s] "#{col} #{index.orders[col.to_s].upcase}" else col.to_s.gsub("\r", '\r').gsub("\n", '\n') end end end |
.index_unique_info(index, format = :default) ⇒ Object
387 388 389 |
# File 'lib/annotate/annotate_models.rb', line 387 def index_unique_info(index, format = :default) index.unique ? " #{INDEX_CLAUSES[:unique][format]}" : '' end |
.index_using_info(index, format = :default) ⇒ Object
400 401 402 403 404 405 406 407 |
# File 'lib/annotate/annotate_models.rb', line 400 def index_using_info(index, format = :default) value = index.try(:using) && index.using.try(:to_sym) if !value.blank? && value != :btree " #{INDEX_CLAUSES[:using][format]} #{value}" else '' end end |
.index_where_info(index, format = :default) ⇒ Object
391 392 393 394 395 396 397 398 |
# File 'lib/annotate/annotate_models.rb', line 391 def index_where_info(index, format = :default) value = index.try(:where).try(:to_s) if value.blank? '' else " #{INDEX_CLAUSES[:where][format]} #{value}" end end |
.list_model_files_from_argument ⇒ Object
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 |
# File 'lib/annotate/annotate_models.rb', line 696 def list_model_files_from_argument return [] if ARGV.empty? specified_files = ARGV.map { |file| File.(file) } model_files = model_dir.flat_map do |dir| absolute_dir_path = File.(dir) specified_files .find_all { |file| file.start_with?(absolute_dir_path) } .map { |file| [dir, file.sub("#{absolute_dir_path}/", '')] } end if model_files.size != specified_files.size puts "The specified file could not be found in directory '#{model_dir.join("', '")}'." puts "Call 'annotate --help' for more info." exit 1 end model_files end |
.magic_comments_as_string(content) ⇒ Object
561 562 563 564 565 566 567 568 569 |
# File 'lib/annotate/annotate_models.rb', line 561 def magic_comments_as_string(content) magic_comments = content.scan(MAGIC_COMMENT_MATCHER).flatten.compact if magic_comments.any? magic_comments.join else '' end end |
.matched_types(options) ⇒ Object
587 588 589 590 591 592 593 |
# File 'lib/annotate/annotate_models.rb', line 587 def matched_types() types = MATCHED_TYPES.dup types << 'admin' if [:active_admin] =~ Annotate::Constants::TRUE_RE && !types.include?('admin') types << 'additional_file_patterns' if [:additional_file_patterns].present? types end |
.options_with_position(options, position_in) ⇒ Object
position = :position_in_fixture or :position_in_class
662 663 664 |
# File 'lib/annotate/annotate_models.rb', line 662 def (, position_in) .merge(position: ([position_in] || [:position])) end |
.parse_options(options = {}) ⇒ Object
769 770 771 772 |
# File 'lib/annotate/annotate_models.rb', line 769 def ( = {}) self.model_dir = split_model_dir([:model_dir]) if [:model_dir] self.root_dir = [:root_dir] if [:root_dir] end |
.quote(value) ⇒ Object
Simple quoting for the default column value
201 202 203 204 205 206 207 208 209 210 211 212 213 |
# File 'lib/annotate/annotate_models.rb', line 201 def quote(value) case value when NilClass then 'NULL' when TrueClass then 'TRUE' when FalseClass then 'FALSE' when Float, Integer then value.to_s # BigDecimals need to be output in a non-normalized form and quoted. when BigDecimal then value.to_s('F') when Array then value.map { |v| quote(v) } else value.inspect end end |
.remove_annotation_of_file(file_name, options = {}) ⇒ Object
571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 |
# File 'lib/annotate/annotate_models.rb', line 571 def remove_annotation_of_file(file_name, = {}) if File.exist?(file_name) content = File.read(file_name) return false if content =~ /#{SKIP_ANNOTATION_PREFIX}.*\n/ wrapper_open = [:wrapper_open] ? "# #{[:wrapper_open]}\n" : '' content.sub!(/(#{wrapper_open})?#{annotate_pattern()}/, '') File.open(file_name, 'wb') { |f| f.puts content } true else false end end |
.remove_annotations(options = {}) ⇒ Object
830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 |
# File 'lib/annotate/annotate_models.rb', line 830 def remove_annotations( = {}) () deannotated = [] deannotated_klass = false get_model_files().each do |file| file = File.join(file) begin klass = get_model_class(file) if klass < ActiveRecord::Base && !klass.abstract_class? model_name = klass.name.underscore table_name = klass.table_name model_file_name = file deannotated_klass = true if remove_annotation_of_file(model_file_name, ) get_patterns(, matched_types()) .map { |f| resolve_filename(f, model_name, table_name) } .each do |f| if File.exist?(f) remove_annotation_of_file(f, ) deannotated_klass = true end end end deannotated << klass if deannotated_klass rescue StandardError => e $stderr.puts "Unable to deannotate #{File.join(file)}: #{e.}" $stderr.puts "\t" + e.backtrace.join("\n\t") if [:trace] end end puts "Removed annotations from: #{deannotated.join(', ')}" end |
.resolve_filename(filename_template, model_name, table_name) ⇒ Object
863 864 865 866 867 868 |
# File 'lib/annotate/annotate_models.rb', line 863 def resolve_filename(filename_template, model_name, table_name) filename_template .gsub('%MODEL_NAME%', model_name) .gsub('%PLURALIZED_MODEL_NAME%', model_name.pluralize) .gsub('%TABLE_NAME%', table_name || model_name.pluralize) end |
.retrieve_indexes_from_table(klass) ⇒ Object
219 220 221 222 223 224 225 226 227 228 229 |
# File 'lib/annotate/annotate_models.rb', line 219 def retrieve_indexes_from_table(klass) table_name = klass.table_name return [] unless table_name indexes = klass.connection.indexes(table_name) return indexes if indexes.any? || !klass.table_name_prefix # Try to search the table without prefix table_name_without_prefix = table_name.to_s.sub(klass.table_name_prefix, '') klass.connection.indexes(table_name_without_prefix) end |
.scaffold_files(root_directory) ⇒ Object
130 131 132 133 134 135 136 137 |
# File 'lib/annotate/annotate_models.rb', line 130 def scaffold_files(root_directory) [ File.join(root_directory, CONTROLLER_TEST_DIR, "%PLURALIZED_MODEL_NAME%_controller_test.rb"), File.join(root_directory, CONTROLLER_SPEC_DIR, "%PLURALIZED_MODEL_NAME%_controller_spec.rb"), File.join(root_directory, REQUEST_SPEC_DIR, "%PLURALIZED_MODEL_NAME%_spec.rb"), File.join(root_directory, ROUTING_SPEC_DIR, "%PLURALIZED_MODEL_NAME%_routing_spec.rb") ] end |
.schema_default(klass, column) ⇒ Object
215 216 217 |
# File 'lib/annotate/annotate_models.rb', line 215 def schema_default(klass, column) quote(klass.column_defaults[column.name]) end |
.serialize_files(root_directory) ⇒ Object
156 157 158 159 160 161 162 |
# File 'lib/annotate/annotate_models.rb', line 156 def serialize_files(root_directory) [ File.join(root_directory, SERIALIZERS_DIR, "%MODEL_NAME%_serializer.rb"), File.join(root_directory, SERIALIZERS_TEST_DIR, "%MODEL_NAME%_serializer_test.rb"), File.join(root_directory, SERIALIZERS_SPEC_DIR, "%MODEL_NAME%_serializer_spec.rb") ] end |
.split_model_dir(option_value) ⇒ Object
774 775 776 777 |
# File 'lib/annotate/annotate_models.rb', line 774 def split_model_dir(option_value) option_value = option_value.is_a?(Array) ? option_value : option_value.split(',') option_value.map(&:strip).reject(&:empty?) end |
.test_files(root_directory) ⇒ Object
113 114 115 116 117 118 119 |
# File 'lib/annotate/annotate_models.rb', line 113 def test_files(root_directory) [ File.join(root_directory, UNIT_TEST_DIR, "%MODEL_NAME%_test.rb"), File.join(root_directory, MODEL_TEST_DIR, "%MODEL_NAME%_test.rb"), File.join(root_directory, SPEC_MODEL_DIR, "%MODEL_NAME%_spec.rb") ] end |