poj_org 0.2.1 → 0.3.0

data/lib/poj_org.rb

@@ -1,17 +1,19 @@
 require 'net/http'
+require 'open-uri'
+require 'cgi'
 
 # This class provides authetication methods
 class PojOrg
-  # Stored user name of poj.org
+  # Stored user name
   @@user = nil
 
-  # Stored password of poj.org
+  # Stored password
   @@password = nil
 
   # Status of last authentication
   @@authenticated = false
 
-  # Autheticate and store user name and password of poj.org
+  # Autheticate and store user name and password
   #
   # @example
   #   >> PojOrg::authenticate('poj_library', 'secret')
@@ -19,10 +21,10 @@ class PojOrg
   #   >> PojOrg::authenticate('poj_library', 'wrong_password')
   #   => false
   #
-  # @param [String] user      user name of poj.org
-  # @param [String] password  password of poj.org
+  # @param  [String]  user      user name of poj.org
+  # @param  [String]  password  password of poj.org
   #
-  # @return [Boolean] whether authenticated or not
+  # @return [Boolean]           whether authenticated or not
   def self.authenticate(user, password)
     if login(user, password)
       @@user = user
@@ -33,6 +35,13 @@ class PojOrg
     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
@@ -40,6 +49,63 @@ class PojOrg
     @@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)
@@ -51,6 +117,3 @@ class PojOrg
     login.body['failed'] ? false : login['set-cookie']
   end
 end
-
-require 'poj_org/problem'
-require 'poj_org/code'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: poj_org
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-23 00:00:00.000000000 Z
+date: 2013-01-25 00:00:00.000000000 Z
 dependencies: []
 description: Fetch details of problems and codes from poj.org
 email: pinepara@gmail.com
@@ -17,9 +17,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/poj_org.rb
-- lib/poj_org/problem.rb
-- lib/poj_org/code.rb
+- ./lib/poj_org.rb
 homepage: https://github.com/pinepara/poj_org
 licenses: []
 post_install_message: 
@@ -45,4 +43,3 @@ signing_key:
 specification_version: 3
 summary: API for poj.org
 test_files: []
-has_rdoc: 

data/lib/poj_org/code.rb

@@ -1,56 +0,0 @@
-require 'open-uri'
-require 'cgi'
-
-# This class handles codes on poj.org
-class PojOrg::Code
-  # Get the url of code via specified ID on poj.org
-  # 
-  # @param [Integer] id  ID of code
-  #
-  # @return [String] Url of code
-  def self.url_for(id)
-    "http://poj.org/showsource?solution_id=#{id}"
-  end
-
-  # @return [Integer] ID of code
-  attr_accessor :id
-  
-  # @return [Integer] Author's user name
-  attr_accessor :user
-
-  # @return [Integer] Time costed in ms
-  attr_accessor :time_in_ms
-
-  # @return [Integer] Memory costed in KB
-  attr_accessor :memory_in_kb
-
-  # @return [String] Programming Language in which code is writen
-  attr_accessor :language
-
-  # @return [String] Source code
-  attr_accessor :content
-
-  # Get a code from poj.org via ID.
-  # Need to be autheticated.
-  #
-  # @param [Integer] id  ID of code
-  def initialize(id)
-    self.id = id
-    if PojOrg::authenticated?
-      cookie = PojOrg::login
-      html = open(url, 'Cookie' => cookie).read
-      self.user         = (/<td><b>User:.+?<a.+?>(.+?)<\/a><\/td>/).match(html)[1]
-      self.time_in_ms   = (/<td><b>Time:.+?(\d+)MS<\/td>/).match(html)[1].to_i
-      self.memory_in_kb = (/<td><b>Memory:.+?(\d+)K<\/td>/).match(html)[1].to_i
-      self.language     = (/<td><b>Language:<\/b>\s*(.+?)<\/td>/).match(html)[1]
-      self.content      = CGI.unescapeHTML((/<pre.+?>(.+)<\/pre>/m).match(html)[1])
-    end
-  end
-
-  # Get the url of code on poj.org
-  #
-  # @return [String] Url of code
-  def url
-    PojOrg::Code.url_for id
-  end
-end

data/lib/poj_org/problem.rb

@@ -1,41 +0,0 @@
-# This class handles problems on poj.org
-class PojOrg::Problem
-  # Get the url of problem via specified ID on poj.org
-  # 
-  # @param [Integer] id  ID of problem
-  #
-  # @return [String] Url of problem
-  def self.url_for(id)
-    "http://poj.org/problem?id=#{id}"
-  end
-
-  # @return [Integer] ID of problem
-  attr_accessor :id
-
-  # @return [String] Title
-  attr_accessor :title
-  
-  # @return [Integer] Time limit in ms
-  attr_accessor :time_limit_in_ms
-
-  # @return [Integer] Memory limit in KB
-  attr_accessor :memory_limit_in_kb
-
-  # Get a problem for poj.org via ID
-  # 
-  # @param [Integer] id  ID of problem
-  def initialize(id)
-    self.id = id
-    html = open(url).read
-    self.title              = html[/(?<=<div class="ptt" lang="en-US">).+?(?=<\/div>)/]
-    self.time_limit_in_ms   = html[/(?<=<b>Time Limit:<\/b>)\s*\d+/].to_i
-    self.memory_limit_in_kb = html[/(?<=<b>Memory Limit:<\/b>)\s*\d+/].to_i
-  end
-
-  # Get the url of problem on poj.org
-  #
-  # @return [String] Url of problem
-  def url
-    PojOrg::Problem.url_for id
-  end
-end