bars 0.0.2 → 0.0.3

data/bin/bars

@@ -4,5 +4,18 @@
 
 require 'bars'
 
-bars = Bars.new(ARGV)
+options = {}
+ARGV.each_with_index do |arg, index|
+  if arg == '-p'
+    options[:print_mode] = :percentage
+  elsif arg == '-n'
+    if ARGV[index+1]
+      options[:bar_size] = ARGV[index+1]
+    else
+      raise "you need to enter a number"
+    end
+  end
+end
+
+bars = Bars.new(options)
 bars.print

data/lib/bars.rb

@@ -1,13 +1,9 @@
 require 'yaml'
 
 class Bars
-  def initialize(args=[])
-    if !args.empty? && args.first == '-p'
-      @print_mode = :percentage
-    else
-      @print_mode = :bars
-    end
-
+  def initialize(options={})
+    @print_mode = options.delete(:print_mode) { :bars }
+    @bar_size = options.delete(:bar_size) { 20 }
     @timeframes = YAML.load(File.open(File.expand_path('~/bars.yml')))
   end
 
@@ -16,13 +12,21 @@ class Bars
     @timeframes.each do |timeframe, tasks|
       puts " #{timeframe.capitalize} ".center(30, "*")
       tasks.each do |task, done, goal|
+        done_percent = (done.to_f/goal)
         if @print_mode == :percentage
-          puts "--> #{task}: #{(done.to_f/goal * 100).round}%"
+          puts "--> #{task}: #{(done_percent * 100).floor}%"
         else
           puts "--> #{task}:"
-          done_count = (done.to_f/goal * 20).to_i
-          todo_count = 20 - done_count
-          puts "    |#{"+" * (done_count)}|#{"-" * todo_count}|"
+          done_string = "+" * (done_percent * @bar_size)
+          todo_string = "-" * (@bar_size - done_string.length)
+          if done == 0
+            splitter = '-'
+          elsif done == goal
+            splitter = '+'
+          else
+            splitter = '|'
+          end
+          puts "    |#{done_string}#{splitter}#{todo_string}|"
         end
       end
       puts

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: bars
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: false
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Robert Fletcher