A string-backed Enum type for DataMapper

dm_string_enum.rb

require 'dm-types/support/flags'

module DataMapper
  class Property
    class StringEnum < String
      min 1

      include Flags

      def initialize(model, name, options = {})
        flags.each do |flag|
          fail "Found flag of type #{flag.class}. Only symbols are allowed."  unless flag.is_a? Symbol
        end

        super
      end

      def load(value)
        value.to_sym
      end

      def dump(value)
        value.to_s
      end

      def typecast_to_primitive(value)
        value.to_sym
        fail "#{value} is not a valid option"  unless flags.include? value
      end

    end # class Enum
  end # class Property
end # module DataMapper