lib/dynamodb/streams/client.rb in dynamodb-streams-client-0.0.2 vs lib/dynamodb/streams/client.rb in dynamodb-streams-client-0.0.3

@@ -5,11 +5,14 @@
 require 'time'
 require 'stringio'
 require 'zlib'
 require 'uri'
 
+require 'thor'
+
 require 'dynamodb/streams/client/version'
+require 'dynamodb/streams/client/cli'
 
 class DynamoDB::Streams::Client
   class Error < StandardError
     attr_reader :data
 
@@ -108,13 +111,15 @@
     res_data = JSON.parse(res_body)
     __type = res_data['__type']
 
     if res_code != 200 or __type
       errmsg = if __type
+                 res_data_msg = res_data['message'] || res_data['Message'] || __type
+
                  if @debug
-                   "#{__type}: #{res_data['message'] || res_data['Message']}"
+                   "#{__type}: #{res_data_msg}"
                  else
-                   "#{res_data['message'] || res_data['Message']}"
+                   "#{res_data_msg}"
                  end
                else
                  "#{res_code} #{res_msg}"
                end