require 'bundler/gem_tasks'
require 'rspec/core/rake_task'
require 'rubygems/package_task'
require 'rake/extensiontask'

RSpec::Core::RakeTask.new(:spec)

task default: :spec

Rake::ExtensionTask.new('character_set') do |ext|
  ext.lib_dir = 'lib/character_set'
end

namespace :java do
  java_gemspec = eval File.read('./character_set.gemspec')
  java_gemspec.platform = 'java'
  java_gemspec.extensions = []

  java_gemspec.add_runtime_dependency 'range_compressor', '~> 1.0'

  Gem::PackageTask.new(java_gemspec) do |pkg|
    pkg.need_zip = true
    pkg.need_tar = true
    pkg.package_dir = 'pkg'
  end
end

task package: 'java:gem'

desc 'Download relevant ruby/spec tests, adapt to CharacterSet and its variants'
task :sync_ruby_spec do
  require 'fileutils'

  variants = {
    'CharacterSet'       => './spec/ruby-spec/library/character_set',
    'CharacterSet::Pure' => './spec/ruby-spec/library/character_set_pure',
  }
  variants.each do |_, dir|
    FileUtils.rm_rf(dir) if File.exist?(dir)
    `svn export https://github.com/ruby/spec/trunk/library/set/sortedset #{dir}`
  end

  base = variants.first[1]
  variants.each_value { |dir| FileUtils.copy_entry(base, dir) unless dir == base }

  variants.each.with_index do |(class_name, dir), i|
    Dir["#{dir}/**/*.rb"].each do |spec|
      # remove some tests that do not apply or are covered otherwise
      if spec =~ %r{/(flatten|initialize|pretty_print)}
        File.delete(spec)
        next
      end

      # some examples w. Strings must be adapted, "mspec" made rspec-compatible,
      # and `i` added to shared example names or they'll override each other
      adapted_content =
        File
        .read(spec)
        .gsub('SortedSet', class_name)
        .gsub('sorted_set_', "sorted_set_#{i}_")
        .gsub(/describe (.*), shared.*$/, 'shared_examples \1 do |method|')
        .gsub(/1\.0|"cat"|"dog"|"hello"|"test"/, '0')
        .gsub('"one"', '1')
        .gsub('"two"', '2')
        .gsub('"three"', '3')
        .gsub('"four"', '4')
        .gsub('"five"', '5')
        .gsub('@method', 'method')
        .gsub(/be_(false|true)/, 'be \1')
        .gsub('mock', 'double')

      File.open(spec, 'w') { |f| f.puts adapted_content }
    end
  end
end

desc 'Download unicode casefold data and write new C header file'
task :sync_casefold_data do
  src_path = './CaseFolding.txt'
  dst_path = './ext/character_set/unicode_casefold_table.h'

  `wget http://www.unicode.org/Public/UNIDATA/CaseFolding.txt`

  mapping = File.foreach(src_path).each_with_object({}) do |line, hash|
    from, type, to = line.split(/\s*;\s*/).first(3)
    # type 'C' stands for 'common', excludes mappings to multiple chars
    hash[from] = to if type == 'C'
  end.sort

  File.open(dst_path, 'w') do |f|
    f.puts <<-C
// THIS FILE IS GENERATED BY $ rake sync_casefold_data - DO NOT EDIT'

typedef struct casefold_mapping {
  unsigned long from;
  unsigned long to;
} casefold_mapping;

#define CASEFOLD_COUNT #{mapping.size}

static const casefold_mapping unicode_casefold_table[CASEFOLD_COUNT] = {
    C

    mapping.each { |from, to| f.puts "{0x#{from},0x#{to}}," }

    f.puts '};'
  end

  File.unlink(src_path)
end

desc 'Run all IPS benchmarks'
task :benchmark do
  Dir['./benchmarks/*.rb'].sort.each { |file| require file }
end

namespace :benchmark do
  desc 'Run all IPS benchmarks and store the comparison results in BENCHMARK.md'
  task :write_to_file do
    $store_comparison_results = {}

    Rake.application[:benchmark].invoke

    File.open('BENCHMARK.md', 'w') do |f|
      f.puts "Results of `rake:benchmark` on #{RUBY_DESCRIPTION}", ''

      $store_comparison_results.each do |caption, result|
        f.puts '```', caption, '',
               result.strip.gsub(/(same-ish).*$/, '\1').lines[1..-1], '```'
      end
    end
  end
end

unless RUBY_PLATFORM =~ /java/
  # recompile before benchmarking or running specs
  task(:benchmark).enhance([:compile])
  task(:spec).enhance([:compile])
end