unpolymorph 0.1.2 → 0.2.0

data/lib/unpolymorph/version.rb

@@ -1,3 +1,3 @@
 module Unpolymorph
-  VERSION = "0.1.2"
+  VERSION = "0.2.0"
 end

data/lib/unpolymorph.rb

@@ -2,21 +2,35 @@ require "unpolymorph/version"
 
 module Unpolymorph
 
-  def unpolymorph( name, associations )
+  def unpolymorph( name, associations, options = {} )
 
     scope "for_#{name}", -> obj { where( "#{obj.class.base_class.to_s.underscore}_id" => obj.id ) }
 
+    if options[:create_associations]
+      associations.each do |assoc|
+        belongs_to assoc
+      end
+    end
+
     define_method( name ) do
+      var_name = "@_unpolymorph_#{name}"
+      value = instance_variable_get( var_name )
+      return value if value
       associations.each do |assoc|
-        associated = send( assoc )
-        return associated if associated
+        value = send( assoc )
+        if value
+          instance_variable_set( var_name, value )
+          return value
+        end
       end
+      return nil
     end
 
     define_method( "#{name}=" ) do |value|
       associations.each do |assoc|
         if value.class.base_class.to_s.underscore == assoc.to_s
           send( "#{assoc}=", value )
+          instance_variable_set( "@_unpolymorph_#{name}", value )
           return
         end
       end
@@ -25,9 +39,10 @@ module Unpolymorph
     
     # For use in migrations
     define_singleton_method( "update_sql_#{name}_check" ) do |migration|
-      constraint = "exactly_one_unpolymorphic_#{name}"
+      required = options.has_key?( :required ) ? options[:required] : true
+      constraint = required ? "exactly_one_unpolymorphic_#{name}" : "at_most_one_unpolymorphic_#{name}"
       migration.execute( "ALTER TABLE #{table_name} DROP CONSTRAINT IF EXISTS #{constraint}" )
-      migration.execute( "ALTER TABLE #{table_name} ADD CONSTRAINT #{constraint} CHECK (#{associations.collect { |a| "(#{a}_id IS NOT NULL)::int" }.join( " + ")} = 1)" )
+      migration.execute( "ALTER TABLE #{table_name} ADD CONSTRAINT #{constraint} CHECK (#{associations.collect { |a| "(#{a}_id IS NOT NULL)::int" }.join( " + ")} #{required ? "=" : "<="} 1)" )
     end
 
     define_singleton_method( "possible_#{name}?" ) do |object|

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: unpolymorph
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 23
   prerelease: 
   segments: 
   - 0
-  - 1
   - 2
-  version: 0.1.2
+  - 0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Marek Janukowicz
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2013-06-06 00:00:00 +02:00
+date: 2013-06-25 00:00:00 +02:00
 default_executable: 
 dependencies: []