require 'active_record'
require 'active_record/fixtures'
require 'active_support/multibyte' # needed for Ruby 1.9.1
require 'stringio'

$query_count = 0
$query_sql = []

ignore_sql = /
    ^(
      PRAGMA | SHOW\ max_identifier_length |
      SELECT\ (currval|CAST|@@IDENTITY|@@ROWCOUNT) |
      SHOW\ ((FULL\ )?FIELDS|TABLES)
    )\b |
    \bFROM\ (sqlite_master|pg_tables|pg_attribute)\b
  /x

ActiveSupport::Notifications.subscribe(/^sql\./) do |*args|
  payload = args.last
  unless payload[:name] =~ /^Fixture/ or payload[:sql] =~ ignore_sql
    $query_count += 1
    $query_sql << payload[:sql]
  end
end

module ActiverecordTestConnector
  extend self

  attr_accessor :able_to_connect
  attr_accessor :connected

  FIXTURES_PATH = File.expand_path('../../fixtures', __FILE__)

  Fixtures = defined?(ActiveRecord::FixtureSet) ? ActiveRecord::FixtureSet :
             defined?(ActiveRecord::Fixtures) ? ActiveRecord::Fixtures :
             ::Fixtures

  # Set our defaults
  self.connected = false
  self.able_to_connect = true

  def setup
    unless self.connected || !self.able_to_connect
      setup_connection
      load_schema
      add_load_path FIXTURES_PATH
      self.connected = true
    end
  rescue Exception => e  # errors from ActiveRecord setup
    $stderr.puts "\nSkipping ActiveRecord tests: #{e}\n\n"
    self.able_to_connect = false
  end

  private

  def add_load_path(path)
    dep = defined?(ActiveSupport::Dependencies) ? ActiveSupport::Dependencies : ::Dependencies
    dep.autoload_paths.unshift path
  end

  def setup_connection
    db = ENV['DB'].blank?? 'sqlite3' : ENV['DB']

    configurations = YAML.load_file(File.expand_path('../../support/database.yml', __FILE__))
    raise "no configuration for '#{db}'" unless configurations.key? db
    configuration = configurations[db]

    # ActiveRecord::Base.logger = Logger.new(STDOUT) if $0 == 'irb'

    ActiveRecord::Base.configurations = { db => configuration }
    ActiveRecord::Base.establish_connection(db.to_sym)
    ActiveRecord::Base.default_timezone = :utc
  end

  def load_schema
    begin
      $stdout = StringIO.new
      ActiveRecord::Migration.verbose = false
      load File.join(FIXTURES_PATH, 'schema.rb')
    ensure
      $stdout = STDOUT
    end
  end

  module FixtureSetup
    def fixtures(*tables)
      cattr_accessor :loaded_fixtures
      cattr_accessor :fixture_cache
      cattr_accessor :fixture_table_names

      table_names = tables.map { |t| t.to_s }
      self.fixture_table_names = table_names

      define_method(:setup) do
        fixtures = Fixtures.create_fixtures ActiverecordTestConnector::FIXTURES_PATH, self.class.fixture_table_names
        self.class.loaded_fixtures = {}
        self.class.fixture_cache = {}

        unless fixtures.nil?
          if fixtures.instance_of?(Fixtures)
            self.class.loaded_fixtures[fixtures.table_name] = fixtures
          else
            fixtures.each { |f| self.class.loaded_fixtures[f.table_name] = f }
          end
        end
      end

      define_method(:teardown) do
        Fixtures.reset_cache
      end

      table_names.each do |table_name|
        define_method(table_name) do |*fixtures|
          self.class.fixture_cache[table_name] ||= {}

          instances = fixtures.map do |fixture|
            if self.class.loaded_fixtures[table_name][fixture.to_s]
              self.class.fixture_cache[table_name][fixture] ||= self.class.loaded_fixtures[table_name][fixture.to_s].find
            else
              raise StandardError, "No fixture with name '#{fixture}' found for table '#{table_name}'"
            end
          end

          instances.size == 1 ? instances.first : instances
        end
      end
    end
  end
end