ojagent 0.0.1 → 0.0.4

data/bin/quicksubmit

@@ -11,10 +11,13 @@ require 'highline'
 
 ME = 'quicksubmit'
 
-Agents = {
-  :livearchive => OJAgent::LiveArchiveAgent,
-  :timus       => OJAgent::TimusAgent
-}
+Agents = {}
+OJAgent.all.each do |agent|
+  name = agent.name.downcase
+  name.sub!(/.*::/, '')
+  name.sub!(/agent$/, '')
+  Agents[name.to_sym] = agent
+end
 
 Ext2Lang = {
   '.pas'  => :pascal,
@@ -138,7 +141,8 @@ $*.uniq.each do |path|
     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
+    level = status[:status] =~ /correct|accepted/i ? Logger::INFO : Logger::WARN
+    LOG.add level, status
   rescue Exception => ex
     LOG.error ex
   end

data/examples/hdoj3337.rb

@@ -0,0 +1,57 @@
+#!/usr/bin/ruby
+
+require 'ojagent'
+
+# Change these two lines
+USER = ENV['OJ_USERNAME'] || 'ojagent'
+PASS = ENV['OJ_PASSWORD'] || 'ppnn13%dkstFeb.1st'
+
+HDOJ = OJAgent::HDOJAgent.new
+HDOJ.login USER, PASS
+
+def code(off)
+  return <<-CODE
+#include <time.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+int main(void) {
+  int n = #{off};
+  clock_t cpu;
+  size_t mem;
+  char* p = NULL;
+
+  while (n-- > 0) {
+    getchar();
+  }
+  n = getchar();
+  if (n == EOF) {
+    n = 0;
+  }
+
+  cpu = ((n >> 4) * 100 + 10) * (CLOCKS_PER_SEC / 1000);
+  mem = ((n & 0xF) * 1000 + 10) * 1024;
+  p = malloc(mem);
+  memset(p, 0xff, mem);
+  while (clock() < cpu) {
+    p[rand() % mem] = (char)cpu;
+  }
+  putchar(p[0]);
+  free(p);
+
+  return 0;
+}
+  CODE
+end
+
+input = ''
+50.times do
+  id = HDOJ.submit! '3337', code(input.size), :c
+  status = HDOJ.status! id
+  ascii = (status[:time].to_i / 100) * 16 + (status[:mem].to_i / 1000)
+  break if ascii.zero?
+  input << ascii.chr
+  puts input
+end
+p input

data/examples/rosalind.rb

@@ -0,0 +1,66 @@
+#!/usr/bin/ruby
+
+BEGIN {
+  require 'ojagent'
+
+  $filename = 'rosalind.cookie'
+  $rosalind = OJAgent::RosalindAgent.new
+  $rosalind.agent.cookie_jar.load $filename if File.readable? $filename
+}
+
+END {
+  $rosalind.agent.cookie_jar.save_as $filename
+  FileUtils.chmod 0600, $filename
+}
+
+require 'highline'
+
+begin
+  require 'wirble'
+  def pp(status)
+    puts Wirble::Colorize.colorize(status.inspect)
+  end
+rescue LoadError
+  alias :pp :p
+end
+
+def usage
+  print <<-USAGE
+Usage:
+  rosalind.rb login
+  rosalind.rb $pid.pl dl
+  rosalind.rb $pid.pl up
+USAGE
+  exit 1
+end
+
+$pl, $op = *$*
+
+if $pl =~ /login/i
+  hl = HighLine.new
+  user = hl.ask("Username: ")
+  pass = hl.ask("Password: "){|q| q.echo = false}
+  $rosalind.login user, pass
+elsif $pl =~ /[a-z]*/
+  id = $&
+  if $op =~ /dl|download|open/i
+    input = $rosalind.open! id
+    IO.write input.filename, input.content
+    system %{perl #{$pl} #{input.filename} | tee output}
+  elsif $op =~ /up|upload|submit/i
+    output = IO.read('output')
+    source = IO.read($pl)
+    status = $rosalind.judge! id, [output, 'output'], [source, $pl]
+    pp status
+    if status && status[:status] == 'Correct'
+      FileUtils.mv $pl, 'pl/', :verbose => true
+      FileUtils.mv "rosalind_#{id}.txt", 'txt/', :verbose => true
+      FileUtils.rm "rosalind_#{id}.txt.index", :force => true
+    end
+  else
+    usage
+  end
+else
+  usage
+end
+

data/lib/ojagent.rb

@@ -1,8 +1,17 @@
 require 'ojagent/version'
 require 'ojagent/ojagent.rb'
-require 'ojagent/timus_agent.rb'
-require 'ojagent/livearchive_agent.rb'
+
+libdir = File.join(File.expand_path(File.dirname(__FILE__)), 'ojagent')
+Dir.entries(libdir).each do |filename|
+  require File.join(libdir, filename) if filename.end_with? '_agent.rb'
+end
 
 module OJAgent
-  # Your code goes here...
+  # Return all avialbe agents.
+  def self.all
+    ObjectSpace.each_object(Class).select{|agent| agent < OJAgent}
+  end
+
+  # All avialbe agents at initial time.
+  StandardAgents = all
 end

data/lib/ojagent/hdoj_agent.rb

@@ -0,0 +1,41 @@
+module OJAgent
+  class HDOJAgent < OJAgent
+    def initialize
+      super 'http://acm.hdu.edu.cn',
+        :cpp    => 0,
+        :c      => 1,
+        :mscpp  => 2,
+        :msc    => 3,
+        :pascal => 4,
+        :java   => 5
+    end
+
+    def login(user, pass)
+      @user = user
+      page = get '/'
+      login_form = page.form_with :action => '/userloginex.php?action=login'
+      login_form.set_fields :username => user,
+                            :userpass => pass
+      login_form.submit
+    end
+
+    def submit(pid, code, lang)
+      page = get '/submit.php'
+      submit_form = page.form_with :action => './submit.php?action=submit'
+      submit_form.set_fields :problemid  => pid,
+                             :language   => languages[lang],
+                             :usercode   => code
+      submit_form.submit
+      pid.to_s
+    end
+
+    @@selector = '#fixed_table table.table_text tr'
+    @@thead = [:id, :date, :status, :pid, :time, :mem, :length, :lang, :user]
+
+    def status(pid)
+      parse_status '/status.php?user=' + user, @@selector, @@thead do |tr|
+        tr[3] && tr[3].inner_text == pid
+      end
+    end
+  end
+end

data/lib/ojagent/livearchive_agent.rb

@@ -9,6 +9,7 @@ module OJAgent
     end
 
     def login(user, pass)
+      @user = user
       page = get '/'
       login_form = page.form_with :id => 'mod_loginform'
       login_form.set_fields :username => user,
@@ -17,9 +18,6 @@ module OJAgent
       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'
@@ -38,25 +36,14 @@ module OJAgent
       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}
+    @@selector = '.maincontent table tr'
+    @@thead = [:id, :pid, :pname, :status, :lang, :time, :data]
 
-      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
+    def status(id)
+      url = '/index.php?option=com_onlinejudge&Itemid=9'
+      parse_status url, @@selector, @@thead do |tr|
+        !tr.empty? && tr[0].inner_text == id
+      end
     end
   end
 end

data/lib/ojagent/ojagent.rb

@@ -2,6 +2,14 @@ require 'forwardable'
 require 'mechanize'
 require 'nokogiri'
 
+class Mechanize::Form::FileUpload
+  def file=(file, default_name = 'noname')
+    data, name = *file
+    self.file_data = data
+    self.file_name = name || default_name
+  end
+end
+
 module OJAgent
   class LoginFailureError < RuntimeError
   end
@@ -16,9 +24,9 @@ module OJAgent
 
     attr_accessor :retries, :duration
 
-    attr_reader :agent, :base_uri, :languages
+    attr_reader :agent, :base_uri, :languages, :user
 
-    def initialize(base_uri, languages)
+    def initialize(base_uri = nil, languages = {})
       @agent = Mechanize.new
       @agent.open_timeout = 20
       @agent.read_timeout = 60
@@ -44,6 +52,10 @@ module OJAgent
       raise NotImplementedError
     end
 
+    def open(pid)
+      # noop
+    end
+
     def submit(pid, code, lang)
       raise NotImplementedError
     end
@@ -52,6 +64,43 @@ module OJAgent
       raise NotImplementedError
     end
 
+    def parse_status(url, selector, thead, &block)
+      tbody = get(url) / selector
+      tr = tbody.map{|tr| tr / 'td'}.find{|tr| yield tr}
+      return nil unless tr
+
+      ret = {}
+      thead.zip(tr) do |th, td|
+        next unless th && td
+        ret[th] = td.inner_text.strip.gsub(/\s+/, ' ')
+        if th == :status
+          url = td / 'a[href]'
+          if url.size > 0
+            ret[:url] = (ret[:url] || []) + url.map{|a| a['href']}
+          end
+        end
+      end
+      ret
+    end
+
+    # Open the problem or download the input. For online judges that ask
+    # users to download the input and upload the output in time limit,
+    # user must open (download the input) before submit (upload the output).
+    # For other online judges, it does nothing.
+    def open!(pid, retries = 1)
+      retries ||= self.retries
+      retries.times do
+        begin
+          ret = open pid
+          return ret if ret
+        rescue
+        end
+      end
+      raise OperationFailureError, "Fail to open"
+    end
+
+    # Submit solution and retry on failure. The return value is a hint for
+    # getting status and varys at different online judges.
     def submit!(pid, code, lang, retries = nil)
       retries ||= self.retries
       retries.times do
@@ -64,6 +113,7 @@ module OJAgent
       raise OperationFailureError, "Fail to submit"
     end
 
+    # Get status and retry on failure.
     def status!(id, retries = nil, duration = nil)
       retries ||= self.retries
       duration ||= self.duration
@@ -84,5 +134,11 @@ module OJAgent
       return ret if ret
       raise OperationFailureError, "Fail to get status"
     end
+
+    # Submit solution and get the corresponding status.
+    def judge!(pid, code, lang, retries = nil, duration = nil)
+      id = submit!(pid, code, lang, retries)
+      status!(id, retries, duration)
+    end
   end
 end

data/lib/ojagent/rosalind_agent.rb

@@ -0,0 +1,44 @@
+module OJAgent
+  class RosalindAgent < OJAgent
+    def initialize
+      super 'http://rosalind.info'
+    end
+
+    def login(user, pass)
+      @user = user
+      page = get '/accounts/login/'
+      login_form = page.form_with :id => 'id_form_login'
+      login_form.set_fields :username => user,
+                            :password => pass
+      login_form.submit
+    end
+
+    def open(pid)
+      get "/problems/#{pid}/dataset/"
+    end
+
+    def submit(pid, ans, src = nil)
+      page = get "/problems/#{pid}/"
+      submit_form = page.form_with :id => 'id_form_submission'
+      submit_form.file_upload_with(:name => 'output_file').file = ans
+      submit_form.file_upload_with(:name => 'code').file = src if src
+      submit_form.submit
+    end
+
+    @@close = "\xc3\x97".force_encoding('utf-8')
+
+    def status(page)
+      status = {}
+      [:error, :warning, :info, :success].each do |type|
+        messages = []
+        (page / "div.main div.alert-#{type}.flash-message").each do |alert|
+          message = alert.inner_text.strip.gsub(@@close, '').gsub(/\s+/, ' ')
+          messages << message unless message.empty?
+        end
+        status[type] = messages unless messages.empty?
+      end
+      status[:status] = status[:success] && !status[:error] ? 'Correct' : 'Wrong'
+      status
+    end
+  end
+end

data/lib/ojagent/timus_agent.rb

@@ -1,7 +1,5 @@
 module OJAgent
   class TimusAgent < OJAgent
-    attr_accessor :user, :pass
-
     def initialize
       super 'http://acm.timus.ru',
         :pascal => '3',
@@ -12,43 +10,33 @@ module OJAgent
     end
 
     def login(user, pass)
-      self.user = user
-      self.pass = pass
-    end
-
-    def logout
+      @user = user
+      @pass = pass
     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,
+      submit_form.set_fields :JudgeID    => user + @pass,
                              :Language   => languages[lang],
                              :ProblemNum => pid,
                              :Source     => code
       submit_form.submit
-      pid
+      pid.to_s
     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)}
+    @@selector = 'table.status tr'
+    @@thead = [:id, :date, :user, :pname, :lang, :status, :testid, :time, :mem]
 
-      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+', ' ')
+    def status(pid)
+      url = '/status.aspx?author=' + user
+      status = parse_status url, @@selector, @@thead do |tr|
+        tr[3] && tr[3].inner_text.start_with?(pid)
       end
-      if ret[:pname] =~ /^(\d+).\s*(.*)$/
-        ret[:pid] = $1
-        ret[:pname] = $2
+      if status[:pname] =~ /^(\d+).\s*(.*)$/
+        status[:pid], status[:pname] = $1, $2
       end
-      url = status[5] / 'a[href]'
-      ret[:url] = url.map{|a| a['href']} unless url.empty?
-      ret
+      status
     end
   end
 end

data/lib/ojagent/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ojagent
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-06 00:00:00.000000000 Z
+date: 2013-01-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mechanize
@@ -94,9 +94,13 @@ files:
 - README.md
 - Rakefile
 - bin/quicksubmit
+- examples/hdoj3337.rb
+- examples/rosalind.rb
 - lib/ojagent.rb
+- lib/ojagent/hdoj_agent.rb
 - lib/ojagent/livearchive_agent.rb
 - lib/ojagent/ojagent.rb
+- lib/ojagent/rosalind_agent.rb
 - lib/ojagent/timus_agent.rb
 - lib/ojagent/version.rb
 - ojagent.gemspec