poj_org 0.3.0 → 0.4.4

data/lib/poj_org/code.rb

@@ -0,0 +1,47 @@
+require 'cgi'
+require 'open-uri'
+require 'nokogiri'
+require 'poj_org/errors'
+require 'poj_org/user'
+
+module PojOrg
+  class Code
+    attr_reader :id, :username, :problem_id, :result, :memory_in_kb, :time_in_ms, :language, :length_in_byte, :submitted_at, :content
+
+    def initialize(id)
+      @id = id
+      fetch_details
+    end
+
+    def url
+      "http://poj.org/showsource?solution_id=#{id}"
+    end
+
+    def fetch_content(user)
+      user.login do |cookie|
+        html = Nokogiri::HTML(open(url, 'Cookie' => cookie))
+        if html.xpath('/html/head/title').text == 'Error'
+          raise PojOrg::Errors::CodeNotFound if html.xpath('/html/body/ul/li').text.include?('No such solution')
+          raise PojOrg::Errors::AccessDenied if html.xpath('/html/body/ul/li').text.include?('Source request declined')
+          raise PojOrg::Errors::Error
+        end
+        @content = html.xpath('/html/body/pre').text
+      end
+    end
+
+    private
+
+    def fetch_details
+      html = Nokogiri::HTML(open("http://poj.org/status?top=#{@id+1}"))
+      raise PojOrg::Errors::CodeNotFound if html.xpath('/html/body/table[2]/tr[2]/td[1]').text.to_i != @id
+      @username       = html.xpath('/html/body/table[2]/tr[2]/td[2]').text
+      @problem_id     = html.xpath('/html/body/table[2]/tr[2]/td[3]').text.to_i
+      @result         = html.xpath('/html/body/table[2]/tr[2]/td[4]').text
+      @memory_in_kb   = html.xpath('/html/body/table[2]/tr[2]/td[5]').text.to_i
+      @time_in_ms     = html.xpath('/html/body/table[2]/tr[2]/td[6]').text.to_i
+      @language       = html.xpath('/html/body/table[2]/tr[2]/td[7]').text
+      @length_in_byte = html.xpath('/html/body/table[2]/tr[2]/td[8]').text.to_i
+      @submitted_at   = html.xpath('/html/body/table[2]/tr[2]/td[9]').text
+    end
+  end
+end

data/lib/poj_org/errors.rb

@@ -0,0 +1,9 @@
+module PojOrg
+  module Errors
+    class Error < StandardError; end
+    class Unauthorized < StandardError; end
+    class ProblemNotFound < StandardError; end
+    class CodeNotFound < StandardError; end
+    class AccessDenied < StandardError; end
+  end
+end

data/lib/poj_org/problem.rb

@@ -0,0 +1,22 @@
+require 'open-uri'
+require 'nokogiri'
+require 'poj_org/errors'
+
+module PojOrg
+  class Problem
+    attr_reader :id, :title, :time_limit_in_ms, :memory_limit_in_kb
+
+    def initialize(id)
+      @id = id
+      html = Nokogiri::HTML(open(url))
+      raise PojOrg::Errors::ProblemNotFound if html.xpath('/html/head/title').text == 'Error'
+      @title              = html.xpath('/html/body/table[2]/tr/td/div[2]').text
+      @time_limit_in_ms   = html.xpath('/html/body/table[2]/tr/td/div[3]/table/tr[1]/td[1]').text[/\d+/].to_i
+      @memory_limit_in_kb = html.xpath('/html/body/table[2]/tr/td/div[3]/table/tr[1]/td[3]').text[/\d+/].to_i
+    end
+
+    def url
+      "http://poj.org/problem?id=#{id}"
+    end
+  end
+end

data/lib/poj_org/user.rb

@@ -0,0 +1,70 @@
+require 'net/http'
+require 'open-uri'
+require 'poj_org/errors'
+
+module PojOrg
+  class User
+    attr_reader :username
+
+    def initialize(username, password)
+      @username = username
+      @password  = password
+    end
+
+    def authenticated?
+      login
+      true
+    rescue
+      false
+    end
+
+    def login
+      response = Net::HTTP.post_form(
+        URI('http://poj.org/login'),
+        'user_id1' => username,
+        'password1' => @password
+      )
+      raise PojOrg::Errors::Unauthorized if response.body['failed']
+      yield response['set-cookie'] if block_given?
+      open('http://poj.org/login?action=logout')
+    end
+
+    def code(id)
+      code = Code.new(id)
+      code.fetch_content(self)
+      code
+    end
+
+    def code_ids(options = {})
+      html = Nokogiri::HTML(open(status_url(options)))
+      ids = []
+      2.upto(Float::INFINITY) do |row|
+        element = html.xpath("/html/body/table[2]/tr[#{row}]/td[1]").text
+        break if element.empty?
+        ids << element.to_i
+      end
+      ids += code_ids(options.merge(top: ids.last)) unless ids.empty?
+      ids
+    end
+
+    def codes(options = {})
+      code_ids.map do |id|
+        code(id)
+      end
+    end
+
+    def url
+      "http://poj.org/userstatus?user_id=#{@username}"
+    end
+
+    private
+
+    def status_url(options)
+      options.merge!(user_id: @username)
+      params = options.map { |key, value|
+        "&#{key}=#{value}"
+      }.join('&')
+      "http://poj.org/status?#{params}"
+    end
+  end
+end

data/lib/poj_org.rb

@@ -1,119 +1,4 @@
-require 'net/http'
-require 'open-uri'
-require 'cgi'
-
-# This class provides authetication methods
-class PojOrg
-  # Stored user name
-  @@user = nil
-
-  # Stored password
-  @@password = nil
-
-  # Status of last authentication
-  @@authenticated = false
-
-  # Autheticate and store user name and password
-  #
-  # @example
-  #   >> PojOrg::authenticate('poj_library', 'secret')
-  #   => true
-  #   >> PojOrg::authenticate('poj_library', 'wrong_password')
-  #   => false
-  #
-  # @param  [String]  user      user name of poj.org
-  # @param  [String]  password  password of poj.org
-  #
-  # @return [Boolean]           whether authenticated or not
-  def self.authenticate(user, password)
-    if login(user, password)
-      @@user = user
-      @@password = password
-      @@authenticated = true
-    else
-      @@authenticated = false
-    end
-  end
-
-  # Unauthenticate and erase user name and password
-  def self.unauthenticate
-    @@user = nil
-    @@password = nil
-    @@authenticated = false
-  end
-
-  # Status of last authentication
-  #
-  # @return [Boolean] Whether the last authentication success or not
-  def self.authenticated?
-    @@authenticated
-  end
-
-  # Url of problem
-  # 
-  # @param  [Integer] ID of problem
-  #
-  # @return [String]  Url of problem
-  def self.problem_url(id)
-    "http://poj.org/problem?id=#{id}"
-  end
-
-  # Details of problem
-  # 
-  # @param [Integer] ID of problem
-  #
-  # @return [Hash]   Details of problem or Nil if any error occured
-  def self.problem(id)
-    html = open(problem_url(id)).read
-    return nil if html.include? '<title>Error</title>'
-    {
-      id:                 id,
-      url:                problem_url(id),
-      title:              html[/(?<=<div class="ptt" lang="en-US">).+?(?=<\/div>)/],
-      time_limit_in_ms:   html[/(?<=<b>Time Limit:<\/b>)\s*\d+/].to_i,
-      memory_limit_in_kb: html[/(?<=<b>Memory Limit:<\/b>)\s*\d+/].to_i
-    }
-  end
-
-  # Url of code
-  # 
-  # @param [Integer] ID of code
-  #
-  # @return [String] Url of code
-  def self.code_url(id)
-    "http://poj.org/showsource?solution_id=#{id}"
-  end
-
-  # Details of code
-  # Need to be autheticated
-  #
-  # @param  [Integer] ID of code
-  #
-  # @return [Hash]    Details of code or Nil if any error occured
-  def self.code(id)
-    return nil unless authenticated?
-    html = open(code_url(id), 'Cookie' => login).read
-    return nil if html.include? '<title>Error</title>'
-    {
-      id:           id,
-      url:          code_url(id),
-      problem_id:   (/<td><b>Problem:.+?<a.+?>(.+?)<\/a><\/td>/).match(html)[1].to_i,
-      user:         (/<td><b>User:.+?<a.+?>(.+?)<\/a><\/td>/).match(html)[1],
-      time_in_ms:   (/<td><b>Time:.+?(\d+)MS<\/td>/).match(html)[1].to_i,
-      memory_in_kb: (/<td><b>Memory:.+?(\d+)K<\/td>/).match(html)[1].to_i,
-      language:     (/<td><b>Language:<\/b>\s*(.+?)<\/td>/).match(html)[1],
-      content:      CGI.unescapeHTML((/<pre.+?>(.+)<\/pre>/m).match(html)[1])
-    }
-  end
-
-  private
-
-  def self.login(user = nil, password = nil)
-    login = Net::HTTP.post_form(
-      URI('http://poj.org/login'),
-      'user_id1' => user || @@user,
-      'password1' => password || @@password
-    )
-    login.body['failed'] ? false : login['set-cookie']
-  end
-end
+require 'poj_org/errors'
+require 'poj_org/problem'
+require 'poj_org/user'
+require 'poj_org/code'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: poj_org
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,15 +9,35 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-25 00:00:00.000000000 Z
-dependencies: []
-description: Fetch details of problems and codes from poj.org
+date: 2013-03-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Fetch details of problems, codes and users from poj.org
 email: pinepara@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
 - ./lib/poj_org.rb
+- ./lib/poj_org/problem.rb
+- ./lib/poj_org/errors.rb
+- ./lib/poj_org/code.rb
+- ./lib/poj_org/user.rb
 homepage: https://github.com/pinepara/poj_org
 licenses: []
 post_install_message: