class Pronto::RakeTask::TravisPullRequest

def client
  Octokit::Client.new
end

def initialize(*args, &task_block)
  setup_ivars(args)
  unless ::Rake.application.last_comment
    desc 'Run Pronto on Travis Pull Request'
  end
  task(name, *args) do |_, task_args|
    RakeFileUtils.send(:verbose, verbose) do
      yield(*[self, task_args].slice(0, task_block.arity)) if task_block
      run_task
    end
  end
end

def pull_id
  ENV['TRAVIS_PULL_REQUEST']
end

def repo_slug
  ENV['TRAVIS_REPO_SLUG']
end

def run_task
  return if pull_id.nil? || pull_id == 'false'
  pull_request = client.pull_request(repo_slug, pull_id)
  formatter = ::Pronto::Formatter::GithubFormatter.new
  gem_names = ::Pronto::GemNames.new.to_a
  gem_names.each { |gem_name| require "pronto/#{gem_name}" }
  ::Pronto.run(pull_request.base.sha, '.', formatter)
end

def setup_ivars(args)
  @name = args.shift || :pronto_travis_pull_request
  @verbose = true
end