Rakefile

require 'bundler/gem_tasks'

def mri?
  RUBY_ENGINE == "ruby"
end

def jruby?
  defined?(JRuby)
end

desc "run tests"
task default: [:test]

if mri? || jruby?
  if mri?
    require 'rake/extensiontask'

    Rake::ExtensionTask.new('did_you_mean') do |ext|
      ext.name    = "method_receiver"
      ext.lib_dir = "lib/did_you_mean"
    end
  elsif jruby?
    require 'rake/javaextensiontask'

    Rake::JavaExtensionTask.new('did_you_mean') do |ext|
      ext.name    = "receiver_capturer"
      ext.lib_dir = "lib/did_you_mean"
    end
  end

  desc "Run tests"
  task :test do
    Rake::Task['compile'].reenable
    Rake::Task['compile'].invoke

    begin
      $stdout.puts("\033[33m")
      sh "bundle exec ruby test/all_test.rb"
    ensure
      $stdout.puts("\033[0m")
    end

    Rake::Task['clobber'].execute
  end

  namespace :test do
    desc "Run tests without re-compiling extensions"
    task :without_compile do
      $stdout.puts("\033[33m")
      sh "bundle exec ruby test/all_test.rb"
      $stdout.puts("\033[0m")
    end
  end
else # for Rubinius
  desc "Run tests"
  task :test do
    $stdout.puts("\033[33m")
    sh "bundle exec ruby test/all_test.rb"
    $stdout.puts("\033[0m")
  end
end