throughcheckboxes 0.0.2 → 0.0.3

data/lib/throughcheckboxes/checkboxes_for.rb

@@ -5,19 +5,35 @@ module ThroughCheckboxes
         extend ActiveSupport::Concern
 
         module ClassMethods
-            def checkboxes_for(*args)
-                for relation in args
+
+            def checkboxes_for(*relations)
+
+                klass = self
+
+                ThroughCheckboxes::Core.validate(klass, *relations)
+
+                for relation in relations
+
+                    field_name="#{relation}_ids"
+
+                    class_eval do
+                        attr_accessible field_name.to_sym
+                    end
+
                     instance_eval do
-                        define_method "#{relation}_ids=" do |ids|
-                            self.send("#{relation}=",ids.select {|id| id.present?}.map {|id| relation.to_s.classify.constantize.find(id)})
+                        define_method "#{field_name}=" do |ids|
+                            send("#{relation}=",ids.select { |id| id.present? }.map {|id| ThroughCheckboxes::Core.relation_klass(klass, relation).find(id)})
                         end
 
-                        define_method "#{relation}_ids" do
-                            self.send("#{relation}").map {|e| e.id}
+                        define_method field_name do
+                            send("#{relation}").map {|e| e.id}
                         end
                     end
+
                 end
+
             end
+
         end
 
         module InstanceMethods

data/lib/throughcheckboxes/checkboxes_for_helpers.rb

@@ -7,7 +7,7 @@ module ThroughCheckboxes
             def checkboxes_for(relation)
                 output = ""
                 output << "#{self.label(relation)}<br />"
-                for record in relation.to_s.classify.constantize.all
+                for record in ThroughCheckboxes::Core.relation_klass(self.object.class, relation).all
                     output << @template.check_box_tag("#{@object_name}[#{relation}_ids][]",record.id,self.object.send(relation).include?(record))
                     output << "#{record}<br />"
                 end

data/lib/throughcheckboxes/core.rb

@@ -0,0 +1,26 @@
+module ThroughCheckboxes
+
+    module Core
+
+        class ThroughCheckboxesNameError <  StandardError; end
+
+        def self.validate(klass, *relations)
+            for relation in relations
+                unless (klass.reflections[relation].present? && 
+                        klass.reflections[relation].macro == :has_many &&
+                        klass.reflections[relation].options[:through].present?)
+                    raise ThroughCheckboxesNameError, "#{relation} isn't an has_many :through for model #{klass}, check it out please."
+                end
+            end
+        end
+
+        def self.relation_klass(klass, relation)
+            source=klass.reflections[relation].options[:source]
+            relation_klass = source.present? ? source.to_s : relation.to_s
+            relation_klass.classify.constantize
+        end
+
+
+    end
+    
+end

data/lib/throughcheckboxes/version.rb

@@ -1,3 +1,3 @@
 module Throughcheckboxes
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/throughcheckboxes.rb

@@ -1,3 +1,4 @@
+require 'throughcheckboxes/core'
 require 'throughcheckboxes/checkboxes_for'
 require 'throughcheckboxes/checkboxes_for_helpers'
 

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - lucapette
@@ -35,6 +35,7 @@ files:
 - lib/throughcheckboxes.rb
 - lib/throughcheckboxes/checkboxes_for.rb
 - lib/throughcheckboxes/checkboxes_for_helpers.rb
+- lib/throughcheckboxes/core.rb
 - lib/throughcheckboxes/version.rb
 - throughcheckboxes.gemspec
 has_rdoc: true