git-contest 0.0.1

data/lib/contest/driver.rb

@@ -0,0 +1,15 @@
+#
+# driver.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+# http://judge.u-aizu.ac.jp/onlinejudge/
+require 'contest/driver/aizu_online_judge'
+
+# http://codeforces.com/
+require 'contest/driver/codeforces'
+
+# http://uva.onlinejudge.org/
+require 'contest/driver/uva_online_judge'

data/lib/contest/driver/aizu_online_judge.rb

@@ -0,0 +1,159 @@
+#
+# aizu_online_judge.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+require 'contest/driver/common'
+require 'rexml/document'
+
+module Contest
+  module Driver
+    class AizuOnlineJudge < DriverBase
+      def get_opts_ext
+        define_options do
+          opt(
+            :problem_id,
+            "Problem ID (Ex: 1000, 123, 0123, etc...)",
+            :type => :string,
+            :required => true,
+          )
+        end
+      end
+
+      def get_site_name
+        "AOJ"
+      end
+
+      def get_problem_id(options)
+        "#{options[:problem_id]}"
+      end
+
+      def get_desc
+        "Aizu Online Judge (URL: http://judge.u-aizu.ac.jp)"
+      end
+
+      def resolve_language(label)
+        case label
+        when "clang"
+          return "C"
+        when "cpp"
+          return "C++"
+        when "java"
+          return "JAVA"
+        when "cpp11"
+          return "C++11"
+        when "cs"
+          return "C#"
+        when "dlang"
+          return "D"
+        when "ruby"
+          return "Ruby"
+        when "python"
+          return "Python"
+        when "php"
+          return "PHP"
+        when "javascript"
+          return "JavaScript"
+        else
+          abort "unknown languag"
+        end
+      end
+
+      def submit_ext(config, source_path, options)
+        # start
+        trigger 'start'
+        problem_id = "%04d" % options[:problem_id]
+
+        @client = Mechanize.new {|agent|
+          agent.user_agent_alias = 'Windows IE 7'
+        }
+
+        # submit
+        trigger 'before_submit', options
+        submit_page = @client.get "http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=0000"
+        submit_page.parser.encoding = "SHIFT_JIS"
+        res_page = submit_page.form_with(:action => '/onlinejudge/servlet/Submit') do |form|
+          form.userID = config["user"]
+          form.password = config["password"]
+          form.problemNO = problem_id
+          form.language = options[:language]
+          form.sourceCode = File.read(source_path)
+        end.submit
+        trigger 'after_submit'
+
+        # need to get the newest waiting submissionId
+        trigger 'before_wait'
+        status_page = @client.get "http://judge.u-aizu.ac.jp/onlinejudge/status.jsp"
+        submission_id = get_submission_id status_page.body
+
+        # wait result
+        status = get_status_wait config["user"], submission_id
+        trigger(
+          'after_wait',
+          {
+            :submission_id => submission_id,
+            :status => status,
+            :result => get_commit_message($config["submit_rules"]["message"], status, options),
+          }
+        )
+        trigger 'finish'
+
+        return "AOJ %s: %s\nhttp://judge.u-aizu.ac.jp/onlinejudge/review.jsp?rid=#{submission_id}" % [problem_id, status]
+      end
+
+      def get_status_wait(user_id, submission_id)
+        # wait result
+        5.times do
+          sleep 3
+          status_page = @client.get "http://judge.u-aizu.ac.jp/onlinejudge/webservice/status_log?user_id=#{user_id}&limit=1"
+          status = get_status(submission_id, status_page.body)
+          return status unless is_waiting(submission_id, status_page.body)
+          trigger 'retry'
+        end
+        trigger 'timeout'
+        return 'request timed out'
+      end
+
+      def status_loop(doc)
+        doc.root.elements['//status_list'].elements.each('status') do |elm|
+          run_id = elm.elements['run_id'].text.to_s.strip
+          status = elm.elements['status'].text.to_s.strip
+          yield run_id, status
+        end
+      end
+
+      def is_waiting(submission_id, body)
+        doc = REXML::Document.new body
+        status_loop(doc) do |run_id|
+          return false if run_id == submission_id
+        end
+        return true
+      end
+
+      def get_status(submission_id, body)
+        doc = REXML::Document.new body
+        status_loop(doc) do |run_id, status|
+          return status if run_id == submission_id
+        end
+        return ''
+      end
+
+      def get_submission_id(body)
+        doc = Nokogiri::HTML body
+        doc.xpath('//table[@id="tableRanking"]//tr[@class="dat"]')[0].search('td')[0].text.strip
+      end
+
+      if is_test_mode?
+        attr_writer :client
+      else
+        private :status_loop
+        private :get_status_wait
+        private :is_waiting
+        private :get_status
+        private :get_submission_id
+      end
+    end
+  end
+end

data/lib/contest/driver/base.rb

@@ -0,0 +1,103 @@
+#
+# base.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+require 'contest/driver/common'
+require 'rexml/document'
+require 'contest/driver/driver_event'
+
+module Contest
+  module Driver
+    class DriverBase < DriverEvent
+      def get_opts_ext
+        # Example:
+        # define_options do
+        #   opt(
+        #     :problem_id,
+        #     "Problem ID (Ex: 1000, 123, 0123, etc...)",
+        #     :type => :string,
+        #     :required => true,
+        #   )
+        # end
+        raise 'TODO: Implement'
+      end
+
+      def get_site_name
+        raise 'TODO: Implement'
+      end
+
+      def get_problem_id(options)
+        raise 'TODO: Implement'
+      end
+
+      def resolve_language(label)
+        raise 'TODO: Implement'
+      end
+
+      def submit_ext(config, source_path, options)
+        raise 'TODO: Implement'
+      end
+
+      def get_desc
+        ''
+      end
+
+      def define_options &block
+        @blocks ||= []
+        @blocks.push(block)
+      end
+
+      def get_opts
+        get_opts_ext
+        define_options do
+          opt(
+            :source,
+            "Specify submitted code (Ex: main.cpp)",
+            :type => :string,
+            :required => false,
+          )
+          opt(
+            :language,
+            "Specify programming language (Ex: C++, Java or etc...)",
+            :type => :string,
+            :required => false,
+          )
+        end
+        Trollop::options ARGV, @blocks do |blocks|
+          version "git-contest driver"
+          blocks.each do |b|
+            instance_eval &b
+          end
+        end
+      end
+
+      def get_commit_message_ext
+        nil
+      end
+
+      def get_commit_message rule, status, options
+        message = rule
+        message = message.gsub '${site}', get_site_name
+        message = message.gsub '${problem-id}', get_problem_id(options)
+        message = message.gsub '${status}', status
+        message = "\n#{get_commit_message_ext}" unless get_commit_message_ext.nil?
+        return message
+      end
+
+      def submit(config, source_path, options)
+        $config = get_config
+        $config["submit_rules"] ||= {}
+        $config["submit_rules"]["message"] ||= "${site} ${problem-id}: ${status}"
+        source_path = Utils.resolve_path(options[:source] || $config["submit_rules"]["source"] || source_path)
+        options[:source] = source_path
+        options[:language] ||= Utils.resolve_language(source_path)
+        options[:language] = resolve_language Utils.normalize_language(options[:language])
+        status = submit_ext(config, source_path, options)
+        get_commit_message($config["submit_rules"]["message"], status, options)
+      end
+    end
+  end
+end

data/lib/contest/driver/codeforces.rb

@@ -0,0 +1,196 @@
+#
+# codeforces.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+require 'contest/driver/common'
+
+module Contest
+  module Driver
+    class Codeforces < DriverBase 
+      def get_opts_ext
+        define_options do
+          opt(
+            :contest_id,
+            "Contest ID (Ex: 100, 234, etc...)",
+            :type => :string,
+            :required => true,
+          )
+          opt(
+            :problem_id,
+            "Problem ID (Ex: A, B, etc...)",
+            :type => :string,
+            :required => true,
+          )
+        end
+      end
+
+      def get_site_name
+        "Codeforces"
+      end
+
+      def get_problem_id(options)
+        "#{options[:contest_id]}#{options[:problem_id]}"
+      end
+
+      def get_desc
+        "Codeforces (URL: http://codeforces.com/)"
+      end
+
+      def resolve_language(label)
+        case label
+        when "clang"
+          return "10"
+        when "cpp"
+          return "1"
+        when "cpp11"
+          return "16"
+        when "cs"
+          return "9"
+        when "dlang"
+          return "28"
+        when "golang"
+          return "32"
+        when "haskell"
+          return "12"
+        when "java"
+          return "23"
+        when "ocaml"
+          return "19"
+        when "delphi"
+          return "3"
+        when "pascal"
+          return "4"
+        when "perl"
+          return "13"
+        when "php"
+          return "6"
+        when "python2"
+          return "7"
+        when "python3"
+          return "31"
+        when "ruby"
+          return "8"
+        when "scala"
+          return "20"
+        else
+          abort "unknown languag"
+        end
+      end
+
+      def submit_ext(config, source_path, options)
+        # start
+        trigger 'start'
+        contest_id = options[:contest_id]
+        problem_id = options[:problem_id]
+
+        @client = Mechanize.new {|agent|
+          agent.user_agent_alias = 'Windows IE 7'
+        }
+
+        # login
+        trigger 'before_login'
+        login_page = @client.get 'http://codeforces.com/enter'
+        login_page.form_with(:action => '') do |form|
+          form.handle = config["user"]
+          form.password = config["password"]
+        end.submit
+        trigger 'after_login'
+
+        # submit
+        trigger 'before_submit', options
+        # retry once
+        retries = 1
+        begin
+          submit_page = @client.get "http://codeforces.com/contest/#{contest_id}/submit"
+          res_page = submit_page.form_with(:class => 'submit-form') do |form|
+            form.submittedProblemIndex = problem_id
+            form.programTypeId = options[:language]
+            form.source = File.read(source_path)
+          end.submit
+        rescue => e
+          raise if retries == 0
+          retries -= 1
+          # may not be registered practice
+          if /submittedProblemIndex/ =~ e.to_s
+            contest_page = @client.get "http://codeforces.com/contest/#{contest_id}"
+            contest_page.form_with(:action => "") do |form|
+              flag_need_to_register = false
+              form.field_with(:value => "registerForPractice") do |field|
+                flag_need_to_register = true
+              end
+              form.click_button if flag_need_to_register
+            end
+            sleep 3
+            retry
+          end
+          raise
+        end
+        trigger 'after_submit'
+
+        # need to get the newest waiting submissionId
+        trigger 'before_wait',
+          submission_id = get_submission_id(res_page.body)
+
+        # wait result
+        status = get_status_wait(contest_id, submission_id)
+        trigger(
+          'after_wait',
+          {
+            :submission_id => submission_id,
+            :status => status,
+            :result => get_commit_message($config["submit_rules"]["message"], status, options),
+          }
+        )
+
+        trigger 'finish'
+        get_commit_message($config["submit_rules"]["message"], status, options)
+      end
+
+      def get_status_wait(contest_id, submission_id)
+        contest_id = contest_id.to_s
+        submission_id = submission_id.to_s
+        # wait result
+        5.times do
+          sleep 3
+          my_page = @client.get "http://codeforces.com/contest/#{contest_id}/my"
+          status = get_status(submission_id, my_page.body)
+          return status unless is_waiting(submission_id, my_page.body)
+          trigger 'retry'
+        end
+        trigger 'timeout'
+        return 'timeout'
+      end
+
+      def is_waiting(submission_id, body)
+        doc = Nokogiri::HTML(body)
+        element = doc.xpath('//td[@submissionid="' + submission_id + '"]')[0]
+        element["waiting"] == "true"
+      end
+
+      def get_status(submission_id, body)
+        doc = Nokogiri::HTML(body)
+        element = doc.xpath('//td[@submissionid="' + submission_id + '"]')[0]
+        element.text().strip
+      end
+
+      def get_submission_id(body)
+        doc = Nokogiri::HTML(body)
+        # get td.status-cell
+        elements = doc.xpath('//td[contains(concat(" ",@class," "), " status-cell ")][@waiting="true"]')
+        elements[0].attributes()["submissionid"].value.strip
+      end
+
+      if is_test_mode?
+        attr_writer :client
+      else
+        private :get_status_wait
+        private :is_waiting
+        private :get_status
+        private :get_submission_id
+      end
+    end
+  end
+end