Contents

module BelongsToEnum
  class Enum
    def Enum.add_item(key,value)
      @hash ||= {}
      @hash[key]=value
    end

    def Enum.method_missing(key, *args)
      @hash[key] || super
    end
    
    def Enum.respond_to?(key)
      return true if super(key)
      return @hash && @hash[key]
    end

    def Enum.each
      @hash.each {|key,value| yield(key,value)}
    end

    def Enum.map
      @hash.map {|key,value| yield(key,value)}
    end

    def Enum.keys
      @hash.values
    end

    def Enum.items_for_select
      @hash.sort{|a,b| a[1]<=>b[1]}.map{|i|[i[0].to_s.gsub("_", " ").capitalize,i[1]]}
    end
    
    def Enum.items
      warn "DEPRECIATED belongs_to_enum: Use items_for_select instead"
      items_for_select
    end

    def Enum.get(value)
      @hash.key(value)
    end

    def Enum.display(value)
      get(value).to_s.gsub("_", " ").titleize
    end

    def Enum.raw_hash
      @hash
    end

    def Enum.valid?(value)
      @hash.has_value?(value.to_i)
    end

    def Enum.default
      @hash.values.first
    end

    def self.create(name, parent, keys, options = {})
      
      # Check to see if this has already been defined...
      return if parent.const_defined?(name.to_s.camelize)

      klass = Class.new(Enum)
      parent.const_set(name.to_s.camelize, klass)

      if keys.respond_to? :raw_hash
        keys.raw_hash.each do |k,v|
          klass.add_item(k,v)
        end

      elsif(keys.is_a? Hash)
        keys.each do |k,v|
          klass.add_item(k,v)
        end
      else
        keys.each_with_index do |key, index|
          klass.add_item(key, index + 1)
        end
      end
      
      # Create scopes if required
      if options[:create_scopes]
        klass.raw_hash.each do |key, value|
          parent.instance_exec do
            scope key.to_s, where("#{name.to_s.underscore}_id = #{value}")
          end
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
belongs_to_enum-0.4.0	lib/belongs_to_enum/enum.rb
belongs_to_enum-0.3.1	lib/belongs_to_enum/enum.rb