database_consistency 2.1.3 → 3.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 67c90295e24b6b49a4e010a612729f86951d2dda7daba5ac06b091e6b435e040
-  data.tar.gz: f1958934302e46a7b21d600a9418bbf58bef2921a5b4d5c2dd5fcccc5a4acda8
+  metadata.gz: 3d92e907cbb8ca8c49b6b4015614f9bc55074d37cb96130b97e62ef5266211d1
+  data.tar.gz: c97b5ff7fb08271812b489ea695f20b0d45c56f5109bd1ec246e1704917b33cc
 SHA512:
-  metadata.gz: 3bb2a7b1dc68923711cad75c25c2e27a0208ac625a911f61b6e9a6459b5c71640800329f920f14373c4202f100f36b2bf05f2a683772bace47ebc5dd3b3b4ea4
-  data.tar.gz: e6a7b06b4bd9d7b3ff002e24c8a320b952c208e864f07ec4f0c255de23b032f14610bf893cc737d3ca2713f1357ee0e0307d91d4a611c48b693e330b1145891e
+  metadata.gz: 004c32a1588bc7cb59299352e09a1866f20229735d56b1290d337ce2322ecd4ff20918e817e26118461356a47b6a81580ede89f695be2ec9fdf2739f44a38064
+  data.tar.gz: 52d25e2ee4da71b6130d442a2443c7f3e02ad84674f7d19d67270356ebe28889db57581907d5d65cc9579013fb9d3b073461e210f0320a90db14cad8bab5e024

data/lib/database_consistency/checkers/association_checkers/foreign_key_cascade_checker.rb

@@ -28,6 +28,7 @@ module DatabaseConsistency
       def preconditions
         !association.polymorphic? &&
           !association.belongs_to? &&
+          association.association_primary_key.present? &&
           foreign_key &&
           DEPENDENT_OPTIONS.include?(dependent_option)
       rescue StandardError
@@ -68,18 +69,18 @@ module DatabaseConsistency
           association.klass
                      .connection
                      .foreign_keys(association.klass.table_name)
-                     .find { |fk| fk.column == association.foreign_key.to_s }
+                     .find { |fk| (Helper.extract_columns(association.foreign_key) - Array.wrap(fk.column)).empty? }
       end
 
-      def report_template(status, error_slug: nil)
+      def report_template(status, error_slug: nil) # rubocop:disable Metrics/AbcSize
         Report.new(
           status: status,
           error_message: nil,
           error_slug: error_slug,
           primary_table: association.table_name.to_s,
-          primary_key: association.association_primary_key.to_s,
+          primary_key: Helper.extract_columns(association.association_primary_key).join('+'),
           foreign_table: association.active_record.table_name.to_s,
-          foreign_key: association.foreign_key.to_s,
+          foreign_key: Helper.extract_columns(association.foreign_key).join('+'),
           cascade_option: required_foreign_key_cascade,
           **report_attributes
         )

data/lib/database_consistency/checkers/association_checkers/missing_dependent_destroy_checker.rb

@@ -15,7 +15,7 @@ module DatabaseConsistency
       private
 
       def preconditions
-        association.belongs_to? && foreign_key
+        association.belongs_to? && !association.polymorphic? && foreign_key
       end
 
       def check

data/lib/database_consistency/checkers/column_checkers/missing_index_find_by_checker.rb

@@ -0,0 +1,187 @@
+# frozen_string_literal: true
+
+begin
+  require 'prism'
+rescue LoadError
+  # Prism is not available; this checker will be disabled on Ruby < 3.3
+end
+
+module DatabaseConsistency
+  module Checkers
+    # This class checks for columns used in find_by queries that are missing a database index.
+    # It uses the Prism parser (Ruby stdlib since 3.3) to traverse the AST of all project
+    # source files (found by iterating loaded constants and excluding gem paths) and detect
+    # calls such as find_by_<column>, find_by(column: ...) and find_by("column" => ...).
+    # The checker is automatically skipped on Ruby versions where Prism is not available.
+    class MissingIndexFindByChecker < ColumnChecker
+      Report = ReportBuilder.define(
+        DatabaseConsistency::Report,
+        :source_location,
+        :total_findings_count
+      )
+
+      private
+
+      # We skip check when:
+      #  - Prism is not available (Ruby < 3.3)
+      #  - column is the primary key (always indexed)
+      #  - column name does not appear in any find_by call across project source files
+      def preconditions
+        defined?(Prism) && !primary_key_column? && find_by_used?
+      end
+
+      # Table of possible statuses
+      # | index    | status |
+      # | -------- | ------ |
+      # | present  | ok     |
+      # | missing  | fail   |
+      def check
+        if indexed?
+          report_template(:ok)
+        else
+          report_template(:fail, error_slug: :missing_index_find_by)
+        end
+      end
+
+      def report_template(status, error_slug: nil)
+        Report.new(
+          status: status,
+          error_slug: error_slug,
+          error_message: nil,
+          source_location: (status == :fail ? @find_by_location : nil),
+          total_findings_count: (status == :fail ? @find_by_count : nil),
+          **report_attributes
+        )
+      end
+
+      def find_by_used?
+        entry = PrismHelper.find_by_calls_index.dig(model.name.to_s, column.name.to_s)
+        return false unless entry
+
+        @find_by_location = entry[:first_location]
+        @find_by_count = entry[:total_findings_count]
+        true
+      end
+
+      def indexed?
+        model.connection.indexes(model.table_name).any? do |index|
+          Helper.extract_index_columns(index.columns).first == column.name.to_s
+        end
+      end
+
+      def primary_key_column?
+        column.name.to_s == model.primary_key.to_s
+      end
+
+      if defined?(Prism)
+        # Prism AST visitor that collects ALL find_by calls from a source file into a results hash.
+        # Key: [model_name, column_name] -- model_name is derived from the explicit receiver or the
+        # lexical class/module scope for bare calls. Bare calls outside any class are ignored.
+        # Value: "file:line" location of the first matching call.
+        #
+        # Handles:
+        #  - find_by_<col>(<value>) / Model.find_by_<col>!  (dynamic finder)
+        #  - find_by(col: <value>) / Model.find_by col:     (symbol-key hash)
+        #  - find_by("col" => <value>)                      (string-key hash)
+        #
+        # Defined only when Prism is available (Ruby 3.3+).
+        class FindByCollector < Prism::Visitor
+          # Matches the full column name from a dynamic finder method name.
+          # e.g. find_by_email -> "email", find_by_first_name -> "first_name"
+          # Multi-column patterns like find_by_name_and_email extract "name_and_email"
+          # which won't match any single-column name, so there are no false positives.
+          DYNAMIC_FINDER_RE = /\Afind_by_(.+?)!?\z/.freeze
+
+          attr_reader :results
+
+          def initialize(file)
+            super()
+            @file = file
+            @results = {}
+            @scope_stack = []
+          end
+
+          def visit_class_node(node)
+            @scope_stack.push(constant_path_name(node.constant_path))
+            super
+          ensure
+            @scope_stack.pop
+          end
+
+          def visit_module_node(node)
+            @scope_stack.push(constant_path_name(node.constant_path))
+            super
+          ensure
+            @scope_stack.pop
+          end
+
+          def visit_call_node(node)
+            name = node.name.to_s
+            if (match = DYNAMIC_FINDER_RE.match(name))
+              model_key = receiver_to_model_key(node.receiver)
+              store(model_key, match[1], node) unless model_key == :skip
+            elsif name == 'find_by' && node.arguments
+              col = single_hash_column(node.arguments)
+              model_key = receiver_to_model_key(node.receiver)
+              store(model_key, col, node) if col && model_key != :skip
+            end
+            super
+          end
+
+          private
+
+          def current_scope
+            @scope_stack.empty? ? nil : @scope_stack.join('::')
+          end
+
+          def store(model_key, col, node)
+            key = [model_key, col]
+            @results[key] ||= []
+            @results[key] << "#{@file}:#{node.location.start_line}"
+          end
+
+          def receiver_to_model_key(receiver)
+            case receiver
+            when nil then current_scope || :skip
+            when Prism::ConstantReadNode, Prism::ConstantPathNode
+              constant_path_name(receiver)
+            when Prism::CallNode
+              scoped_receiver_model(receiver)
+            else
+              :skip
+            end
+          end
+
+          def scoped_receiver_model(call_node)
+            return :skip unless %w[unscoped includes].include?(call_node.name.to_s)
+
+            rec = call_node.receiver
+            return :skip unless rec.is_a?(Prism::ConstantReadNode) || rec.is_a?(Prism::ConstantPathNode)
+
+            constant_path_name(rec)
+          end
+
+          def constant_path_name(node)
+            case node
+            when Prism::ConstantReadNode then node.name.to_s
+            when Prism::ConstantPathNode then "#{constant_path_name(node.parent)}::#{node.name}"
+            end
+          end
+
+          def single_hash_column(arguments_node)
+            arguments_node.arguments.each do |arg|
+              next unless arg.is_a?(Prism::KeywordHashNode) && arg.elements.size == 1
+
+              assoc = arg.elements.first
+              next unless assoc.is_a?(Prism::AssocNode)
+
+              key = assoc.key
+              return key.unescaped if key.is_a?(Prism::SymbolNode) || key.is_a?(Prism::StringNode)
+            end
+            nil
+          end
+        end
+      end
+    end
+  end
+end

data/lib/database_consistency/checkers/index_checkers/unique_index_checker.rb

@@ -29,10 +29,13 @@ module DatabaseConsistency
       def valid?
         uniqueness_validators = model.validators.select { |validator| validator.kind == :uniqueness }
 
-        uniqueness_validators.any? do |validator|
-          validator.attributes.any? do |attribute|
-            sorted_index_columns == Helper.sorted_uniqueness_validator_columns(attribute, validator, model)
-          end
+        uniqueness_validators.any? { |validator| validator_matches?(validator) }
+      end
+
+      def validator_matches?(validator)
+        validator.attributes.any? do |attribute|
+          sorted_index_columns == Helper.sorted_uniqueness_validator_columns(attribute, validator, model) &&
+            Helper.conditions_match_index?(model, validator.options[:conditions], index.where)
         end
       end
 

data/lib/database_consistency/checkers/model_checkers/view_primary_key_checker.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module DatabaseConsistency
+  module Checkers
+    # This class checks that a model pointing to a view has a primary_key set and that column exists
+    class ViewPrimaryKeyChecker < ModelChecker
+      private
+
+      def preconditions
+        ActiveRecord::VERSION::MAJOR >= 5 &&
+          !model.abstract_class? &&
+          model.connection.view_exists?(model.table_name)
+      end
+
+      # Table of possible statuses
+      # | primary_key set | column exists | status |
+      # | --------------- | ------------- | ------ |
+      # | no              | -             | fail   |
+      # | yes             | no            | fail   |
+      # | yes             | yes           | ok     |
+      def check
+        if model.primary_key.blank?
+          report_template(:fail, error_slug: :view_missing_primary_key)
+        elsif !primary_key_column_exists?
+          report_template(:fail, error_slug: :view_primary_key_column_missing)
+        else
+          report_template(:ok)
+        end
+      end
+
+      def primary_key_column_exists?
+        Array(model.primary_key).all? { |key| model.column_names.include?(key.to_s) }
+      end
+    end
+  end
+end

data/lib/database_consistency/checkers/validator_checkers/missing_unique_index_checker.rb

@@ -47,9 +47,13 @@ module DatabaseConsistency
       end
 
       def unique_index
-        @unique_index ||= model.connection.indexes(model.table_name).find do |index|
-          index.unique && Helper.extract_index_columns(index.columns).sort == sorted_uniqueness_validator_columns
-        end
+        @unique_index ||= model.connection.indexes(model.table_name).find { |index| index_matches?(index) }
+      end
+
+      def index_matches?(index)
+        index.unique &&
+          Helper.extract_index_columns(index.columns).sort == sorted_uniqueness_validator_columns &&
+          Helper.conditions_match_index?(model, validator.options[:conditions], index.where)
       end
 
       def primary_key_covers_validation?

data/lib/database_consistency/files_helper.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+module DatabaseConsistency
+  # The module contains file system helper methods for locating project source files.
+  module FilesHelper
+    module_function
+
+    # Returns all unique project source file paths (non-gem Ruby files from loaded constants).
+    # Memoized so the file system walk happens once per database_consistency run.
+    def project_source_files
+      @project_source_files ||=
+        if Module.respond_to?(:const_source_location)
+          collect_source_files
+        else
+          []
+        end
+    end
+
+    def collect_source_files
+      files = []
+      ObjectSpace.each_object(Module) { |mod| files << source_file_path(mod) }
+      files.compact.uniq
+    end
+
+    def source_file_path(mod)
+      return unless (name = mod.name)
+
+      file, = Module.const_source_location(name)
+      return unless file && File.exist?(file)
+      return if excluded_source_file?(file)
+
+      file
+    rescue NameError, ArgumentError
+      nil
+    end
+
+    def excluded_source_file?(file)
+      return true if defined?(Bundler) && file.include?(Bundler.bundle_path.to_s)
+      return true if defined?(Gem) && file.include?(Gem::RUBYGEMS_DIR)
+
+      excluded_by_ruby_stdlib?(file)
+    end
+
+    def excluded_by_ruby_stdlib?(file)
+      return false unless defined?(RbConfig)
+
+      file.include?(RbConfig::CONFIG['rubylibdir']) ||
+        file.include?(RbConfig::CONFIG['bindir']) ||
+        file.include?(RbConfig::CONFIG['sbindir'])
+    end
+  end
+end

data/lib/database_consistency/helper.rb

@@ -144,6 +144,38 @@ module DatabaseConsistency
       model._reflect_on_association(attribute)&.foreign_key || attribute
     end
 
+    # Returns the normalized WHERE SQL produced by a conditions proc, or nil if
+    # it cannot be determined (complex proc, unsupported AR version, etc.).
+    def conditions_where_sql(model, conditions)
+      sql = model.unscoped.instance_exec(&conditions).to_sql
+      where_part = sql.split(/\bWHERE\b/i, 2).last
+      return nil unless where_part
+
+      normalize_sql(where_part.gsub("#{model.quoted_table_name}.", '').gsub('"', ''))
+    rescue StandardError
+      nil
+    end
+
+    # Returns true when validator conditions and index WHERE clause are a valid
+    # pairing: both absent means a match; exactly one present means no match;
+    # when both present the normalized SQL is compared.
+    def conditions_match_index?(model, conditions, index_where)
+      return true if conditions.nil? && index_where.blank?
+      return false if conditions.nil? || index_where.blank?
+
+      conditions_sql = conditions_where_sql(model, conditions)
+      # Strip one level of outer parentheses that some databases (e.g. PostgreSQL)
+      # add when storing/returning the index WHERE clause.
+      normalized_where = normalize_sql(index_where.sub(/\A\s*\((.+)\)\s*\z/m, '\1'))
+      conditions_sql&.casecmp?(normalized_where)
+    end
+
+    def normalize_sql(sql)
+      sql.gsub(/\bTRUE\b/i, '1').gsub(/\bFALSE\b/i, '0')
+         .gsub(/ = 't'/, ' = 1').gsub(/ = 'f'/, ' = 0')
+         .strip
+    end
+
     # @return [String]
     def wrapped_attribute_name(attribute, validator, model)
       attribute = foreign_key_or_attribute(model, attribute)

data/lib/database_consistency/prism_helper.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module DatabaseConsistency
+  # The module contains Prism AST helper methods for scanning project source files.
+  module PrismHelper
+    module_function
+
+    # Returns a memoized index: {model_name => {column_name => "file:line"}}.
+    # Built once per run by scanning all project source files with Prism (Ruby 3.3+).
+    # Bare find_by calls are resolved to their lexical class/module scope.
+    def find_by_calls_index
+      return {} unless defined?(Prism)
+
+      @find_by_calls_index ||= build_find_by_calls_index
+    end
+
+    def build_find_by_calls_index
+      FilesHelper.project_source_files.each_with_object({}) do |file, index|
+        collector = Checkers::MissingIndexFindByChecker::FindByCollector.new(file)
+        collector.visit(Prism.parse_file(file).value)
+        merge_collector_results(collector.results, index)
+      rescue StandardError
+        nil
+      end
+    end
+
+    def merge_collector_results(results, index)
+      results.each do |(model_key, col), locations|
+        index[model_key] ||= {}
+        if (entry = index[model_key][col])
+          entry[:total_findings_count] += locations.size
+        else
+          index[model_key][col] = { first_location: locations.first, total_findings_count: locations.size }
+        end
+      end
+    end
+  end
+end

data/lib/database_consistency/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DatabaseConsistency
-  VERSION = '2.1.3'
+  VERSION = '3.0.0'
 end

data/lib/database_consistency/writers/simple/missing_index_find_by.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module DatabaseConsistency
+  module Writers
+    module Simple
+      class MissingIndexFindBy < Base # :nodoc:
+        private
+
+        def template
+          'column is used in find_by but is missing an index%<source_location>s'
+        end
+
+        def attributes
+          if report.source_location
+            count = report.total_findings_count || 1
+            count_str = count > 1 ? ", and #{count - 1} more occurrences" : ''
+            { source_location: " (found at #{report.source_location}#{count_str})" }
+          else
+            { source_location: '' }
+          end
+        end
+
+        def unique_attributes
+          {
+            table_or_model_name: report.table_or_model_name,
+            column_or_attribute_name: report.column_or_attribute_name
+          }
+        end
+      end
+    end
+  end
+end

data/lib/database_consistency/writers/simple/view_missing_primary_key.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module DatabaseConsistency
+  module Writers
+    module Simple
+      class ViewMissingPrimaryKey < Base # :nodoc:
+        private
+
+        def template
+          'model pointing to a view should have primary_key set'
+        end
+
+        def unique_attributes
+          {
+            table_or_model_name: report.table_or_model_name
+          }
+        end
+      end
+    end
+  end
+end

data/lib/database_consistency/writers/simple/view_primary_key_column_missing.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module DatabaseConsistency
+  module Writers
+    module Simple
+      class ViewPrimaryKeyColumnMissing < Base # :nodoc:
+        private
+
+        def template
+          'model pointing to a view has a non-existent primary_key column set'
+        end
+
+        def unique_attributes
+          {
+            table_or_model_name: report.table_or_model_name
+          }
+        end
+      end
+    end
+  end
+end

data/lib/database_consistency.rb

@@ -4,6 +4,8 @@ require 'active_record'
 
 require 'database_consistency/version'
 require 'database_consistency/helper'
+require 'database_consistency/files_helper'
+require 'database_consistency/prism_helper'
 require 'database_consistency/configuration'
 require 'database_consistency/rescue_error'
 require 'database_consistency/errors'
@@ -44,6 +46,9 @@ require 'database_consistency/writers/simple/missing_association_class'
 require 'database_consistency/writers/simple/missing_table'
 require 'database_consistency/writers/simple/implicit_order_column_missing'
 require 'database_consistency/writers/simple/missing_dependent_destroy'
+require 'database_consistency/writers/simple/missing_index_find_by'
+require 'database_consistency/writers/simple/view_missing_primary_key'
+require 'database_consistency/writers/simple/view_primary_key_column_missing'
 require 'database_consistency/writers/simple_writer'
 
 require 'database_consistency/writers/autofix/helpers/migration'
@@ -77,6 +82,7 @@ require 'database_consistency/checkers/enum_checkers/enum_type_checker'
 
 require 'database_consistency/checkers/model_checkers/model_checker'
 require 'database_consistency/checkers/model_checkers/missing_table_checker'
+require 'database_consistency/checkers/model_checkers/view_primary_key_checker'
 
 require 'database_consistency/checkers/association_checkers/association_checker'
 require 'database_consistency/checkers/association_checkers/missing_index_checker'
@@ -93,6 +99,7 @@ require 'database_consistency/checkers/column_checkers/primary_key_type_checker'
 require 'database_consistency/checkers/column_checkers/enum_value_checker'
 require 'database_consistency/checkers/column_checkers/three_state_boolean_checker'
 require 'database_consistency/checkers/column_checkers/implicit_ordering_checker'
+require 'database_consistency/checkers/column_checkers/missing_index_find_by_checker'
 
 require 'database_consistency/checkers/validator_checkers/validator_checker'
 require 'database_consistency/checkers/validator_checkers/missing_unique_index_checker'

metadata

@@ -1,13 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: database_consistency
 version: !ruby/object:Gem::Version
-  version: 2.1.3
+  version: 3.0.0
 platform: ruby
 authors:
 - Evgeniy Demin
+autorequire:
 bindir: bin
 cert_chain: []
-date: 1980-01-02 00:00:00.000000000 Z
+date: 2026-03-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -135,6 +136,7 @@ dependencies:
     - - ">"
       - !ruby/object:Gem::Version
         version: '1.3'
+description:
 email:
 - lawliet.djez@gmail.com
 executables:
@@ -156,6 +158,7 @@ files:
 - lib/database_consistency/checkers/column_checkers/enum_value_checker.rb
 - lib/database_consistency/checkers/column_checkers/implicit_ordering_checker.rb
 - lib/database_consistency/checkers/column_checkers/length_constraint_checker.rb
+- lib/database_consistency/checkers/column_checkers/missing_index_find_by_checker.rb
 - lib/database_consistency/checkers/column_checkers/null_constraint_checker.rb
 - lib/database_consistency/checkers/column_checkers/primary_key_type_checker.rb
 - lib/database_consistency/checkers/column_checkers/three_state_boolean_checker.rb
@@ -167,6 +170,7 @@ files:
 - lib/database_consistency/checkers/index_checkers/unique_index_checker.rb
 - lib/database_consistency/checkers/model_checkers/missing_table_checker.rb
 - lib/database_consistency/checkers/model_checkers/model_checker.rb
+- lib/database_consistency/checkers/model_checkers/view_primary_key_checker.rb
 - lib/database_consistency/checkers/validator_checkers/case_sensitive_unique_validation_checker.rb
 - lib/database_consistency/checkers/validator_checkers/missing_unique_index_checker.rb
 - lib/database_consistency/checkers/validator_checkers/validator_checker.rb
@@ -178,7 +182,9 @@ files:
 - lib/database_consistency/databases/types/sqlite.rb
 - lib/database_consistency/debug_context.rb
 - lib/database_consistency/errors.rb
+- lib/database_consistency/files_helper.rb
 - lib/database_consistency/helper.rb
+- lib/database_consistency/prism_helper.rb
 - lib/database_consistency/processors/associations_processor.rb
 - lib/database_consistency/processors/base_processor.rb
 - lib/database_consistency/processors/columns_processor.rb
@@ -227,6 +233,7 @@ files:
 - lib/database_consistency/writers/simple/missing_dependent_destroy.rb
 - lib/database_consistency/writers/simple/missing_foreign_key.rb
 - lib/database_consistency/writers/simple/missing_foreign_key_cascade.rb
+- lib/database_consistency/writers/simple/missing_index_find_by.rb
 - lib/database_consistency/writers/simple/missing_table.rb
 - lib/database_consistency/writers/simple/missing_unique_index.rb
 - lib/database_consistency/writers/simple/missing_uniqueness_validation.rb
@@ -239,6 +246,8 @@ files:
 - lib/database_consistency/writers/simple/redundant_unique_index.rb
 - lib/database_consistency/writers/simple/small_primary_key.rb
 - lib/database_consistency/writers/simple/three_state_boolean.rb
+- lib/database_consistency/writers/simple/view_missing_primary_key.rb
+- lib/database_consistency/writers/simple/view_primary_key_column_missing.rb
 - lib/database_consistency/writers/simple_writer.rb
 - lib/database_consistency/writers/todo_writer.rb
 homepage: https://github.com/djezzzl/database_consistency
@@ -273,7 +282,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.7.2
+rubygems_version: 3.4.19
+signing_key:
 specification_version: 4
 summary: Provide an easy way to check the consistency of the database constraints
   with the application validations.