ojagent 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in ojagent.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Zejun Wu
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,51 @@
+# OJAgent
+
+OJAgent is a client to submit solutions and query status at different online
+judges. It provides a uniformed interface to a lot of famous online judges.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'ojagent'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install ojagent
+
+## Usage
+
+OJ Agent:
+
+    require 'ojagent'
+
+    oj = OJAgent::AnyAnget.new
+    oj.login($user, $pass)
+    oj.submit!($pid, $code, $lang)
+    p oj.status!
+
+Quick Submit:
+
+    $ quicksubmit -j pat -u admin -i 1001 -s pat/1001/ -t pat/1001/test/
+    $ export OJ_JUDGE=zoj
+    $ export OJ_USERNAME=watashi
+    $ export OJ_PASSWORD=*******
+    $ quicksubmit 1001.c
+
+For more information:
+
+    $ quicksubmit -h
+    $ quicksubmit -l
+    $ ri OJAgent
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/quicksubmit

@@ -0,0 +1,147 @@
+#!/usr/bin/ruby
+
+require 'ojagent'
+
+require 'find'
+require 'logger'
+require 'stringio'
+
+require 'slop'
+require 'highline'
+
+ME = 'quicksubmit'
+
+Agents = {
+  :livearchive => OJAgent::LiveArchiveAgent,
+  :timus       => OJAgent::TimusAgent
+}
+
+Ext2Lang = {
+  '.pas'  => :pascal,
+  '.c'    => :c,
+  '.cs'   => :cs,
+  '.cpp'  => :cpp,
+  '.java' => :java
+}
+
+def to_comment(status, lang)
+  if lang == :pascal
+    open, close, inline = '{', '}', nil
+  else
+    open, close, inline = '/*', '*/', nil
+  end
+
+  ios = StringIO.new('', 'w')
+  ios.puts
+  ios.puts open if open
+  status.each do |k, v|
+    ios.puts "#{k} => #{v}"
+  end
+  ios.puts close if close
+  ios.string
+end
+
+# opts
+opts = Slop.new :strict => true, :help => true do
+  banner <<-BANNER
+#{ME} -j judge [options] [-s src]
+#{ME} -j judge [options] [files ...]
+#{ME} -j judge [options] -i pid [-e lang] file
+#{ME} -l
+  BANNER
+
+  on :l, :list,      'List all available online judges.'
+  on :j, :judge=,    'The remote online judge name, required.',
+    :default => ENV['OJ_JUDGE'], :required => ENV['OJ_JUDGE'].nil?, :as => :symbol
+  on :s, :source=,   'The directory of files to processe.'
+  on :t, :dest=,     'The directory for storing processed files.'
+  on :r, :retries=,  'The number of retries for each operation.',   :as => :integer
+  on :d, :duration=, 'The time to wait before each retry.',         :as => :float
+  on :b, :break=,    'The break between submissions.',              :as => :float
+  on :u, :user=,     'Override $OJ_USERNAME, usually required.',
+    :default => ENV['OJ_USERNAME'], :required => ENV['OJ_USERNAME'].nil?
+  on :p, :pass=,     'Override $OJ_PASSWORD, prompt to ask when required.',
+    :default => ENV['OJ_PASSWORD']
+  on :i, :pid=,      'Problem id, extract from filename when omitted.'
+  on :e, :lang=,     'Program language, extract from extension when omitted.',
+    :as => :symbol
+end
+
+begin
+  opts.parse!
+rescue Slop::Error
+  raise unless opts[:help] || opts[:list]
+end
+
+# help
+exit 0 if opts[:help]
+
+# list
+if opts[:list]
+  Agents.to_a.sort.each do |name, agent|
+    puts '%-12s %s' % [name, agent.new.base_uri]
+  end
+  exit 0
+end
+
+# init
+judge = opts[:judge]
+if Agents[judge].nil?
+  raise Slop::InvalidArgumentError, "#{judge} is not a supported judge"
+else
+  OJ = Agents[judge].new
+end
+
+OJ.retries = opts[:retries] if opts[:retries]
+OJ.duration = opts[:duration] if opts[:duration]
+
+user = opts[:user]
+pass = opts[:pass] ||
+  HighLine.new.ask("#{user}@#{judge}'s password: "){|q| q.echo = false}
+OJ.login(user, pass)
+
+DESTINATION = opts[:dest] || '.'
+
+BREAK = opts[:break] ? opts[:break] : 10
+
+LOG = Logger.new($stderr)
+
+# main
+if opts[:source] || $*.empty?
+  Find.find(opts[:source] || '.') do |path|
+    $* << path if File.file?(path)
+  end
+end
+
+$*.uniq.each do |path|
+  base = File.basename(path)
+  pid = opts[:pid] || base[/^\d{4}/]
+
+  ext = File.extname(path)
+  lang = opts[:lang] || Ext2Lang[ext]
+
+  if pid && lang
+    LOG.info "PROCESS #{path}"
+  else
+    LOG.warn "IGNORE #{path}"
+    next
+  end
+
+  begin
+    code = IO.read(path)
+    id = OJ.submit! pid, code, lang
+    status = OJ.status! id
+    code += to_comment(status, lang)
+
+    FileUtils.mkdir_p(DESTINATION)
+    path = "#{DESTINATION}/#{base}"
+    path.sub!(/#{ext}$/, ".#{Time.now.to_i}#{ext}") if File.exists? path
+    IO.write(path, code)
+
+    LOG.add status[:status] =~ /accepted/i ? Logger::INFO : Logger::WARN, status
+  rescue Exception => ex
+    LOG.error ex
+  end
+
+  sleep BREAK
+end

data/lib/ojagent.rb

@@ -0,0 +1,8 @@
+require 'ojagent/version'
+require 'ojagent/ojagent.rb'
+require 'ojagent/timus_agent.rb'
+require 'ojagent/livearchive_agent.rb'
+
+module OJAgent
+  # Your code goes here...
+end

data/lib/ojagent/livearchive_agent.rb

@@ -0,0 +1,62 @@
+module OJAgent
+  class LiveArchiveAgent < OJAgent
+    def initialize
+      super 'http://icpcarchive.ecs.baylor.edu',
+        :c      => '1',
+        :java   => '2',
+        :cpp    => '3',
+        :pascal => '4'
+    end
+
+    def login(user, pass)
+      page = get '/'
+      login_form = page.form_with :id => 'mod_loginform'
+      login_form.set_fields :username => user,
+                            :passwd   => pass
+      login_form.checkbox_with(:name => 'remember').check
+      result = login_form.submit
+    end
+
+    def logout
+    end
+
+    def submit(pid, code, lang)
+      page = get '/index.php?option=com_onlinejudge&Itemid=25'
+      action = 'index.php?option=com_onlinejudge&Itemid=25&page=save_submission'
+      submit_form = page.form_with :action => action
+      submit_form.set_fields :localid  => pid,
+                             :code     => code
+      submit_form.radiobutton_with(
+        :name => 'language', :value => languages[lang]).check
+      result = submit_form.submit
+
+      message = result / '.maincontent div.message'
+      if message && message.inner_text =~ /Submission received with ID (\d+)/
+        $1
+      else
+        nil
+      end
+    end
+
+    def status(id)
+      page = get '/index.php?option=com_onlinejudge&Itemid=9'
+      status = (page / '.maincontent table tr').
+        map{|tr| tr / 'td'}.
+        find{|td| !td.empty? && td[0].inner_text == id}
+
+      return nil unless status
+      ret = {
+        :id     => status[0].inner_text,
+        :pid    => status[1].inner_text,
+        :pname  => status[2].inner_text,
+        :status => status[3].inner_text.strip,
+        :lang   => status[4].inner_text,
+        :time   => status[5].inner_text,
+        :date   => status[6].inner_text
+      }
+      url = status[3] / 'a[href]'
+      ret[:url] = url.map{|a| a['href']} unless url.empty?
+      ret
+    end
+  end
+end

data/lib/ojagent/ojagent.rb

@@ -0,0 +1,88 @@
+require 'forwardable'
+require 'mechanize'
+require 'nokogiri'
+
+module OJAgent
+  class LoginFailureError < RuntimeError
+  end
+
+  class OperationFailureError < RuntimeError
+  end
+
+  class OJAgent
+    extend Forwardable
+
+    def_delegators :agent, :submit
+
+    attr_accessor :retries, :duration
+
+    attr_reader :agent, :base_uri, :languages
+
+    def initialize(base_uri, languages)
+      @agent = Mechanize.new
+      @agent.open_timeout = 20
+      @agent.read_timeout = 60
+      @base_uri = base_uri
+      @languages = languages
+      @retries = 10
+      @duration = 2
+    end
+
+    def get(uri, parameters = [], referer = nil, headers = {})
+      agent.get base_uri + uri, parameters, referer
+    end
+
+    def post(uri, query={}, headers={})
+      agent.post base_uri + uri, query, headers
+    end
+
+    def login(user, pass)
+      raise NotImplementedError
+    end
+
+    def logout
+      raise NotImplementedError
+    end
+
+    def submit(pid, code, lang)
+      raise NotImplementedError
+    end
+
+    def status(id)
+      raise NotImplementedError
+    end
+
+    def submit!(pid, code, lang, retries = nil)
+      retries ||= self.retries
+      retries.times do
+        begin
+          ret = submit pid, code, lang
+          return ret if ret
+        rescue
+        end
+      end
+      raise OperationFailureError, "Fail to submit"
+    end
+
+    def status!(id, retries = nil, duration = nil)
+      retries ||= self.retries
+      duration ||= self.duration
+      ret = nil
+      retries.times do
+        begin
+          ret = status id
+          if ret &&
+             ret[:status] &&
+             ret[:status] =~ /[a-z]/i &&
+             ret[:status] !~ /sent to judge|(?:pend|queu|compil|link|runn|judg)ing/i
+            return ret
+          end
+        rescue
+        end
+        sleep duration
+      end
+      return ret if ret
+      raise OperationFailureError, "Fail to get status"
+    end
+  end
+end

data/lib/ojagent/timus_agent.rb

@@ -0,0 +1,54 @@
+module OJAgent
+  class TimusAgent < OJAgent
+    attr_accessor :user, :pass
+
+    def initialize
+      super 'http://acm.timus.ru',
+        :pascal => '3',
+        :java   => '7',
+        :c      => '9',
+        :cpp    => '10',
+        :cs     => '11'
+    end
+
+    def login(user, pass)
+      self.user = user
+      self.pass = pass
+    end
+
+    def logout
+    end
+
+    def submit(pid, code, lang)
+      page = get '/submit.aspx'
+      submit_form = page.form_with :action => 'submit.aspx?space=1'
+      submit_form.set_fields :JudgeID    => user + pass,
+                             :Language   => languages[lang],
+                             :ProblemNum => pid,
+                             :Source     => code
+      submit_form.submit
+      pid
+    end
+
+    def status(pid)
+      page = get '/status.aspx?author=' + user
+      status = (page / 'table.status tr').
+        map{|tr| tr / 'td'}.
+        find{|td| td[3] && td[3].inner_text.start_with?(pid)}
+
+      return nil unless status
+      ret = {}
+      ths = [:id, :date, :user, :pname, :lang, :status, :testid, :time, :mem]
+      ths.each_with_index do |k, i|
+        ret[k] = status[i].inner_text.strip.gsub('\s+', ' ')
+      end
+      if ret[:pname] =~ /^(\d+).\s*(.*)$/
+        ret[:pid] = $1
+        ret[:pname] = $2
+      end
+      url = status[5] / 'a[href]'
+      ret[:url] = url.map{|a| a['href']} unless url.empty?
+      ret
+    end
+  end
+end

data/lib/ojagent/version.rb

@@ -0,0 +1,3 @@
+module OJAgent
+  VERSION = "0.0.1"
+end

data/ojagent.gemspec

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ojagent/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "ojagent"
+  gem.version       = OJAgent::VERSION
+  gem.authors       = ["Zejun Wu"]
+  gem.email         = ["zejun.wu@gmail.com"]
+  gem.description   = <<-EOF
+OJAgent is a client to submit solutions and query status at different online
+judges. It provides a uniformed interface to a lot of famous online judges.
+  EOF
+  gem.summary       = %q{Agent/Bot/Proxy for Online Judges.}
+  gem.homepage      = "https://github.com/watashi/ojagent"
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency 'mechanize', '~> 2.5.1'
+  gem.add_dependency 'nokogiri',  '~> 1.5.5'
+  gem.add_dependency 'slop',      '~> 3.3.3'
+  gem.add_dependency 'highline',  '~> 1.6.15'
+end

metadata

@@ -0,0 +1,127 @@
+--- !ruby/object:Gem::Specification
+name: ojagent
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Zejun Wu
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mechanize
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.5.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.5.1
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.5
+- !ruby/object:Gem::Dependency
+  name: slop
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.3.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.3.3
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.15
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.15
+description: ! 'OJAgent is a client to submit solutions and query status at different
+  online
+
+  judges. It provides a uniformed interface to a lot of famous online judges.
+
+'
+email:
+- zejun.wu@gmail.com
+executables:
+- quicksubmit
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/quicksubmit
+- lib/ojagent.rb
+- lib/ojagent/livearchive_agent.rb
+- lib/ojagent/ojagent.rb
+- lib/ojagent/timus_agent.rb
+- lib/ojagent/version.rb
+- ojagent.gemspec
+homepage: https://github.com/watashi/ojagent
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Agent/Bot/Proxy for Online Judges.
+test_files: []