active_mongoid 0.1.0

data/gemfiles/mongoid_3.1.gemfile

@@ -0,0 +1,7 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "mongoid", "~> 3.1.0"
+
+gemspec :path => "../"

data/gemfiles/mongoid_3.1.gemfile.lock

@@ -0,0 +1,108 @@
+PATH
+  remote: ../
+  specs:
+    active_mongoid (0.0.1)
+      activerecord
+      activesupport
+      after_do
+      bson_ext
+      mongo
+      mongoid
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    activemodel (3.2.21)
+      activesupport (= 3.2.21)
+      builder (~> 3.0.0)
+    activerecord (3.2.21)
+      activemodel (= 3.2.21)
+      activesupport (= 3.2.21)
+      arel (~> 3.0.2)
+      tzinfo (~> 0.3.29)
+    activesupport (3.2.21)
+      i18n (~> 0.6, >= 0.6.4)
+      multi_json (~> 1.0)
+    after_do (0.3.1)
+    appraisal (1.0.2)
+      bundler
+      rake
+      thor (>= 0.14.0)
+    arel (3.0.3)
+    bson (1.11.1)
+    bson_ext (1.11.1)
+      bson (~> 1.11.1)
+    builder (3.0.4)
+    coderay (1.1.0)
+    coveralls (0.7.1)
+      multi_json (~> 1.3)
+      rest-client
+      simplecov (>= 0.7)
+      term-ansicolor
+      thor
+    database_cleaner (1.3.0)
+    diff-lcs (1.2.5)
+    docile (1.1.5)
+    i18n (0.6.11)
+    method_source (0.8.2)
+    mime-types (2.4.3)
+    mongo (1.11.1)
+      bson (= 1.11.1)
+    mongoid (3.1.6)
+      activemodel (~> 3.2)
+      moped (~> 1.4)
+      origin (~> 1.0)
+      tzinfo (~> 0.3.29)
+    moped (1.5.2)
+    multi_json (1.10.1)
+    netrc (0.8.0)
+    origin (1.1.0)
+    pry (0.10.1)
+      coderay (~> 1.1.0)
+      method_source (~> 0.8.1)
+      slop (~> 3.4)
+    rake (10.3.2)
+    rest-client (1.7.2)
+      mime-types (>= 1.16, < 3.0)
+      netrc (~> 0.7)
+    rspec (3.1.0)
+      rspec-core (~> 3.1.0)
+      rspec-expectations (~> 3.1.0)
+      rspec-mocks (~> 3.1.0)
+    rspec-core (3.1.7)
+      rspec-support (~> 3.1.0)
+    rspec-expectations (3.1.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.1.0)
+    rspec-mocks (3.1.3)
+      rspec-support (~> 3.1.0)
+    rspec-support (3.1.2)
+    simplecov (0.9.1)
+      docile (~> 1.1.0)
+      multi_json (~> 1.0)
+      simplecov-html (~> 0.8.0)
+    simplecov-gem-adapter (1.0.1)
+      simplecov
+    simplecov-html (0.8.0)
+    slop (3.6.0)
+    sqlite3 (1.3.10)
+    term-ansicolor (1.3.0)
+      tins (~> 1.0)
+    thor (0.19.1)
+    tins (1.3.3)
+    tzinfo (0.3.42)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  active_mongoid!
+  appraisal
+  coveralls
+  database_cleaner
+  mongoid (~> 3.1.0)
+  pry
+  rspec
+  simplecov
+  simplecov-gem-adapter
+  sqlite3

data/lib/active_mongoid.rb

@@ -0,0 +1,8 @@
+require "active_mongoid/associations"
+require "active_mongoid/bson_id"
+require "active_mongoid/finder_proxy"
+require "active_mongoid/finders"
+
+module ActiveMongoid
+  BSON = ::Gem::Version.new(Mongoid::VERSION) < ::Gem::Version.new("3.0.0") ? ::BSON : ::Moped::BSON
+end

data/lib/active_mongoid/associations.rb

@@ -0,0 +1,82 @@
+require "active_mongoid/associations/metadata"
+require "active_mongoid/associations/builder"
+require "active_mongoid/associations/binding"
+require "active_mongoid/associations/proxy"
+require "active_mongoid/associations/one"
+require "active_mongoid/associations/many"
+require "active_mongoid/associations/builders/in"
+require "active_mongoid/associations/builders/one"
+require "active_mongoid/associations/builders/many"
+require "active_mongoid/associations/document_relation/associations"
+require "active_mongoid/associations/record_relation/associations"
+require "active_mongoid/associations/targets/enumerable"
+require 'after_do'
+
+module ActiveMongoid
+  module Associations
+    extend ActiveSupport::Concern
+
+    included do
+      extend ::AfterDo
+      class_attribute :am_relations
+      self.am_relations = {}
+
+      if defined?(::ActiveRecord::Base) && self <= ::ActiveRecord::Base
+        include DocumentRelation::Associations
+      elsif defined?(::Mongoid::Document) && self.included_modules.include?(::Mongoid::Document)
+        include RecordRelation::Associations
+      else
+        raise
+      end
+
+      before :reload do
+        am_relations.each_pair do |name, meta|
+          if instance_variable_defined?("@#{name}")
+            if instance_variable_get("@#{name}")
+              remove_instance_variable("@#{name}")
+            end
+          end
+        end
+      end
+    end
+
+    def reflect_on_am_association(name)
+      self.class.reflect_on_am_association(name.to_s)
+    end
+
+    def referenced_many_records?
+      __metadata__ && __metadata__.macro == :has_many_records
+    end
+
+    def referenced_one_record?
+      __metadata__ && __metadata__.macro == :has_one_record
+    end
+
+    def referenced_many_documents?
+      __metadata__ && __metadata__.macro == :has_many_documents
+    end
+
+    def referenced_one_document?
+      __metadata__ && __metadata__.macro == :has_one_document
+    end
+
+    module ClassMethods
+
+      def reflect_on_am_association(name)
+        am_relations[name.to_s]
+      end
+
+      private
+
+      def characterize_association(name, relation, options)
+        ActiveMongoid::Associations::Metadata.new({
+          :relation => relation,
+          :inverse_class_name => self.name,
+          :name => name
+        }.merge(options))
+      end
+
+    end
+
+  end
+end

data/lib/active_mongoid/associations/binding.rb

@@ -0,0 +1,77 @@
+require 'mongoid/threaded'
+
+module ActiveMongoid
+  module Associations
+    class Binding
+      include ::Mongoid::Threaded::Lifecycle
+
+      attr_reader :base, :target, :__metadata__
+
+      def initialize(base, target, metadata)
+        @base = base
+        @target = target
+        @__metadata__ = metadata
+      end
+
+      private
+
+      def check_inverse!(object)
+        # check for inverse relation and raise exception when not
+        true
+      end
+
+      def bind_foreign_key(base, id)
+        return if base.send(__metadata__.foreign_key) == id
+        base.send(__metadata__.foreign_key_setter, id)
+      end
+
+      def bind_polymorphic_type(base, name)
+        if __metadata__.type
+          base.send(__metadata__.type_setter, name)
+        end
+      end
+
+      def bind_polymorphic_inverse_type(base, name)
+        if __metadata__.inverse_type
+          base.send(__metadata__.inverse_type_setter, name)
+        end
+      end
+
+      def bind_inverse(object, inverse)
+        unless _binding?
+          _binding do
+            if object.respond_to?(__metadata__.inverse_setter)
+              object.send(__metadata__.inverse_setter, inverse)
+            end
+          end
+        end
+      end
+
+      def record_id(base)
+        base.send(__metadata__.primary_key)
+      end
+
+      def set_base_metadata
+        inverse_metadata = __metadata__.inverse_metadata(target)
+        if inverse_metadata != __metadata__ && !inverse_metadata.nil?
+          base.__metadata__ = inverse_metadata
+        end
+      end
+
+      def bind_from_relational_parent(object)
+        check_inverse!(object)
+        bind_foreign_key(object, record_id(base))
+        bind_polymorphic_type(object, base.class.name)
+        bind_inverse(object, base)
+      end
+
+      def unbind_from_relational_parent(object)
+        check_inverse!(object)
+        bind_foreign_key(object, nil)
+        bind_polymorphic_type(object, nil)
+        bind_inverse(object, nil)
+      end
+
+    end
+  end
+end

data/lib/active_mongoid/associations/builder.rb

@@ -0,0 +1,26 @@
+module ActiveMongoid
+  module Associations
+    class Builder
+
+      attr_reader :base, :__metadata__, :object
+
+      def initialize(base, metadata, object)
+        @base = base
+        @__metadata__ = metadata
+        @object = object
+      end
+
+      protected
+
+      def klass
+        @klass ||= __metadata__.klass
+      end
+
+      def query?
+        obj = Array(object).first
+        !obj.is_a?(::Mongoid::Document) && !obj.is_a?(::ActiveRecord::Base) && !obj.nil?
+      end
+
+    end
+  end
+end

data/lib/active_mongoid/associations/builders/in.rb

@@ -0,0 +1,17 @@
+module ActiveMongoid
+  module Associations
+    module Builders
+      class In < Builder
+
+        def build(type = nil)
+          return object unless query?
+          return nil if object.blank?
+          type = type.constantize if type and type.is_a?(String)
+          model = type ? type : __metadata__.klass
+          __metadata__.criteria(object, model).first
+        end
+
+      end
+    end
+  end
+end

data/lib/active_mongoid/associations/builders/many.rb

@@ -0,0 +1,15 @@
+module ActiveMongoid
+  module Associations
+    module Builders
+      class Many < Builder
+
+        def build(type = nil)
+          return object unless query?
+          return [] if object.is_a?(Array)
+          __metadata__.criteria(object, base.class)
+        end
+
+      end
+    end
+  end
+end

data/lib/active_mongoid/associations/builders/one.rb

@@ -0,0 +1,15 @@
+module ActiveMongoid
+  module Associations
+    module Builders
+      class One < Builder
+
+        def build(type = nil)
+          return object unless query?
+          return nil if base.new_record?
+          __metadata__.criteria(object, base.class).first
+        end
+
+      end
+    end
+  end
+end

data/lib/active_mongoid/associations/document_relation/accessors.rb

@@ -0,0 +1,100 @@
+module ActiveMongoid
+  module Associations
+    module DocumentRelation
+      module Accessors
+        extend ActiveSupport::Concern
+
+        def build_document(name, object, metadata)
+          relation = create_document_relation(object, metadata)
+          set_document_relation(name, relation)
+        end
+
+        def set_document_relation(name, object)
+          instance_variable_set("@#{name}", object)
+        end
+
+        def create_document_relation(object, metadata)
+          type = self.attributes[metadata.inverse_type]
+          target = metadata.builder(self, object).build(type)
+          target ? metadata.relation.new(self, target, metadata) : nil
+        end
+
+        private
+
+        def get_document_relation(name, metadata, object, reload = false)
+          if !reload && (value = instance_variable_get("@#{name}")) != nil
+            value
+          else
+            if object && needs_no_database_query?(object, metadata)
+              build_document(name, object, metadata)
+            else
+              build_document(name, self.send(metadata.key), metadata)
+            end
+          end
+        end
+
+        def needs_no_database_query?(object, metadata)
+          object.is_a?(::Mongoid::Document) && object.id.to_s == attributes[metadata.key].to_s
+        end
+
+
+        module ClassMethods
+
+          private
+
+          def existence_check(name)
+            module_eval <<-END
+              def #{name}?
+                !__send__(:#{name}).blank?
+              end
+              alias :has_#{name}? :#{name}?
+            END
+            self
+          end
+
+          # Getters
+
+          def document_getter(name, metadata)
+            self.instance_eval do
+              define_method(name) do |reload = false|
+                get_document_relation(name, metadata, nil, reload)
+              end
+            end
+          end
+
+          def document_id_getter(name, metadata)
+            self.instance_eval do
+              define_method("#{name}_id") do
+                attribute = read_attribute("#{name}_id")
+                attribute.nil? ? nil : ::ActiveMongoid::BSON::ObjectId.from_string(attribute)
+              end
+            end
+          end
+
+          def document_setter(name, metadata)
+            self.instance_eval do
+              define_method("#{name}=") do |object|
+                if value = get_document_relation(name, metadata, object)
+                  set_document_relation(name, value.substitute(object))
+                else
+                  build_document(name, object, metadata)
+                end
+              end
+            end
+          end
+
+          def document_id_setter(name, metadata)
+            self.instance_eval do
+              define_method("#{name}_id=") do |bson_id|
+                attribute = bson_id.nil? ? nil : bson_id.to_s
+                write_attribute("#{name}_id", attribute)
+              end
+            end
+          end
+
+        end
+
+      end
+    end
+  end
+end