require 'rspec/core/rake_task'

###########################################################
#
#   Documentation Tasks
#
###########################################################
DOCS_DAEMON = 'yard server --reload --daemon --server thin'
FOREGROUND_SERVER = 'bundle exec yard server --reload --verbose --server thin lib/beaker'

def running?(cmdline)
  ps = `ps -ef`
  found = ps.lines.grep(/#{Regexp.quote(cmdline)}/)
  raise StandardError, "Found multiple YARD Servers. Don't know what to do." if found.length > 1

  yes = found.empty? ? false : true
  [yes, found.first]
end

def pid_from(output)
  output.squeeze(' ').strip.split(' ')[1]
end

desc 'Start the documentation server in the foreground'
task docs: 'docs:clear' do
  original_dir = Dir.pwd
  Dir.chdir(__dir__)
  sh FOREGROUND_SERVER
  Dir.chdir(original_dir)
end

namespace :docs do
  desc 'Clear the generated documentation cache'
  task :clear do
    original_dir = Dir.pwd
    Dir.chdir(__dir__)
    sh 'rm -rf docs'
    Dir.chdir(original_dir)
  end

  desc 'Generate static documentation'
  task gen: 'docs:clear' do
    original_dir = Dir.pwd
    Dir.chdir(__dir__)
    output = `bundle exec yard doc`
    puts output
    raise 'Errors/Warnings during yard documentation generation' if output =~ /\[warn\]|\[error\]/

    Dir.chdir(original_dir)
  end

  desc 'Run the documentation server in the background, alias `bg`'
  task background: 'docs:clear' do
    yes, output = running?(DOCS_DAEMON)
    if yes
      puts 'Not starting a new YARD Server...'
      puts "Found one running with pid #{pid_from(output)}."
    else
      original_dir = Dir.pwd
      Dir.chdir(__dir__)
      sh "bundle exec #{DOCS_DAEMON}"
      Dir.chdir(original_dir)
    end
  end

  task(:bg) { Rake::Task['docs:background'].invoke }

  desc 'Check the status of the documentation server'
  task :status do
    yes, output = running?(DOCS_DAEMON)
    if yes
      pid = pid_from(output)
      puts "Found a YARD Server running with pid #{pid}"
    else
      puts 'Could not find a running YARD Server.'
    end
  end

  desc 'Stop a running YARD Server'
  task :stop do
    yes, output = running?(DOCS_DAEMON)
    if yes
      pid = pid_from(output)
      puts "Found a YARD Server running with pid #{pid}"
      `kill #{pid}`
      puts 'Stopping...'
      yes, output = running?(DOCS_DAEMON)
      if yes
        `kill -9 #{pid}`
        yes, output = running?(DOCS_DAEMON)
        if yes
          puts 'Could not Stop Server!'
        else
          puts 'Server stopped.'
        end
      else
        puts 'Server stopped.'
      end
    else
      puts 'Could not find a running YARD Server'
    end
  end
end