rsim/oracle-enhanced

# frozen_string_literal: true

require "active_record/base"

module ActiveRecord
  module ConnectionAdapters
    class OracleEnhancedAdapter
      class DatabaseTasks
        delegate :connection, :establish_connection, to: ActiveRecord::Base

        def initialize(config)
          @config = config
        end

        def create
          system_password = ENV.fetch("ORACLE_SYSTEM_PASSWORD") {
            print "Please provide the SYSTEM password for your Oracle installation (set ORACLE_SYSTEM_PASSWORD to avoid this prompt)\n>"
            $stdin.gets.strip
          }
          establish_connection(@config.merge(username: "SYSTEM", password: system_password))
          begin
            connection.execute "CREATE USER #{@config[:username]} IDENTIFIED BY #{@config[:password]}"
          rescue => e
            if /ORA-01920/.match?(e.message) # user name conflicts with another user or role name
              connection.execute "ALTER USER #{@config[:username]} IDENTIFIED BY #{@config[:password]}"
            else
              raise e
            end
          end

          OracleEnhancedAdapter.permissions.each do |permission|
            connection.execute "GRANT #{permission} TO #{@config[:username]}"
          end
        end

        def drop
          establish_connection(@config)
          connection.execute_structure_dump(connection.full_drop)
        end

        def purge
          drop
          connection.execute("PURGE RECYCLEBIN") rescue nil
        end

        def structure_dump(filename, extra_flags)
          establish_connection(@config)
          File.open(filename, "w:utf-8") { |f| f << connection.structure_dump }
          if @config[:structure_dump] == "db_stored_code"
            File.open(filename, "a") { |f| f << connection.structure_dump_db_stored_code }
          end
        end

        def structure_load(filename, extra_flags)
          establish_connection(@config)
          connection.execute_structure_dump(File.read(filename))
        end
      end
    end
  end
end

ActiveRecord::Tasks::DatabaseTasks.register_task(/(oci|oracle)/, ActiveRecord::ConnectionAdapters::OracleEnhancedAdapter::DatabaseTasks)