horario 0.0.1 → 0.0.2

data/bin/horario

@@ -1,45 +1,59 @@
 #!/usr/bin/env ruby
+# encoding: utf-8
 
 require 'colorize'
 
-# first_number = ARGV[0].to_i
-# second_number = ARGV[1].to_i
-
-puts
-puts
-print '        Digite a hora de entrada: '.green
-hour = gets.chomp
-
-puts
-print '        Digite os minutos de entrada: '.green
-minutes = gets.chomp
-
-puts
-print '        Digite as horas adicionais: '.green
-hours_add = gets.chomp
-
-puts
-print '        Digite os minutos adicionais: '.green
-minutes_add = gets.chomp
+if (ARGV.size == 0)
+	puts
+	print '        Horario diário (h:m): '.green
+	total = gets.chomp
+
+	puts
+	print '        Hora de entrada (h:m): '.green
+	hour = gets.chomp
+
+	puts
+	print '        Tempo adicional (h:m): '.green
+	hour_add = gets.chomp
+elsif (ARGV.size == 2)
+	total = ARGV[0]
+	hour = ARGV[1]
+elsif (ARGV.size == 3)
+	total = ARGV[0]
+	hour = ARGV[1]
+	hour_add = ARGV[2]
+end
+
+total = total.split(':') if total
+hour = hour.split(':') if hour
+hour_add = hour_add.split(':') if hour_add
 
 now = Time.now
 year = now.year
 month = now.month
 day = now.day
-hour = Integer(hour)
-minutes = Integer(minutes)
-hours_add = Integer(hours_add)
-hours_add = (60 * 60 * hours_add)
-minutes_add = Integer(minutes_add)
-minutes_add = (60 * 60 * (100/60.0*minutes_add/100))
- 
+minutes = Integer(hour[1])
+hour = Integer(hour[0])
+
+if (hour_add)
+	minutes_add = Integer(hour_add[1])
+	hour_add = Integer(hour_add[0])
+	
+	hour_add = (60 * 60 * hour_add)
+	minutes_add = (60 * 60 * (100/60.0*minutes_add/100))
+else
+	hour_add = 0
+	minutes_add = 0
+end
+
+total_hours = Integer(total[0])
+total_minutes = Integer(total[1])
 
 time = Time.new(year, month, day, hour, minutes, 0)
-interval = (60 * 60 * (9 + (100/60.0*48/100)))
-result = (time+interval+hours_add+minutes_add).strftime('%H:%M')
+interval = (60 * 60 * (total_hours + (100/60.0*total_minutes/100)))
+result = (time+interval+hour_add+minutes_add).strftime('%H:%M')
 
 puts
 puts "        Seu horario de saida e #{result}".yellow
 puts
-puts
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: horario
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-12-10 00:00:00.000000000 Z
+date: 2012-12-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: colorize