Sha256: 76a5a767692cb0b99cf89f3dcf134c1d4fd5a4b8ca81ce2dc317eb0cd9ad497e
Contents?: true
Size: 1.97 KB
Versions: 10
Compression:
Stored size: 1.97 KB
Contents
# frozen_string_literal: true
module RuboCop
module Cop
module Rails
# Checks for add_column call with NOT NULL constraint
# in migration file.
#
# @example
# # bad
# add_column :users, :name, :string, null: false
# add_reference :products, :category, null: false
#
# # good
# add_column :users, :name, :string, null: true
# add_column :users, :name, :string, null: false, default: ''
# add_reference :products, :category
# add_reference :products, :category, null: false, default: 1
class NotNullColumn < Base
MSG = 'Do not add a NOT NULL column without a default value.'
RESTRICT_ON_SEND = %i[add_column add_reference].freeze
def_node_matcher :add_not_null_column?, <<~PATTERN
(send nil? :add_column _ _ $_ (hash $...))
PATTERN
def_node_matcher :add_not_null_reference?, <<~PATTERN
(send nil? :add_reference _ _ (hash $...))
PATTERN
def_node_matcher :null_false?, <<~PATTERN
(pair (sym :null) (false))
PATTERN
def_node_matcher :default_option?, <<~PATTERN
(pair (sym :default) !nil)
PATTERN
def on_send(node)
check_add_column(node)
check_add_reference(node)
end
private
def check_add_column(node)
add_not_null_column?(node) do |type, pairs|
return if type.respond_to?(:value) && (type.value == :virtual || type.value == 'virtual')
check_pairs(pairs)
end
end
def check_add_reference(node)
add_not_null_reference?(node) do |pairs|
check_pairs(pairs)
end
end
def check_pairs(pairs)
return if pairs.any? { |pair| default_option?(pair) }
null_false = pairs.find { |pair| null_false?(pair) }
return unless null_false
add_offense(null_false)
end
end
end
end
end
Version data entries
10 entries across 10 versions & 4 rubygems