Merge pull request #109 from kidpollo/uniform_protected_api

Provide a uniform api between Simple, KeyValue and Chain stores

lib/i18n/backend/chain.rb

@@ -26,6 +26,15 @@ def initialize(*backends)
           self.backends = backends
         end
 
+        def initialized?
+          backends.all? do |backend|
+            backend.instance_eval do
+              return false unless initialized?
+            end
+          end
+          true
+        end
+
         def reload!
           backends.each { |backend| backend.reload! }
         end
@@ -74,6 +83,19 @@ def localize(locale, object, format = :default, options = EMPTY_HASH)
         end
 
         protected
+          def init_translations
+            backends.each do |backend|
+              backend.send(:init_translations)
+            end
+          end
+
+          def translations
+            backends.first.instance_eval do
+              init_translations unless initialized?
+              translations
+            end
+          end  
+
           def namespace_lookup?(result, options)
             result.is_a?(Hash) && !options.has_key?(:count)
           end

lib/i18n/backend/key_value.rb

@@ -76,6 +76,10 @@ def initialize(store, subtrees=true)
           @store, @subtrees = store, subtrees
         end
 
+        def initialized?
+          !@store.nil?
+        end
+
         def store_translations(locale, data, options = EMPTY_HASH)
           escape = options.fetch(:escape, true)
           flatten_translations(locale, data, escape, @subtrees).each do |key, value|
@@ -105,6 +109,26 @@ def available_locales
 
       protected
 
+        # Queries the translations from the key-value store and converts
+        # them into a hash such as the one returned from loading the
+        # haml files
+        def translations
+          @translations = @store.keys.clone.map do |main_key|
+            main_value = JSON.decode(@store[main_key])
+            main_key.to_s.split(".").reverse.inject(main_value) do |value, key|
+              {key.to_sym => value}
+            end
+          end.inject{|hash, elem| hash.deep_merge!(elem)}.deep_symbolize_keys
+        end
+
+        def init_translations
+          # NO OP
+          # This call made also inside Simple Backend and accessed by
+          # other plugins like I18n-js and babilu and
+          # to use it along with the Chain backend we need to
+          # provide a uniform API even for protected methods :S
+        end
+
         def subtrees?
           @subtrees
         end

test/backend/chain_test.rb

@@ -81,6 +81,39 @@ def setup
     I18n.backend.store_translations :foo, {:bar => :baz}, {:option => 'persists'}
   end
 
+  test 'store should call initialize on all backends and return true if all initialized' do
+    @first.send :init_translations
+    @second.send :init_translations
+    assert I18n.backend.initialized?
+  end
+
+  test 'store should call initialize on all backends and return false if one not initialized' do
+    @first.reload!
+    @second.send :init_translations
+    assert !I18n.backend.initialized?
+  end
+
+  test 'should reload all backends' do
+    @first.send :init_translations
+    @second.send :init_translations
+    I18n.backend.reload!
+    assert !@first.initialized?
+    assert !@second.initialized?
+  end
+
+  test 'should be able to get all translations of the first backend' do
+    assert_equal I18n.backend.send(:translations),
+                 en: {
+                   foo: 'Foo',
+                   formats: {
+                     short: 'short',
+                     subformats: { short: 'short' }
+                   },
+                   plural_1: { one: "%{count}" },
+                   dates: { a: 'A' }
+                 }
+  end
+
   protected
 
     def backend(translations)

test/backend/key_value_test.rb

@@ -49,6 +49,19 @@ def assert_flattens(expected, nested, escape=true, subtree=true)
     end
   end
 
+  test 'initialized? checks that a store is available' do
+    setup_backend!
+    I18n.backend.reload!
+    assert_equal I18n.backend.initialized?, true
+  end
+
+  test 'translations gets the translations from the store' do
+    setup_backend!
+    I18n.backend.send(:translations)
+    expected = { :en => {:foo => { :bar => 'bar', :baz => 'baz' }} }
+    assert_equal expected, translations
+  end 
+
   test "subtrees enabled: given incomplete pluralization data it raises I18n::InvalidPluralizationData" do
     setup_backend!
     store_translations(:en, :bar => { :one => "One" })