fides 0.0.6 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 11810cdba23bbe7edf554320cab120f33c466d94
-  data.tar.gz: 0af72bcf6dcaf6a0d9264884f952d29735d9a0de
+  metadata.gz: 7a7f704749dcd05ef1186b0a46f0cad900ab00ed
+  data.tar.gz: 9f933f47a0b8fd70fe54fb1cb816a376cd42730d
 SHA512:
-  metadata.gz: 640ee69bf5632a32caffd9195d41e8b86374c9f8d77fd7b1c024bc22104cca22a96b359ea1ec91886531b0bfd28dd04097b8a80af82c42d6ef7bda2f55326876
-  data.tar.gz: 9a250b1639d0a41c949ce6e9dcb9d8767b0d7e956e76aeec3f9ba46ea67cfc615a51c599edcec508d7698b73c4438ff2cd17a9bf958647ace9ad40831a785ce7
+  metadata.gz: dbe549233fb1bd8f5bdcd42dea5c2ab84e4a5e958b7322db74a35ae5c2ac439dd4c691ed4fa5c879a73efde0b671adf458c847ec8bda676b09411789f7dd4a5f
+  data.tar.gz: 692e61dafb6e55e1367bcb66ca473c1a80957bcb22f0770fe76539c60561fa74c1fcae0761ed3a44a4d1648d77c2f8cac297a9486a64a569e90c49717ad5697f

data/lib/fides/version.rb

@@ -1,3 +1,3 @@
 module Fides
-  VERSION = "0.0.6"
+  VERSION = "0.0.9"
 end

data/lib/fides.rb

@@ -7,118 +7,115 @@ module Fides
 
   extend ActiveSupport::Concern
 
-  module ClassMethods
+  def add_polymorphic_triggers(opts)
+    raise ArgumentError, "missing :associated_models from options hash" if !opts.has_key?(:associated_models) 
+    raise ArgumentError, "missing :polymorphic_model from options hash" if !opts.has_key?(:polymorphic_model)
+    associated_models = opts[:associated_models]
+    polymorphic_model = opts[:polymorphic_model]
+    interface_name = opts.has_key?(:interface_name) ? opts[:interface_name] : interface_name(polymorphic_model)
 
-    def add_polymorphic_triggers(opts)
-      raise ArgumentError, "missing :associated_models from options hash" if !opts.has_key?(:associated_models) 
-      raise ArgumentError, "missing :polymorphic_model from options hash" if !opts.has_key?(:polymorphic_model)
-      associated_models = opts[:associated_models]
-      polymorphic_model = opts[:polymorphic_model]
-      interface_name = opts.has_key?(:interface_name) ? opts[:interface_name] : interface_name(polymorphic_model)
+    sql = get_create_function_sql(interface_name, associated_models, polymorphic_model)
+    sql << get_delete_function_sql(interface_name, associated_models, polymorphic_model)
 
-      sql = get_create_function_sql(interface_name, associated_models, polymorphic_model)
-      sql << get_delete_function_sql(interface_name, associated_models, polymorphic_model)
+    execute sql
+  end
 
-      execute sql
-    end
+  def remove_polymorphic_triggers(opts)
+    polymorphic_model = opts[:polymorphic_model]
+    interface_name = opts.has_key?(:interface_name) ? opts[:interface_name] : interface_name(polymorphic_model)
 
-    def remove_polymorphic_triggers(opts)
-      polymorphic_model = opts[:polymorphic_model]
-      interface_name = opts.has_key?(:interface_name) ? opts[:interface_name] : interface_name(polymorphic_model)
+    execute %{
+      DROP FUNCTION IF EXISTS check_#{interface_name}_create_integrity() CASCADE;
+      DROP FUNCTION IF EXISTS check_#{interface_name}_delete_integrity() CASCADE;
+    }
+  end
 
-      execute %{
-        DROP FUNCTION IF EXISTS check_#{interface_name}_create_integrity() CASCADE;
-        DROP FUNCTION IF EXISTS check_#{interface_name}_delete_integrity() CASCADE;
-      }
-    end
+  private
 
-    private
+  # TODO: Is it safe to just grab the first polymorphic association?
+  def interface_name(model_name)
+    model_name.constantize.reflect_on_all_associations.select { |r| r if r.options[:polymorphic] }.first.name
+  end
 
-    # TODO: Is it safe to just grab the first polymorphic association?
-    def interface_name(model_name)
-      model_name.constantize.reflect_on_all_associations.select { |r| r if r.options[:polymorphic] }.first.name
-    end
+  def get_create_function_sql(interface_name, models, polymorphic_model)
+    sql = "DROP FUNCTION IF EXISTS check_#{interface_name}_create_integrity() CASCADE;"
+    
+    sql << %{
+      CREATE FUNCTION check_#{interface_name}_create_integrity() RETURNS TRIGGER AS '
+        BEGIN
+          IF NEW.#{interface_name}_type = ''#{models[0]}'' AND EXISTS (
+              SELECT id FROM #{models[0].constantize.table_name} WHERE id = NEW.#{interface_name}_id) THEN
+            RETURN NEW;
+    }
 
-    def get_create_function_sql(interface_name, models, polymorphic_model)
-      sql = "DROP FUNCTION IF EXISTS check_#{interface_name}_create_integrity() CASCADE;"
-      
+    models[1..-1].each do |model|
       sql << %{
-        CREATE FUNCTION check_#{interface_name}_create_integrity() RETURNS TRIGGER AS '
-          BEGIN
-            IF NEW.#{interface_name}_type = ''#{models[0]}'' AND EXISTS (
-                SELECT id FROM #{models[0].constantize.table_name} WHERE id = NEW.#{interface_name}_id) THEN
-              RETURN NEW;
+        ELSEIF NEW.#{interface_name}_type = ''#{model}'' AND EXISTS (
+            SELECT id FROM #{model.constantize.table_name} WHERE id = NEW.#{interface_name}_id) THEN
+          RETURN NEW;
       }
+    end
+    
+    sql << %{
+          ELSE
+            RAISE EXCEPTION ''No % model with id %.'', NEW.#{interface_name}_type, NEW.#{interface_name}_id;
+            RETURN NULL;
+          END IF;
+        END'
+        LANGUAGE plpgsql;
 
-      models[1..-1].each do |model|
-        sql << %{
-          ELSEIF NEW.#{interface_name}_type = ''#{model}'' AND EXISTS (
-              SELECT id FROM #{model.constantize.table_name} WHERE id = NEW.#{interface_name}_id) THEN
-            RETURN NEW;
-        }
-      end
-      
-      sql << %{
-            ELSE
-              RAISE EXCEPTION ''No % model with id %.'', NEW.#{interface_name}_type, NEW.#{interface_name}_id;
-              RETURN NULL;
-            END IF;
-          END'
-          LANGUAGE plpgsql;
-
-          CREATE TRIGGER check_#{interface_name}_create_integrity_trigger 
-            BEFORE INSERT OR UPDATE ON #{polymorphic_model.constantize.table_name} 
-            FOR EACH ROW EXECUTE PROCEDURE check_#{interface_name}_create_integrity();
-      }
+        CREATE TRIGGER check_#{interface_name}_create_integrity_trigger 
+          BEFORE INSERT OR UPDATE ON #{polymorphic_model.constantize.table_name} 
+          FOR EACH ROW EXECUTE PROCEDURE check_#{interface_name}_create_integrity();
+    }
 
-      return sql
-    end
+    return sql
+  end
 
-    def get_delete_function_sql(interface_name, models, polymorphic_model)
-      polymorphic_model_table_name = polymorphic_model.constantize.table_name
-      
-      sql = ""
+  def get_delete_function_sql(interface_name, models, polymorphic_model)
+    polymorphic_model_table_name = polymorphic_model.constantize.table_name
+    
+    sql = ""
+    sql << %{
+      CREATE FUNCTION check_#{interface_name}_delete_integrity() RETURNS TRIGGER AS '
+        BEGIN
+          IF TG_TABLE_NAME = ''#{models[0].constantize.table_name}'' AND EXISTS (
+            SELECT id FROM #{polymorphic_model_table_name} 
+            WHERE #{interface_name}_type = ''#{models[0]}'' AND #{interface_name}_id = OLD.id) THEN
+            RAISE EXCEPTION ''There are records in #{polymorphic_model_table_name} that refer to %. You must delete those records first.'', OLD;
+    }
+
+    models[1..-1].each do |model|
       sql << %{
-        CREATE FUNCTION check_#{interface_name}_delete_integrity() RETURNS TRIGGER AS '
-          BEGIN
-            IF TG_TABLE_NAME = ''#{models[0].constantize.table_name}'' AND EXISTS (
-              SELECT id FROM #{polymorphic_model_table_name} 
-              WHERE #{interface_name}_type = ''#{models[0]}'' AND #{interface_name}_id = OLD.id) THEN
-              RAISE EXCEPTION ''There are records in #{polymorphic_model_table_name} that refer to %. You must delete those records first.'', OLD;
+        ELSEIF TG_TABLE_NAME = ''#{model.constantize.table_name}'' AND EXISTS (
+          SELECT id FROM #{polymorphic_model_table_name} 
+          WHERE #{interface_name}_type = ''#{model}'' AND #{interface_name}_id = OLD.id) THEN
+            RAISE EXCEPTION ''There are records in #{polymorphic_model_table_name} that refer to %. You must delete those records first.'', OLD;
       }
+    end
 
-      models[1..-1].each do |model|
-        sql << %{
-          ELSEIF TG_TABLE_NAME = ''#{model.constantize.table_name}'' AND EXISTS (
-            SELECT id FROM #{polymorphic_model_table_name} 
-            WHERE #{interface_name}_type = ''#{model}'' AND #{interface_name}_id = OLD.id) THEN
-              RAISE EXCEPTION ''There are records in #{polymorphic_model_table_name} that refer to %. You must delete those records first.'', OLD;
-        }
-      end
+    sql << %{
+          ELSE
+            RETURN NULL;
+          END IF;
+        END'
+      LANGUAGE plpgsql;
+    }
 
+    models.each do |model|
+      table_name = model.constantize.table_name
+      
       sql << %{
-            ELSE
-              RETURN NULL;
-            END IF;
-          END'
-        LANGUAGE plpgsql;
+        CREATE TRIGGER check_#{table_name}_delete_integrity_trigger 
+          BEFORE DELETE ON #{table_name} 
+          FOR EACH ROW EXECUTE PROCEDURE check_#{interface_name}_delete_integrity();
       }
-
-      models.each do |model|
-        table_name = model.constantize.table_name
-        
-        sql << %{
-          CREATE TRIGGER check_#{table_name}_delete_integrity_trigger 
-            BEFORE DELETE ON #{table_name} 
-            FOR EACH ROW EXECUTE PROCEDURE check_#{interface_name}_delete_integrity();
-        }
-      end
-
-      return sql
     end
 
+    return sql
   end
-
 end
 
-ActiveRecord::Base.send(:include, Fides)
+class ActiveRecord::Migration
+  include Fides
+end

data/test/lib/fides/fides_test.rb

@@ -15,14 +15,15 @@ describe Fides do
       end
     end
     @my_test_association = MyTestAssociaiton.new
+    @my_test_migration = MyTestMigration.new
   end
 
   it "responds to add_polymorphic_triggers" do
-    assert_respond_to ActiveRecord::Migration, :add_polymorphic_triggers
+    assert_respond_to @my_test_migration, :add_polymorphic_triggers
   end
 
   it "responds to remove_polymorphic_triggers" do
-    assert_respond_to ActiveRecord::Migration, :remove_polymorphic_triggers
+    assert_respond_to @my_test_migration, :remove_polymorphic_triggers
   end
 
   it "includes the ability to use of the constantize method" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fides
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.9
 platform: ruby
 authors:
 - Martin Kraft