polymorphic_constraints 1.0.0

data/lib/polymorphic_constraints/migration/command_recorder.rb

@@ -0,0 +1,20 @@
+module PolymorphicConstraints
+  module Migration
+    module CommandRecorder
+      def add_polymorphic_constraints(*args)
+        record(:add_polymorphic_constraints, args)
+      end
+
+      def remove_polymorphic_constraints(*args)
+        record(:remove_polymorphic_constraints, args)
+      end
+
+      def invert_add_polymorphic_constraints(args)
+        relation, associated_model, options = *args
+        [:remove_polymorphic_constraints, relation]
+      end
+
+      alias_method :update_polymorphic_constraints, :add_polymorphic_constraints
+    end
+  end
+end

data/lib/polymorphic_constraints/railtie.rb

@@ -0,0 +1,27 @@
+module PolymorphicConstraints
+  class Railtie < Rails::Railtie
+    initializer 'polymorphic_constraints.load_migration' do
+      ActiveSupport.on_load :active_record do
+        class ActiveRecord::ReferenceViolation < ActiveRecord::ActiveRecordError
+        end
+
+        ActiveRecord::ConnectionAdapters.module_eval do
+          include PolymorphicConstraints::ConnectionAdapters::SchemaStatements
+        end
+
+        if defined?(ActiveRecord::Migration::CommandRecorder)
+          ActiveRecord::Migration::CommandRecorder.class_eval do
+            include PolymorphicConstraints::Migration::CommandRecorder
+          end
+        end
+
+        PolymorphicConstraints::Adapter.load!
+      end
+
+      ActiveSupport.on_load :action_controller do
+        require 'polymorphic_constraints/utils/polymorphic_error_handler'
+        include PolymorphicConstraints::Utils::PolymorphicErrorHandler
+      end
+    end
+  end
+end

data/lib/polymorphic_constraints/utils/polymorphic_error_handler.rb

@@ -0,0 +1,19 @@
+module PolymorphicConstraints
+  module Utils
+    module PolymorphicErrorHandler
+      extend ActiveSupport::Concern
+
+      included do
+        rescue_from ActiveRecord::StatementInvalid do |exception|
+          if exception.message =~ /Polymorphic record not found./
+            raise ActiveRecord::RecordNotFound, exception.message
+          elsif exception.message =~ /Polymorphic reference exists./
+            raise ActiveRecord::ReferenceViolation, exception.message
+          else
+            raise exception
+          end
+        end
+      end
+    end
+  end
+end

data/lib/polymorphic_constraints/utils/polymorphic_model_finder.rb

@@ -0,0 +1,19 @@
+module PolymorphicConstraints
+  module Utils
+    module PolymorphicModelFinder
+      def get_polymorphic_models(relation)
+        Rails.application.eager_load!
+        ActiveRecord::Base.descendants.select do |klass|
+          contains_polymorphic_relation?(klass, relation)
+        end
+      end
+
+      private
+
+      def contains_polymorphic_relation?(model_class, relation)
+        associations = model_class.reflect_on_all_associations
+        associations.map{ |r| r.options[:as] }.include?(relation.to_sym)
+      end
+    end
+  end
+end

data/lib/polymorphic_constraints/utils/sql_string.rb

@@ -0,0 +1,9 @@
+module PolymorphicConstraints
+  module Utils
+    module SqlString
+      def strip_non_essential_spaces(string_to_strip)
+        string_to_strip.gsub(/\s{2,}|\\n/, " ").strip
+      end
+    end
+  end
+end

data/lib/polymorphic_constraints/version.rb

@@ -0,0 +1,3 @@
+module PolymorphicConstraints
+  VERSION = "1.0.0"
+end

data/polymorphic_constraints.gemspec

@@ -0,0 +1,30 @@
+$:.push File.expand_path('../lib', __FILE__)
+
+# Maintain your gem's version:
+require 'polymorphic_constraints/version'
+
+# Describe your gem and declare its dependencies:
+Gem::Specification.new do |s|
+  s.name        = 'polymorphic_constraints'
+  s.version     = PolymorphicConstraints::VERSION
+  s.authors     = ['Ahmad Musaffa']
+  s.email       = ['musaffa_csemm@yahoo.com']
+  s.homepage    = 'https://github.com/musaffa/polymorphic_constraints'
+  s.summary     = 'Database agnostic referential integrity enforcer for Rails polymorphic associations using triggers.'
+  s.description = 'Helps to maintain referential integrity for Rails polymorphic associations.'
+  s.license     = 'MIT'
+
+  s.files         = `git ls-files`.split($/)
+  s.executables   = s.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  s.test_files    = s.files.grep(%r{^spec/})
+  s.require_paths  = ['lib']
+
+  s.required_ruby_version = '>= 1.9.3'
+  s.add_dependency 'rails'
+
+  s.add_development_dependency 'sqlite3'
+  s.add_development_dependency 'pg'
+  s.add_development_dependency 'mysql2'
+  s.add_development_dependency 'rspec-rails', '~> 3.0.2'
+  s.add_development_dependency 'coveralls'
+end

data/spec/dummy/Rakefile

@@ -0,0 +1,6 @@
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
+require File.expand_path('../config/application', __FILE__)
+
+Rails.application.load_tasks

data/spec/dummy/app/assets/javascripts/application.js

@@ -0,0 +1,13 @@
+// This is a manifest file that'll be compiled into application.js, which will include all the files
+// listed below.
+//
+// Any JavaScript/Coffee file within this directory, lib/assets/javascripts, vendor/assets/javascripts,
+// or vendor/assets/javascripts of plugins, if any, can be referenced here using a relative path.
+//
+// It's not advisable to add code directly here, but if you do, it'll appear at the bottom of the
+// compiled file.
+//
+// Read Sprockets README (https://github.com/sstephenson/sprockets#sprockets-directives) for details
+// about supported directives.
+//
+//= require_tree .

data/spec/dummy/app/assets/stylesheets/application.css

@@ -0,0 +1,15 @@
+/*
+ * This is a manifest file that'll be compiled into application.css, which will include all the files
+ * listed below.
+ *
+ * Any CSS and SCSS file within this directory, lib/assets/stylesheets, vendor/assets/stylesheets,
+ * or vendor/assets/stylesheets of plugins, if any, can be referenced here using a relative path.
+ *
+ * You're free to add application-wide styles to this file and they'll appear at the bottom of the
+ * compiled file so the styles you add here take precedence over styles defined in any styles
+ * defined in the other CSS/SCSS files in this directory. It is generally better to create a new
+ * file per style scope.
+ *
+ *= require_tree .
+ *= require_self
+ */

data/spec/dummy/app/controllers/application_controller.rb

@@ -0,0 +1,5 @@
+class ApplicationController < ActionController::Base
+  # Prevent CSRF attacks by raising an exception.
+  # For APIs, you may want to use :null_session instead.
+  protect_from_forgery with: :exception
+end

data/spec/dummy/app/helpers/application_helper.rb

@@ -0,0 +1,2 @@
+module ApplicationHelper
+end

data/spec/dummy/app/models/employee.rb

@@ -0,0 +1,3 @@
+class Employee < ActiveRecord::Base
+  has_many :pictures, as: :imageable, dependent: :destroy
+end

data/spec/dummy/app/models/picture.rb

@@ -0,0 +1,3 @@
+class Picture < ActiveRecord::Base
+  belongs_to :imageable, polymorphic: true
+end

data/spec/dummy/app/models/product.rb

@@ -0,0 +1,3 @@
+class Product < ActiveRecord::Base
+  has_many :pictures, as: :imageable
+end

data/spec/dummy/app/views/layouts/application.html.erb

@@ -0,0 +1,14 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Dummy</title>
+  <%= stylesheet_link_tag    'application', media: 'all', 'data-turbolinks-track' => true %>
+  <%= javascript_include_tag 'application', 'data-turbolinks-track' => true %>
+  <%= csrf_meta_tags %>
+</head>
+<body>
+
+<%= yield %>
+
+</body>
+</html>

data/spec/dummy/bin/bundle

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../Gemfile', __FILE__)
+load Gem.bin_path('bundler', 'bundle')

data/spec/dummy/bin/rails

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+APP_PATH = File.expand_path('../../config/application',  __FILE__)
+require_relative '../config/boot'
+require 'rails/commands'

data/spec/dummy/bin/rake

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require_relative '../config/boot'
+require 'rake'
+Rake.application.run

data/spec/dummy/config.ru

@@ -0,0 +1,4 @@
+# This file is used by Rack-based servers to start the application.
+
+require ::File.expand_path('../config/environment',  __FILE__)
+run Rails.application

data/spec/dummy/config/application.rb

@@ -0,0 +1,30 @@
+require File.expand_path('../boot', __FILE__)
+
+# Pick the frameworks you want:
+require "active_record/railtie"
+require "action_controller/railtie"
+require "action_mailer/railtie"
+require "action_view/railtie"
+require "sprockets/railtie"
+# require "rails/test_unit/railtie"
+
+Bundler.require(*Rails.groups)
+require "polymorphic_constraints"
+
+module Dummy
+  class Application < Rails::Application
+    # Settings in config/environments/* take precedence over those specified here.
+    # Application configuration should go into files in config/initializers
+    # -- all .rb files in that directory are automatically loaded.
+
+    # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
+    # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
+    # config.time_zone = 'Central Time (US & Canada)'
+
+    # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
+    # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]
+    # config.i18n.default_locale = :de
+    config.active_record.schema_format = :sql
+  end
+end
+

data/spec/dummy/config/boot.rb

@@ -0,0 +1,5 @@
+# Set up gems listed in the Gemfile.
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../../../Gemfile', __FILE__)
+
+require 'bundler/setup' if File.exist?(ENV['BUNDLE_GEMFILE'])
+$LOAD_PATH.unshift File.expand_path('../../../../lib', __FILE__)

data/spec/dummy/config/database.mysql.yml

@@ -0,0 +1,6 @@
+test:
+  adapter: mysql2
+  encoding: utf8
+  username: root
+  database: polymorphic_constraints
+  host: localhost

data/spec/dummy/config/database.postgresql.yml

@@ -0,0 +1,6 @@
+test:
+  adapter: postgresql
+  encoding: utf8
+  username: postgres
+  database: polymorphic_constraints
+  host: localhost

data/spec/dummy/config/database.sqlite.yml

@@ -0,0 +1,5 @@
+test:
+  adapter: sqlite3
+  database: db/test.sqlite3
+  pool: 5
+  timeout: 5000

data/spec/dummy/config/database.yml

@@ -0,0 +1,5 @@
+test:
+  adapter: sqlite3
+  database: db/test.sqlite3
+  pool: 5
+  timeout: 5000

data/spec/dummy/config/environment.rb

@@ -0,0 +1,5 @@
+# Load the Rails application.
+require File.expand_path('../application', __FILE__)
+
+# Initialize the Rails application.
+Rails.application.initialize!

data/spec/dummy/config/environments/development.rb

@@ -0,0 +1,37 @@
+Dummy::Application.configure do
+  # Settings specified here will take precedence over those in config/application.rb.
+
+  # In the development environment your application's code is reloaded on
+  # every request. This slows down response time but is perfect for development
+  # since you don't have to restart the web server when you make code changes.
+  config.cache_classes = false
+
+  # Do not eager load code on boot.
+  config.eager_load = false
+
+  # Show full error reports and disable caching.
+  config.consider_all_requests_local       = true
+  config.action_controller.perform_caching = false
+
+  # Don't care if the mailer can't send.
+  config.action_mailer.raise_delivery_errors = false
+
+  # Print deprecation notices to the Rails logger.
+  config.active_support.deprecation = :log
+
+  # Raise an error on page load if there are pending migrations.
+  config.active_record.migration_error = :page_load
+
+  # Debug mode disables concatenation and preprocessing of assets.
+  # This option may cause significant delays in view rendering with a large
+  # number of complex assets.
+  config.assets.debug = true
+
+  # Adds additional error checking when serving assets at runtime.
+  # Checks for improperly declared sprockets dependencies.
+  # Raises helpful error messages.
+  config.assets.raise_runtime_errors = true
+
+  # Raises error for missing translations
+  # config.action_view.raise_on_missing_translations = true
+end

data/spec/dummy/config/environments/production.rb

@@ -0,0 +1,78 @@
+Dummy::Application.configure do
+  # Settings specified here will take precedence over those in config/application.rb.
+
+  # Code is not reloaded between requests.
+  config.cache_classes = true
+
+  # Eager load code on boot. This eager loads most of Rails and
+  # your application in memory, allowing both threaded web servers
+  # and those relying on copy on write to perform better.
+  # Rake tasks automatically ignore this option for performance.
+  config.eager_load = true
+
+  # Full error reports are disabled and caching is turned on.
+  config.consider_all_requests_local       = false
+  config.action_controller.perform_caching = true
+
+  # Enable Rack::Cache to put a simple HTTP cache in front of your application
+  # Add `rack-cache` to your Gemfile before enabling this.
+  # For large-scale production use, consider using a caching reverse proxy like nginx, varnish or squid.
+  # config.action_dispatch.rack_cache = true
+
+  # Disable Rails's static asset server (Apache or nginx will already do this).
+  config.serve_static_assets = false
+
+  # Compress JavaScripts and CSS.
+  config.assets.js_compressor = :uglifier
+  # config.assets.css_compressor = :sass
+
+  # Do not fallback to assets pipeline if a precompiled asset is missed.
+  config.assets.compile = false
+
+  # Generate digests for assets URLs.
+  config.assets.digest = true
+
+  # `config.assets.precompile` and `config.assets.version` have moved to config/initializers/assets.rb
+
+  # Specifies the header that your server uses for sending files.
+  # config.action_dispatch.x_sendfile_header = "X-Sendfile" # for apache
+  # config.action_dispatch.x_sendfile_header = 'X-Accel-Redirect' # for nginx
+
+  # Force all access to the app over SSL, use Strict-Transport-Security, and use secure cookies.
+  # config.force_ssl = true
+
+  # Set to :debug to see everything in the log.
+  config.log_level = :info
+
+  # Prepend all log lines with the following tags.
+  # config.log_tags = [ :subdomain, :uuid ]
+
+  # Use a different logger for distributed setups.
+  # config.logger = ActiveSupport::TaggedLogging.new(SyslogLogger.new)
+
+  # Use a different cache store in production.
+  # config.cache_store = :mem_cache_store
+
+  # Enable serving of images, stylesheets, and JavaScripts from an asset server.
+  # config.action_controller.asset_host = "http://assets.example.com"
+
+  # Ignore bad email addresses and do not raise email delivery errors.
+  # Set this to true and configure the email server for immediate delivery to raise delivery errors.
+  # config.action_mailer.raise_delivery_errors = false
+
+  # Enable locale fallbacks for I18n (makes lookups for any locale fall back to
+  # the I18n.default_locale when a translation cannot be found).
+  config.i18n.fallbacks = true
+
+  # Send deprecation notices to registered listeners.
+  config.active_support.deprecation = :notify
+
+  # Disable automatic flushing of the log to improve performance.
+  # config.autoflush_log = false
+
+  # Use default logging formatter so that PID and timestamp are not suppressed.
+  config.log_formatter = ::Logger::Formatter.new
+
+  # Do not dump schema after migrations.
+  config.active_record.dump_schema_after_migration = false
+end