basic version that works with activemodel

Author: lest

.gitignore

@@ -15,3 +15,4 @@ spec/reports
 test/tmp
 test/version_tmp
 tmp
+vendor/bundle

Gemfile

@@ -1,4 +1,5 @@
 source 'https://rubygems.org'
 
-# Specify your gem's dependencies in enumerize.gemspec
 gemspec
+
+gem 'minitest', '~> 2.11.0'

Rakefile

@@ -1,2 +1,11 @@
 #!/usr/bin/env rake
 require "bundler/gem_tasks"
+
+require 'rake/testtask'
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+  t.pattern = 'test/*_test.rb'
+  t.verbose = true
+end
+
+task :default => :test

enumerize.gemspec

@@ -14,4 +14,6 @@ Gem::Specification.new do |gem|
   gem.name          = "enumerize"
   gem.require_paths = ["lib"]
   gem.version       = Enumerize::VERSION
+
+  gem.add_dependency('activesupport', '>= 3.1.3')
 end

lib/enumerize.rb

@@ -1,5 +1,16 @@
-require "enumerize/version"
+require 'active_support/concern'
+require 'i18n'
+require 'enumerize/version'
 
 module Enumerize
-  # Your code goes here...
+  autoload :Attribute, 'enumerize/attribute'
+  autoload :Value,     'enumerize/value'
+
+  extend ActiveSupport::Concern
+
+  module ClassMethods
+    def enumerize(*args, &block)
+      Attribute.new(self, *args).attach!
+    end
+  end
 end

lib/enumerize/attribute.rb

@@ -0,0 +1,45 @@
+module Enumerize
+  class Attribute
+    attr_reader :name, :values
+
+    def initialize(klass, name, options={})
+      raise ArgumentError, ':in option is required' unless options[:in]
+
+      @klass  = klass
+      @name   = name
+      @values = Array(options[:in]).map { |v| Value.new(self, v) }
+    end
+
+    def attach!
+      attr = self
+      @klass.singleton_class.class_eval do
+        define_method("#{attr.name}") { attr }
+      end
+
+      @klass.class_eval <<-RUBY, __FILE__, __LINE__ + 1
+        attr_reader :#{name}
+
+        def #{name}_text
+          #{name}.text
+        end
+
+        def #{name}=(new_value)
+          @#{name} = self.class.#{name}.find_value(new_value)
+        end
+      RUBY
+    end
+
+    def find_value(value)
+      value = value.to_s
+      values.find { |v| v == value }
+    end
+
+    def i18n_suffix
+      @klass.model_name.i18n_key if @klass.respond_to?(:model_name)
+    end
+
+    def options
+      values.map { |v| [v.text, v.to_s] }
+    end
+  end
+end

lib/enumerize/value.rb

@@ -0,0 +1,25 @@
+module Enumerize
+  class Value < String
+    def initialize(attr, value)
+      singleton_class.class_eval do
+        define_method(:attr) { attr }
+      end
+
+      super(value.to_s)
+      freeze
+    end
+
+    def text
+      i18n_keys = ['']
+      i18n_keys.unshift "#{attr.i18n_suffix}." if attr.i18n_suffix
+      i18n_keys.map! { |k| :"enumerize.#{k}#{attr.name}.#{self}" }
+      I18n.t(i18n_keys.shift, :default => i18n_keys)
+    end
+
+    if YAML::ENGINE.yamler == 'syck'
+      def to_yaml(opts={})
+        to_s.to_yaml(opts)
+      end
+    end
+  end
+end

test/basic_test.rb

@@ -0,0 +1,59 @@
+require 'test_helper'
+require 'syck'
+
+describe Enumerize do
+  let(:klass) do
+    Class.new do
+      include Enumerize
+    end
+  end
+
+  let(:object) { klass.new }
+
+  it 'defines method that returns nil' do
+    klass.enumerize(:foo, :in => [:a, :b])
+    object.foo.must_equal nil
+  end
+
+  it 'defines setter method' do
+    klass.enumerize(:foo, :in => [:a, :b])
+    object.foo = :a
+    object.foo.must_equal 'a'
+  end
+
+  it 'returns translation' do
+    I18n.backend.store_translations(:en, :enumerize => {:foo => {:a => 'a text'}})
+    klass.enumerize(:foo, :in => [:a, :b])
+    object.foo = :a
+    object.foo.text.must_equal 'a text'
+    object.foo_text.must_equal 'a text'
+  end
+
+  it 'scopes translation by i18 key' do
+    I18n.backend.store_translations(:en, :enumerize => {:example_class => {:foo => {:a => 'a text scoped'}}})
+    def klass.model_name
+      name = "ExampleClass"
+      def name.i18n_key
+        'example_class'
+      end
+
+      name
+    end
+    klass.enumerize(:foo, :in => [:a, :b])
+    object.foo = :a
+    object.foo.text.must_equal 'a text scoped'
+    object.foo_text.must_equal 'a text scoped'
+  end
+
+  it 'returns options for select' do
+    I18n.backend.store_translations(:en, :enumerize => {:foo => {:a => 'a text', :b => 'b text'}})
+    klass.enumerize(:foo, :in => [:a, :b])
+    klass.foo.options.must_equal [['a text', 'a'], ['b text', 'b']]
+  end
+
+  it 'dumpes value to yaml using' do
+    klass.enumerize(:foo, :in => [:a, :b])
+    object.foo = :a
+    YAML.dump(object.foo).must_equal YAML.dump('a')
+  end
+end

test/test_helper.rb

@@ -0,0 +1,7 @@
+require 'minitest/autorun'
+require 'minitest/spec'
+
+$VERBOSE=true
+YAML::ENGINE.yamler = ENV['YAMLER'] if ENV['YAMLER']
+
+require 'enumerize'

test/value_test.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+require 'yaml'
+
+describe Enumerize::Value do
+  let(:value) { value = Enumerize::Value.new(nil, 'test_value') }
+
+  it 'is a string' do
+    value.must_be_kind_of String
+  end
+
+  it 'is compared to string' do
+    value.must_be :==, 'test_value'
+  end
+
+  it 'is frozen' do
+    value.must_be :frozen?
+  end
+end