query_methods_extend 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9c4039add270257fd9f5c8c1799aec94af2603eb
-  data.tar.gz: 7a45d215f5a8f2307e60ccbc8d7feb594f0e3ff5
+  metadata.gz: c1c09d9a5b6095985c753cae8723a064f231c8ad
+  data.tar.gz: 51d97ba8f08c24129ddd7c43ff2ec35d0836f3d3
 SHA512:
-  metadata.gz: 329c27fa87fc6f178f55ddb12d07fe2a5e69c11f74cd391d397ee1a0b5ce649a3041a99f55937290d4b79635bd08b3981d2e140881028a2a3eb6f9519114b660
-  data.tar.gz: bb9845d99f036e524aaf7037df007351ee02e63e8200927a7d2c6a3c6879ee641a7202c697e90f81f1ffefacaaa73d2c56c7b5076af73f7f883a2131958d623e
+  metadata.gz: 0e5a639e1397544856539f123a5a8a1fcce44bdf4557a8f53e067933bde27112180e9906f35f0ca225d23127bc4dd88f350373ecd408ab8852a97ddc4da27ddb
+  data.tar.gz: f38a9b1a4cb249ea2f1fd0396b3aee65bb267c0204a44078f0f10add77b84d2d7622a72c9acbd1838639e5891f6dfa917f6d8d3349bdc8d4407dd2d5fee4f78f

data/lib/query_methods_extend/association.rb

@@ -13,10 +13,32 @@ module QueryMethodsExtend
         association = self.try(:reflect_on_association, method_name)
         result = false
         if association && association.try(:macro) == :has_many
-          model_reflect = association.inverse_of
-          model = model_reflect.active_record
-          query = self.select(association.primary_key_column.name)
-          result = model.where("#{model.table_name}.#{model_reflect.foreign_key.to_sym} IN (#{query.to_sql})")
+          options = association.options
+
+          if options[:through]
+            @check_association_exist_flag = false
+            return self.all.send(options[:through]).send(method_name)
+          end
+
+          query = self.select(association.active_record.primary_key)
+
+          if options[:as] && !association.inverse_of && association.klass
+            polymorphic_as = association.klass.reflect_on_association(options[:as])
+            if polymorphic_as.polymorphic? && association.active_record
+              polymorphic_model = polymorphic_as.active_record
+              model = association.active_record
+              @check_association_exist_flag = false
+              return polymorphic_as.active_record
+                .where("#{polymorphic_model.table_name}.#{polymorphic_as.foreign_type}".to_sym => model.name)
+                .where("#{polymorphic_model.table_name}.#{polymorphic_as.foreign_key}".to_sym => query)
+            end
+          end
+
+          model = association.options[:class] || association.inverse_of.active_record
+          foreign_key = association.options[:foreign_key] || association.inverse_of.foreign_key
+
+          result = model.where("#{model.table_name}.#{foreign_key.to_s}".to_sym => query)
+
         end
 
         @check_association_exist_flag = false

data/lib/query_methods_extend/version.rb

@@ -1,3 +1,3 @@
 module QueryMethodsExtend
-  VERSION = '1.0.0'
+  VERSION = '1.0.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: query_methods_extend
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Alicuche
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-02-07 00:00:00.000000000 Z
+date: 2015-02-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -121,6 +121,7 @@ files:
 - lib/query_methods_extend/spec/union_spec.rb
 - lib/query_methods_extend/union.rb
 - lib/query_methods_extend/version.rb
+- query_methods_extend-1.0.0.gem
 - query_methods_extend.gemspec
 - spec/models/book.rb
 - spec/models/category.rb