ripple 0.9.5 → 1.0.0.beta

data/.gitignore

@@ -0,0 +1,32 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage/**/*
+rdoc/**/*
+pkg/**/*
+
+## PROJECT::SPECIFIC
+doc/*
+.yardoc
+.bundle
+spec/support/test_server.yml
+Gemfile.lock
+**/bin
+*.rbc
+.rvmrc
+
+.ripplenode
+.riaktest

data/Gemfile

@@ -1,16 +1,18 @@
 source :rubygems
 
-gem 'bundler'
-gem 'riak-client', :path => "../riak-client"
-gem 'activemodel', '~>3.0.0'
-gem 'rspec', '~>2.4.0'
-gem 'tzinfo'
-gem 'rake'
+gemspec
 
-platforms :mri do
-  gem 'yajl-ruby'
+if ENV['RAILS31']
+  gem 'activemodel', '~> 3.1.0'
+else
+  gem 'activemodel', '~> 3.0.10'
 end
 
+gem 'riak-client', :path => "../riak-client"
+gem 'guard-rspec'
+gem 'rb-fsevent'
+gem 'growl'
+
 platforms :mri_18, :jruby do
   gem 'ruby-debug'
 end
@@ -20,6 +22,5 @@ platforms :mri_19 do
 end
 
 platforms :jruby do
-  gem 'json'
   gem 'jruby-openssl'
 end

data/Guardfile

@@ -0,0 +1,15 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+gemset = ENV['RVM_GEMSET'] || 'ripple'
+gemset = "@#{gemset}" unless gemset.to_s == ''
+
+rvms = %w[ 1.8.7 1.9.2 jruby ].map do |version|
+  "#{version}@#{gemset}"
+end
+
+guard 'rspec', :cli => '--profile', :rvm => rvms do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/(.+)\.rb$})     { |m| "spec/#{m[1]}_spec.rb" }
+  watch('spec/spec_helper.rb')  { "spec/ripple" }
+end
+

data/Rakefile

@@ -1,48 +1,22 @@
 require 'rubygems'
-require 'rake/gempackagetask'
-
-version = File.read('../VERSION').strip
-
-gemspec = Gem::Specification.new do |gem|
-  gem.name = "ripple"
-  gem.summary = %Q{ripple is an object-mapper library for Riak, the distributed database by Basho.}
-  gem.description = %Q{ripple is an object-mapper library for Riak, the distributed database by Basho.  It uses ActiveModel to provide an experience that integrates well with Rails 3 applications.}
-  gem.version = version
-  gem.email = "sean@basho.com"
-  gem.homepage = "http://seancribbs.github.com/ripple"
-  gem.authors = ["Sean Cribbs"]
-  gem.add_development_dependency "rspec", "~>2.4.0"
-  gem.add_dependency "riak-client", "~>#{version}"
-  gem.add_dependency "activesupport", "~>3.0.0"
-  gem.add_dependency "activemodel", "~>3.0.0"
-
-  files = FileList["**/*"]
-  # Editor and O/S files
-  files.exclude ".DS_Store", "*~", "\#*", ".\#*", "*.swp", "*.tmproj", "tmtags"
-  # Generated artifacts
-  files.exclude "coverage", "rdoc", "pkg", ".bundle", "*.rbc", ".rvmrc", ".watchr", ".rspec"
-  # Project-specific
-  files.exclude "Gemfile.lock", %r{spec/support/test_server.yml$}, "bin"
-  # Remove directories
-  files.exclude {|d| File.directory?(d) }
-
-  gem.files = files.to_a
+require 'rubygems/package_task'
+require 'rspec/core'
+require 'rspec/core/rake_task'
 
-  gem.test_files = gem.files.grep(/_spec\.rb$/)
+def gemspec
+  $ripple_gemspec ||= Gem::Specification.load("ripple.gemspec")
 end
 
-# Gem packaging tasks
-Rake::GemPackageTask.new(gemspec) do |pkg|
+Gem::PackageTask.new(gemspec) do |pkg|
   pkg.need_zip = false
   pkg.need_tar = false
 end
 
 task :gem => :gemspec
 
-desc %{Build the gemspec file.}
+desc %{Validate the gemspec file.}
 task :gemspec do
   gemspec.validate
-  File.open("#{gemspec.name}.gemspec", 'w'){|f| f.write gemspec.to_ruby }
 end
 
 desc %{Release the gem to RubyGems.org}
@@ -50,24 +24,21 @@ task :release => :gem do
   system "gem push pkg/#{gemspec.name}-#{gemspec.version}.gem"
 end
 
-require 'rspec/core'
-require 'rspec/core/rake_task'
-
 desc "Run Unit Specs Only"
 RSpec::Core::RakeTask.new(:spec) do |spec|
-  spec.pattern = "spec/ripple/**/*_spec.rb"
+  spec.rspec_opts = %w[--profile --tag ~integration]
 end
 
 namespace :spec do
   desc "Run Integration Specs Only"
   RSpec::Core::RakeTask.new(:integration) do |spec|
-    spec.pattern = "spec/integration/**/*_spec.rb"
+    spec.rspec_opts = %w[--profile --tag integration]
   end
 
   desc "Run All Specs"
   RSpec::Core::RakeTask.new(:all) do |spec|
-    spec.pattern = "spec/{ripple,integration}/**/*_spec.rb"
+    spec.rspec_opts = %w[--profile]
   end
 end
 
-task :default => :spec
+task :default => "spec:all"

data/lib/rails/generators/ripple/configuration/configuration_generator.rb

@@ -1,16 +1,3 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
 require 'rails/generators/ripple_generator'
 
 module Ripple

data/lib/rails/generators/ripple/configuration/templates/ripple.yml

@@ -2,7 +2,7 @@
 development:
   http_port: 8098
   pb_port: 8087
-  host: localhost
+  host: 127.0.0.1
 
 # The test environment has additional keys for configuring the
 # Riak::TestServer for your test/spec suite:
@@ -14,11 +14,11 @@ development:
 test:
   http_port: 9000
   pb_port: 9002
-  host: localhost
-  bin_dir: /usr/local/bin   # Default for Homebrew.
+  host: 127.0.0.1
+  source: /usr/local/bin   # Default for Homebrew.
   js_source_dir: <%%= Rails.root + "app/mapreduce" %>
 
 production:
   http_port: 8098
   pb_port: 8087
-  host: localhost
+  host: 127.0.0.1

data/lib/rails/generators/ripple/js/js_generator.rb

@@ -1,16 +1,3 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
 require 'rails/generators/ripple_generator'
 
 module Ripple

data/lib/rails/generators/ripple/js/templates/js/contrib.js

@@ -1,22 +1,5 @@
 // -------------------------------------------------------------------
 //
-//
-// This file is provided to you under the Apache License,
-// Version 2.0 (the "License"); you may not use this file
-// except in compliance with the License.  You may obtain
-// a copy of the License at
-//
-//  http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing,
-// software distributed under the License is distributed on an
-// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-// KIND, either express or implied.  See the License for the
-// specific language governing permissions and limitations
-// under the License.
-//
-// -------------------------------------------------------------------
-//
 // This file contains Javascript MapReduce functions copied from the
 // "Riak Function Contrib" project.
 //

data/lib/rails/generators/ripple/js/templates/js/ripple.js

@@ -1,16 +1,3 @@
-// Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-//
-//    Licensed under the Apache License, Version 2.0 (the "License");
-//    you may not use this file except in compliance with the License.
-//    You may obtain a copy of the License at
-//
-//        http://www.apache.org/licenses/LICENSE-2.0
-//
-//    Unless required by applicable law or agreed to in writing, software
-//    distributed under the License is distributed on an "AS IS" BASIS,
-//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-//    See the License for the specific language governing permissions and
-//    limitations under the License.
 
 /*
  * Namespace for Ripple's built-in MapReduce functions.

data/lib/rails/generators/ripple/model/model_generator.rb

@@ -1,17 +1,3 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
-
 require 'rails/generators/ripple_generator'
 
 module Ripple

data/lib/rails/generators/ripple/model/templates/model.rb

@@ -1,7 +1,7 @@
 class <%= class_name %><%= " < #{options[:parent].classify}" if options[:parent] %>
 <% unless options[:parent] -%>
   include Ripple::<%= "Embedded" if options[:embedded] || options[:embedded_in] %>Document
-  <% if options[:embedded_in] -%>embedded_in :<%= options[:embedded_in].underscore %><% end -%> 
+  <% if options[:embedded_in] -%>embedded_in :<%= options[:embedded_in].underscore %><% end -%>
 <% end -%>
 
 <% attributes.reject{|attr| attr.reference?}.each do |attribute| -%>

data/lib/rails/generators/ripple/observer/observer_generator.rb

@@ -1,17 +1,3 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
-
 require 'rails/generators/ripple_generator'
 
 module Ripple

data/lib/rails/generators/ripple/test/test_generator.rb

@@ -1,16 +1,3 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
 require 'rails/generators/ripple_generator'
 
 module Ripple
@@ -20,7 +7,6 @@ module Ripple
       # Cucumber
       def create_cucumber_file
         if File.directory?(Rails.root + "features/support")
-          template 'test_server.rb', 'features/support/ripple.rb'
           insert_into_file 'features/support/ripple.rb', "\n\nAfter do\n  Ripple::TestServer.clear\nend", :after => "Ripple::TestServer.setup"
         end
       end
@@ -28,9 +14,12 @@ module Ripple
       # RSpec
       def create_rspec_file
         if File.file?(Rails.root + 'spec/spec_helper.rb')
-          template 'test_server.rb', 'spec/support/ripple.rb'
+          inject_into_file 'spec/spec_helper.rb', :before => /R[Ss]pec\.configure do \|config\|/ do
+            "require 'ripple/test_server'\n"
+          end
           inject_into_file 'spec/spec_helper.rb', :after => /R[Ss]pec\.configure do \|config\|/ do
-            "\n  config.after(:each) do\n    Ripple::TestServer.clear\n  end\n"
+            "\n  config.before(:all){ Ripple::TestServer.setup }" +
+            "\n  config.after(:each){ Ripple::TestServer.clear }\n"
           end
         end
       end
@@ -38,12 +27,11 @@ module Ripple
       # Test::Unit
       def create_test_unit_file
         if File.file?(Rails.root + 'test/test_helper.rb')
-          template 'test_server.rb', 'test/ripple_test_helper.rb'
           inject_into_file "test/test_helper.rb", :before => "class ActiveSupport::TestCase" do
-            "# Setup in-memory test server for Riak\nrequire File.expand_path('../ripple_test_helper.rb', __FILE__)\n\n"
+            "# Setup in-memory test server for Riak\nrequire 'ripple/test_server'\n\n"
           end
           inject_into_class "test/test_helper.rb", ActiveSupport::TestCase do
-            "  teardown { Ripple::TestServer.clear }\n\n"
+            "  setup { Ripple::TestServer.setup }\n  teardown { Ripple::TestServer.clear }\n\n"
           end
         end
       end

data/lib/rails/generators/ripple_generator.rb

@@ -1,16 +1,3 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
 require "rails/generators/named_base"
 require "rails/generators/active_model"
 
@@ -43,7 +30,7 @@ module Ripple
     # Generator for a {Ripple::Document} model
     class ActiveModel < ::Rails::Generators::ActiveModel
       def self.all(klass)
-        "#{klass}.all"
+        "#{klass}.list"
       end
 
       def self.find(klass, params=nil)

data/lib/ripple.rb

@@ -1,17 +1,3 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
-
 require 'riak'
 require 'erb'
 require 'yaml'
@@ -68,6 +54,7 @@ module Ripple
       config_file = File.expand_path(config_file)
       config_hash = YAML.load(ERB.new(File.read(config_file)).result).with_indifferent_access
       config_keys.each {|k| config_hash = config_hash[k]}
+      configure_ports(config_hash)
       self.config = config_hash || {}
     rescue Errno::ENOENT
       raise Ripple::MissingConfiguration.new(config_file)
@@ -78,6 +65,12 @@ module Ripple
     def client_config
       config.slice(*Riak::Client::VALID_OPTIONS)
     end
+
+    def configure_ports(config)
+      return unless config && config[:min_port]
+      config[:http_port] ||= (config[:min_port].to_i)
+      config[:pb_port] ||= (config[:min_port].to_i + 1)
+    end
   end
 
   # Exception raised when the path passed to

data/lib/ripple/associations.rb

@@ -1,18 +1,5 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
-
 require 'active_support/concern'
+require 'active_support/dependencies'
 require 'riak/walk_spec'
 require 'ripple/translation'
 require 'ripple/associations/proxy'
@@ -26,11 +13,15 @@ require 'ripple/associations/one_embedded_proxy'
 require 'ripple/associations/many_embedded_proxy'
 require 'ripple/associations/one_linked_proxy'
 require 'ripple/associations/many_linked_proxy'
+require 'ripple/associations/many_stored_key_proxy'
+require 'ripple/associations/one_key_proxy'
+require 'ripple/associations/one_stored_key_proxy'
+require 'ripple/associations/many_reference_proxy'
 
 module Ripple
   # Adds associations via links and embedding to {Ripple::Document}
   # models. Examples:
-  # 
+  #
   #   # Documents can contain embedded documents, and link to other standalone documents
   #   # via associations using the many and one class methods.
   #   class Person
@@ -40,14 +31,14 @@ module Ripple
   #     many :friends, :class_name => "Person"
   #     one :account
   #   end
-  #    
+  #
   #   # Account and Address are embeddable documents
   #   class Account
   #     include Ripple::EmbeddedDocument
   #     property :paid_until, Time
   #     embedded_in :person # Adds "person" method to get parent document
   #   end
-  #    
+  #
   #   class Address
   #     include Ripple::EmbeddedDocument
   #     property :street, String
@@ -55,7 +46,7 @@ module Ripple
   #     property :state, String
   #     property :zip, String
   #   end
-  #    
+  #
   #   person = Person.find("adamhunter")
   #   person.friends << Person.find("seancribbs") # Links to people/seancribbs with tag "friend"
   #   person.addresses << Address.new(:street => "100 Main Street") # Adds an embedded address
@@ -64,6 +55,7 @@ module Ripple
     extend ActiveSupport::Concern
 
     module ClassMethods
+      include Translation
       # @private
       def inherited(subclass)
         super
@@ -77,22 +69,39 @@ module Ripple
 
       # Associations of embedded documents
       def embedded_associations
-        associations.values.select(&:embeddable?)
+        associations.values.select(&:embedded?)
+      end
+
+      # Associations of linked documents
+      def linked_associations
+        associations.values.select(&:linked?)
+      end
+
+      # Associations of stored_key documents
+      def stored_key_associations
+        associations.values.select(&:stored_key?)
       end
 
       # Creates a singular association
       def one(name, options={})
+        configure_for_key_correspondence if options[:using] === :key
         create_association(:one, name, options)
       end
 
       # Creates a plural association
       def many(name, options={})
+        raise ArgumentError, t('many_key_association') if options[:using] === :key
         create_association(:many, name, options)
       end
 
+      def configure_for_key_correspondence
+        include Ripple::Associations::KeyDelegator
+      end
+
       private
       def create_association(type, name, options={})
         association = associations[name] = Association.new(type, name, options)
+        association.setup_on(self)
 
         define_method(name) do
           get_proxy(association)
@@ -121,6 +130,13 @@ module Ripple
         proxy
       end
 
+      # @private
+      def reset_associations
+        self.class.associations.each do |name, assoc_object|
+          send(name).reset
+        end
+      end
+
       # Adds embedded documents to the attributes
       # @private
       def attributes_for_persistence
@@ -137,6 +153,39 @@ module Ripple
           attrs
         end
       end
+
+      def propagate_callbacks_to_embedded_associations(name, kind)
+        self.class.embedded_associations.each do |association|
+          documents = instance_variable_get(association.ivar)
+          # We must explicitly check #nil? (rather than just saying `if documents`)
+          # because documents can be an association proxy that is proxying nil.
+          # In this case ruby treats documents as true because it is not _really_ nil,
+          # but #nil? will tell us if it is proxying nil.
+          next if documents.nil?
+
+          Array(documents).each do |doc|
+            doc.send("_#{name}_callbacks").each do |callback|
+              next unless callback.kind == kind
+              doc.send(callback.filter)
+            end
+          end
+        end
+      end
+
+      # Propagates callbacks (save/create/update/destroy) to embedded associated documents.
+      # This is necessary so that when a parent is saved, the embedded child's before_save
+      # hooks are run as well.
+      # @private
+      def run_callbacks(name, *args, &block)
+        # validation is already propagated to embedded documents via the
+        # AssociatedValidator.  We don't need to duplicate the propgation here.
+        return super if name == :validation
+
+        propagate_callbacks_to_embedded_associations(name, :before)
+        return_value = super
+        propagate_callbacks_to_embedded_associations(name, :after)
+        return_value
+      end
     end
   end
 
@@ -169,7 +218,7 @@ module Ripple
 
     # @return [Class] The class of the associated object(s)
     def klass
-      @klass ||= options[:class] || class_name.constantize
+      @klass ||= discover_class
     end
 
     # @return [true,false] Is the cardinality of the association > 1
@@ -183,7 +232,7 @@ module Ripple
     end
 
     # @return [true,false] Is the associated class an EmbeddedDocument
-    def embeddable?
+    def embedded?
       klass.embeddable?
     end
 
@@ -198,6 +247,11 @@ module Ripple
       using == :linked
     end
 
+    # @return [true,false] Does the association use stored_key
+    def stored_key?
+      using == :stored_key
+    end
+
     # @return [String] the instance variable in the owner where the association will be stored
     def ivar
       "@_#{name}"
@@ -224,13 +278,16 @@ module Ripple
       linked? ? Array(link_spec).first.tag : nil
     end
 
+    def bucket_name
+      polymorphic? ? '_' : klass.bucket_name
+    end
+
     # @return [Riak::WalkSpec] when #linked? is true, a specification for which links to follow to retrieve the associated documents
     def link_spec
       # TODO: support transitive linked associations
       if linked?
         tag = name.to_s
-        bucket = polymorphic? ? '_' : klass.bucket_name
-        Riak::WalkSpec.new(:tag => tag, :bucket => bucket)
+        Riak::WalkSpec.new(:tag => tag, :bucket => bucket_name)
       else
         nil
       end
@@ -238,7 +295,7 @@ module Ripple
 
     # @return [Symbol] which method is used for representing the association - currently only supports :embedded and :linked
     def using
-      @using ||= options[:using] || (embeddable? ? :embedded : :linked)
+      @using ||= options[:using] || (embedded? ? :embedded : :linked)
     end
 
     # @raise [ArgumentError] if the value does not match the class of the association
@@ -258,10 +315,56 @@ module Ripple
       when polymorphic?
         one? || Array === value
       when many?
-        Array === value && value.all? {|d| (embeddable? && Hash === d) || klass === d }
+        Array === value && value.all? {|d| (embedded? && Hash === d) || klass === d }
       when one?
-        value.nil? || (embeddable? && Hash === value) || value.kind_of?(klass)
+        value.nil? || (embedded? && Hash === value) || value.kind_of?(klass)
+      end
+    end
+
+    def uses_search?
+      (options[:using] == :reference)
+    end
+
+    def setup_on(model)
+      @model = model
+      define_callbacks_on(model)
+      if uses_search?
+        klass.before_save do |o|
+          unless o.class.bucket.is_indexed?
+            o.class.bucket.enable_index!
+          end
+        end
       end
     end
+
+    def define_callbacks_on(klass)
+      _association = self
+
+      klass.before_save do
+        if _association.linked? && !@_in_save_loaded_documents_callback
+          @_in_save_loaded_documents_callback = true
+
+          begin
+            send(_association.name).loaded_documents.each do |document|
+              document.save if document.new? || document.changed?
+            end
+          ensure
+            remove_instance_variable(:@_in_save_loaded_documents_callback)
+          end
+        end
+      end
+    end
+
+    private
+    def discover_class
+      options[:class] || (@model && find_class(@model, class_name)) || class_name.constantize
+    end
+
+    def find_class(scope, class_name)
+      return nil if class_name.include?("::")
+      class_sym = class_name.to_sym
+      parent_scope = scope.parents.unshift(scope).find {|s| ActiveSupport::Dependencies.local_const_defined?(s, class_sym) }
+      parent_scope.const_get(class_sym) if parent_scope
+    end
   end
 end