boj-solvedac 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b00f463812297cb9cdb55639a511e4bc309f305836fd489d639c089ebeefca66
-  data.tar.gz: 788ccc75ecab48a649d22db13e4fe86d16774aafec56d344a90c189f03602ff2
+  metadata.gz: 81197416100863f9824b775ede3d0dc02f9ce0272f079a162aecbfe3bcca4bb7
+  data.tar.gz: 952823c2c47e8ff96e9c0bbab2d16ad1f56deb68443232c6551350d15d235d87
 SHA512:
-  metadata.gz: cecc8e9cebc4aa1eadccceeea49e78202cbb94662cbcd8ea40e927a88c8cdb037503e1234c50cba5451605eefacb8110d75f74cc6dc5dac5a68321c22317f650
-  data.tar.gz: 976b1be13f808d0addc5cc57be1ed05802b8b3eebb4ae8c3f154ba2b9c4fb3b1c8f18625b9c37c6da00bfabc925f64976be0bdf2c96538c44fbb20ad694e6390
+  metadata.gz: 89a0fb10502412bae5674189a08968ff3d76c5a469c1c21924bc2bdaa97ea5376856129b91c36c87ae0a331a30b97f077acc08051541c9d9751d67a001f1355d
+  data.tar.gz: 1d8f03d833a141f10d224ecbcb772cd8ea6b3de3e7483e73e582cf9138d08e1eddfd7c5fd1bf87ba425eaf156ea4c5ac77cd2c421c9c1d61627f9d8308ea9afa

data/lib/boj-solvedac.rb

@@ -1,4 +1,5 @@
 require 'data/data.rb'
+require 'colorize/colorize.rb'
 
 module BOJ
   class BOJSolvedAC < BOJData
@@ -44,15 +45,15 @@ module BOJ
     def command_stat(level)
       @prev = ''
       if level == "all"
-        puts "%-10s %8s %8s %8s" % ["Level", "Unsolved", "Solved", "Total"]
+        puts "%-12s%8s%8s%8s" % ["Level", "Unsolved", "Solved", "Total"]
         @stats.each_pair do |k, v|
           stat = @levels[v.to_sym]
-          puts "%-10s %8s %8s %8s" % [k, stat[:unsolved], stat[:solved], stat[:total]]
+          puts "%-17s%10s%8s%8s" % [color(k), stat[:unsolved], stat[:solved], stat[:total]]
         end
       elsif %w(bronze silver gold platinum diamond ruby).include? level
         5.downto(1) do |i|
           stat = @levels[@stats[(level+(i.to_s)).to_sym].to_sym]
-          puts "%-10s unsolved(%3s) solved(%3s) total(%3s)" % [level+(i.to_s), stat[:unsolved], stat[:solved], stat[:total]]
+          puts "%-15s unsolved(%3s) solved(%3s) total(%3s)" % [color(level+(i.to_s)), stat[:unsolved], stat[:solved], stat[:total]]
         end
       else
         if !level or !@stats[level.to_sym]
@@ -60,7 +61,7 @@ module BOJ
           return
         end
         stat = @levels[@stats[level.to_sym].to_sym]
-        puts "%-10s unsolved(%3s) solved(%3s) total(%3s)" % [level, stat[:unsolved], stat[:solved], stat[:total]]
+        puts "%-15s unsolved(%3s) solved(%3s) total(%3s)" % [color(level), stat[:unsolved], stat[:solved], stat[:total]]
       end
     end
 
@@ -71,7 +72,7 @@ module BOJ
         return
       end
 
-      puts "%-10s %-15s" % ["ID", "Title"]
+      puts "%-10s %-10s %-15s" % ["Level", "ID", "Title"]
 
       if op == "solved"
         level_min = 0
@@ -90,7 +91,7 @@ module BOJ
         solved = solved.sort
         i = 0
         solved.each do |prob|
-          puts "%-10s%-10s%-15s" % [prob[1][:level], prob[0], prob[1][:title]]
+          puts "%-15s%-10s%-15s" % [color(prob[1][:level]), prob[0], prob[1][:title]]
           i+=1
           if (i+1)%10 == 0
             print("..... [q=quit] ") 
@@ -99,6 +100,7 @@ module BOJ
           end
         end
       else
+        puts "%-10s %-15s" % ["ID", "Title"]
         list = @levels[@stats[op.to_sym].to_sym][:prob]
         size = list.size
         i = 0
@@ -116,14 +118,21 @@ module BOJ
     end
 
     def  command_random(level)
-      problems = @levels[@stats[level.to_sym].to_sym][:prob] unless level==nil
-      level_str = level
-
       if level == nil
         max_level = 31
         level = rand(max_level).to_s
         problems = @levels[level.to_sym][:prob]
         level_str = @stats.keys[level.to_i]
+      elsif %w(bronze silver gold platinum diamond ruby).include? level
+        level_str = level + (rand(5)+1).to_s
+        problems = @levels[@stats[level_str.to_sym].to_sym][:prob]
+        lever_str = @stats.keys[level.to_i]
+      elsif @stats[level.to_sym] == nil
+        puts "'#{level}' level does not exist.."
+        return
+      else
+        problems = @levels[@stats[level.to_sym].to_sym][:prob] unless level==nil
+        level_str = level
       end
       
       unsolved = []
@@ -132,15 +141,15 @@ module BOJ
       end
 
       if unsolved.size == 0
-        puts "You solved every problems in '#{level_str}'"
+        puts "You solved every problems in '#{color(level_str)}'"
         return
       end
       id = unsolved[rand(unsolved.size)]
       title = problems[id][:title]
 
       puts
-      puts "Level: #{level_str}"
-      puts "ID   : #{id} (https://www.acmicpc.net/problem/#{id})"
+      puts "Level: #{color(level_str)}"
+      puts "ID   : #{id} (https://www.acmicpc.net/problem/#{id})".default
       puts "Title: #{title}"
       puts
     end
@@ -192,7 +201,7 @@ module BOJ
 
     def start
       while true do
-        print("boj-solvedac>  ")
+        print "%s" % "boj-solvedac>  ".default
         command(gets.chomp)
       end
     end

data/lib/colorize/colorize.rb

@@ -0,0 +1,59 @@
+class String
+  def color_code(code)
+    "\e[#{code}m#{self}\e0"
+  end
+
+  def default
+    color_code(39)
+  end
+
+  def unrated
+    # red
+    color_code(31)
+  end
+
+  def bronze
+    # default
+    color_code(39)
+  end
+
+  def silver
+    # white
+    color_code(97)
+  end
+
+  def gold
+    # yellow
+    color_code(93)
+  end
+
+  def platinum
+    # light cyan
+    color_code(96)
+  end
+
+  def diamond
+    # blue
+    color_code(34)
+  end
+
+  def ruby
+    # magenta
+    color_code(35)
+  end
+end
+
+def color(str)
+  temp = str.to_s.scan(/[a-z]/).join
+  str = str.to_s
+  case temp
+  when 'unrated' then str.unrated
+  when 'bronze' then str.bronze
+  when 'silver' then str.silver
+  when 'gold' then str.gold
+  when 'platinum' then str.platinum
+  when 'diamond' then str.diamond
+  when 'ruby' then str.ruby
+  end
+end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: boj-solvedac
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Jione Eu
@@ -10,13 +10,14 @@ bindir: bin
 cert_chain: []
 date: 2020-05-27 00:00:00.000000000 Z
 dependencies: []
-description: "  version 0.3.x -> 'solved [ID]' method added. \n"
+description: "  version 0.4.x -> colors added to each level\n"
 email: 
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
 - lib/boj-solvedac.rb
+- lib/colorize/colorize.rb
 - lib/data/data.rb
 homepage: https://solved.ac/
 licenses: