Contents

module Annotator
  module InitialDescription

    # Initial descriptions for columns associated with belongs_to
    class BelongsTo < Base

      def check
        # check if there is belongs to association where this column is a foreign key
        @model.reflect_on_all_associations.each do |reflection|
          if reflection.foreign_key == @column && reflection.macro == :belongs_to
            @reflection = reflection
            return true
          end
        end

        # Polymorphic association type column
        if @column.ends_with? '_type'
          return true if @reflection = @model.reflect_on_association(@column.match(/(.*?)_type$/)[1].to_sym) 
        end

        return false
      end

      def text
        "belongs to :#{@reflection.name}#{@reflection.options[:polymorphic] ? ' (polymorphic)' : ''}"
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
annotator-0.0.6	lib/annotator/initial_description/belongs_to.rb
annotator-0.0.5	lib/annotator/initial_description/belongs_to.rb