activerecord-not_nil 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '08d8a27f8b2a4efc4245621db5111cb7277f19cece3b527079e2f288279db3aa'
-  data.tar.gz: f2a674a41f89ff788a577633a7939c3ee456b74243bacd3eede901852d6feb9e
+  metadata.gz: 3cfd9baad29246455c854ab457c5fe3f7a1af1721f5fe66f0894f3f2b4f5b0e3
+  data.tar.gz: 2b6c5da6ca453551ccabb613f5966a5e4c1ff21e673df8dc9a6a1985c2ed1015
 SHA512:
-  metadata.gz: 915b1856fca4648527dbd50fde7a58d3d1d690102424d3f095cfbad56416503979a6ef8c07cc6588b1496cedab90b60118010da2836053e0e44d42ee1a445ce1
-  data.tar.gz: '087edd6a1f578e6d7ecf559a1df9e5332d0e7328b6dcd151bdfbaececde9945f82a5de34ee6ddcab5eadfdd7056e5f95b4b2663a5baade634e735f51302e0e37'
+  metadata.gz: 5710661d1a73e6e02485981145bd51c1ea145cb4bd84503921daffc88f316e7171fc6664e9b96eeb9ebd5fc4c8e4f84b80747335787019524d58bbe8ba17cf2d
+  data.tar.gz: c57ead36b09896eeeab2db1411e7a9ed86cd6fb20f8274e521726ef4b82c5267fa58eec8ff6c42702f227df2fc363899ee2699e5ea4d6870df40220f7ff943b4

data/lib/activerecord-not_nil/active_record/connection_adapters/quoting.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module ConnectionAdapters
+    module Quoting
+      alias origin_type_cast type_cast
+      def type_cast(value)
+        raise ArgumentError, 'Cannot assign not_nil as a value' if value.is_a?(::NotNilClass)
+
+        origin_type_cast(value)
+      end
+    end
+  end
+end

data/lib/activerecord-not_nil/arel/visitors/to_sql.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Arel
+  module Visitors
+    # Module to override Arel::Visitors::ToSql#visit_Arel_Nodes_Equality
+    module NotNilOverride
+      def visit_Arel_Nodes_Equality(o, collector) # rubocop:disable Naming/MethodName
+        return super if o.right.is_a?(Arel::Attributes::Attribute)
+        return super if o.right.is_a?(Arel::Nodes::NamedFunction)
+        return super unless o.right.value.is_a?(::NotNilClass)
+
+        collector = visit o.left, collector
+        collector << ' IS NOT NULL'
+      end
+    end
+
+    class ToSql < Arel::Visitors::Visitor
+      prepend NotNilOverride
+    end
+  end
+end

data/lib/activerecord-not_nil.rb

@@ -4,7 +4,7 @@
 require 'active_record'
 require 'not_niller'
 
-require_relative 'activerecord-not_nil/activerecord/connection_adapters/quoting'
+require_relative 'activerecord-not_nil/active_record/connection_adapters/quoting'
 require_relative 'activerecord-not_nil/arel/visitors/to_sql'
 
 # rubocop:enable Naming/FileName

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-not_nil
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Payt devs
@@ -52,6 +52,8 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/activerecord-not_nil.rb
+- lib/activerecord-not_nil/active_record/connection_adapters/quoting.rb
+- lib/activerecord-not_nil/arel/visitors/to_sql.rb
 homepage: https://github.com/payt/activerecord-not_nil
 licenses:
 - MIT