Sha256: ebf5dce4acadb1165c63b9da8fc707fd16f516d285006ea1512842463347356d

Contents?: true

Size: 1.42 KB

Versions: 6

Compression:

Stored size: 1.42 KB

Contents

module ActiveEnum

  module ActsAsEnum

    def self.included(base)
      base.extend MacroMethods
    end

    module MacroMethods

      def acts_as_enum(options={})
        extend ClassMethods
        class_inheritable_accessor :active_enum_options
        self.active_enum_options = options.reverse_merge(:name_column => 'name')
        scope :enum_values, select("#{primary_key}, #{active_enum_options[:name_column]}").
                            where(active_enum_options[:conditions]).
                            order("#{primary_key} #{active_enum_options[:order]}")
      end

    end

    module ClassMethods

      def ids
        enum_values.map {|v| v.id }
      end

      def names
        enum_values.map {|v| v.send(active_enum_options[:name_column]) }
      end

      def to_select
        enum_values.map {|v| [v.send(active_enum_options[:name_column]), v.id] }
      end

      def [](index)
        if index.is_a?(Fixnum)
          v = lookup_by_id(index)
          v.send(active_enum_options[:name_column]) unless v.blank?
        else
          v = lookup_by_name(index)
          v.id unless v.blank?
        end
      end

      private

      def lookup_by_id(index)
        enum_values.find_by_id(index)
      end

      def lookup_by_name(index)
        enum_values.where("#{active_enum_options[:name_column]} like ?", index.to_s).first
      end

    end

  end
end

ActiveRecord::Base.send :include, ActiveEnum::ActsAsEnum

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
active_enum-0.9.0 lib/active_enum/acts_as_enum.rb
active_enum-0.8.2 lib/active_enum/acts_as_enum.rb
active_enum-0.8.1 lib/active_enum/acts_as_enum.rb
active_enum-0.8.0 lib/active_enum/acts_as_enum.rb
active_enum-0.7.1 lib/active_enum/acts_as_enum.rb
active_enum-0.7.0 lib/active_enum/acts_as_enum.rb