git-contest 0.0.1

data/lib/contest/driver/common.rb

@@ -0,0 +1,72 @@
+#
+# common.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+require 'mechanize'
+require 'nokogiri'
+require 'trollop'
+require 'contest/driver/base'
+
+module Contest
+  module Driver
+    module Utils
+      def self.resolve_path path
+        path = `ls #{path} | cat | head -n 1`
+        path.strip
+      end
+
+      def self.resolve_language path
+        regexp = /\.([a-z0-9]+)$/
+        if path.match(regexp)
+          return path.match(regexp)[1]
+        else
+          return nil
+        end
+      end
+
+      def self.normalize_language label
+        case label
+        when "c", "C"
+          return "clang"
+        when "cpp", "C++", "c++"
+          return "cpp"
+        when "c++11", "C++11"
+          return "cpp11"
+        when "cs", "c#", "C#"
+          return "cs"
+        when "d", "D", "dlang"
+          return "dlang"
+        when "go", "golang"
+          return "golang"
+        when "hs", "haskell", "Haskell"
+          return "haskell"
+        when "java", "Java"
+          return "java"
+        when "ocaml", "ml", "OCaml"
+          return "ocaml"
+        when "Delphi", "delphi"
+          return "delphi"
+        when "pascal", "Pascal"
+          return "pascal"
+        when "perl", "Perl", "pl"
+          return "perl"
+        when "php", "PHP"
+          return "php"
+        when "python2"
+          return "python2"
+        when "python3", "python", "Python", "py"
+          return "python3"
+        when "ruby", "rb", "Ruby"
+          return "ruby"
+        when "scala", "Scala"
+          return "scala"
+        else
+          abort "unknown language @ normalize language"
+        end
+      end
+    end
+  end
+end

data/lib/contest/driver/driver_event.rb

@@ -0,0 +1,30 @@
+#
+# driver_event.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+module Contest
+  module Driver
+    class DriverEvent
+      def initialize
+        @callbacks = {}
+      end
+      def on(type, proc)
+        @callbacks[type] = [] unless @callbacks.has_key?(type)
+        @callbacks[type].push proc
+      end
+      def off(type, proc)
+        @callbacks[type] = [] unless @callbacks.has_key?(type)
+        @callbacks[type].delete proc
+      end
+      def trigger(type, *params)
+        @callbacks[type] = [] unless @callbacks.has_key?(type)
+        @callbacks[type].each do |proc|
+          proc.call *params
+        end
+      end
+    end
+  end
+end

data/lib/contest/driver/uva_online_judge.rb

@@ -0,0 +1,131 @@
+#
+# uva_online_judge.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+require 'contest/driver/common'
+
+module Contest
+  module Driver
+    class UvaOnlineJudge < DriverBase
+      def get_opts_ext
+        define_options do
+          opt(
+            :problem_id,
+            "Problem ID (Ex: 100, 200, etc...)",
+            :type => :string,
+            :required => true,
+          )
+        end
+      end
+
+      def get_site_name
+        "UVa"
+      end
+
+      def get_desc
+        "UVa Online Judge (URL: http://uva.onlinejudge.org/)"
+      end
+
+      def resolve_language(label)
+        case label
+        when "c"
+          return "1"
+        when "cpp"
+          return "3"
+        when "java"
+          return "2"
+        when "pascal"
+          return "4"
+        else
+          abort "unknown language"
+        end
+      end
+
+      def submit_ext(config, source_path, options)
+        trigger 'start'
+        problem_id = options[:problem_id]
+
+        @client = Mechanize.new {|agent|
+          agent.user_agent_alias = 'Windows IE 7'
+        }
+
+        # submit
+        trigger 'before_login'
+        login_page = @client.get 'http://uva.onlinejudge.org/'
+        login_page.form_with(:id => 'mod_loginform') do |form|
+          form.username = config["user"]
+          form.passwd = config["password"]
+        end.submit
+        trigger 'after_login'
+
+        trigger 'before_submit', options
+        submit_page = @client.get 'http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=25'
+        res_page = submit_page.form_with(:action => 'index.php?option=com_onlinejudge&Itemid=25&page=save_submission') do |form|
+          form.localid = problem_id
+          form.radiobutton_with(:name => 'language', :value => options[:language]).check
+          form.code = File.read(source_path)
+        end.submit
+        trigger 'after_submit'
+
+        # <div class="message">Submission received with ID 12499981</div>
+        trigger 'before_wait'
+        submission_id = get_submission_id(res_page.body)
+        status = get_status_wait(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(submission_id)
+        submission_id = submission_id.to_s
+        # wait result
+        12.times do
+          sleep 10
+          my_page = @client.get 'http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=9'
+          status = get_submission_status(submission_id, my_page.body)
+          return status unless status == 'Sent to judge' || status == ''
+          trigger 'retry'
+        end
+        trigger 'timeout'
+        return 'timeout'
+      end
+
+      def get_submission_id(body)
+        doc = Nokogiri::HTML(body)
+        text = doc.xpath('//div[@class="message"]')[0].text().strip
+        # Submission received with ID 12500010
+        text.match(/Submission received with ID ([0-9]+)/)[1]
+      end
+
+      def get_submission_status(submission_id, body)
+        doc = Nokogiri::HTML(body)
+        doc.xpath('//tr[@class="sectiontableheader"]/following-sibling::node()').search('tr').each do |elm|
+          td_list = elm.search('td')
+          item_submission_id = td_list[0].text.strip
+          if item_submission_id == submission_id
+            item_problem_id = td_list[1].text.strip
+            item_status = td_list[3].text.strip
+            return item_status
+          end
+        end
+        'timeout'
+      end
+
+      if is_test_mode?
+        attr_writer :client
+      else
+      end
+    end
+  end
+end

data/lib/git/contest.rb

@@ -0,0 +1,14 @@
+#
+# contest.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+require "git/contest/version"
+
+module Git
+  module Contest
+    # Your code goes here...
+  end
+end

data/lib/git/contest/common.rb

@@ -0,0 +1,48 @@
+#
+# common.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+require 'git/contest/version'
+require 'git/contest/test'
+require 'git/contest/git'
+require 'contest/driver'
+require 'yaml'
+
+def init
+  init_global
+  init_home
+end
+
+def init_global
+  $GIT_CONTEST_HOME   = File.expand_path(ENV['GIT_CONTEST_HOME'] || "~/.git-contest")
+  $GIT_CONTEST_CONFIG = File.expand_path(ENV['GIT_CONTEST_CONFIG'] || "#{$GIT_CONTEST_HOME}/config.yml")
+  if git_do_no_echo 'branch'
+    $MASTER = git_do 'config --get git.contest.branch.master'
+    $PREFIX = git_do 'config --get git.contest.branch.prefix'
+    $ORIGIN = git_do 'config --get git.contest.origin'
+    if $ORIGIN == ''
+      $ORIGIN = 'origin'
+    end
+    $GIT_CONTEST_GIT_OK = true
+  else
+    $GIT_CONTEST_GIT_OK = false
+  end
+end
+
+def init_home
+  if ! FileTest.exists? $GIT_CONTEST_HOME
+    FileUtils.mkdir $GIT_CONTEST_HOME
+  end
+  if ! FileTest.exists? $GIT_CONTEST_CONFIG
+    FileUtils.touch $GIT_CONTEST_CONFIG
+  end
+end
+
+def get_config
+  config_path = File.expand_path($GIT_CONTEST_CONFIG)
+  YAML.load_file config_path
+end
+

data/lib/git/contest/git.rb

@@ -0,0 +1,189 @@
+#
+# git.rb
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+def git_do(*args)
+  puts "git #{args.join(' ')}" if ENV['GIT_CONTEST_DEBUG'] == 'ON'
+  return `git #{args.join(' ')} 2>&1`.strip
+end
+
+# use return value
+def git_do_no_echo(*args)
+  puts "git #{args.join(' ')}" if ENV['GIT_CONTEST_DEBUG'] == 'ON'
+  system "git #{args.join(' ')} >/dev/null 2>&1"
+end
+
+#
+def git_contest_is_initialized
+  git_contest_has_master_configured &&
+    git_contest_has_prefix_configured &&
+    git_do('config --get git.contest.branch.master') != git_do('config --get git.contest.branch.develop')
+end
+
+def git_contest_has_master_configured
+  master = (git_do 'config --get git.contest.branch.master').strip
+  master != '' && git_local_branches().include?(master)
+end
+
+def git_contest_has_develop_configured
+  develop = (git_do 'config --get git.contest.branch.develop').strip
+  develop != '' && git_local_branches().include?(develop)
+end
+
+def git_contest_has_prefix_configured
+  git_do_no_echo 'config --get git.contest.branch.prefix'
+end
+
+def git_contest_resolve_nameprefix name, prefix
+  if git_local_branch_exists "#{prefix}/#{name}"
+    return name
+  end
+  branches = git_local_branches().select {|branch| branch.start_with? "#{prefix}/#{name}" }
+  if branches.size == 0
+    abort "No branch matches prefix '#{name}'"
+  else
+    if branches.size == 1
+      return branches[0][prefix.length..-1]
+    else
+      abort "Multiple branches match prefix '#{name}'"
+    end
+  end
+end
+
+#
+def git_remote_branch_exists(branch_name)
+  git_remote_branches().include?(branch_name)
+end
+
+def git_local_branch_exists(branch_name)
+  git_local_branches().include?(branch_name)
+end
+
+def git_branch_exists(branch_name)
+  git_all_branches().include?(branch_name)
+end
+
+def git_remote_branches
+  cmd_ret = git_do 'branch -r --no-color'
+  cmd_ret.lines.map {|line|
+    line.gsub(/^[*]?\s*/, '').gsub(/\s*$/, '').strip
+  }
+end
+
+def git_local_branches
+  cmd_ret = git_do 'branch --no-color'
+  cmd_ret.lines.map {|line|
+    line.gsub(/^[*]?\s*/, '').gsub(/\s*$/, '').strip
+  }
+end
+
+def git_all_branches
+  cmd_ret1 = git_do 'branch --no-color'
+  cmd_ret2 = git_do 'branch -r --no-color'
+  lines = ( cmd_ret1 + cmd_ret2 ).lines
+  lines.map {|line|
+    line.gsub(/^[*]?\s*/, '').gsub(/\s*$/, '').strip
+  }
+end
+
+def git_current_branch
+  ret = git_do('branch --no-color').lines
+  ret = ret.grep /^\*/
+  ret[0].gsub(/^[* ] /, '').strip
+end
+
+def git_is_clean_working_tree
+  if ! git_do_no_echo 'diff --no-ext-diff --ignore-submodules --quiet --exit-code'
+    return 1
+  elsif ! git_do_no_echo 'diff-index --cached --quiet --ignore-submodules HEAD --'
+    return 2
+  else
+    return 0
+  end
+end
+
+def git_repo_is_headless
+  ! git_do_no_echo 'rev-parse --quiet --verify HEAD'
+end
+
+#
+# 0: same
+# 1: first branch needs ff
+# 2: second branch needs ff
+# 3: branch needs merge
+# 4: there is no merge
+#
+def git_compare_branches first, second
+  commit1 = git_do "rev-parse \"#{first}\""
+  commit2 = git_do "rev-parse \"#{second}\""
+  if commit1 != commit2
+    if git_do_no_echo("merge-base \"#{commit1}\" \"#{commit2}\"") > 0
+      return 4
+    else
+      base = git_do "merge-base \"#{commit1}\" \"#{commit2}\""
+      if commit1 == base
+        return 1
+      elsif commit2 == base
+        return 2
+      else
+        return 3
+      end
+    end
+  else
+    return 0
+  end
+end
+
+def require_branch(branch)
+  if ! git_all_branches().include?(branch)
+    abort "Branch #{branch} does not exist."
+  end
+end
+
+def require_branch_absent(branch)
+  if git_all_branches().include?(branch)
+    abort "Branch #{branch} already exists. Pick another name."
+  end
+end
+
+def require_clean_working_tree
+  ret = git_is_clean_working_tree
+  if ret == 1
+    abort "fatal: Working tree contains unstaged changes. Aborting."
+  end
+  if ret == 2
+    abort "fatal: Index contains uncommited changes. Aborting."
+  end
+end
+
+def require_local_branch branch
+  if ! git_local_branch_exists branch
+    abort "fatal: Local branch '#{branch}' does not exist and is required."
+  end
+end
+
+def require_remote_branch branch
+  if ! git_remote_branch_exists branch
+    abort "fatal: Remote branch '#{branch}' does not exist and is required."
+  end
+end
+
+def require_branches_equal local, remote
+  require_local_branch local
+  require_remote_branch remote
+  ret = git_compare_branches local, remote
+  if ret > 0
+    puts "Branches '#{local}' and '#{remote}' have diverged."
+    if ret == 1
+      abort "And branch #{local} may be fast-forwarded."
+    elsif ret == 2
+      puts "And local branch #{local} is ahead of #{remote}"
+    else
+      abort "Branches need merging first."
+    end
+  end
+end
+