# frozen_string_literal: true

module SaferRailsConsole
  module Patches
    module Sandbox
      module TransactionReadOnly
        module ActiveRecord
          module ConnectionAdapters
            module PostgreSQLAdapter
              def begin_db_transaction
                super
                execute 'SET TRANSACTION READ ONLY'
              end
            end
          end
        end
      end
    end
  end
end

if defined?(::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter)
  ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.prepend(SaferRailsConsole::Patches::Sandbox::TransactionReadOnly::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter)

  # Ensure transaction is read-only if it was began before this patch was loaded
  connection = ::ActiveRecord::Base.connection
  connection.execute 'SET TRANSACTION READ ONLY' if connection.open_transactions > 0
end