boj-solvedac 0.1.0 → 0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ecbc0d0383d854651a218a98a9587de98f18197c71a101782fbbaaf1502e1a16
-  data.tar.gz: 42b528416563938285ff1e9bf00a9d5bed8b75faa2230f19d0edf545d410bee3
+  metadata.gz: e97f59decd24f566582a2d122146f25f29a438fe5b315960fc113344a8ef4ef1
+  data.tar.gz: 316233507d002926dd8df309526ff9dc9f042545bbd628f339b30cf8c1b89370
 SHA512:
-  metadata.gz: 5ff573b1aad9e851ea7e7925db17137404e6d04b71c112de28c71d2f7c8a85bd42eb47e854107add7b9c7bc3e4a0be06d4ccccc859f1e3e2fca049ea1270512c
-  data.tar.gz: 7e2d8720c33f7c416155b1c1f532723e1074b8ee687c02a058eaf12294ec6d70e92ac95b537a32b05516d05c90cd37f045164aebc80785cc30a3661b870ef32f
+  metadata.gz: 263f4f56e06ffbd11c6153597997ccfc7655ece984a5247009e5d6e977a09ebeb7df140203c2e03d648a65372999fb29cf6ee6cdf53dfc79dd57e265f45c8a1c
+  data.tar.gz: d5479ff94c96cc2bf9f7dae499668c2c6428bdcd40833336b98f9fb6baf3c05a9e73f07e0bdb4f131ad0d3cccbe8d31a29714f0caa86983cfddf0ce229bfe75a

data/lib/boj-solvedac.rb

@@ -1,4 +1,5 @@
 require 'data/data.rb'
+require 'colorize/colorize.rb'
 
 module BOJ
   class BOJSolvedAC < BOJData
@@ -30,6 +31,7 @@ module BOJ
       when 'stat' then command_stat(level)
       when 'prob' then command_prob(level)
       when 'random' then command_random(level)
+      when 'solved' then command_solved(level)
       when 'clear' then system('clear')
       when 'help' then command_help
       when /exit|quit/ then exit(1)
@@ -43,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]
@@ -59,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
 
@@ -70,9 +72,8 @@ module BOJ
         return
       end
 
-      puts "%-10s %-15s" % ["ID", "Title"]
-
       if op == "solved"
+        puts "%-10s %-10s %-15s" % ["Level", "ID", "Title"]
         level_min = 0
         level_max = 30
         solved = {}
@@ -89,7 +90,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] ") 
@@ -98,6 +99,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
@@ -115,26 +117,53 @@ 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 = []
+      problems.each_pair do |k, v|
+        unsolved.push(k) if v[:solved] == false 
       end
 
-      id    = problems.keys[rand(problems.size)]
+      if unsolved.size == 0
+        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
 
+    def command_solved(id)
+      path = 'stats/solved-problems.dat'
+      File.open(path, 'r+') # create one if one doesn't exist
+      problems = IO.read(path).chomp
+      problems += " #{id}"
+      File.open(path, "r+") do |f|
+        f.puts problems
+      end
+      fetch_solved_data
+    end
+
     def command_help
       puts '''
       levels you can specify:
@@ -157,6 +186,9 @@ module BOJ
              random         --> randomly pick a problem from all levels
              random [LEVEL] --> randomly pick a problem from [LEVEL]
              * hit [ENTER] to repeat the previous \'random\' command
+        solved:
+             solved [ID] --> adds [ID] to \'solved-problems.dat\' and
+                             updates the stat.
         clear:
              clear --> clears the screen
         exit:
@@ -168,7 +200,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
+    # magenta
+    color_code(35)
+  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
+    # red
+    color_code(31)
+  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
+

data/lib/data/data.rb

@@ -74,7 +74,7 @@ class BOJData
         end
       end
       @@levels[i.to_s.to_sym][:solved] = solved_prob
-      @@levels[i.to_s.to_sym][:unsolved] = ((@@levels[i.to_s.to_sym][:unsolved].to_i) - solved_prob).to_s
+      @@levels[i.to_s.to_sym][:unsolved] = ((@@levels[i.to_s.to_sym][:total].to_i) - solved_prob).to_s
 
       break if solved_prob == solved_size
     end

metadata

@@ -1,22 +1,26 @@
 --- !ruby/object:Gem::Specification
 name: boj-solvedac
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.4.2
 platform: ruby
 authors:
 - Jione Eu
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-05-25 00:00:00.000000000 Z
+date: 2020-05-29 00:00:00.000000000 Z
 dependencies: []
-description: 
+description: |2
+    version 0.4.0 -> colors added to each level
+    version 0.4.1 -> 'solved' command overwriting 'solved-problems.dat' bug fixed.
+    version 0.4.2 -> 'ruby' and 'unrated' color swapped.
 email: 
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
 - lib/boj-solvedac.rb
+- lib/colorize/colorize.rb
 - lib/data/data.rb
 homepage: https://solved.ac/
 licenses: