sh19910711/codeforces-api.rb

module Codeforces::Helper

  def problems
    @problems ||= create_problems(api.problemset.problems.map {|p| create_problem p })
  end

  def contest(query)
    create_contest resolve_contest(query)
  end

  def rounds(*args)
    contests.send :rounds, *args
  end

  def contests
    create_contests api.contest.list.map {|c| create_contest c }
  end

  def each_contest
    contests.each {|contest| yield(contest) if block_given? }
  end

  def user(handle)
    create_user api.user.info(handle).first
  end

  def users
    create_users api.user.rated_list.map {|u| create_user u }
  end

  def recent_status
    api.problemset.recent_status.map {|s| create_submission s }
  end

  def each_status
    recent_status.each {|status| yield(status) if block_given? }
  end

  private

  def resolve_contest(query)
    if query.is_a?(Integer)
      contests.select {|c| query === c.id }.first
    else
      ret = contests.select {|c| query === c.name }
      raise "Error: contest is not found" if ret.empty?
      raise "Error: query is ambiguous" if ret.length > 1
      ret.first
    end
  end

end