cf0d7c1c46
Invalid enum values would previously result in a Sequel::DatabaseError (wrapping PG::InvalidTextRepresentation) when saving. This changes the behavior from an unhandled exception 500 error to a nice 400 error. We previously had manually validations in LoadBalancer for the enum values. This removes those. Only change in the tests is to an error message, and I think the new error message is nicer. In order for this to work, the Database needs to use the pg_enum plugin, so this uses that and regenerates the schema cache with it.
103 lines
3.0 KiB
Ruby
103 lines
3.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "db"
|
|
require "sequel/model"
|
|
|
|
if ENV["RACK_ENV"] == "development"
|
|
Sequel::Model.cache_associations = false
|
|
end
|
|
|
|
Sequel::Model.plugin :auto_validations
|
|
Sequel::Model.plugin :require_valid_schema
|
|
Sequel::Model.plugin :singular_table_names
|
|
Sequel::Model.plugin :subclasses unless ENV["RACK_ENV"] == "development"
|
|
Sequel::Model.plugin :column_encryption do |enc|
|
|
enc.key 0, Config.clover_column_encryption_key
|
|
end
|
|
Sequel::Model.plugin :many_through_many
|
|
Sequel::Model.plugin :insert_conflict
|
|
Sequel::Model.plugin :inspect_pk
|
|
Sequel::Model.plugin :static_cache_cache, "cache/static_cache.cache"
|
|
Sequel::Model.plugin :pg_auto_constraint_validations, cache_file: "cache/pg_auto_constraint_validations.cache"
|
|
Sequel::Model.plugin :pg_auto_validate_enums, message: proc { |valid_values| "is not one of the supported values (#{valid_values.sort.join(", ")})" }
|
|
Sequel::Model.plugin :pg_eager_any_typed_array
|
|
|
|
def Sequel.any_type(array, type)
|
|
Sequel.function(:ANY, Sequel.pg_array(array, type))
|
|
end
|
|
|
|
def Sequel.any_uuid(array)
|
|
if array.is_a?(Array)
|
|
any_type(array, :uuid)
|
|
else
|
|
array
|
|
end
|
|
end
|
|
|
|
if (level = Config.database_logger_level) || Config.test?
|
|
require "logger"
|
|
LOGGER = Logger.new($stdout, level: level || "fatal")
|
|
DB.loggers << LOGGER
|
|
end
|
|
|
|
if ENV["CHECK_LOGGED_SQL"]
|
|
require "logger"
|
|
File.unlink("sql.log")
|
|
f = File.open("sql.log", "ab")
|
|
|
|
# Remove optimization that does not use parameterization
|
|
def (Sequel::Model).reset_fast_pk_lookup_sql = nil
|
|
|
|
# Hack to make specs pass that mock Time.now and depend
|
|
# on certain number of Time.now calls
|
|
time = Time.now
|
|
def time.now
|
|
self
|
|
end
|
|
Logger.const_set(:Time, time)
|
|
|
|
sql_logger = Logger.new(f, level: :INFO)
|
|
sql_logger.formatter = proc do |sev, _, _, msg|
|
|
"#{sev} -- : #{msg}\0"
|
|
end
|
|
|
|
DB.loggers << sql_logger
|
|
end
|
|
|
|
module SequelExtensions
|
|
def delete(force: false, &)
|
|
# Do not error if this is a plain dataset that does not respond to destroy
|
|
return super(&) unless respond_to?(:destroy)
|
|
|
|
caller_lines = caller
|
|
rodauth_in_callstack = !caller_lines.grep(/rodauth/).empty?
|
|
destroy_in_callstack = !caller_lines.grep(/sequel\/model\/base.*_destroy_delete/).empty?
|
|
|
|
# This can happen when fast instance deletes are disabled (when CHECK_LOGGED_SQL
|
|
# environment variable is set)
|
|
callee_in_callstack = !caller_lines.grep(/#{Regexp.escape(__FILE__)}.*delete/).empty?
|
|
|
|
unless rodauth_in_callstack || destroy_in_callstack || callee_in_callstack || force
|
|
raise "Calling delete is discouraged as it skips hooks such as before_destroy, which " \
|
|
"we use to archive records. Use destroy instead. If you know what you are doing " \
|
|
"and still want to use delete, you can pass force: true to trigger delete."
|
|
end
|
|
|
|
if is_a?(Sequel::Dataset)
|
|
super(&)
|
|
else
|
|
super()
|
|
end
|
|
end
|
|
end
|
|
|
|
module Sequel
|
|
class Dataset
|
|
prepend SequelExtensions
|
|
end
|
|
|
|
class Model
|
|
prepend SequelExtensions
|
|
end
|
|
end
|