Add columns managed by Globalize gem

Globalize hooks into the model and removes the translated columns
from the `klass.columns`. This commit checks if globalize is
hooked into the model and adds the necessary columns to the
annotation array.

Author: peterfication

lib/annotate/annotate_models.rb

@@ -244,16 +244,7 @@ def get_schema_info(klass, header, options = {})
         info << "# #{ '-' * ( max_size + md_names_overhead ) } | #{'-' * md_type_allowance} | #{ '-' * 27 }\n"
       end
 
-      cols = if ignore_columns = options[:ignore_columns]
-               klass.columns.reject do |col|
-                 col.name.match(/#{ignore_columns}/)
-               end
-             else
-               klass.columns
-             end
-
-      cols = cols.sort_by(&:name) if options[:sort]
-      cols = classified_sort(cols) if options[:classified_sort]
+      cols = columns(klass, options)
       cols.each do |col|
         col_type = get_col_type(col)
         attrs = []
@@ -905,13 +896,15 @@ def with_comments?(klass, options)
     end
 
     def max_schema_info_width(klass, options)
+      cols = columns(klass, options)
+
       if with_comments?(klass, options)
-        max_size = klass.columns.map do |column|
+        max_size = cols.map do |column|
           column.name.size + (column.comment ? width(column.comment) : 0)
         end.max || 0
         max_size += 2
       else
-        max_size = klass.column_names.map(&:size).max
+        max_size = cols.map(&:name).map(&:size).max
       end
       max_size += options[:format_rdoc] ? 5 : 1
 
@@ -939,6 +932,55 @@ def mb_chars_ljust(string, length)
     def non_ascii_length(string)
       string.to_s.chars.reject(&:ascii_only?).length
     end
+
+    def columns(klass, options)
+      cols = klass.columns
+      cols += translated_columns(klass)
+
+      if ignore_columns = options[:ignore_columns]
+        cols = cols.reject do |col|
+          col.name.match(/#{ignore_columns}/)
+        end
+      end
+
+      cols = cols.sort_by(&:name) if options[:sort]
+      cols = classified_sort(cols) if options[:classified_sort]
+
+      cols
+    end
+
+    ##
+    # Add columns managed by the globalize gem if this gem is being used.
+    def translated_columns(klass)
+      return [] unless klass.respond_to? :translation_class
+
+      ignored_cols = ignored_translation_table_colums(klass)
+      klass.translation_class.columns.reject do |col|
+        ignored_cols.include? col.name.to_sym
+      end
+    end
+
+    ##
+    # These are the columns that the globalize gem needs to work but
+    # are not necessary for the models to be displayed as annotations.
+    def ignored_translation_table_colums(klass)
+      # Construct the foreign column name in the translations table
+      # eg. Model: Car, foreign column name: car_id
+      foreign_column_name = [
+        klass.translation_class.to_s
+             .gsub('::Translation', '').gsub('::', '_')
+             .downcase,
+        '_id'
+      ].join.to_sym
+
+      [
+        :id,
+        :created_at,
+        :updated_at,
+        :locale,
+        foreign_column_name
+      ]
+    end
   end
 
   class BadModelFileError < LoadError

spec/lib/annotate/annotate_models_spec.rb

@@ -788,6 +788,39 @@ def mock_column(name, type, options = {})
 EOS
   end
 
+  it 'should work with the Globalize gem', :focus do
+    klass = mock_class(:posts,
+                       :id,
+                       [
+                         mock_column(:id, :integer, limit: 8),
+                         mock_column(:author_name, :string, limit: 50),
+                       ])
+
+    options = {
+      to_s: 'Post::Translation',
+      columns: [
+        mock_column(:id, :integer, limit: 8),
+        mock_column(:post_id, :integer, limit: 8),
+        mock_column(:locale, :string, limit: 50),
+        mock_column(:title, :string, limit: 50),
+      ]
+    }
+    translation_klass = double('Post::Translation', options)
+    allow(klass).to receive(:translation_class).and_return(translation_klass)
+
+    expected_schema_info = <<~EOS
+      # Schema Info
+      #
+      # Table name: posts
+      #
+      #  id          :integer          not null, primary key
+      #  author_name :string(50)       not null
+      #  title       :string(50)       not null
+      #
+    EOS
+    expect(AnnotateModels.get_schema_info(klass, 'Schema Info')).to eql(expected_schema_info)
+  end
+
   describe '#set_defaults' do
     it 'should default show_complete_foreign_keys to false' do
       expect(Annotate.true?(ENV['show_complete_foreign_keys'])).to be(false)