ninja-model 0.4.2

data/lib/ninja_model/railtie.rb

@@ -0,0 +1,27 @@
+require 'ninja_model'
+require 'rails/all'
+
+module NinjaModel
+  class Railtie < Rails::Railtie
+
+    config.ninja_model = NinjaModel::Configuration.create
+
+    #config.generators.orm :ninja_model, :migration => false
+
+    initializer 'ninja_model.logger' do |app|
+      NinjaModel::set_logger(Rails.logger)
+    end
+
+    config.after_initialize do |app|
+      config_path = File.join(Rails.root, app.config.ninja_model.config_file_path)
+      if File.exists?(config_path)
+        require 'erb'
+        require 'yaml'
+        app.config.ninja_model.specs = YAML::load(ERB.new(IO.read(config_path)).result)
+        NinjaModel::Base.set_adapter
+      else
+        NinjaModel.logger.warn "[ninja-model] *WARNING* Unable to find configuration file at #{config_path}"
+      end
+    end
+  end
+end

data/lib/ninja_model/reflection.rb

@@ -0,0 +1,118 @@
+module NinjaModel
+  module Reflection
+    extend ActiveSupport::Concern
+
+    module ClassMethods
+      def create_reflection(macro, name, options, ninja_model)
+        case macro
+        when :has_many, :belongs_to, :has_one
+          reflection = AssociationReflection.new(macro, name, options, ninja_model)
+        when :composed_of
+        end
+        write_inheritable_hash :reflections, name => reflection
+        reflection
+      end
+
+      def ninja_model?(macro, association)
+        klass = association.to_s.camelize
+        klass = klass.singularize unless [:has_one, :belongs_to].include?(macro)
+        klass = klass.constantize
+        defined?(klass) && klass.ancestors.include?(NinjaModel::Base)
+      end
+
+      def reflections
+        read_inheritable_attribute(:reflections) || write_inheritable_attribute(:reflections, {})
+      end
+
+      def reflect_on_association(association)
+        reflections[association].is_a?(AssociationReflection) ? reflections[association] : nil
+      end
+    end
+
+    class MacroReflection
+      def initialize(macro, name, options, ninja_model)
+        @macro, @name, @options, @ninja_model = macro, name, options, ninja_model
+      end
+
+      attr_reader :ninja_model, :name, :macro, :options
+
+      def klass
+        @klass ||= class_name.constantize
+      end
+
+      def class_name
+        @class_name ||= options[:class_name] || derive_class_name
+      end
+
+      private
+
+      def derive_class_name
+        name.to_s.camelize
+      end
+    end
+
+    class AssociationReflection < MacroReflection
+      def initialize(macro, name, options, ninja_model)
+        super
+        @collection = [:has_many].include?(macro)
+      end
+
+      def build_association(*options)
+        klass.new(*options)
+      end
+
+      def create_association(*options)
+        klass.create(*options)
+      end
+
+      def create_association!(*options)
+        klass.create!(*options)
+      end
+
+      def primary_key_name
+        @primary_key_name ||= options[:foreign_key] || derive_primary_key_name
+      end
+
+      def association_foreign_key
+        @association_foreign_key ||= @options[:association_foreign_key] || class_name.foreign_key
+      end
+
+      def check_validity!
+        check_validity_of_inverse!
+      end
+
+      def check_validity_of_inverse!
+      end
+
+      def collection?
+        @collection
+      end
+
+      def validate?
+        !options[:validate].nil? ? options[:validate] : (options[:autosave] == true || macro == :has_many)
+      end
+
+      private
+
+      def belongs_to?
+        macro == :belongs_to
+      end
+
+      def derive_class_name
+        class_name = name.to_s.camelize
+        class_name = class_name.singularize if collection?
+        class_name
+      end
+
+      def derive_primary_key_name
+        if belongs_to?
+          "#{name}_id"
+        elsif options[:as]
+          "#{options[:as]}_id"
+        else
+          ninja_model.name.foreign_key
+        end
+      end
+    end
+  end
+end

data/lib/ninja_model/relation/finder_methods.rb

@@ -0,0 +1,74 @@
+module NinjaModel
+  module FinderMethods
+    def first(*args)
+      if args.any?
+      else
+        find_first
+      end
+    end
+
+    def last(*args)
+      if args.any?
+      else
+      end
+    end
+
+    def all(*args)
+      args.any? ? apply_finder_options(args.first).to_a : to_a
+    end
+
+    def find(*args)
+      options = args.extract_options!
+
+      if options.present?
+        apply_finder_options(options).find(*args)
+      else
+        case args.first
+        when :first, :last, :all
+          send(args.first)
+        else
+          find_with_ids(*args)
+        end
+      end
+    end
+
+    def exists?(id)
+      where(primary_key.to_sym => id).limit(1)
+      relation.first ? true : false
+    end
+
+    protected
+
+    def find_with_ids(*ids)
+      expects_array = ids.first.kind_of?(Array)
+
+      return ids.first if expects_array && ids.first.empty?
+
+      ids = ids.flatten.compact.uniq
+
+      case ids.size
+      when 0
+        raise RecordNotFound, "Couldn't find #{@klass.name} without an ID"
+      when 1
+        result = find_one(ids.first)
+        expects_array ? [result] : result
+      else
+        find_some(ids)
+      end
+    end
+
+    def find_one(id)
+      id = id.id if NinjaModel::Base === id
+
+      where(primary_key.to_sym => id).first
+    end
+
+    def find_first
+      if loaded?
+        @records.first
+      else
+        @first ||= limit(1).to_a[0]
+      end
+    end
+  end
+end

data/lib/ninja_model/relation/query_methods.rb

@@ -0,0 +1,62 @@
+require 'active_support'
+
+module NinjaModel
+  module QueryMethods
+    extend ActiveSupport::Concern
+
+    def order(*args)
+      relation = clone
+      relation.ordering += args.flatten unless args.blank?
+      relation
+    end
+
+    def where(opts, *rest)
+      relation = clone
+      relation.predicates += build_predicates(opts, rest)
+      relation
+    end
+
+    def limit(value)
+      relation = clone
+      relation.limit_value = value
+      relation
+    end
+
+    private
+
+    def build_predicates(opts, other = [])
+      case opts
+      when String
+        raise ArgumentError,
+          "NinjaModel doesn't work with strings...yet. You'll need to use a predicate (see NinjaModel::Predicate::PREDICATES for a list)."
+      when Array
+        opts.collect do |o|
+          build_predicates(o)
+        end.flatten!
+      when Hash
+        opts.to_a.map do |o|
+          raise ArgumentError, "Not sure what to do with #{o}" unless o.length.eql?(2)
+          k = o[0]
+          v = o[1]
+
+          case k
+          when NinjaModel::Predicate
+            k.value = klass.model_attributes_hash[k.attribute].convert(v)
+            k
+          when Symbol, String
+            raise ArgumentError, "#{klass} doesn't have an attribute #{k}." unless klass.attribute_method?(k)
+            p = NinjaModel::Predicate.new(k.to_sym, :eq)
+            p.value = klass.model_attributes_hash[p.attribute].convert(v)
+            p
+          else
+            raise ArgumentError, "#{k} isn't a predicate or a symbol."
+          end
+        end
+      when NinjaModel::Predicate
+        [opts]
+      else
+        raise ArgumentError, "Unknown argument to #{self}.where: #{opts.inspect}"
+      end
+    end
+  end
+end

data/lib/ninja_model/relation/spawn_methods.rb

@@ -0,0 +1,59 @@
+require 'active_support'
+
+module NinjaModel
+  module SpawnMethods
+
+    def merge(r)
+      merged_relation = clone
+      return merged_relation unless r
+      return to_a & r if r.is_a?(Array)
+
+      order_value = r.ordering
+      if order_value.present?
+        merged_relation.ordering = merged_relation.ordering + order_value
+      end
+
+      merged_predicates = @predicates + r.predicates
+
+      unless @predicates.empty?
+        seen = []
+        merged_predicates = merged_predicates.reverse.reject { |w|
+          nuke = false
+          if w.respond_to?(:operator) && w.operator == :==
+            attribute = w.attribute
+            nuke = seen[attribute]
+            seen[attribute] = true
+          end
+          nuke
+        }.reverse
+      end
+      merged_relation.predicates = merged_predicates
+
+      Relation::SINGLE_VALUE_ATTRS.reject { |m| m == :lock }.each do |method|
+        value = r.send(:"#{method}_value")
+        merged_relation.send(:"#{method}_value=", value) unless value.nil?
+      end
+
+      merged_relation
+
+    end
+
+    VALID_FIND_OPTIONS = [:conditions, :limit, :offset, :order]
+
+    def apply_finder_options(options)
+      relation = clone
+      return relation unless options
+
+      options.assert_valid_keys(VALID_FIND_OPTIONS)
+      finders = options.dup
+      finders.delete_if { |key, value| value.nil? }
+
+      ([:order, :limit, :offset] & finders.keys).each do |finder|
+        relation = relation.send(finder, finders[finder])
+      end
+
+      relation = relation.where(finders[:conditions]) if options.key?(:conditions)
+      relation
+    end
+  end
+end

data/lib/ninja_model/relation.rb

@@ -0,0 +1,80 @@
+require 'active_support/core_ext/object/blank'
+
+module NinjaModel
+  class Relation
+    include QueryMethods, FinderMethods, SpawnMethods
+
+    delegate :length, :each, :map, :collect, :all?, :include?, :to => :to_a
+
+    attr_reader :klass, :loaded
+
+    attr_accessor :ordering, :predicates, :limit_value, :offset_value
+
+    alias :loaded? :loaded
+
+    SINGLE_VALUE_ATTRS = [:limit, :offset]
+    MULTI_VALUE_ATTRS = [:ordering, :predicates]
+
+    def initialize(klass)
+      @klass  = klass
+      @loaded = false
+
+      SINGLE_VALUE_ATTRS.each do |v|
+        instance_variable_set("@#{v}_value".to_sym, nil)
+      end
+
+      MULTI_VALUE_ATTRS.each do |v|
+        instance_variable_set("@#{v}".to_sym, [])
+      end
+    end
+
+    def to_a
+      @records ||= begin
+        records = @klass.adapter.read(self)
+        @loaded = true
+        records
+      end
+    end
+    alias :to_ary :to_a
+
+    def scoping
+      @klass.scoped_methods << self
+      begin
+        yield
+      ensure
+        @klass.scoped_methods.pop
+      end
+    end
+
+    def size
+      to_a.length
+    end
+
+    def blank?
+      empty?
+    end
+
+    def empty?
+      size.zero?
+    end
+
+    protected
+
+    alias :inspect! :inspect
+    def inspect
+      to_a.inspect
+    end
+
+    def method_missing(method, *args, &block)
+      if Array.method_defined?(method)
+        to_a.send(method, *args, &block)
+      elsif @klass.scopes[method]
+        merge(@klass.send(method, *args, &block))
+      elsif @klass.respond_to?(method)
+        scoping { @klass.send(method, *args, &block) }
+      else
+        super
+      end
+    end
+  end
+end

data/lib/ninja_model/scoping.rb

@@ -0,0 +1,50 @@
+require 'active_support'
+require 'active_support/core_ext/array'
+require 'active_support/core_ext/kernel/singleton_class'
+
+module NinjaModel
+  module Scoping
+    extend ActiveSupport::Concern
+
+    module ClassMethods
+      def scoped(options = nil)
+        if options
+          scoped.apply_finder_options(options)
+        else
+          current_scoped_methods ? relation.merge(current_scoped_methods) : relation.clone
+        end
+      end
+
+      def scopes
+        read_inheritable_attribute(:scopes) || write_inheritable_attribute(:scopes, {})
+      end
+
+      def scope(name, scope_options = {}, &block)
+        name = name.to_sym
+        valid_scope_name?(name)
+        extension = Module.new(&block) if block_given?
+
+        scopes[name] = lambda do |*args|
+          options = scope_options.is_a?(Proc) ? scope_options.call(*args) : scope_options
+          relation = if options.is_a?(Hash)
+                       scoped.apply_finder_options(options)
+                     elsif options
+                       scoped.merge(options)
+                     else
+                       scoped
+                     end
+
+          extension ? relation.extending(extension) : relation
+        end
+
+        singleton_class.send(:redefine_method, name, &scopes[name])
+      end
+
+      def valid_scope_name?(name)
+        if !scopes[name] && respond_to?(name, true)
+          logger.warn "Creating scope :#{name}.  Overwriting existing method #{self.name}.#{name}."
+        end
+      end
+    end
+  end
+end

data/lib/ninja_model/validation.rb

@@ -0,0 +1,38 @@
+require 'active_model'
+
+module NinjaModel
+  module Validation
+    extend ActiveSupport::Concern
+    include ActiveModel::Validations
+
+    def save(options={})
+      run_callbacks :validation do
+        perform_validations(options) ? super : false
+      end
+    end
+
+    def valid?(context = nil)
+      context ||= (persisted? ? :update : :create)
+      output = super(context)
+      errors.empty? && output
+    end
+
+    protected
+
+    def perform_validations(options={})
+      perform_validation = case options
+        when Hash
+          options[:validate] != false
+        else
+          ActiveSupport::Deprecation.warn "save(#{options}) is deprecated, please give save(:validate => #{options}) instead", caller
+          options
+        end
+
+      if perform_validation
+        valid?(options.is_a?(Hash) ? options[:context] : nil)
+      else
+        true
+      end
+    end
+  end
+end

data/lib/ninja_model/version.rb

@@ -0,0 +1,3 @@
+module NinjaModel
+ VERSION = "0.4.2"
+end

data/lib/ninja_model.rb

@@ -0,0 +1,38 @@
+require 'active_support'
+require 'active_support/core_ext/hash/indifferent_access'
+
+module NinjaModel
+  extend ActiveSupport::Autoload
+
+  autoload :Base
+  autoload :Adapters
+  autoload :Associations
+  autoload :Attributes
+  autoload :Callbacks
+  autoload :Configuration
+  autoload :Identity
+  autoload :Persistence
+  autoload :Predicate
+  autoload :Reflection
+  autoload :Relation
+  autoload :Scoping
+  autoload :Validation
+
+  autoload_under 'relation' do
+    autoload :QueryMethods
+    autoload :FinderMethods
+    autoload :SpawnMethods
+  end
+
+
+  class << self
+    attr_accessor :logger
+
+    def set_logger(logger)
+      ::NinjaModel.logger = logger
+    end
+  end
+end
+
+require 'ninja_model/railtie'
+require 'ninja_model/errors'

data/spec/ninja_model/adapters/adapter_pool_spec.rb

@@ -0,0 +1,72 @@
+require 'spec_helper'
+require 'ninja_model'
+
+describe NinjaModel::Adapters::AdapterPool do
+  before(:each) do
+    @spec = mock('AdapterSpecification') do
+      stubs(:adapter_method).returns(:dummy_adapter)
+      stubs(:config)
+    end
+
+    @adapter = mock('AbstractAdapter') do
+      stubs(:verify!)
+    end
+
+    @pool = NinjaModel::Adapters::AdapterPool.new(@spec)
+    @thread = mock('thread') do
+      stubs('object_id').returns(123)
+    end
+    Thread.stubs(:current).returns(@thread)
+    NinjaModel::Base.stubs(:dummy_adapter).returns(@adapter)
+  end
+
+  it 'should properly respond to connected?' do
+    @pool.connected?.should be_false
+    @pool.instance
+    @pool.connected?.should be_true
+  end
+
+  it 'should initialize from an AdapterSpecification' do
+    spec = mock('AdapterSpecification')
+    NinjaModel::Adapters::AdapterPool.new(spec).spec.should eql(spec)
+  end
+
+  it 'should return a valid instance_id' do
+    (@pool.send :current_instance_id).should eql(@thread.object_id)
+  end
+
+  describe 'when creating an instance' do
+    it 'should create a valid instance' do
+      (@pool.send :new_instance).should eql(@adapter)
+    end
+
+    it 'should store the instance' do
+      @pool.stubs(:new_instance).returns(@adapter)
+      @pool.instance
+      @pool.instances.should include(@adapter)
+    end
+
+    it 'should assign the instance to the current_thread' do
+      @pool.stubs(:new_instance).returns(@adapter)
+      @pool.instance
+      @pool.stubs(:checkout).returns("new_instance")
+      @pool.instance.should_not eql("new_instance")
+      @pool.instance.should eql(@adapter)
+    end
+
+    it 'should reuse an unallocated instance' do
+      @pool.stubs(:current_instance_id).returns(123)
+      inst = @pool.instance
+      @pool.release_instance(123)
+      @pool.instance.should eql(inst)
+    end
+  end
+
+  it 'should properly allocate/deallocate instances' do
+    @pool.stubs(:current_instance_id).returns(123)
+    @pool.expects(:checkout).returns(@adapter)
+    @pool.expects(:checkin)
+    @pool.with_instance do |instance|
+    end
+  end
+end

data/spec/ninja_model/adapters_spec.rb

@@ -0,0 +1,8 @@
+require 'spec_helper'
+
+describe NinjaModel::Adapters do
+  describe 'set_adapter' do
+
+  end
+
+end