database_validations 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 75cc8e49376be81dd56be2da6de60dd9e4763243af4aef1ecb3bc5d5324bfabc
-  data.tar.gz: 7d091827be83bcfa46e89bdbaa7ff5a1964c063272a3b3099b1169e678d5d1b0
+  metadata.gz: a7c3f11bd0b6a20ebd9cbcbe9134ae84e049c53e13d0a67faaa3da570994adc8
+  data.tar.gz: 1a45078331bccc6a9fdd6e59426b766dce95375ec9ffa7ac3130bb91df86f761
 SHA512:
-  metadata.gz: c60c9a3eb7ea1783235b4636d7c9baa0af690a2218593ba96853e6f18647a01feb0449676c920d9afb6e73145d3d44df0ce1f02f153c4a161a22983204823102
-  data.tar.gz: 478bcddd2b83c208b34632bf782ba27e9c948c878fb7c6d751f65fb08c6ac206f1d3a7c6b28706299499f592e827dd6b8381a306014e03dd8200776ccb24ec60
+  metadata.gz: 75563aa5561d03288ae9dbe9b38e5fc8d87955f9227150d0c3d782eb42bf02e12d259d2614c4e35d2dd344958bb306ba87a66f46747b04608b5082c5f430210e
+  data.tar.gz: 00f656712648b20f62d90d72db0b233e8f69d6a552e9adc7eb42c45cd85d2f17622812776223990dae9aea21ed42d586719cdff4a030eff25218cfde513c8b7e

data/lib/database_validations/adapters.rb

@@ -0,0 +1,20 @@
+require 'database_validations/adapters/base_adapter'
+require 'database_validations/adapters/sqlite_adapter'
+require 'database_validations/adapters/postgresql_adapter'
+require 'database_validations/adapters/mysql_adapter'
+
+module DatabaseValidations
+  module Adapters
+    module_function
+
+    def factory(model)
+      case (database = model.connection.adapter_name.downcase.to_sym)
+      when :sqlite then Adapters::SqliteAdapter.new(model)
+      when :postgresql then Adapters::PostgresqlAdapter.new(model)
+      when :mysql2 then Adapters::MysqlAdapter.new(model)
+      else
+        raise Errors::UnknownDatabase.new(database)
+      end
+    end
+  end
+end

data/lib/database_validations/adapters/base_adapter.rb

@@ -0,0 +1,18 @@
+module DatabaseValidations
+  module Adapters
+    class BaseAdapter
+      attr_reader :model
+
+      def initialize(model)
+        @model = model
+      end
+
+      def index_columns(index_name)
+        model.connection
+          .indexes(model.table_name)
+          .find { |index| index.name == index_name  }
+          .columns
+      end
+    end
+  end
+end

data/lib/database_validations/adapters/mysql_adapter.rb

@@ -0,0 +1,10 @@
+module DatabaseValidations
+  module Adapters
+    class MysqlAdapter < BaseAdapter
+      def error_columns(error_message)
+        index_name = error_message[/key '([^']+)'/, 1]
+        index_columns(index_name)
+      end
+    end
+  end
+end

data/lib/database_validations/adapters/postgresql_adapter.rb

@@ -0,0 +1,10 @@
+module DatabaseValidations
+  module Adapters
+    class PostgresqlAdapter < BaseAdapter
+      def error_columns(error_message)
+        index_name = error_message[/unique constraint "([^"]+)"/, 1]
+        index_columns(index_name)
+      end
+    end
+  end
+end

data/lib/database_validations/adapters/sqlite_adapter.rb

@@ -0,0 +1,9 @@
+module DatabaseValidations
+  module Adapters
+    class SqliteAdapter < BaseAdapter
+      def error_columns(error_message)
+        error_message.scan(/#{model.table_name}\.([^,:]+)/).flatten
+      end
+    end
+  end
+end

data/lib/database_validations/errors.rb

@@ -0,0 +1,23 @@
+module DatabaseValidations
+  module Errors
+    class Base < StandardError; end
+
+    class IndexNotFound < Base
+      attr_reader :columns
+
+      def initialize(columns)
+        @columns = columns.map(&:to_s)
+        super "No unique index found with columns: #{self.columns}"
+      end
+    end
+
+    class UnknownDatabase < Base
+      attr_reader :database
+
+      def initialize(database)
+        @database = database
+        super "Unknown database: #{self.database}"
+      end
+    end
+  end
+end

data/lib/database_validations/helpers.rb

@@ -0,0 +1,29 @@
+module DatabaseValidations
+  module Helpers
+    module_function
+
+    def raise_if_index_missed!(model, columns)
+      index = model.connection
+                .indexes(model.table_name)
+                .select(&:unique)
+                .find { |index| index.columns.map(&:to_s).sort == columns }
+
+      raise Errors::IndexNotFound.new(columns) unless index
+    end
+
+    def handle_unique_error(instance, error)
+      columns = DatabaseValidations::Adapters
+                  .factory(instance.class)
+                  .error_columns(error.message)
+                  .map!(&:to_s)
+                  .sort!
+
+      options = instance.class.validates_db_uniqueness[columns]
+
+      error_options = options.except(:case_sensitive, :scope, :conditions, :attributes)
+      error_options[:value] = instance.public_send(options[:attributes])
+
+      instance.errors.add(options[:attributes], :taken, error_options)
+    end
+  end
+end

data/lib/database_validations/uniqueness_validator.rb

@@ -0,0 +1,64 @@
+module DatabaseValidations
+  module DatabaseUniquenessValidator
+    extend ActiveSupport::Concern
+
+    included do
+      alias_method :valid_without_uniqueness?, :valid?
+
+      def valid?(context = nil)
+        output = super(context)
+
+        self.class.validates_db_uniqueness.each_value do |opts|
+          validates_with(ActiveRecord::Validations::UniquenessValidator, opts.merge(allow_nil: true))
+        end
+
+        errors.empty? && output
+      end
+
+      alias_method :validate, :valid?
+    end
+
+    def save(options = {})
+      super
+    rescue ActiveRecord::RecordNotUnique => e
+      DatabaseValidations::Helpers.handle_unique_error(self, e)
+      false
+    end
+
+    def save!(options = {})
+      super
+    rescue ActiveRecord::RecordNotUnique => e
+      DatabaseValidations::Helpers.handle_unique_error(self, e)
+      raise ActiveRecord::RecordInvalid, self
+    end
+
+    private
+
+    def perform_validations(options = {})
+      options[:validate] == false || valid_without_uniqueness?(options[:context])
+    end
+  end
+
+  module ClassMethods
+    def validates_db_uniqueness_of(*attributes)
+      @validates_db_uniqueness ||= {}
+
+      options = attributes.extract_options!
+
+      attributes.each do |attribute|
+        columns = [attribute, Array.wrap(options[:scope])].flatten!.map!(&:to_s).sort!
+
+        DatabaseValidations::Helpers.raise_if_index_missed!(self, columns)
+
+        @validates_db_uniqueness[columns] = options.merge(attributes: attribute)
+      end
+
+      include(DatabaseUniquenessValidator)
+    end
+
+    def validates_db_uniqueness
+      derived = superclass.respond_to?(:validates_db_uniqueness) ? superclass.validates_db_uniqueness : {}
+      derived.merge(@validates_db_uniqueness || {})
+    end
+  end
+end

data/lib/database_validations/version.rb

@@ -0,0 +1,3 @@
+module DatabaseValidations
+  VERSION = "0.3.1"
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: database_validations
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
 platform: ruby
 authors:
 - Evgeniy Demin
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-08-31 00:00:00.000000000 Z
+date: 2018-09-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -144,6 +144,15 @@ files:
 - LICENSE.txt
 - README.md
 - lib/database_validations.rb
+- lib/database_validations/adapters.rb
+- lib/database_validations/adapters/base_adapter.rb
+- lib/database_validations/adapters/mysql_adapter.rb
+- lib/database_validations/adapters/postgresql_adapter.rb
+- lib/database_validations/adapters/sqlite_adapter.rb
+- lib/database_validations/errors.rb
+- lib/database_validations/helpers.rb
+- lib/database_validations/uniqueness_validator.rb
+- lib/database_validations/version.rb
 homepage: https://github.com/toptal/database_validations
 licenses:
 - MIT