require 'bundler'
begin
  Bundler.setup
rescue Bundler::BundlerError => e
  $stderr.puts e.message
  $stderr.puts "Run `bundle install` to install missing gems"
  exit e.status_code
end
require 'rake'
require 'rake/clean'

$:.unshift(File.join(File.dirname(__FILE__), './lib'))

require 'anystyle/version'

task :default
task :build => [:clean] do
  system 'gem build anystyle-parser.gemspec'
end

task :release => [:build] do
  system "git tag #{AnyStyle::VERSION}"
  system "gem push anystyle-parser-#{AnyStyle::VERSION}.gem"
end

task :check_warnings do
  $VERBOSE = true
  require 'anystyle/parser'

  puts AnyStyle::VERSION
end

require 'rspec/core'
require 'rspec/core/rake_task'
RSpec::Core::RakeTask.new(:spec) do |spec|
  spec.pattern = FileList['spec/**/*_spec.rb']
end

begin
  require 'coveralls/rake/task'
  Coveralls::RakeTask.new
  task :test_with_coveralls => [:spec, 'coveralls:push']
rescue LoadError
  # ignore
end if ENV['CI']

task :default => :spec

begin
  require 'yard'
  YARD::Rake::YardocTask.new
rescue LoadError
  # ignore
end

desc 'Run an IRB session with AnyStyle loaded'
task :console, [:script] do |t, args|
  ARGV.clear
  require 'irb'
  require 'anystyle'
  IRB.conf[:SCRIPT] = args.script
  IRB.start
end

desc 'Update model using latest source and training data'
task :train, :model, :threads do |t, args|
  model = args[:model] || 'parser'
  threads = args[:threads] || 4
  require 'anystyle'
  Wapiti.debug!
  case model
  when 'finder'
    AnyStyle::Finder.defaults[:threads] = threads
    AnyStyle.finder.train
    AnyStyle.finder.model.save
  else
    AnyStyle::Parser.defaults[:threads] = threads
    AnyStyle.parser.train
    AnyStyle.parser.model.save
  end
end


desc 'Check all tagged datasets'
task :check, :model do |t, args|
  model = args[:model] || 'parser'
  require 'anystyle'
  case model
  when 'finder'
    Dir['./res/finder/*.ttx'].sort.each do |ttx|
      print 'Checking %.25s' % "#{File.basename(ttx)}....................."
      start = Time.now
      stats = AnyStyle.finder.check ttx.untaint
      report stats, Time.now - start
    end
  else
    Dir['./res/parser/*.xml'].sort.each do |xml|
      print 'Checking %.25s' % "#{File.basename(xml)}....................."
      start = Time.now
      stats = AnyStyle.parser.check xml.untaint
      report stats, Time.now - start
    end
  end
end

desc "Save delta of a tagged dataset with itself"
task :delta, :input do |t, args|
  require 'anystyle'
  input = args[:input].untaint
  if File.directory?(input)
    files = Dir.entries(input)
      .reject { |f| f.start_with?('.') }
      .map { |f| File.join(input, f).untaint }
  else
    files = [input]
  end
  files.each do |file|
    extn = File.extname(file)
    print 'Checking %.25s' % "#{File.basename(file)}....................."
    case extn
    when '.ttx'
      input = Wapiti::Dataset.new([AnyStyle::Document.open(file)])
      output = AnyStyle.finder.label input
      format = 'txt'
    else
      input = Wapiti::Dataset.open(file)
      output = AnyStyle.parser.label input
      format = 'xml'
    end
    delta = output - input
    if delta.length == 0
      puts ' ✓'
    else
      name = File.basename(file, extn)
      delta.save "delta_#{name}#{extn}", indent: 2, tagged: true, format: format
      puts "delta saved to delta_#{name}#{extn} (#{delta.length})"
    end
  end
end

desc "Find references in document"
task :find, :input do |t, args|
  require 'anystyle'
  file = args[:input].untaint
  refs = AnyStyle.finder.find(file, format: :references)[0]
  break unless refs.length > 0
  output = AnyStyle.parser.label refs.join("\n")
  puts output.to_xml(indent: 2)
end

def report(stats, time)
  if stats[:token][:errors] == 0
    puts '   ✓                               %2ds' % time
  else
    puts '%4d seq %6.2f%% %6d tok %5.2f%% %2ds' % [
      stats[:sequence][:errors],
      stats[:sequence][:rate],
      stats[:token][:errors],
      stats[:token][:rate],
      time
    ]
  end
end

CLEAN.include('*.gem')
CLEAN.include('*.rbc')