uva-tools 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 600ae5264d13d0c083a5ad143e01f6cd5cd3b180
+  data.tar.gz: c13486119816dc8ad210072915efac90142d6256
+SHA512:
+  metadata.gz: 5d565deb6a25120c1f19a66d640f7e060055b28ba31a6abc1ee01457ff0c5da25bf08b9678e51f309ca0c9be57fb05a7329f296e3b125638b4853711c63c9e86
+  data.tar.gz: 55be55d88c394e2570361d3508dcc506a626308f55fdc0136968ca98beed12bc8a2efa1c52cda3d9af0055b876a84a299194eb627eae2f9468d4d33657dd1d20

data/.gitignore

@@ -0,0 +1,3 @@
+doc
+pkg
+*.gem

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Henning Koch
+
+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 @@
+# UVa Tools

data/lib/uva-tools/problem.rb

@@ -0,0 +1,124 @@
+=begin
+ 0: Problem ID
+ 1: Problem Number
+ 2: Problem Title
+ 3: Number of Distinct Accepted User (DACU)
+ 4: Best Runtime of an Accepted Submission
+ 5: Best Memory used of an Accepted Submission
+ 6: Number of No Verdict Given (can be ignored)
+ 7: Number of Submission Error
+ 8: Number of Can't be Judged
+ 9: Number of In Queue
+10: Number of Compilation Error
+11: Number of Restricted Function
+12: Number of Runtime Error
+13: Number of Output Limit Exceeded
+14: Number of Time Limit Exceeded
+15: Number of Memory Limit Exceeded
+16: Number of Wrong Answer
+17: Number of Presentation Error
+18: Number of Accepted
+19: Problem Run-Time Limit (milliseconds)
+20: Problem Status (0 = unavailable, 1 = normal, 2 = special judge)
+=end
+
+module UVaTools
+  class Problem
+    def initialize(array)
+      @info = array
+    end
+
+    def id
+      @info[0]
+    end
+
+    def number
+      @info[1]
+    end
+
+    def title
+      @info[2]
+    end
+
+    def info
+      {
+        id: id,
+        number: number,
+        title: title,
+        run_time: @info[19]
+      }
+    end
+
+    def submitted
+      @info[6..18].inject(:+)
+    end
+
+    def accepted
+      @info[18]
+    end
+
+    def dacu
+      @info[3]
+    end
+
+    def submission_info
+      {
+        submitted: submitted,
+        accepted: accepted,
+        dacu: dacu
+      }
+    end
+
+    def download
+      unless downloaded?
+        FileUtils::mkdir_p dir_name
+        Dir.chdir(dir_name) do
+          uri = URI("https://uva.onlinejudge.org/external/#{number/100}/#{number}.html")
+          source = Net::HTTP.get(uri)
+          if source.include? "URL=p#{number}.pdf"
+            File.open("#{number.to_s}.pdf", 'w') do |pdf|
+              pdf.write open("https://uva.onlinejudge.org/external/#{number/100}/p#{number}.pdf").read
+            end
+          else
+            File.open("#{number.to_s}.html", 'w') { |f| f.write source }
+
+            source.scan(/#{number}img\d{1,3}\.[a-zA-Z]+/).uniq.each do |picture|
+              File.open(picture, 'wb') do |pic_file|
+                pic_file.write open("https://uva.onlinejudge.org/external/#{number/100}/#{picture}").read
+              end
+            end
+          end
+        end
+        puts "downloaded: #{number}"
+        true
+      else
+        puts "#{number} was already downloaded"
+        false
+      end
+    end
+
+    def remove
+      if downloaded?
+        Dir.chdir(dir_name) do
+          Dir.entries('.').each do |file_name|
+            if file_name =~ /^#{number}.*/ && !File.directory?(file_name)
+              File.delete(file_name)
+            end
+          end
+        end
+        puts "removed #{number}"
+      else
+        puts "#{number} wasn't downloaded"
+      end
+    end
+
+    def downloaded?
+      File.directory?(dir_name) && (File.exists?("#{dir_name}/#{number}.html") || File.exists?("#{dir_name}/#{number}.pdf"))
+    end
+
+    private
+    def dir_name
+      "#{ENV['HOME']}/uva-tools/#{number/100}"
+    end
+  end
+end

data/lib/uva-tools/user.rb

@@ -0,0 +1,80 @@
+module UVaTools
+  class User
+    def initialize(username)
+      @username = username
+    end
+
+    def solved
+      solved_hash.values
+    end
+
+    def solved_by_number
+      solved_hash
+    end
+
+    def unsolved
+      @unsolved ||= begin
+        prob_by_id = UVaTools.problems_by_id
+        unsolved_pids.map do |pid|
+          prob_by_id[pid]
+        end.sort{ |a, b| b.dacu <=> a.dacu }
+      end
+    end
+
+    def has_solved?(prob_nums)
+      res = Array(prob_nums).map do |prob_num|
+        solved_by_number.has_key? prob_num
+      end
+      res.length < 2 ? res[0] : res
+    end
+
+    def reload
+      @solved_pids = nil
+      @solved = nil
+      @unsolved = nil
+      self
+    end
+
+    private
+    def uid
+      @uid ||= begin
+        uri = URI("http://uhunt.felix-halim.net/api/uname2uid/#{@username}")
+        Net::HTTP.get(uri)
+      end
+    end
+
+    def solved_pids
+      @solved_pids ||= get_solved_pids
+    end
+
+    def solved_hash
+      @solved ||= begin
+        prob_by_id = UVaTools.problems_by_id
+        h = {}
+        solved_pids.each do |pid|
+          problem = prob_by_id[pid]
+          h[problem.number] = problem
+        end
+        h
+      end
+    end
+
+    def reload_solved_pids
+      @solved_pids = get_solved_pids
+    end
+
+    def get_solved_pids
+      uri = URI("http://uhunt.felix-halim.net/api/solved-bits/#{uid}")
+      res = JSON.parse(Net::HTTP.get(uri))[0]['solved']
+      res.each_with_index.map do |num, j|
+        num.to_s(2).reverse.scan(/./).each_with_index.map do |c, i|
+          j*32 + i if c == '1'
+        end.compact
+      end.flatten.sort
+    end
+
+    def unsolved_pids
+      UVaTools.problems.map(&:id) - solved_pids
+    end
+  end
+end

data/lib/uva-tools/uva-tools.rb

@@ -0,0 +1,108 @@
+module UVaTools
+  class << self
+    def problems
+      problem_hash.values
+    end
+
+    def problems_by_number
+      problem_hash
+    end
+
+    def problems_by_id
+      h = {}
+      problem_hash.values.each{ |p| h[p.id] = p }
+      h
+    end
+
+    def download_multiple(prob_nums, worker_count = 4)
+      to_download = Array(prob_nums).map do |prob_num|
+        problems_by_number[prob_num]
+      end
+
+      if to_download.length > 0
+        worker_count = [worker_count, to_download.length].min
+        workers = []
+
+        worker_count.times do
+          workers << worker(to_download)
+        end
+
+        reads = workers.map{|worker| worker[:read]}
+        writes = workers.map{|worker| worker[:write]}
+
+        index = 0
+        finished = 0
+
+        loop do
+          break if finished >= to_download.size
+
+          ready = IO.select(reads, writes)
+
+          ready[0].each do |readable|
+            data = Marshal.load(readable)
+            # assets.merge! data["assets"]
+            # files.merge! data["files"]
+            # paths_with_errors.merge! data["errors"]
+            finished += 1
+          end
+
+          ready[1].each do |write|
+            break if index >= to_download.size
+
+            Marshal.dump(index, write)
+            index += 1
+          end
+        end
+
+        workers.each do |worker|
+          worker[:read].close
+          worker[:write].close
+        end
+
+        workers.each do |worker|
+          Process.wait worker[:pid]
+        end
+      end
+
+      nil
+    end
+
+    private
+    def worker(problems)
+      child_read, parent_write = IO.pipe
+      parent_read, child_write = IO.pipe
+
+      pid = fork do
+        begin
+          parent_write.close
+          parent_read.close
+
+          while !child_read.eof?
+            problem = problems[Marshal.load(child_read)]
+            problem.download
+            Marshal.dump(problem.number, child_write)
+          end
+        ensure
+          child_read.close
+          child_write.close
+        end
+      end
+
+      child_read.close
+      child_write.close
+
+      {:read => parent_read, :write => parent_write, :pid => pid}
+    end
+
+    def problem_hash
+      @@problems ||= begin
+        uri = URI("http://uhunt.felix-halim.net/api/p")
+        res = JSON.parse(Net::HTTP.get(uri))
+
+        h = {}
+        res.each{ |row| h[row[1]] = UVaTools::Problem.new(row) }
+        h
+      end
+    end
+  end
+end

data/lib/uva-tools/version.rb

@@ -0,0 +1,3 @@
+module UVaTools
+  VERSION = '0.0.1'
+end

data/lib/uva-tools.rb

@@ -0,0 +1,7 @@
+require 'net/http'
+require 'open-uri'
+require 'json'
+require 'fileutils'
+require 'uva-tools/uva-tools'
+require 'uva-tools/user'
+require 'uva-tools/problem'

data/uva-tools.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+$: << File.expand_path('../lib', __FILE__)
+require 'uva-tools/version'
+
+Gem::Specification.new do |gem|
+  gem.authors       = "Dave Allie"
+  gem.email         = "dave@daveallie.com"
+  gem.description   = "Set of UVa Online tools"
+  gem.summary       = gem.description
+  gem.homepage      = "https://github.com/daveallie/uva-tools"
+  gem.license       = 'MIT'
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "uva-tools"
+  gem.require_paths = ["lib"]
+  gem.version       = UVaTools::VERSION
+end

metadata

@@ -0,0 +1,52 @@
+--- !ruby/object:Gem::Specification
+name: uva-tools
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Dave Allie
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-10-05 00:00:00.000000000 Z
+dependencies: []
+description: Set of UVa Online tools
+email: dave@daveallie.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- MIT-LICENSE
+- README.md
+- lib/uva-tools.rb
+- lib/uva-tools/problem.rb
+- lib/uva-tools/user.rb
+- lib/uva-tools/uva-tools.rb
+- lib/uva-tools/version.rb
+- uva-tools.gemspec
+homepage: https://github.com/daveallie/uva-tools
+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.4.5.1
+signing_key: 
+specification_version: 4
+summary: Set of UVa Online tools
+test_files: []