Add WorkflowBase base class for ItemProcessor/Workflow

lib/floe.rb

@@ -7,6 +7,7 @@
 
 require_relative "floe/runner"
 
+require_relative "floe/workflow_base"
 require_relative "floe/workflow"
 require_relative "floe/workflow/catcher"
 require_relative "floe/workflow/choice_rule"

lib/floe/workflow.rb

@@ -4,7 +4,7 @@
 require "json"
 
 module Floe
-  class Workflow
+  class Workflow < Floe::WorkflowBase
     include Logging
 
     class << self

lib/floe/workflow/item_processor.rb

@@ -2,20 +2,12 @@
 
 module Floe
   class Workflow
-    class ItemProcessor
-      attr_reader :processor_config, :payload, :states, :start_at
+    class ItemProcessor < Floe::WorkflowBase
+      attr_reader :processor_config
 
       def initialize(payload, name = nil)
-        @payload = payload
-        @name    = name
-
-        raise Floe::InvalidWorkflowError, "Missing field \"States\" for state [#{name}]"  if payload["States"].nil?
-        raise Floe::InvalidWorkflowError, "Missing field \"StartAt\" for state [#{name}]" if payload["StartAt"].nil?
-        raise Floe::InvalidWorkflowError, "\"StartAt\" not in the \"States\" field for state [#{name}]" unless payload["States"].key?(payload["StartAt"])
-
+        super
         @processor_config = payload.fetch("ProcessorConfig", "INLINE")
-        @states           = payload["States"].to_a.map { |state_name, state| State.build!(self, state_name, state) }
-        @states_by_name   = @states.to_h { |state| [state.name, state] }
       end
 
       def value(_context, input = {})

lib/floe/workflow_base.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Floe
+  class WorkflowBase
+    attr_reader :payload, :states, :start_at
+
+    def initialize(payload, name = nil)
+      @payload = payload
+      @name    = name
+
+      raise Floe::InvalidWorkflowError, "Missing field \"States\" for state [#{name}]"  if payload["States"].nil?
+      raise Floe::InvalidWorkflowError, "Missing field \"StartAt\" for state [#{name}]" if payload["StartAt"].nil?
+      raise Floe::InvalidWorkflowError, "\"StartAt\" not in the \"States\" field for state [#{name}]" unless payload["States"].key?(payload["StartAt"])
+
+      @states           = payload["States"].to_a.map { |state_name, state| Floe::Workflow::State.build!(self, state_name, state) }
+      @states_by_name   = @states.to_h { |state| [state.name, state] }
+    end
+  end
+end