ruby-sfn-local 0.1.5 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1d87496719dd8f50c7c11aabea2f8491490bf9de3d343c71a8bf81d977061711
-  data.tar.gz: 971cf0f7a5bc42bac1bc696a88c70493675e868d3e02005765c215a171211eea
+  metadata.gz: 031c92b9bd2cca0fe86da9f393fd5c8881dbc04bee6c43b189585bdbbd9aa35a
+  data.tar.gz: d6e51f7f87a2278c8f7f12781f1d795d3afb324265287744c0d030d53f70efd1
 SHA512:
-  metadata.gz: d3a80927e8f087ad773be0eb0c64e4b4be550a10929be43b45a79efad5923325f8bfc579906b1bb6e4520ea56e2a14e18010e538fa80e8a16b2f43eee57410cc
-  data.tar.gz: f639bd302f6622747d3743745ed1264aca5d96103716d9e678f9d8a5e0b5bbf8b40d3959e19824da34ab71881bbcede1f7509d031c37b4dd913ae1a814746743
+  metadata.gz: 0dc87f86c63cd1fe745db17dab08afbd440b76b7d81a6f766b1cfae830acbaf7e475d480aff53a22dd8f015fc73c724951c9ee970ac76c09953970ab7bc768b3
+  data.tar.gz: a31fbba0e5c16e68fd792f959ca88494f3abde0f9caea5a0ca7a1b38671b7ad9bcacb07d5eeeb580579f6f0d8e4626029f8095a4687214ded011b9692a87658a

data/lib/sfn/aws_cli.rb

@@ -13,11 +13,13 @@ module Sfn
                         "--#{k}=#{v}"
                       end.join(' ')} \
                       --no-cli-pager"
-      stdout, stderr, _status = Open3.capture3(cmd)
+      if command == 'get-execution-history'
+        stdout = `#{cmd}`
+      else
+        stdout, stderr, _status = Open3.capture3(cmd)
+        raise raise ExecutionError, "#{stderr.strip}\n#{debug_info}" unless stderr.strip.empty?
+      end
 
-      raise raise ExecutionError, "#{stderr.strip}\n#{debug_info}" unless stderr.strip.empty?
-      
-      stdout = `#{cmd}` if command == 'get-execution-history'
       unless key.nil?
         data = JSON.parse(stdout)
         return data[key].strip if data.key?(key)

data/lib/sfn/execution_log.rb

@@ -15,11 +15,13 @@ module Sfn
     attr_accessor :event
 
     EVENTS = %w[stateEnteredEventDetails stateExitedEventDetails executionSucceededEventDetails
-                executionFailedEventDetails].freeze
+                executionFailedEventDetails taskScheduledEventDetails].freeze
+
     def self.parse(execution_arn)
       profile = {}
       output = nil
       error = nil
+      state_name = nil
       events_json = AwsCli.run('stepfunctions', 'get-execution-history',
                                { 'execution-arn': execution_arn.to_s, query: "'events[?#{EVENTS.join(' || ')}]'" })
       JSON.parse(events_json).each do |event|
@@ -27,13 +29,17 @@ module Sfn
 
         output ||= parsed_event.output
         error  ||= parsed_event.error(events_json)
+        last_state_name = state_name
         state_name = parsed_event.state_name
 
-        next if state_name.nil?
-
-        profile[state_name] ||= { input: [], output: [] }
-        profile[state_name][:input] << parsed_event.profile[:input] unless parsed_event.profile[:input].nil?
-        profile[state_name][:output] << parsed_event.profile[:output] unless parsed_event.profile[:output].nil?
+        unless state_name.nil?
+          profile[state_name] ||= { 'input' => [], 'output' => [], 'parameters' => [] }
+          profile[state_name]['input'] << parsed_event.profile['input'] unless parsed_event.profile['input'].nil?
+          profile[state_name]['output'] << parsed_event.profile['output'] unless parsed_event.profile['output'].nil?
+        end
+        if !last_state_name.nil? && !parsed_event.profile['parameters'].nil?
+          profile[last_state_name]['parameters'] << parsed_event.profile['parameters']
+        end
       end
       [output, profile]
     end
@@ -60,8 +66,9 @@ module Sfn
 
     def profile
       {
-        input: try_parse(event.dig('stateEnteredEventDetails', 'input')),
-        output: try_parse(event.dig('stateExitedEventDetails', 'output'))
+        'input' => try_parse(event.dig('stateEnteredEventDetails', 'input')),
+        'output' => try_parse(event.dig('stateExitedEventDetails', 'output')),
+        'parameters' => try_parse(event.dig('taskScheduledEventDetails', 'parameters'))
       }.compact
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-sfn-local
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.8
 platform: ruby
 authors:
 - Gianni Mazza
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-05-04 00:00:00.000000000 Z
+date: 2022-05-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-release