require 'logger'
require 'pathname'
require_relative '../gemfile_selector'

module SchemaDev
  module Rspec
    module Db
      extend self

      def setup
        @db = GemfileSelector.infer_db
        set_logger
        connect
        RSpec.configure do |config|
          config.include Helpers
          config.filter_run_excluding :postgresql => :only unless Helpers.postgresql?
          config.filter_run_excluding :postgresql => :skip if Helpers.postgresql?
          config.filter_run_excluding :mysql => :only unless Helpers.mysql?
          config.filter_run_excluding :mysql => :skip if Helpers.mysql?
          config.filter_run_excluding :sqlite3 => :only unless Helpers.sqlite3?
          config.filter_run_excluding :sqlite3 => :skip if Helpers.sqlite3?
        end
      end

      def tmproot
        @tmproot ||= Pathname.new('tmp').tap { |path| path.mkpath }
      end

      def logroot
        @logroot ||= Pathname.new('log').tap { |path| path.mkpath }
      end

      def configuration
        case @db
        when 'mysql'
          {
            :adapter => 'mysql',
            :database => 'schema_plus_test',
            :username => ENV.fetch('MYSQL_DB_USER', 'schema_plus'),
            :encoding => 'utf8',
            :min_messages => 'warning'
          }
        when 'mysql2'
          {
            :adapter => 'mysql2',
            :database => 'schema_plus_test',
            :username => ENV.fetch('MYSQL_DB_USER', 'schema_plus'),
            :encoding => 'utf8',
            :min_messages => 'warning'
          }
        when 'postgresql'
          {
            :adapter => 'postgresql',
            :username => ENV['POSTGRESQL_DB_USER'],
            :database => 'schema_plus_test',
            :min_messages => 'warning'
          }
        when 'sqlite3'
          {
            :adapter => 'sqlite3',
            :database => tmproot.join('schema_plus.sqlite3').to_s
          }
        else
          raise "Unknown db adapter #{@db.inspect}"
        end
      end

      def connect
        ActiveRecord::Base.configurations = { 'schema_dev' => configuration }
        ActiveRecord::Base.establish_connection :schema_dev
        case @db
        when 'sqlite3'
          ActiveRecord::Base.connection.execute "PRAGMA synchronous = OFF"
        end
      end

      def set_logger
        ruby = "#{RUBY_ENGINE}-#{RUBY_VERSION}"
        activerecord = "activerecord-#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}"
        ActiveRecord::Base.logger = Logger.new(logroot.join("#{ruby}.#{activerecord}.#{@db}.log").open("w"))
      end

      module Helpers
        extend self

        def mysql?
          ActiveRecord::Base.connection.adapter_name =~ /^mysql/i
        end

        def postgresql?
          ActiveRecord::Base.connection.adapter_name =~ /^postgresql/i
        end

        def sqlite3?
          ActiveRecord::Base.connection.adapter_name =~ /^sqlite/i
        end
      end
    end
  end
end