lib/floe/workflow/states/wait.rb in floe-0.4.0 vs lib/floe/workflow/states/wait.rb in floe-0.4.1

@@ -1,19 +1,24 @@
 # frozen_string_literal: true
 
+require 'time'
+
 module Floe
   class Workflow
     module States
       class Wait < Floe::Workflow::State
         attr_reader :end, :next, :seconds, :input_path, :output_path
 
         def initialize(workflow, name, payload)
           super
 
-          @next    = payload["Next"]
-          @end     = !!payload["End"]
-          @seconds = payload["Seconds"].to_i
+          @next           = payload["Next"]
+          @end            = !!payload["End"]
+          @seconds        = payload["Seconds"]&.to_i
+          @timestamp      = payload["Timestamp"]
+          @timestamp_path = Path.new(payload["TimestampPath"]) if payload.key?("TimestampPath")
+          @seconds_path   = Path.new(payload["SecondsPath"]) if payload.key?("SecondsPath")
 
           @input_path  = Path.new(payload.fetch("InputPath", "$"))
           @output_path = Path.new(payload.fetch("OutputPath", "$"))
         end
 
@@ -21,23 +26,27 @@
           super
           input = input_path.value(context, input)
 
           context.output     = output_path.value(context, input)
           context.next_state = end? ? nil : @next
+          please_hold(input)
         end
 
         def running?
-          now = Time.now.utc
-          if now > (Time.parse(context.state["EnteredTime"]) + @seconds)
-            context.state["FinishedTime"] = now.iso8601
-            false
-          else
-            true
-          end
+          waiting?
         end
 
         def end?
           @end
+        end
+
+        private
+
+        def please_hold(input)
+          wait(
+            :seconds => @seconds_path ? @seconds_path.value(context, input).to_i : @seconds,
+            :time    => @timestamp_path ? @timestamp_path.value(context, input) : @timestamp
+          )
         end
       end
     end
   end
 end