# -*- mode: ruby; -*-
require 'rubygems'
require 'rubygems/specification'
require 'fileutils'
require 'rake'
require 'rake/testtask'
require 'rake/gempackagetask'
begin
  require 'rake/contrib/rubyforgepublisher'
rescue LoadError
end
require 'rbconfig'
include Config
ENV['TEST_MODE'] = 'TRUE'

desc "Test the MongoDB Ruby driver."
task :test do
  puts "\nThis option has changed."
  puts "\nTo test the driver with the c-extensions:\nrake test:c\n"
  puts "To test the pure ruby driver: \nrake test:ruby"
end

namespace :test do

  desc "Test the driver with the c extension enabled."
  task :c do
    ENV['C_EXT'] = 'TRUE'
    Rake::Task['test:unit'].invoke
    Rake::Task['test:functional'].invoke
    Rake::Task['test:pooled_threading'].invoke
    Rake::Task['test:drop_databases'].invoke
    ENV['C_EXT'] = nil
  end

  desc "Test the driver using pure ruby (no c extension)"
  task :ruby do
    ENV['C_EXT'] = nil
    Rake::Task['test:unit'].invoke
    Rake::Task['test:functional'].invoke
    Rake::Task['test:pooled_threading'].invoke
    Rake::Task['test:drop_databases'].invoke
  end

  Rake::TestTask.new(:unit) do |t|
    t.test_files = FileList['test/unit/*_test.rb']
    t.verbose    = true
  end

  Rake::TestTask.new(:functional) do |t|
    t.test_files = FileList['test/test*.rb']
    t.verbose    = true
  end

  Rake::TestTask.new(:pooled_threading) do |t|
    t.test_files = FileList['test/threading/*.rb']
    t.verbose    = true
  end

  Rake::TestTask.new(:pair_count) do |t|
    t.test_files = FileList['test/replica/count_test.rb']
    t.verbose    = true
  end

  Rake::TestTask.new(:pair_insert) do |t|
    t.test_files = FileList['test/replica/insert_test.rb']
    t.verbose    = true
  end

  Rake::TestTask.new(:pooled_pair_insert) do |t|
    t.test_files = FileList['test/replica/pooled_insert_test.rb']
    t.verbose    = true
  end

  Rake::TestTask.new(:pair_query) do |t|
    t.test_files = FileList['test/replica/query_test.rb']
    t.verbose    = true
  end

  task :drop_databases do |t|
    puts "Dropping test database..."
    require File.join(File.dirname(__FILE__), 'lib', 'mongo')
    include Mongo
    con = Connection.new(ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
      ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT)
    con.drop_database('ruby-mongo-test')
  end
end

desc "Generate documentation"
task :rdoc do
  version = eval(File.read("mongo-ruby-driver.gemspec")).version
  out = File.join('html', version.to_s)
  FileUtils.rm_rf('html')
  system "rdoc --main README.rdoc --op #{out} --inline-source --quiet README.rdoc `find lib -name '*.rb'`"
end

desc "Generate YARD documentation"
task :ydoc do
  require File.join(File.dirname(__FILE__), 'lib', 'mongo')
  out = File.join('ydoc', Mongo::VERSION)
  FileUtils.rm_rf('ydoc')
  system "yardoc lib/**/*.rb lib/mongo/**/*.rb -p templates -o #{out} --title MongoRuby-#{Mongo::VERSION}"
end

desc "Publish documentation to mongo.rubyforge.org"
task :publish => [:rdoc] do
  # Assumes docs are in ./html
  Rake::RubyForgePublisher.new(GEM, RUBYFORGE_USER).upload
end

namespace :gem do

  desc "Install the gem locally"
  task :install do
    sh "gem build mongo-ruby-driver.gemspec"
    sh "gem install mongo-*.gem"
    sh "rm mongo-*.gem"
  end

  desc "Install the optional c extensions"
  task :install_extensions do
    sh "gem build mongo-extensions.gemspec"
    sh "gem install mongo_ext-*.gem"
    sh "rm mongo_ext-*.gem"
  end

end

task :default => :list

task :list do
  system 'rake -T'
end