aoj 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: b5e7eb9e935d083673c1106654ebf87b884edf9e
+  data.tar.gz: 58c4bf1e4f68be8f3aad5dd02d31c9885da9b82c
+SHA512:
+  metadata.gz: 31627f6169e330dbdb128c69d33a82ac4d958b3262d082f3c799e1421a18938a010656f51c21e5c9ebe6daf85aae609ca57d31d47648ba4b75a3955101f213c5
+  data.tar.gz: e505e9861441cd5c176d2e68c553cc38a4110a1cb8ccb9b4ddb4fc11ae919221e4df14ffa98a75295ebe1bf1664a64968e53aa1ab3c659de1506a876339f299c

data/.aoj

@@ -0,0 +1,8 @@
+{
+  "username": "YOUR AOJ USERNAME",
+  "password": "YOUR AOJ PASSWORD",
+  "tw_consumer_key"        : "YOUR TWITTER CONSUMER KEY", 
+  "tw_consumer_secret"     : "YOUR TWITTER CONSUMER SECRET",
+  "tw_access_token"        : "YOUR TWITTER ACCESS TOKEN",
+  "tw_access_token_secret" : "YOUR TWITTER ACCESS TOKEN SECRET"
+}

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 aoj.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 TODO: Write your name
+
+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,90 @@
+# AOJ
+
+AOJ is a command-line tool for [AOJ(Aizu Online Judge)](http://judge.u-aizu.ac.jp/onlinejudge/). You can submit your source to AOJ and check the result of judgement without opening browser or copy & paste.
+
+This program is forked from aizuzia's [AOJToolkit](http://d.hatena.ne.jp/Tayama/20101207/1291727425). 
+
+## Installation
+
+Please make sure that Ruby (version 2.0 or later) is installed.
+
+You can install AOJ via RubyGems.org
+```
+$ gem install aoj
+```
+
+## Usage
+
+Problem ID and Language are auto-detected by filename. You should make your source file be like `[ProblemID].(cpp|rb|...)`.
+
+#### Submit simply
+
+```
+$ aoj submit 0001.cpp
+```
+
+#### Problem ID and Language can be manually specified
+
+```
+$ aoj submit mysource.cpp -p 0001 -l cpp11
+```
+
+#### Submit and Tweet result
+
+```
+$ aoj submit 0001.cpp -t
+```
+
+#### List available languages
+
+```
+$ aoj lang
+List available languages:
+  c, cpp, java, cpp11, csharp, d, ruby, python, python3, php, js
+
+Auto-detect extensions:
+  [ext]     [lang]
+  .c        c
+  .cpp      cpp
+  .cc       cpp
+  .C        cpp
+  .java     java
+  .rb       ruby
+  .cs       csharp
+  .d        d
+  .py       python
+  .php      php
+  .js       javascript
+```
+
+You can check all other commands and options by typing `aoj` and enter.
+
+## Configuration
+
+AOJ supports interactive configuration. Your account information once used will be stored at your home directory `~/.aojrc`.
+
+If you need to use an HTTP proxy to access the Internet, export the HTTP_PROXY or http_proxy environment variable: `export $HTTP_PROXY="http://user:password@proxy:8080"`.
+
+AOJ also support some extra configuration.
+
+#### Custom mapping from file extension to language
+
+To use custom mapping, please edit `~/.aojrc` directly.
+
+For example:
+
+```
+extname:
+  ".py": python3
+  ".cpp": cpp11
+```
+
+which means, `hoge.py` file will be submitted as Python3 and `hoge.cpp` is C++11.
+
+## Contributing
+
+1. Fork it ( http://github.com/na-o-sss/aoj/fork )
+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/aoj.gemspec

@@ -0,0 +1,28 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'aoj/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "aoj"
+  spec.version       = AOJ::VERSION
+  spec.authors       = ["na-o-ys"]
+  spec.email         = ["naoyoshi0511@gmail.com"]
+  spec.summary       = %q{aoj submitter}
+  spec.description   = %q{This is a submitter program which submits your source to AOJ(Aizu Online Judge) and retrieves result of judgement.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  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_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "rake", "~> 10.4"
+  spec.add_development_dependency "thor", "~> 0.19"
+  spec.add_development_dependency "activesupport", "~> 4.2"
+  spec.add_development_dependency "rspec", "~> 3.2"
+  spec.add_development_dependency "oauth", "~> 0.4"
+  spec.add_development_dependency "twitter", "~> 5.14"
+end

data/bin/aoj

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require "aoj"
+
+AOJ::CLI.start

data/lib/aoj.rb

@@ -0,0 +1,7 @@
+require "active_support"
+require "active_support/core_ext"
+require "active_support/dependencies"
+require "active_support/rescuable"
+ActiveSupport::Dependencies.autoload_paths << File.dirname(__FILE__)
+
+module AOJ; end

data/lib/aoj/api.rb

@@ -0,0 +1,150 @@
+require "rexml/document"
+
+module AOJ
+  module API
+    SUBMISSION_DATE_GAP_SEC = 30 # to identify judge result of a submission
+    RETRY_SEC = 30 # max retry sec fetching result
+    FETCH_STATUS_WAIT_SEC = 5
+
+    class << self
+
+      def problem_search(problem_id)
+        base_url = 'http://judge.u-aizu.ac.jp/onlinejudge/webservice/problem'
+        params = { id: problem_id }
+        response = AOJ::HTTP.get(
+          URI.parse("#{base_url}?#{params.to_query}")
+        )
+        doc = REXML::Document.new(response)
+
+        if doc.elements['problem/id'].nil?
+          raise AOJ::Error::APIError, "Problem id may be invalid (#{problem_id})"
+        end
+
+        AOJ::Problem.new.tap do |p|
+          v = ->(k) { doc.get_text(k).try(:value).try(:strip) }
+          p.id           = v['problem/id']
+          p.name         = v['problem/name']
+          p.available    = v['problem/available'] == '1'
+          p.time_limit   = v['problem/problemtimelimit'].to_i
+          p.memory_limit = v['problem/problemmemorylimit'].to_i
+        end
+      end
+
+      def status_log_search(problem, credential)
+        base_url = 'http://judge.u-aizu.ac.jp/onlinejudge/webservice/status_log'
+
+        statuses = []
+        start = 0, limit = 100
+        loop do
+          params = {
+            'user_id'    => credential.username,
+            'problem_id' => problem.id,
+            'start'      => start,
+            'limit'      => limit
+          }
+          # TODO: non-blocking
+          uri = URI.parse("#{base_url}?#{params.to_query}")
+          res =
+            REXML::Document.new(AOJ::HTTP.get(uri))
+              .get_elements('status_list/status')
+          statuses += res
+
+          break if res.size < limit
+          start += limit
+        end
+
+        statuses.map do |s|
+          v = -> (k) { s.get_text(k).try(:value).try(:strip) }
+          AOJ::Status.new.tap { |o|
+            o.run_id          = v['run_id'].to_i
+            o.user_id         = v['user_id']
+            o.problem_id      = v['problem_id'].to_i
+            o.submission_date = Time.at(v['submission_date'].to_i/1000)
+            o.status          = v['status']
+            o.language        = v['language']
+            o.cputime         = v['cputime']
+            o.memory          = v['memory']
+            o.code_size       = v['code_size']
+          }
+        end
+      end
+
+      def user(user_id)
+        url = "http://judge.u-aizu.ac.jp/onlinejudge/webservice/user"
+        params = { "id" => user_id }
+        uri = URI.parse("#{url}?#{params.to_query}")
+        doc = REXML::Document.new(AOJ::HTTP.get(uri))
+        solved_problems = doc.get_elements('user/solved_list/problem')
+        {
+          id:   doc.get_text('user/id').value.strip,
+          name: doc.get_text('user/name').value.strip,
+          solved_list: solved_problems.map { |item|
+            {
+              id: item.get_text('id').value.strip
+            }
+          }
+        }
+      end
+
+      def problem_list(volume)
+        url = "http://judge.u-aizu.ac.jp/onlinejudge/webservice/problem_list"
+        params = { "volume" => volume }
+        uri = URI.parse("#{url}?#{params.to_query}")
+        doc = REXML::Document.new(AOJ::HTTP.get(uri))
+        problems = doc.get_elements('problem_list/problem')
+        problems.map do |problem|
+          {
+            id:   problem.get_text('id').value.strip,
+            name: problem.get_text('name').value.strip
+          }
+        end
+      end
+
+      def submit(solution, credential)
+        uri = URI.parse('http://judge.u-aizu.ac.jp/onlinejudge/servlet/Submit')
+        params = {
+          'userID'     => credential.username,
+          'password'   => credential.password,
+          'problemNO'  => solution.problem.id,
+          'sourceCode' => solution.source,
+          'language'   => solution.language.submit_name
+        }
+
+        if AOJ::HTTP.post_form(uri, params).code != '200'
+          raise AOJ::Error::APIError, 'Failed to submit'
+        end
+        # TODO: when userid or password is wrong
+        solution
+      end
+
+      def judge_result(solution, submission_date, credential)
+        start_time = Time.now
+        loop do
+          if Time.now - start_time > RETRY_SEC
+            raise AOJ::Error::FetchResultError, 'Failed to fetch result'
+          end
+
+          sleep FETCH_STATUS_WAIT_SEC
+
+          hit = status_log_search(solution.problem, credential)
+            .select { |s|
+              (s.submission_date - submission_date).abs < SUBMISSION_DATE_GAP_SEC
+            }
+            .max_by(&:submission_date)
+
+          return hit if hit
+        end
+      end
+
+      def loginable?(credential)
+        uri = URI.parse("http://judge.u-aizu.ac.jp/onlinejudge/index.jsp")
+        params = {
+          'loginUserID'   => credential.username,
+          'loginPassword' => credential.password
+        }
+        res = AOJ::HTTP.post_form(uri, params)
+        !res.body.include?('Wrong User ID.') and !res.body.include?('Wrong Password.')
+      end
+    end
+  end
+end

data/lib/aoj/cli.rb

@@ -0,0 +1,99 @@
+require 'thor'
+
+module AOJ
+  class CLI < Thor
+    include Helper
+
+    option :twitter, type: :boolean, aliases: :t
+    option :problem, aliases: :p
+    option :lang, aliases: :l
+    desc "submit SOURCE_FILE", "Submit your source code"
+    def submit(file)
+      unless conf.has_credential?
+        input_credentials
+        puts
+      end
+
+      solution = Solution.new.tap do |s|
+        s.source   = read_file(file)
+        s.problem  = detect_problem(file, options[:problem])
+        s.language = detect_language(file, options[:lang])
+      end
+      print_solution_info solution
+      puts
+
+      puts "Submitting..."
+      API.submit(solution, Credential.get)
+      puts
+
+      puts "Fetching judge result..."
+      result = API.judge_result(solution, Time.now, Credential.get)
+      puts
+
+      print_result result
+
+      if options[:twitter]
+        unless conf.has_twitter_credential?
+          puts
+          twitter_auth
+        end
+        Twitter.instance.post(solution, result)
+      end
+    rescue AOJ::Error::LanguageDetectionError,
+           AOJ::Error::APIError,
+           AOJ::Error::FetchResultError,
+           AOJ::Error::FileOpenError => e
+      puts e.message
+    end
+
+    desc "omikuji", "今日の 1 問"
+    def omikuji
+      problem = AOJ::Problem.random_icpc(conf['username'])
+      title = "ID #{problem.id}"
+      width = title.size
+      line = "-" * (width + 4)
+      title_line = "| #{title} |"
+      lspace = " " * ((width + 1)/2)
+      rspace = " " * (width + 1 - (width + 1)/2)
+      body = [problem.name.split("")]
+        .transpose
+        .map { |s| "|#{lspace}#{s[0]}#{rspace}|" }
+
+      outer_width = problem.url.size
+      outer_lspace = " " * (outer_width/2 - line.size/2 - 2)
+      puts [line, title_line, line, body, line]
+        .flatten
+        .map { |s| outer_lspace + s }
+        .join("\n")
+      puts
+      puts problem.url
+      puts
+    rescue AOJ::Error::NoProblemLeftError
+      puts "No problem left. You are crazy!"
+    end
+
+    desc "setting", "Setup login credentials"
+    def setting
+      input_credentials
+    end
+
+    desc "twitter", "Auth twitter account"
+    def twitter
+      twitter_auth
+    end
+
+    desc "langs", "List available languages"
+    def langs
+      puts "List available languages:"
+      print "  "
+      puts Language.languages.map(&:key).join(", ")
+
+      puts
+      puts "Auto-detect extensions:"
+      puts "  " + "[ext]".ljust(10) + "[lang]"
+      Language.extnames.each do |k, v|
+        puts "  " + k.ljust(10) + v.to_s
+      end
+    end
+  end
+end

data/lib/aoj/cli/helper.rb

@@ -0,0 +1,85 @@
+require "io/console"
+
+module AOJ
+  module CLI::Helper
+    def detect_language(file, opt)
+      language =
+        if opt
+          AOJ::Language.find(opt.to_sym)
+        else
+          AOJ::Language.find_by_ext(File.extname(file))
+        end
+
+      unless language
+        raise AOJ::Error::LanguageDetectionError, "Failed to detect language, check `aoj langs`"
+      end
+      language
+    end
+
+    def detect_problem(file, opt)
+      problem_id = opt || File.basename(file, ".*")
+      API.problem_search problem_id
+    end
+
+    def read_file(file)
+      File.read(file)
+    rescue Errno::ENOENT => e
+      raise AOJ::Error::FileOpenError, e.message
+    end
+
+    def print_solution_info(solution)
+      puts "Language: #{solution.language.submit_name}"
+      puts "Problem:"
+      puts "  id:   #{solution.problem.id}"
+      puts "  name: #{solution.problem.name}"
+    end
+
+    def print_result(result)
+      puts "Judge result:"
+      puts "  Status:    #{result.status}"     if result.status
+      puts "  Code Size: #{result.code_size}"  if result.code_size
+      puts "  CPU Time:  #{result.cputime}"    if result.cputime
+      puts "  Memory:    #{result.memory}"     if result.memory
+      puts "  URL:       #{result.review_url}" if result.review_url
+    end
+
+    def input_credentials
+      credential = Credential.new
+
+      loop do
+        puts "Input AOJ username:"
+        credential.username = STDIN.gets.strip
+        puts "Input AOJ password:"
+        credential.password = STDIN.noecho(&:gets).strip
+        puts
+        break if credential.valid?
+        puts "Invalid username/password."
+      end
+
+      conf['username'] = credential.username
+      conf['password'] = credential.password
+      conf.save
+      puts "Your credential is saved at #{conf.rcfile_path}"
+    end
+
+    def twitter_auth
+      request_token = Twitter.instance.request_token
+      puts "Connect to twitter account. Please access to url."
+      puts request_token.authorize_url
+      puts "Input PIN:"
+      pin = STDIN.gets.strip
+      puts
+      access_token = request_token.get_access_token(oauth_verifier: pin)
+      conf['twitter'] = {
+        'access_token'        => access_token.token,
+        'access_token_secret' => access_token.secret
+      }
+      conf.save
+      puts "Your credential is saved at #{conf.rcfile_path}"
+    end
+
+    def conf
+      Conf.instance
+    end
+  end
+end

data/lib/aoj/conf.rb

@@ -0,0 +1,42 @@
+require "yaml"
+require "forwardable"
+
+module AOJ
+  class Conf
+    include Singleton
+    extend Forwardable
+    RCFILE = ".aojrc"
+
+    def_delegators :@data, :[], :[]=
+    attr_accessor :data
+
+    def initialize
+      @data = load_file
+    end
+
+    def rcfile_path
+      File.join(Dir.home, RCFILE)
+    end
+
+    def load_file
+      YAML.load_file rcfile_path
+    rescue Errno::ENOENT
+      {}
+    end
+
+    def save
+      open(rcfile_path, "w", 0600) do |file|
+        file.write @data.to_yaml
+      end
+    end
+
+    def has_credential?
+      self['username'] and self['password']
+    end
+
+    def has_twitter_credential?
+      (twitter = self['twitter']) and
+        twitter['access_token'] and twitter['access_token_secret']
+    end
+  end
+end

data/lib/aoj/credential.rb

@@ -0,0 +1,16 @@
+module AOJ
+  class Credential
+    attr_accessor :username, :password
+
+    def valid?
+      AOJ::API.loginable?(self)
+    end
+
+    def self.get
+      Credential.new.tap do |c|
+        c.username = AOJ::Conf.instance['username']
+        c.password = AOJ::Conf.instance['password']
+      end
+    end
+  end
+end

data/lib/aoj/error/api_error.rb

@@ -0,0 +1,5 @@
+module AOJ
+  module Error
+    class APIError < StandardError; end
+  end
+end

data/lib/aoj/error/fetch_result_error.rb

@@ -0,0 +1,5 @@
+module AOJ
+  module Error
+    class FetchResultError < StandardError; end
+  end
+end

data/lib/aoj/error/file_open_error.rb

@@ -0,0 +1,5 @@
+module AOJ
+  module Error
+    class FileOpenError < StandardError; end
+  end
+end

data/lib/aoj/error/invalid_problem_id_error.rb

@@ -0,0 +1,5 @@
+module AOJ
+  module Error
+    class InvalidProblemIdError < StandardError; end
+  end
+end

data/lib/aoj/error/language_detection_error.rb

@@ -0,0 +1,5 @@
+module AOJ
+  module Error
+    class LanguageDetectionError < StandardError; end
+  end
+end

data/lib/aoj/error/no_problem_left_error.rb

@@ -0,0 +1,5 @@
+module AOJ
+  module Error
+    class NoProblemLeftError < StandardError; end
+  end
+end

data/lib/aoj/http.rb

@@ -0,0 +1,35 @@
+require 'net/http'
+
+module AOJ
+
+  HTTP = -> do
+    # set proxy if specified
+    # [http://][user:pass@]host:port
+    env = ENV["HTTP_PROXY"] || ENV["http_proxy"]
+    break ::Net::HTTP unless env
+
+    info = parse_proxy_info(env)
+    ::Net::HTTP::Proxy(
+      info[:host],
+      info[:port],
+      info[:user],
+      info[:pass]
+    )
+  end.call
+
+  private
+  # TODO: test
+  def self.parse_proxy_info(str)
+    {}.tap do |i|
+      str.sub!(/http:\/\//, "")
+      str.sub!(/\/+$/, "")
+      if str.include?("@")
+        auth, str = str.split("@")
+        i[:user], i[:pass] = auth.split(":")
+      end
+      i[:host], port_str = str.split(":")
+      i[:port] = port_str.sub(/\//, "").to_i
+      i[:uri] = "http://" + str
+    end
+  end
+end

data/lib/aoj/language.rb

@@ -0,0 +1,57 @@
+module AOJ
+  class Language
+    attr_accessor :key, :submit_name
+
+    langs = { # TODO: yml config
+      c:       "C",
+      cpp:     "C++",
+      java:    "JAVA",
+      cpp11:   "C++11",
+      csharp:  "C#",
+      d:       "D",
+      ruby:    "Ruby",
+      python:  "Python",
+      python3: "Python3",
+      php:     "PHP",
+      js:      "JavaScript"
+    } 
+
+    @languages = langs.each.map { |key, submit_name|
+      Language.new.tap { |l|
+        l.key         = key
+        l.submit_name = submit_name
+      }
+    }.freeze
+
+    @extnames = { #TODO: yml config
+      ".c"    => :c,
+      ".cpp"  => :cpp,
+      ".cc"   => :cpp,
+      ".C"    => :cpp,
+      ".java" => :java,
+      ".rb"   => :ruby,
+      ".cs"   => :csharp,
+      ".d"    => :d,
+      ".py"   => :python,
+      ".php"  => :php,
+      ".js"   => :javascript
+    }
+    if custom_ext = AOJ::Conf.instance['extname']
+      custom_ext.transform_values!(&:to_sym)
+      @extnames.merge! custom_ext
+    end
+    @extnames.freeze
+
+    class << self
+      attr_reader :languages, :extnames
+
+      def find(key)
+        languages.find { |l| l.key == key }
+      end
+
+      def find_by_ext(ext)
+        languages.find { |l| l.key == extnames[ext] }
+      end
+    end
+  end
+end

data/lib/aoj/problem.rb

@@ -0,0 +1,30 @@
+module AOJ
+  class Problem
+    attr_accessor :id, :name, :available, :time_limit, :memory_limit
+
+    def url
+      base_url = "http://judge.u-aizu.ac.jp/onlinejudge/description.jsp"
+      params = { "id" => id }
+      base_url + "?" + params.to_query
+    end
+
+    ICPC_VOLUMES = [
+      '10', '11', '12', '13', '20', '21', '22', '23', '24', '25', '26'
+    ].freeze
+
+    class << self
+      def random_icpc(user_id)
+        volumes = ICPC_VOLUMES
+        solved_ids = AOJ::API.user(user_id)[:solved_list].map { |s| s[:id] }
+        volumes.shuffle.each do |volume|
+          problem_list = AOJ::API.problem_list(volume).map { |e| e[:id] } - solved_ids
+          unless problem_list.empty?
+            problem_id = problem_list.sample
+            return AOJ::API.problem_search(problem_id)
+          end
+        end
+        raise AOJ::Error::NoProblemLeftError
+      end
+    end
+  end
+end

data/lib/aoj/solution.rb

@@ -0,0 +1,5 @@
+module AOJ
+  class Solution
+    attr_accessor :problem, :source, :language
+  end
+end

data/lib/aoj/status.rb

@@ -0,0 +1,12 @@
+module AOJ
+  class Status
+    attr_accessor :run_id, :user_id, :problem_id, :submission_date, :status, :language, :cputime, :memory, :code_size
+
+    def review_url
+      nil unless run_id
+      base_url = "http://judge.u-aizu.ac.jp/onlinejudge/review.jsp"
+      params = { "rid" => run_id }
+      base_url + "?" + params.to_query
+    end
+  end
+end

data/lib/aoj/twitter.rb

@@ -0,0 +1,57 @@
+require "twitter"
+require "oauth"
+
+module AOJ
+  class Twitter
+    include Singleton
+
+    CONSUMER_KEY = "0T4d46LUm3jlf5xDPeRRQIUam"
+    CONSUMER_SECRET = "knA6biWCQ6oBpyVSEjHhFz3Mn8Z9UsJc7TD9h5DdZoRxBl0KBp"
+
+    def post(solution, status)
+      client.update create_text(solution, status)
+    end
+
+    def request_token
+      consumer = OAuth::Consumer.new(
+        CONSUMER_KEY,
+        CONSUMER_SECRET,
+        { site: "https://api.twitter.com" }
+      )
+      consumer.get_request_token
+    end
+
+    private
+    def create_text(solution, status)
+      text = <<-"EOS"
+#{status.status} #AOJ
+[#{solution.problem.id}:%s] %s
+SOURCE: %s
+LANG: #{solution.language.submit_name}
+      EOS
+
+      # Shortening problem title
+      url_len = 2 * client.configuration.short_url_length
+      title = solution.problem.name
+      wordcount = text.length - 2 * 3 + url_len
+      if wordcount > 140
+        title = title[0, 140 - wordcount - 2] + ".."
+      end
+
+      text % [title, solution.problem.url, status.review_url]
+    end
+
+    def client
+      @client ||= ::Twitter::REST::Client.new do |c|
+        c.consumer_key        = CONSUMER_KEY
+        c.consumer_secret     = CONSUMER_SECRET
+        c.access_token        = conf['twitter']['access_token']
+        c.access_token_secret = conf['twitter']['access_token_secret']
+      end
+    end
+
+    def conf
+      AOJ::Conf.instance
+    end
+  end
+end

data/lib/aoj/version.rb

@@ -0,0 +1,3 @@
+module AOJ 
+  VERSION = "0.0.3"
+end

data/spec/lib/aoj/api_spec.rb

@@ -0,0 +1,88 @@
+require "spec_helper"
+
+describe AOJ::API do
+  describe 'problem_search' do
+    it 'should return valid problem' do
+      v = AOJ::API.problem_search('0000').instance_values.symbolize_keys
+      expect(v).to eq({
+        id: '0000',
+        name: 'QQ',
+        available: true,
+        time_limit: 1,
+        memory_limit: 65536
+      })
+    end
+
+    it 'should raise error' do
+      expect{AOJ::API.problem_search('114514')}.to raise_error(AOJ::Error::APIError)
+    end
+  end
+
+  describe 'status_log_search' do
+    it 'should return an array of AOJ::Status' do
+      problem = AOJ::Problem.new.tap { |p| p.id = '0000' }
+      credential = AOJ::Credential.new.tap { |c| c.username = 'na_o_sss' }
+      statuses = AOJ::API.status_log_search(problem, credential)
+      expect(statuses[0]).to be_a AOJ::Status
+    end
+  end
+
+  describe 'user' do
+    it 'should return valid hash' do
+      h = AOJ::API.user('shioshiota')
+      expect(h[:solved_list].size).to be > 0
+      expect(h[:solved_list][0][:id]).to be_a String
+    end
+  end
+
+  describe 'problem_list' do
+    it 'should return valid array' do
+      a = AOJ::API.problem_list('10')
+      expect(a.size).to be > 0
+      expect(a[0][:id]).to be_a String
+    end
+  end
+
+  describe 'submit'
+
+  describe 'judge_result' do
+    context 'target status exists' do
+      it 'should return valid status' do
+        module AOJ::API
+          def self.status_log_search(problem, credential)
+            [
+              AOJ::Status.new.tap { |s| s.submission_date = Time.now.ago(15) }
+            ]
+          end
+        end
+        status = AOJ::API.judge_result(AOJ::Solution.new, Time.now, nil)
+        expect(status).to be_a AOJ::Status
+      end
+    end
+
+    context 'target status does not exist' do
+      it 'should raise error' do
+        module AOJ::API
+          remove_const :RETRY_SEC
+          RETRY_SEC = 1
+          def self.status_log_search(problem, credential)
+            [
+              AOJ::Status.new.tap { |s| s.submission_date = Time.now.ago(45) }
+            ]
+          end
+        end
+        expect{AOJ::API.judge_result(AOJ::Solution.new, Time.now, nil)}.to raise_error(AOJ::Error::FetchResultError)
+      end
+    end
+  end
+
+  describe 'loginable?' do
+    it 'should return false with invalid credential' do
+      credential = AOJ::Credential.new.tap { |c|
+        c.username = 'dummy_user'
+        c.password = '1111'
+      }
+      expect(AOJ::API.loginable?(credential)).to be false
+    end
+  end
+end

data/spec/lib/aoj/conf_spec.rb

@@ -0,0 +1,35 @@
+require "spec_helper"
+
+describe AOJ::Conf do
+  before { @conf = AOJ::Conf.instance }
+
+  describe 'has_credential?' do
+    it 'should return true' do
+      @conf['username'] = 'hoge'
+      @conf['password'] = 'fuga'
+      expect(@conf.has_credential?).to be_truthy
+    end
+
+    it 'should return false' do
+      @conf['username'] = 'hoge'
+      @conf.data.delete 'password'
+      expect(@conf.has_credential?).to be_falsy
+    end
+  end
+
+  describe 'has_twitter_credential?' do
+    it 'should return true' do
+      @conf['twitter'] = {
+        'access_token'        => 'hoge',
+        'access_token_secret' => 'fuga'
+      }
+      expect(@conf.has_twitter_credential?).to be_truthy
+    end
+
+    it 'should return false' do
+      @conf['twitter'] = { 'access_token' => 'hoge' }
+      @conf['twitter'].delete 'access_token_secret'
+      expect(@conf.has_twitter_credential?).to be_falsy
+    end
+  end
+end

data/spec/lib/aoj/language_spec.rb

@@ -0,0 +1,19 @@
+require "spec_helper"
+
+describe AOJ::Language do
+  describe 'find' do
+    it 'should return valid language' do
+      l = AOJ::Language.find(:cpp)
+      expect(l.key).to eq(:cpp)
+      expect(l.submit_name).to eq("C++")
+    end
+  end
+
+  describe 'find_by_ext' do
+    it 'should return valid language' do
+      l = AOJ::Language.find_by_ext(".rb")
+      expect(l.key).to eq(:ruby)
+      expect(l.submit_name).to eq("Ruby")
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,5 @@
+require "rubygems"
+require "bundler/setup"
+require "rspec"
+require "active_support"
+require File.join(File.dirname(__FILE__), "..", "lib/aoj.rb")

metadata

@@ -0,0 +1,179 @@
+--- !ruby/object:Gem::Specification
+name: aoj
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- na-o-ys
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-02-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.4'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.4'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.19'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.19'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+- !ruby/object:Gem::Dependency
+  name: oauth
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.4'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.4'
+- !ruby/object:Gem::Dependency
+  name: twitter
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.14'
+description: This is a submitter program which submits your source to AOJ(Aizu Online
+  Judge) and retrieves result of judgement.
+email:
+- naoyoshi0511@gmail.com
+executables:
+- aoj
+extensions: []
+extra_rdoc_files: []
+files:
+- ".aoj"
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- aoj.gemspec
+- bin/aoj
+- lib/aoj.rb
+- lib/aoj/api.rb
+- lib/aoj/cli.rb
+- lib/aoj/cli/helper.rb
+- lib/aoj/conf.rb
+- lib/aoj/credential.rb
+- lib/aoj/error/api_error.rb
+- lib/aoj/error/fetch_result_error.rb
+- lib/aoj/error/file_open_error.rb
+- lib/aoj/error/invalid_problem_id_error.rb
+- lib/aoj/error/language_detection_error.rb
+- lib/aoj/error/no_problem_left_error.rb
+- lib/aoj/http.rb
+- lib/aoj/language.rb
+- lib/aoj/problem.rb
+- lib/aoj/solution.rb
+- lib/aoj/status.rb
+- lib/aoj/twitter.rb
+- lib/aoj/version.rb
+- spec/lib/aoj/api_spec.rb
+- spec/lib/aoj/conf_spec.rb
+- spec/lib/aoj/language_spec.rb
+- spec/spec_helper.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: aoj submitter
+test_files:
+- spec/lib/aoj/api_spec.rb
+- spec/lib/aoj/conf_spec.rb
+- spec/lib/aoj/language_spec.rb
+- spec/spec_helper.rb