ruby-sfn-local 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1d87496719dd8f50c7c11aabea2f8491490bf9de3d343c71a8bf81d977061711
-  data.tar.gz: 971cf0f7a5bc42bac1bc696a88c70493675e868d3e02005765c215a171211eea
+  metadata.gz: 5fe17e00c8d4e14a4fce01894eb055a416e43abe8bcc9e7c9335a7af385e20af
+  data.tar.gz: d5fbd90070b32dd0c8d3afd1b6c7096842679b97028d8538efd19c0a70d29b98
 SHA512:
-  metadata.gz: d3a80927e8f087ad773be0eb0c64e4b4be550a10929be43b45a79efad5923325f8bfc579906b1bb6e4520ea56e2a14e18010e538fa80e8a16b2f43eee57410cc
-  data.tar.gz: f639bd302f6622747d3743745ed1264aca5d96103716d9e678f9d8a5e0b5bbf8b40d3959e19824da34ab71881bbcede1f7509d031c37b4dd913ae1a814746743
+  metadata.gz: fcde4ca78448650c0dfd234c9783a01f7ef9e90e819f3d678a9094a8702a2b3c3affdbcf99bd1450bfc04e74715f1c2e3de98bbe20098826758b71aa10eec79b
+  data.tar.gz: 2fdd25a501f35fd1d9b76198e1a3b71328eddc1b1c9dd470f8cf225655eb41f61271496cd6ba1bc3b0d1890c18be939e6d005d93d5fcf066a7077eec28148a73

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
@@ -61,7 +67,8 @@ module Sfn
     def profile
       {
         input: try_parse(event.dig('stateEnteredEventDetails', 'input')),
-        output: try_parse(event.dig('stateExitedEventDetails', 'output'))
+        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.6
 platform: ruby
 authors:
 - Gianni Mazza
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-05-04 00:00:00.000000000 Z
+date: 2022-05-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-release