lib/mongoid/i18n.rb in mongoid_i18n-0.2.1 vs lib/mongoid/i18n.rb in mongoid_i18n-0.3.0

@@ -1,30 +1,55 @@
 require 'mongoid/i18n/localized_field'
 require 'mongoid/i18n/criterion/selector'
+require 'mongoid/i18n/localized_validator'
 
+# add english load path by default
+I18n.load_path << File.join(File.dirname(__FILE__), "..", "..", "config", "locales", "en.yml")
+
 module Mongoid
   module I18n
     extend ActiveSupport::Concern
 
     module ClassMethods
       def localized_field(name, options = {})
-        field name, options.merge(:type => LocalizedField)
+        field name, options.merge(:type => LocalizedField, :default => LocalizedField.new)
       end
 
+      def validates_default_locale(names, options = {})
+        validates_with LocalizedValidator, options.merge(:mode => :only_default, :attributes => names)
+      end
+
+      def validates_one_locale(names, options = {})
+        validates_with LocalizedValidator, options.merge(:mode => :one_locale,   :attributes => names)
+      end
+
+      def validates_all_locales(names, options = {})
+        validates_with LocalizedValidator, options.merge(:mode => :all_locales,  :attributes => names)
+      end
+
       protected
       def create_accessors(name, meth, options = {})
         if options[:type] == LocalizedField
-          define_method(meth) { read_attribute(name)[::I18n.locale.to_s] rescue '' }
+          if options[:use_default_if_empty]
+            define_method(meth) { read_attribute(name)[::I18n.locale.to_s] || read_attribute(name)[::I18n.default_locale.to_s] rescue ''}
+          else
+            define_method(meth) { read_attribute(name)[::I18n.locale.to_s] rescue '' }
+          end
           define_method("#{meth}=") do |value|
             value = if value.is_a?(Hash)
               (@attributes[name] || {}).merge(value)
             else
               (@attributes[name] || {}).merge(::I18n.locale.to_s => value)
             end
+            value = value.delete_if { |key, value| value.blank? } if options[:clear_empty_values]
             write_attribute(name, value)
           end
           define_method("#{meth}_translations") { read_attribute(name) }
-          define_method("#{meth}_translations=") { |value| write_attribute(name, value) }
+          if options[:clear_empty_values]
+            define_method("#{meth}_translations=") { |value| write_attribute(name, value.delete_if { |key, value| value.blank? }) }
+          else
+            define_method("#{meth}_translations=") { |value| write_attribute(name, value) }
+          end
         else
           super
         end
       end
     end
\ No newline at end of file