class RuboCop::Cop::Rails::ThreeStateBooleanColumn

def table_node(node)
  case node.method_name
  when :add_column
    node.first_argument
  when :column, :boolean
    ancestor = node.each_ancestor(:block).find do |n|
      n.method?(:create_table) || n.method?(:change_table)
    end
    ancestor&.send_node&.first_argument
  end
end