sleeper 0.0.0 → 0.1.0

data/lib/sleeper/timer.rb

@@ -1,29 +1,48 @@
 module Sleeper
   # TODO: StateMachine?
+  # TODO: Need to be able to add and remove schedules?
   class Timer
-    attr_reader :schedule
+    attr_reader :schedule, :default
 
-    def initialize(schedule, cyclic=false)
-      @cyclic = cyclic
+    def initialize(schedule, opts={})
+      @cyclic = opts[:cyclic] || false
+      @default = opts[:default] || nil
       from_hash(schedule) if schedule.class == Hash
       from_array(schedule) if schedule.class == Array
     end
 
-    # TODO: Should raise exception if @schedule[key] is nil
-    # TODO: Need two separate method calls here for block or no block
     def run(&block)
-      if block_given?
-        key = block.call
-        sleep(@schedule[key].current)
-        @schedule[key].next
+      block_given? ? schedule = @schedule[block.call] : schedule = @schedule
+      
+      # TODO: If schedule nil then use default or raise exception?
+      follow_schedule(schedule)
+    end
+
+    def follow_schedule(schedule)
+      period = schedule.current
+
+      sleep(period)
+      schedule.next
+
+      period
+    end
+
+    # TODO: Hash reset should be separate method and should handle entire hash or specific keys.
+    def reset(key=nil)
+      return @schedule[key].reset unless key.nil?
+
+      if @schedule.class == Hash
+        @schedule.values.each do |schedule|
+          schedule.reset
+        end
       else
-        sleep(@schedule.current)
-        @schedule.next
+        @schedule.reset
       end
+      #key.nil? ? @schedule.reset : @schedule[key].reset
     end
 
-    def reset(key=nil)
-      key.nil? ?  @schedule.reset : @schedule[key].reset
+    def default=(value)
+      value.nil? ? @default = nil : @default = Schedule.new(Array(value), @cyclic)
     end
 
     private

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sleeper
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-02-15 00:00:00.000000000 Z
+date: 2012-02-20 00:00:00.000000000 Z
 dependencies: []
 description: A small gem for variable sleep intervals
 email: 
@@ -23,7 +23,7 @@ files:
   bGliL3NsZWVwZXIvdGltZXIucmI=
 - !binary |-
   bGliL3NsZWVwZXIucmI=
-homepage: ''
+homepage: https://github.com/LeakyBucket/Sleeper
 licenses: []
 post_install_message: 
 rdoc_options: []