enum_attr 1.0.0 → 1.0.1

data/VERSION

@@ -1 +1 @@
-1.0.0
+1.0.1

data/enum_attr.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{enum_attr}
-  s.version = "1.0.0"
+  s.version = "1.0.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["qichunren"]
@@ -25,6 +25,8 @@ Gem::Specification.new do |s|
      "enum_attr.gemspec",
      "init.rb",
      "lib/enum_attr.rb",
+     "lib/enum_attr_base.rb",
+     "lib/enum_attr_for_active_record.rb",
      "rails/init.rb"
   ]
   s.homepage = %q{https://github.com/qichunren/enum_attr}

data/lib/enum_attr_base.rb

@@ -0,0 +1,44 @@
+# encoding: utf-8
+module EnumAttr
+  module Mixin
+
+    def enum_attr(attr, enums)
+      raise "The secone param must be a array!" unless enums.is_a? Array
+
+      if self.superclass == Object
+        attr_accessor attr.to_sym
+      end
+
+      attr = attr.to_s
+
+      enums.each do |enum|
+        const_set("#{attr.upcase}_#{enum[2].upcase}", enum[1] )
+
+        class_eval(%Q{
+          def #{attr}_#{enum[2]}?
+            #{attr} == #{enum[1]}
+          end
+        })
+
+      end # end: enums.each
+
+      self.class_eval(%Q{
+
+ENUMS_#{attr.upcase} = enums.collect{|item| [item[0], item[1]]}
+
+def self.#{attr}_name_by(arg)
+  ENUMS_#{attr.upcase}.find{|option| option[1] == arg }[0] rescue ""
+end
+
+def #{attr}_name
+  ENUMS_#{attr.upcase}.find{|option| option[1] == #{attr}}[0] rescue "" unless #{attr}.nil?
+end
+
+})
+    end
+
+
+  end
+end
+
+Object.send :extend, EnumAttr::Mixin

data/lib/enum_attr_for_active_record.rb

@@ -0,0 +1,31 @@
+# encoding: utf-8
+require 'enum_attr_base'
+require "active_record"
+
+module EnumAttr
+  module MixinForActiveRecord
+    include EnumAttr::Mixin
+
+    def enum_attr(attr, enums)
+      super(attr, enums)
+      enums.each do |enum|
+        scope "#{attr}_#{enum[2]}".to_sym, where("#{attr}=#{enum[1]}")
+      end # end: enums.each
+
+      self.class_eval(%Q{
+validates_inclusion_of attr, :in => enums.map{|e| e[1].to_i}, :allow_nil => true
+def attr
+  read_attribute(attr.to_sym)
+end
+
+def attr=(value)
+  write_attribute(attr.to_sym, value)
+end
+})
+
+    end # end def
+  end
+end
+
+
+::ActiveRecord::Base.send :extend, EnumAttr::MixinForActiveRecord

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: enum_attr
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
   prerelease: 
 platform: ruby
 authors:
@@ -28,6 +28,8 @@ files:
 - enum_attr.gemspec
 - init.rb
 - lib/enum_attr.rb
+- lib/enum_attr_base.rb
+- lib/enum_attr_for_active_record.rb
 - rails/init.rb
 - spec/enum_spec.rb
 - spec/spec_helper.rb