langalex-couch_potato 0.1 → 0.1.1

data/lib/couch_potato/persistence/belongs_to_property.rb

@@ -7,22 +7,36 @@ module CouchPotato
         self.name = name
         accessors =  <<-ACCESSORS
           def #{name}
-            @#{name} || @#{name}_id ? #{item_class_name}.find(@#{name}_id) : nil
+            return @#{name} if instance_variable_defined?(:@#{name})
+            @#{name} = @#{name}_id ? #{item_class_name}.find(@#{name}_id) : nil
           end
           
           def #{name}=(value)
             @#{name} = value
-            @#{name}_id = value.id
+            if value.nil?
+              @#{name}_id = nil
+            else
+              @#{name}_id = value.id
+            end
+          end
+          
+          def #{name}_id=(id)
+            remove_instance_variable(:@#{name}) if instance_variable_defined?(:@#{name})
+            @#{name}_id = id
           end
         ACCESSORS
         owner_clazz.class_eval accessors
-        owner_clazz.send :attr_accessor, "#{name}_id"
+        owner_clazz.send :attr_reader, "#{name}_id"
       end
       
       def save(object)
         
       end
       
+      def dirty?(object)
+        false
+      end
+      
       def destroy(object)
         
       end

data/lib/couch_potato/persistence/callbacks.rb

@@ -5,6 +5,7 @@ module CouchPotato
         base.extend ClassMethods
         
         base.class_eval do
+          attr_accessor :skip_callbacks
           def self.callbacks
             @@callbacks ||= {}
             @@callbacks[self.name] ||= {:before_validation_on_create => [], 
@@ -16,9 +17,17 @@ module CouchPotato
         end
       end
       
+      def save_without_callbacks
+        self.skip_callbacks = true
+        result = save
+        self.skip_callbacks = false
+        result
+      end
+      
       private
       
       def run_callbacks(name)
+        return if skip_callbacks
         self.class.callbacks[name].each do |callback|
           self.send callback
         end

data/lib/couch_potato/persistence/custom_view.rb

@@ -0,0 +1,41 @@
+module CouchPotato
+  module Persistence
+    module CustomView
+      
+      def self.included(base)
+        base.extend ClassMethods
+      end
+      
+      module ClassMethods
+        
+        def view(name, options)
+          (class << self; self; end).instance_eval do
+            if options[:properties]
+              define_method name do
+                ViewQuery.new(self.name.underscore, name, map_function(options[:key], options[:properties]), nil, {}, {}).query_view!['rows'].map{|doc| self.new(doc['value'].merge(:_id => doc['id']))}
+              end
+            else
+              define_method name do
+                ViewQuery.new(self.name.underscore, name, map_function(options[:key], options[:properties]), nil, {}, {:include_docs =>  true}).query_view!['rows'].map{|doc| self.new(doc['doc'])}
+              end
+            end
+          end
+        end
+        
+        def map_function(key, properties)
+          "function(doc) {
+              emit(doc.#{key}, #{properties_for_map(properties)});
+           }"
+        end
+        
+        def properties_for_map(properties)
+          if properties.nil?
+            'null'
+          else
+            '{' + properties.map{|p| "#{p}: doc.#{p}"}.join(', ') + '}'
+          end
+        end
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/dirty_attributes.rb

@@ -0,0 +1,19 @@
+module CouchPotato
+  module Persistence
+    module DirtyAttributes
+      def save
+        if dirty?
+          super 
+        else
+          valid?
+        end
+      end
+      
+      def dirty?
+        new_document? || self.class.properties.inject(false) do |res, property|
+          res || property.dirty?(self)
+        end
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/external_collection.rb

@@ -13,6 +13,26 @@ module CouchPotato
         @owner_id_attribute_name = owner_id_attribute_name
       end
       
+      def all(options = {}, view_options = {})
+        if options.empty? && view_options.empty?
+          items
+        else
+          Finder.new.find @item_class, options.merge(@owner_id_attribute_name => @owner_id), view_options
+        end
+      end
+      
+      def first(options = {}, view_options = {})
+        if options.empty? && view_options.empty?
+          items.first
+        else
+          Finder.new.find(@item_class, options.merge(@owner_id_attribute_name => @owner_id), view_options).first
+        end
+      end
+      
+      def count(options = {}, view_options = {})
+        Finder.new.count @item_class, options.merge(@owner_id_attribute_name => @owner_id), view_options
+      end
+      
       def build(attributes = {})
         item = @item_class.new(attributes)
         self.<< item
@@ -32,8 +52,17 @@ module CouchPotato
         item
       end
       
+      def dirty?
+        return unless @items
+        @original_item_ids != @items.map(&:_id) || @items.inject(false) {|res, item| res || item.dirty?}
+      end
+      
       def items
-        @items ||= Finder.new.find @item_class, @owner_id_attribute_name => owner_id
+        unless @items
+          @items = Finder.new.find @item_class, @owner_id_attribute_name => owner_id
+          @original_item_ids = @items.map(&:_id)
+        end
+        @items
       end
       
       def save
@@ -44,7 +73,7 @@ module CouchPotato
       end
       
       def destroy
-        @items.each do |item|
+        items.each do |item|
           item.destroy
         end
       end

data/lib/couch_potato/persistence/external_has_many_property.rb

@@ -19,6 +19,10 @@ module CouchPotato
         owner_clazz.class_eval getter
       end
       
+      def dirty?(object)
+        object.send("#{name}").dirty?
+      end
+      
       def save(object)
         object.send(name).owner_id = object._id
         object.send(name).each do |item|

data/lib/couch_potato/persistence/finder.rb

@@ -8,97 +8,53 @@ module CouchPotato
       # value can also be a range which will do a range search with startkey/endkey
       # WARNING: calling this methods creates a new view in couchdb if it's not present already so don't overuse this
       def find(clazz, conditions = {}, view_options = {})
-        params = view_parameters(clazz, conditions, view_options)
-        to_instances clazz, query_view!(params)
+        to_instances clazz, ViewQuery.new(design_document(clazz), view(conditions), map_function(clazz, search_fields(conditions)), nil, conditions, view_options).query_view!
       end
       
       def count(clazz, conditions = {}, view_options = {})
-        params = view_parameters(clazz, conditions, view_options)
-        query_view!(params, '_count')['rows'].first.try(:[], 'value') || 0
+        ViewQuery.new(design_document(clazz), view(conditions) + '_count', map_function(clazz, search_fields(conditions)), count_reduce_function, conditions, view_options).query_view!['rows'].first.try(:[], 'value') || 0
       end
       
       private
       
-      def query_view!(params, view_postfix = nil)
-        begin
-          query_view params, view_postfix
-        rescue RestClient::ResourceNotFound => e
-          create_view params
-          query_view params, view_postfix
-        end
-      end
-      
       def db(name = nil)
         ::CouchPotato::Persistence.Db(name)
       end
       
-      def create_view(params)
-        # in couchdb 0.9 we could use only 1 view and pass reduce=false for find and count with reduce
-        design_doc = db.get "_design/#{params[:design_document]}" rescue nil
-        db.save({
-          "_id" => "_design/#{params[:design_document]}",
-          :views => {
-            params[:view] => {
-              :map => map_function(params)
-            },
-            params[:view] + '_count' => {
-              :map => map_function(params),
-              :reduce => "function(keys, values) {
-                return values.length;
-              }"
-            }
-          }
-        }.merge(design_doc ? {'_rev' => design_doc['_rev']} : {}))
+      def design_document(clazz)
+        clazz.name.underscore
       end
       
-      def map_function(params)
+      def map_function(clazz, search_fields)
         "function(doc) {
-          if(doc.ruby_class == '#{params[:class]}') {
+          if(doc.ruby_class == '#{clazz}') {
             emit(
-              [#{params[:search_fields].map{|attr| "doc[\"#{attr}\"]"}.join(', ')}], doc
+              [#{search_fields.map{|attr| "doc[\"#{attr}\"]"}.join(', ')}], doc
                 );
           }
         }"
       end
       
-      def to_instances(clazz, query_result)
-        query_result['rows'].map{|doc| doc['value']}.map{|json| clazz.json_create json}
-      end
-      
-      def query_view(params, view_postfix)
-        db.view params[:view_url] + view_postfix.to_s, search_keys(params)
+      def count_reduce_function
+        "function(keys, values, combine) {
+          if (combine) {
+            return sum(values);
+          } else {
+            return values.length;
+          }
+        }"
       end
       
-      def search_keys(params)
-        if params[:search_values].select{|v| v.is_a?(Range)}.any?
-          {:startkey => params[:search_values].map{|v| v.is_a?(Range) ? v.first : v}, :endkey => params[:search_values].map{|v| v.is_a?(Range) ? v.last : v}}.merge(params[:view_options])
-        elsif params[:search_values].select{|v| v.is_a?(Array)}.any?
-          {:keys => prepare_multi_key_search(params[:search_values])}.merge(params[:view_options])
-        else
-          {:key => params[:search_values]}.merge(params[:view_options])
-        end
+      def to_instances(clazz, query_result)
+        query_result['rows'].map{|doc| doc['value']}.map{|json| clazz.json_create json}
       end
       
-      def prepare_multi_key_search(values)
-        array = values.select{|v| v.is_a?(Array)}.first
-        index = values.index array
-        array.map do |item|
-          copy = values.dup
-          copy[index] = item
-          copy
-        end
+      def view(conditions)
+        "by_#{view_name(conditions)}"
       end
       
-      def view_parameters(clazz, conditions, view_options)
-        {
-          :class => clazz,
-          :design_document => clazz.name.underscore,
-          :search_fields => conditions.to_a.sort_by{|f| f.first.to_s}.map(&:first),
-          :search_values => conditions.to_a.sort_by{|f| f.first.to_s}.map(&:last),
-          :view_options => view_options,
-          :view => "by_#{view_name(conditions)}",
-          :view_url => "#{clazz.name.underscore}/by_#{view_name(conditions)}"
-        }
+      def search_fields(conditions)
+        conditions.to_a.sort_by{|f| f.first.to_s}.map(&:first)
       end
       
       def view_name(options)

data/lib/couch_potato/persistence/inline_has_many_property.rb

@@ -19,6 +19,10 @@ class InlineHasManyProperty
     end
   end
   
+  def dirty?(object)
+    object.send("#{name}").dirty?
+  end
+  
   def save(object)
     
   end

data/lib/couch_potato/persistence/json.rb

@@ -36,7 +36,7 @@ module CouchPotato
           instance._id = json['_id']
           instance._rev = json['_rev']
           properties.each do |property|
-            property.build(instance, json)
+            property.build(instance, json) unless property.is_a?(ExternalHasManyProperty)
           end
           instance
         end

data/lib/couch_potato/persistence/simple_property.rb

@@ -6,10 +6,34 @@ module CouchPotato
       def initialize(owner_clazz, name, options = {})
         self.name = name
         owner_clazz.class_eval do
-          attr_accessor name
+          attr_reader name, "#{name}_was"
+          
+          def initialize(attributes = {})
+            super attributes
+            attributes.each do |name, value|
+              self.instance_variable_set("@#{name}_was", value)
+            end if attributes
+          end
+          
+          def self.json_create(json)
+            instance = super
+            instance.attributes.each do |name, value|
+              instance.instance_variable_set("@#{name}_was", value)
+            end
+            instance
+          end
+          
+          define_method "#{name}=" do |value|
+            self.instance_variable_set("@#{name}", value)
+          end
+          
           define_method "#{name}?" do
             !self.send(name).nil? && !self.send(name).try(:blank?)
           end
+          
+          define_method "#{name}_changed?" do
+            self.send(name) != self.send("#{name}_was")
+          end
         end
       end
       
@@ -17,6 +41,10 @@ module CouchPotato
         object.send "#{name}=", json.stringify_keys[name.to_s]
       end
       
+      def dirty?(object)
+        object.send("#{name}_changed?")
+      end
+      
       def save(object)
         
       end

data/lib/couch_potato/persistence/view_query.rb

@@ -0,0 +1,81 @@
+module CouchPotato
+  module Persistence
+
+    class ViewQuery
+      def initialize(design_document_name, view_name, map_function, reduce_function = nil, conditions = {}, view_options = {})
+        @design_document_name = design_document_name
+        @view_name = view_name
+        @map_function = map_function
+        @reduce_function = reduce_function
+        @conditions = conditions
+        @view_options = view_options
+      end
+      
+      def query_view!
+        begin
+          query_view
+        rescue RestClient::ResourceNotFound => e
+          create_view
+          query_view
+        end
+      end
+      
+      private
+      
+      def create_view
+        # in couchdb 0.9 we could use only 1 view and pass reduce=false for find and count with reduce
+        design_doc = db.get "_design/#{@design_document_name}" rescue nil
+        design_doc ||= {'views' => {}, "_id" => "_design/#{@design_document_name}"}
+        design_doc['views'][@view_name.to_s] = {
+          'map' => @map_function,
+          'reduce' => @reduce_function
+        }
+        db.save(design_doc)
+      end
+      
+      def db(name = nil)
+        ::CouchPotato::Persistence.Db(name)
+      end
+      
+      def query_view
+        db.view view_url, search_keys
+      end
+      
+      def view_url
+        "#{@design_document_name}/#{@view_name}"
+      end
+      
+      def search_keys
+        if search_values.select{|v| v.is_a?(Range)}.any?
+          {:startkey => search_values.map{|v| v.is_a?(Range) ? v.first : v}, :endkey => search_values.map{|v| v.is_a?(Range) ? v.last : v}}.merge(view_options)
+        elsif search_values.select{|v| v.is_a?(Array)}.any?
+          {:keys => prepare_multi_key_search(search_values)}.merge(view_options)
+        else
+          view_options.merge(search_values.any? ? {:key => search_values} : {})
+        end
+      end
+      
+      def search_values
+        conditions.to_a.sort_by{|f| f.first.to_s}.map(&:last)
+      end
+      
+      def view_options
+        @view_options
+      end
+      
+      def conditions
+        @conditions
+      end
+      
+      def prepare_multi_key_search(values)
+        array = values.select{|v| v.is_a?(Array)}.first
+        index = values.index array
+        array.map do |item|
+          copy = values.dup
+          copy[index] = item
+          copy
+        end
+      end
+    end
+  end
+end