git-contest 0.0.1

data/bin/git-contest-init

@@ -0,0 +1,77 @@
+#!/usr/bin/env ruby
+
+#
+# git-contest-init
+# https://github.com/sh19910711/git-contest
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+$:.unshift File.expand_path('../../lib', __FILE__)
+require 'git/contest/common'
+require 'trollop'
+require 'highline/import'
+
+init
+
+sub_commands = %w()
+options = Trollop::options do
+  version "git-contest #{Git::Contest::VERSION} (c) 2013 Hiroyuki Sano"
+  opt(
+    :defaults,
+    "Use default branch naming conventions.",
+    :type => :flag,
+    :default => false,
+    :required => false,
+  )
+  opt(
+    :force,
+    "force setting of git-contest branches, even if already configured.",
+    :short => :f,
+    :type => :flag,
+    :default => false,
+    :required => false,
+  )
+  stop_on sub_commands
+end
+
+if git_contest_is_initialized && ! options[:force]
+  puts "Already initialized for git-contest."
+  puts "To force reinitialization, use: git contest init -f"
+  exit 0
+end
+
+# run commands
+if ! git_do_no_echo 'rev-parse --git-dir'
+  git_do 'init'
+end
+
+# init main
+if git_contest_has_master_configured
+  master_branch = git_do 'config --get git.contest.branch.master'
+elsif options[:defaults]
+  master_branch = 'master'
+else
+  master_branch = ask('Master branch name: ') do |q|
+    q.default = 'master'
+  end
+end
+
+if options[:defaults]
+  prefix = 'contest'
+else
+  prefix = ask('Prefix of contest branch name:  ') do |q|
+    q.default = 'contest'
+  end
+end
+
+if git_repo_is_headless
+  git_do 'symbolic-ref', 'HEAD', "\"refs/heads/#{master_branch}\""
+  git_do 'commit --allow-empty --quiet -m "Initial commit"'
+end
+
+# save config
+git_do 'config', 'git.contest.branch.master', master_branch
+git_do 'config', 'git.contest.branch.prefix', prefix
+

data/bin/git-contest-rebase

@@ -0,0 +1,76 @@
+#!/usr/bin/env ruby
+
+#
+# git-contest-rebase
+# https://github.com/sh19910711/git-contest
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+$:.unshift File.expand_path('../../lib', __FILE__)
+require 'git/contest/common'
+require 'trollop'
+
+def use_current_branch
+  current_branch = git_current_branch
+  if current_branch.start_with? $PREFIX
+    $BRANCH = current_branch.strip
+    $NAME = $BRANCH[$PREFIX.length+1..-1]
+  else
+    puts "The current HEAD is no feature branch."
+    puts "Please spefcify a <name> argument."
+    abort ''
+  end
+end
+
+def expand_nameprefix_arg name, prefix
+  expanded_name = git_contest_resolve_nameprefix name, prefix
+  exitcode = $?.to_i
+  if $? == 0
+    $NAME = expanded_name
+    $BRANCH = "#{$PREFIX}/#{$NAME}"
+  else
+    return 1
+  end
+end
+
+def expand_nameprefix_arg_or_current
+  if ARGV.length > 0
+    expand_nameprefix_arg ARGV[0], $PREFIX
+    require_branch "#{$PREFIX}/#{$NAME}"
+  else
+    use_current_branch
+  end
+end
+
+init
+
+sub_commands = %w()
+$options = Trollop::options do
+  version "git-contest #{Git::Contest::VERSION} (c) 2013 Hiroyuki Sano"
+  opt(
+    :interactive,
+    "Do an interactive rebase.",
+    :type => :flag,
+    :default => false,
+    :required => false,
+  )
+  stop_on sub_commands
+end
+
+expand_nameprefix_arg_or_current
+
+puts "Will try to rebase '#{$NAME}'..."
+
+require_clean_working_tree
+require_branch $BRANCH
+
+git_do "checkout -q \"#{$BRANCH}\""
+rebase_options = ""
+if $options[:interactive]
+  rebase_options += " -i"
+end
+
+puts git_do "rebase #{rebase_options} #{$MASTER}"
+

data/bin/git-contest-start

@@ -0,0 +1,68 @@
+#!/usr/bin/env ruby
+
+#
+# git-contest-start
+# https://github.com/sh19910711/git-contest
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+$:.unshift File.expand_path('../../lib', __FILE__)
+require 'git/contest/common'
+require 'trollop'
+
+init
+
+sub_commands = %w()
+options = Trollop::options do
+  version "git-contest #{Git::Contest::VERSION} (c) 2013 Hiroyuki Sano"
+  opt(
+    :fetch,
+    "fetch from origin before performing operation.",
+    :type => :flag,
+    :default => false,
+    :required => false,
+  )
+  stop_on sub_commands
+end
+
+if ARGV.length == 0
+  puts "Missing argument <name>"
+  exit 1
+end
+
+base_branch_name = $MASTER
+# specify based branch
+if ARGV.length == 2
+  base_branch_name = ARGV[1]
+end
+contest_branch_name = "#{ARGV[0]}"
+contest_branch = "#{$PREFIX}/#{contest_branch_name}"
+require_branch_absent contest_branch
+
+# fetch origin/master
+if options[:fetch]
+  git_do "fetch -q \"#{$ORIGIN}\""
+end
+
+# require equal
+if git_branch_exists "#{$ORIGIN}/#{$MASTER}"
+  require_branches_equal "#{$MASTER}", "#{$ORIGIN}/#{$MASTER}"
+end
+
+# create branch
+if ! git_do "checkout -b \"#{contest_branch}\" \"#{base_branch_name}\""
+  abort "Could not create contest branch #{contest_branch}"
+end
+
+puts ""
+puts "Summary of actions:"
+puts "- A new branch \"#{contest_branch}\" was created, based on \"#{base_branch_name}\""
+puts "- You are now on branch \"#{contest_branch}\""
+puts ""
+puts "Now, start committing on your contest. When done, use:"
+puts ""
+puts "    git contest finish #{contest_branch_name}"
+puts ""
+

data/bin/git-contest-submit

@@ -0,0 +1,188 @@
+#!/usr/bin/env ruby
+
+#
+# git-contest-submit
+# https://github.com/sh19910711/git-contest
+#
+# Copyright (c) 2013 Hiroyuki Sano <sh19910711 at gmail.com>
+# Licensed under the MIT-License.
+#
+
+$:.unshift File.expand_path('../../lib', __FILE__)
+require 'git/contest/common'
+require 'trollop'
+
+init
+
+$config = get_config() || {}
+$sites  = {}
+if $config.has_key? 'sites'
+  $sites = $config["sites"]
+end
+
+#
+# Load Plugins
+#
+def load_plugins
+  # load drivers
+  Dir.glob("#{$GIT_CONTEST_HOME}/plugins/**") do |path|
+    require path if /\/driver_.*\.rb$/.match path
+  end
+end
+
+load_plugins
+
+#
+# Load Drivers
+#
+def load_drivers
+  driver_names = $sites.keys().map {|key| $sites[key]["driver"] }
+  driver_names.uniq.each do |driver_name|
+    class_name = driver_name.clone
+    class_name.gsub!(/^[a-z]/) {|c| c.upcase }
+    class_name.gsub!(/(_)([a-z])/) {|c, b| $2.upcase }
+    $drivers[driver_name] = Contest::Driver.const_get "#{class_name}"
+  end
+end
+
+$drivers = {}
+load_drivers
+
+#
+# Command Utils
+#
+def usage
+  puts get_banner
+  return 0
+end
+
+def get_banner
+  res = ""
+  res += "usage: git contest submit <site>\n"
+  res += "\n"
+  res += "Available sites are:\n"
+  $sites.keys().each do |site|
+    if $drivers.has_key? $sites[site]["driver"]
+      driver = $drivers[$sites[site]["driver"]].new
+      res += "  %-12s\t#{driver.get_desc}\n" % [site]
+    else
+      # TODO: driver not found
+    end
+  end
+  res += "\n"
+  res += "Try 'git contest submit <site> --help' for details.\n"
+  return res
+end
+
+def get_git_add_target rule
+  str = rule
+  str = str.gsub('${source}', $submit_info[:source])
+  str
+end
+
+# check options
+sub_commands = $sites.keys
+global_opts = Trollop::options do
+  version "git-contest #{Git::Contest::VERSION} (c) 2013 Hiroyuki Sano"
+  banner get_banner
+  stop_on sub_commands
+end
+
+# detect site
+site = ARGV.shift.to_s.strip
+
+unless $sites.has_key?(site)
+  if site != ""
+    puts "site not found"
+  else
+    usage
+  end
+  exit 0
+end
+
+# detect driver
+driver_name = $sites[site]["driver"]
+
+unless $drivers.has_key?(driver_name)
+  puts "driver not found"
+  exit
+end
+
+#
+# Submit Start
+#
+driver = $drivers[driver_name].new
+
+$submit_info = {}
+
+# set events
+driver.on(
+  'start',
+  Proc.new do
+    puts "@start: submit"
+  end
+)
+driver.on(
+  'before_login',
+  Proc.new do
+    puts "@submit: logging in..."
+  end
+)
+driver.on(
+  'after_login',
+  Proc.new do
+    puts "@submit: login ok"
+  end
+)
+driver.on(
+  'before_submit',
+  Proc.new do |submit_info|
+    $submit_info = submit_info
+    puts "@submit: doing..."
+  end
+)
+driver.on(
+  'after_submit',
+  Proc.new do
+    puts "@submit: done"
+  end
+)
+driver.on(
+  'before_wait',
+  Proc.new do
+    print "@result: waiting..."
+  end
+)
+driver.on(
+  'retry',
+  Proc.new do
+    print "."
+  end
+)
+driver.on(
+  'after_wait',
+  Proc.new do |submission_info|
+    puts ""
+    next unless submission_info.is_a?(Hash)
+    puts ""
+    puts "@result: Submission Result"
+    puts "  %s: %s" % ["submission id", "#{submission_info[:submission_id]}"]
+    puts "  %s: %s" % ["status", "#{submission_info[:status]}"]
+    puts ""
+    if git_contest_is_initialized
+      git_do "add #{get_git_add_target($config["submit_rules"]["add"] || ".")}"
+      git_do "commit --allow-empty -m '#{submission_info[:result]}'"
+    end
+  end
+)
+driver.on(
+  'finish',
+  Proc.new do
+    puts "@finish"
+  end
+)
+
+opts = driver.get_opts()
+result = driver.submit $sites[site], "main.*", opts
+
+

data/git-contest.gemspec

@@ -0,0 +1,34 @@
+# coding: utf-8
+
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+require 'git/contest/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "git-contest"
+  spec.version       = Git::Contest::VERSION
+  spec.authors       = ["Hiroyuki Sano"]
+  spec.email         = ["sh19910711@gmail.com"]
+  spec.description   = %q{the Git Extension for online judge (Codeforces, Google Codejam, ICPC, etc...)}
+  spec.summary       = %q{the git extension for online judge}
+  spec.homepage      = "https://github.com/sh19910711/git-contest"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "mechanize"
+  spec.add_dependency "nokogiri"
+  spec.add_dependency "trollop"
+  spec.add_dependency "highline"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "watchr"
+  spec.add_development_dependency "rspec"
+  spec.add_development_dependency "webmock"
+end
+