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,42 @@ 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
+
+      if ignore_columns = options[:ignore_columns]
+        cols = cols.reject do |col|
+          col.name.match(/#{ignore_columns}/)
+        end
+      end
+
+      # Add columns managed by the globalize gem
+      if klass.respond_to? :translation_class
+        # These are the columns that Globalize needs to work but
+        # are not necessary for the models to be displayed as annotations
+        ignored_translation_table_colums = [
+          'id',
+          'created_at',
+          'updated_at',
+          'locale',
+          # Construct the foreign column name in the translations table
+          # eg. Model: Car, foreign column name: car_id
+          klass.translation_class.to_s
+               .gsub('::Translation', '').gsub('::', '_')
+               .downcase + '_id'
+        ]
+
+        cols += klass.translation_class.columns.reject do |col|
+          ignored_translation_table_colums.include? col.name
+        end
+      end
+
+      cols = cols.sort_by(&:name) if options[:sort]
+      cols = classified_sort(cols) if options[:classified_sort]
+
+      cols
+    end
   end
 
   class BadModelFileError < LoadError