rails_optimizer 0.1.2 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 51932ccdf691ba5d2b226ea43f5fb68dc1e45aad
-  data.tar.gz: 9529d9c3dd83c7f08ba150e42197d6dd2986e355
+  metadata.gz: 0dc3f68cf2eb84764efca80777738c551537ee33
+  data.tar.gz: beb196f15f49ee0c8c48d37e6950c490ed4fcbbd
 SHA512:
-  metadata.gz: 2f4f82db45e9fde9d63043102358e65a2b854de76f51c118056cfb237c6f0e779a424e5009759e35e0ffb7327cf6235c9eec46eca3ac7fe2939b9cd6ebd3275a
-  data.tar.gz: 803ddf88add1e53c52b6118676a821e51fe28b6a3efd30918c9774fc6ebd2f082fdcd73f51883aaf2810646003c51b8710be00383bdfa727b502630fcb990343
+  metadata.gz: a35ebf74d2a112055ec3b33b68411a6ff714710b85c44add6d1851980e17346fb2795cb7e94e42f03bb09fd2fc0786122fd2a57454e0282fa40756bdf55b2e43
+  data.tar.gz: 1850a36c5e5532467bc65da93beae6bdb6deccb66c94aeb7d3ac6e215d7914f7c0ce44b98726557e902b96dce0aba391fd93190becdb6958a98b2b3bf657bb39

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rails_optimizer (0.1.1)
+    rails_optimizer (0.1.6)
       activerecord
       activesupport
 

data/lib/rails_optimizer.rb

@@ -13,32 +13,7 @@ module RailsOptimizer
 			end
 		end
 
-		def self.scoped proc
-			return proc.call if proc
-			all
-		end
-
-		def self.finded obj, relation_name
-			fk = foreign_key(obj, relation_name)
-			find(obj.read_attribute(fk.to_sym)) unless obj.read_attribute(fk.to_sym).blank?
-		end
-
-		def self.foreign_key(obj, relation_name)
-			obj.class.reflections[relation_name.to_s].foreign_key
-		end
-
-		def self.belongs_to(name, scope = nil, **options)
-			super
-			define_method name.to_s do |*args|
-				if options[:polymorphic]
-					read_attribute("#{name}_type".to_sym).classify
-				else
-					name.to_s.classify
-				end.constantize.select(:id, *args).scoped(scope).finded(self, name)
-			end
-		end
-
-	  def self.has_one(name, scope = nil, **options)
+		def self.has_one(name, scope = nil, **options)
 			super
 			define_method name.to_s do |*args|
 				fk = self.class.foreign_key(name).to_sym
@@ -49,5 +24,44 @@ module RailsOptimizer
 				end
 			end
 		end
+
+		def self.belongs_to(name, scope = nil, **options)
+			super
+			_define(name, scope, options)
+		end
+
+		private
+
+			def self.scoped proc
+				return proc.call if proc
+				all
+			end
+
+			def self.finded obj, relation_name
+				fk = foreign_key(obj, relation_name)
+				find(obj.read_attribute(fk.to_sym)) unless obj.read_attribute(fk.to_sym).blank?
+			end
+
+			def self.foreign_key(obj, relation_name)
+				obj.class.reflections[relation_name.to_s].foreign_key
+			end
+
+			
+
+			def self._define(name, scope=nil, **options)
+				define_method name.to_s do |*args|
+					klass = if options[:polymorphic]
+						read_attribute("#{name}_type".to_sym).constantize
+					else
+						_reflections[name.to_s].klass
+					end
+					if args.empty?
+						klass.select('*').scoped(scope).finded(self, name)
+					else
+						klass.select(*args).scoped(scope).finded(self, name)
+					end
+				end
+			end
+	  	
 	end
 end

data/lib/rails_optimizer/version.rb

@@ -1,3 +1,3 @@
 module RailsOptimizer
-  VERSION = "0.1.2"
+  VERSION = "0.1.7"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails_optimizer
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.7
 platform: ruby
 authors:
 - Facundo A. Díaz Martínez