couch_potato-rails2 0.5.6

data/lib/couch_potato/persistence/active_model_compliance.rb

@@ -0,0 +1,44 @@
+module CouchPotato
+  module Persistence
+    module ActiveModelCompliance
+      begin
+        require 'active_model'
+
+        def self.included(base)
+          base.extend ClassMethods
+        end
+
+        def to_model
+          self
+        end
+
+        def errors
+          super || {}
+        end
+        
+        def persisted?
+          !self.new?
+        end
+        
+        def to_key
+          persisted? ? [to_param] : nil
+        end
+
+        def destroyed?
+          !!_deleted
+        end
+
+        module ClassMethods
+          def model_name
+            @model_name ||= ::ActiveModel::Name.new(self)
+          end
+        end
+
+      rescue LoadError, NameError
+        # if it's not installed you probably don't want to use it anyway
+      end
+    end
+  end
+end
+
+

data/lib/couch_potato/persistence/attachments.rb

@@ -0,0 +1,31 @@
+module CouchPotato
+  module Attachments
+    def self.included(base) #:nodoc:
+      base.class_eval do
+        attr_writer :_attachments
+        
+        def _attachments
+          @_attachments ||= {}
+        end
+        
+        base.extend ClassMethods
+      end
+    end
+    
+    def to_hash
+      if _attachments && _attachments.any?
+        super.merge('_attachments' => _attachments)
+      else
+        super
+      end
+    end
+    
+    module ClassMethods
+      def json_create(json)
+        instance = super
+        instance._attachments = json['_attachments'] if json
+        instance
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/callbacks.rb

@@ -0,0 +1,29 @@
+require 'active_support/concern'
+require 'active_model/callbacks'
+
+module CouchPotato
+  module Persistence
+    module Callbacks
+      def self.included(base) #:nodoc:
+        base.extend ActiveModel::Callbacks
+
+        base.class_eval do
+          attr_accessor :skip_callbacks
+
+          define_model_callbacks :create, :save, :update, :destroy
+          define_model_callbacks *[:save, :create, :update].map {|c| :"validation_on_#{c}"}
+          define_model_callbacks :validation unless Config.validation_framework == :active_model
+        end
+      end
+
+      # Runs all callbacks on a model with the given name, e.g. :after_create.
+      # 
+      # This method is called by the CouchPotato::Database object when saving/destroying an object 
+      def run_callbacks(name, &block)
+        return if skip_callbacks
+
+        send(:"_run_#{name}_callbacks", &block)
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/dirty_attributes.rb

@@ -0,0 +1,56 @@
+require 'bigdecimal'
+module CouchPotato
+  module Persistence
+    module DirtyAttributes
+      
+      def self.included(base) #:nodoc:
+        base.send :include, ActiveModel::Dirty
+        base.class_eval do
+          after_save :reset_dirty_attributes
+        end
+      end
+      
+      def initialize(attributes = {})
+        super
+      end
+    
+      # returns true if a model has dirty attributes, i.e. their value has changed since the last save
+      def dirty? 
+        changed? || @forced_dirty
+      end
+      
+      # marks a model as dirty
+      def is_dirty
+        @forced_dirty = true
+      end
+      
+      def method_missing(name, *args)
+        if(name.to_s.include?('_will_change!'))
+          self.class.define_attribute_methods self.class.property_names
+          send(name, *args)
+        else
+          super
+        end
+      end
+      
+      private
+      
+      def reset_dirty_attributes
+        @previously_changed = changes
+        @changed_attributes.clear
+        @forced_dirty = nil
+      end
+      
+      def clone_attribute(value)
+        if [Fixnum, Symbol, TrueClass, FalseClass, NilClass, Float, BigDecimal].include?(value.class)
+          value
+        elsif [Hash, Array].include?(value.class)
+          #Deep clone
+          Marshal::load(Marshal::dump(value))
+        else
+          value.clone
+        end
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/ghost_attributes.rb

@@ -0,0 +1,12 @@
+module CouchPotato
+  module GhostAttributes #:nodoc:
+    def method_missing(name, *args)
+      if(value = _document && _document[name.to_s])
+        value
+      else
+        super
+      end
+    end
+  end
+end
+

data/lib/couch_potato/persistence/json.rb

@@ -0,0 +1,47 @@
+module CouchPotato
+  module Persistence
+    module Json
+      def self.included(base) #:nodoc:
+        base.class_eval do
+          extend ClassMethods
+          attr_accessor :_document
+        end
+      end
+      
+      # returns a JSON representation of a model in order to store it in CouchDB
+      def to_json(*args)
+        to_hash.to_json(*args)
+      end
+      
+      # returns all the attributes, the ruby class and the _id and _rev of a model as a Hash
+      def to_hash
+        (self.class.properties).inject({}) do |props, property|
+          property.serialize(props, self)
+          props
+        end.merge(JSON.create_id => self.class.name).merge(id_and_rev_json)
+      end
+      
+      private
+      
+      def id_and_rev_json
+        ['_id', '_rev', '_deleted'].inject({}) do |hash, key|
+          hash[key] = self.send(key) unless self.send(key).nil?
+          hash
+        end
+      end
+      
+      module ClassMethods
+        
+        # creates a model instance from JSON
+        def json_create(json)
+          return if json.nil?
+          instance = self.new
+          instance._id = json[:_id] || json['_id']
+          instance._rev = json[:_rev] || json['_rev']
+          instance._document = HashWithIndifferentAccess.new(json)
+          instance
+        end
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/magic_timestamps.rb

@@ -0,0 +1,23 @@
+require 'active_support/time'
+
+module CouchPotato
+  module MagicTimestamps #:nodoc:
+    def self.included(base)
+      base.instance_eval do
+        property :created_at, :type => Time
+        property :updated_at, :type => Time
+        
+        before_create lambda {|model|
+          model.created_at ||= (Time.zone || Time).now
+          @changed_attributes.delete 'created_at'
+          model.updated_at ||= (Time.zone || Time).now
+          @changed_attributes.delete 'updated_at'
+        }
+        before_update lambda {|model|
+          model.updated_at = (Time.zone || Time).now
+          @changed_attributes.delete 'updated_at'
+        }
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/properties.rb

@@ -0,0 +1,79 @@
+require File.dirname(__FILE__) + '/simple_property'
+
+module CouchPotato
+  module Persistence
+    module Properties
+      class PropertyList
+        include Enumerable
+        
+        attr_accessor :list
+        
+        def initialize(clazz)
+          @clazz = clazz
+          @list = []
+        end
+        
+        def each
+          (list + inherited_properties).each {|property| yield property}
+        end
+        
+        def <<(property)
+          @list << property
+        end
+        
+        def inherited_properties
+          superclazz = @clazz.superclass
+          properties = []
+          while superclazz && superclazz.respond_to?(:properties)
+            properties << superclazz.properties.list
+            superclazz = superclazz.superclass
+          end
+          properties.flatten
+        end
+      end
+      
+      def self.included(base) #:nodoc:
+        base.extend ClassMethods
+        base.class_eval do
+          def self.properties
+            @properties ||= {}
+            @properties[name] ||= PropertyList.new(self)
+          end
+        end
+      end
+      
+      def type_caster #:nodoc:
+        @type_caster ||= TypeCaster.new
+      end
+      
+      module ClassMethods
+        # returns all the property names of a model class that have been defined using the #property method
+        #
+        # example:
+        #  class Book
+        #    property :title
+        #    property :year
+        #  end
+        #  Book.property_names # => [:title, :year]
+        def property_names
+          properties.map(&:name)
+        end
+
+        # Declare a property on a model class. Properties are not typed by default.
+        # You can store anything in a property that can be serialized into JSON.
+        # If you want a property to be of a custom class you have to define it using the :type option.
+        #
+        # example:
+        #  class Book
+        #    property :title
+        #    property :year
+        #    property :publisher, :type => Publisher
+        #  end
+        def property(name, options = {})
+          properties << SimpleProperty.new(self, name, options)
+        end
+
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/simple_property.rb

@@ -0,0 +1,82 @@
+module CouchPotato
+  module Persistence
+    module PropertyMethods
+      private
+      
+      def load_attribute_from_document(name)
+        if (_document || {}).has_key?(name.to_s)
+          property = self.class.properties.find{|property| property.name == name}
+          @skip_dirty_tracking = true
+          value = property.build(self, _document)
+          @skip_dirty_tracking = false
+          value
+        end
+      end
+    end
+    
+    class SimpleProperty  #:nodoc:
+      attr_accessor :name, :type
+      
+      def initialize(owner_clazz, name, options = {})
+        self.name = name
+        self.type = options[:type]
+        @type_caster = TypeCaster.new
+        owner_clazz.send :include, PropertyMethods unless owner_clazz.ancestors.include?(PropertyMethods)
+        
+        define_accessors accessors_module_for(owner_clazz), name, options
+      end
+      
+      def build(object, json)
+        value = json[name.to_s].nil? ? json[name.to_sym] : json[name.to_s]
+        object.send "#{name}=", value
+      end
+      
+      def dirty?(object)
+        object.send("#{name}_changed?")
+      end
+      
+      def serialize(json, object)
+        json[name] = object.send name
+      end
+      alias :value :serialize
+      
+      private
+      
+      def accessors_module_for(clazz)
+        unless clazz.const_defined?('AccessorMethods')
+          accessors_module = clazz.const_set('AccessorMethods', Module.new) 
+          clazz.send(:include, accessors_module)
+        end
+        clazz.const_get('AccessorMethods')
+      end
+      
+      def define_accessors(base, name, options)
+        base.class_eval do
+          include PropertyMethods
+          
+          define_method "#{name}" do
+            load_attribute_from_document(name) unless instance_variable_defined?("@#{name}")
+            value = instance_variable_get("@#{name}")
+            if value.nil? && options[:default]
+              default = clone_attribute(options[:default])
+              self.instance_variable_set("@#{name}", default)
+              default
+            else
+              value
+            end
+          end
+          
+          define_method "#{name}=" do |value|
+            typecasted_value = type_caster.cast(value, options[:type])
+            send("#{name}_will_change!") unless @skip_dirty_tracking || typecasted_value == send(name)
+            self.instance_variable_set("@#{name}", typecasted_value)
+          end
+          
+          define_method "#{name}?" do
+            !self.send(name).nil? && !self.send(name).try(:blank?)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/couch_potato/persistence/type_caster.rb

@@ -0,0 +1,40 @@
+module CouchPotato
+  module Persistence
+    class TypeCaster #:nodoc:
+      def cast(value, type)
+        if type == :boolean
+          cast_boolean(value)
+        else
+          cast_native(value, type)
+        end
+      end
+
+      private
+
+      def cast_boolean(value)
+        if [FalseClass, TrueClass].include?(value.class) || value.nil?
+          value
+        elsif [0, '0', 'false'].include?(value)
+          false
+        else
+          true
+        end
+      end
+
+      def cast_native(value, type)
+        if type && !value.instance_of?(type)
+          if type == Fixnum
+            value.to_s.scan(/-?\d+/).join.to_i unless value.blank?
+          elsif type == Float
+            value.to_s.scan(/-?\d+\.?\d*/).join.to_f unless value.blank?
+          else
+            type.json_create value unless value.blank?
+          end
+        else
+          value
+        end
+      end
+
+    end
+  end
+end