enviroblyd 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e1557473923398525894640e93df45810a6333ede11a9b51974071bf80b48509
-  data.tar.gz: 77a34dabf2f079257158827a8a2f80448c9b19c0d879b194f29dab720ec9de66
+  metadata.gz: c88f9afa8cae744a95f4b3813a6ba92b273fb259b6b58815baadc60a889bc698
+  data.tar.gz: e1e3938c06ffa662ad066cca2b49d3f69500a04178e264721c78faf954523135
 SHA512:
-  metadata.gz: ff373c35e37d10c84fa9d70bde75c12845d99eb89e24f3fb597c9abfad81a20b3646a8be542d4d36b0303893984aa331f3fdfcd19a08abf82509bf21ef53d514
-  data.tar.gz: 8f6819a56c04a72a18e1b41e9aee4ad5ab9f21adbcd032b8df9ee1440464f871edae1734ef9a13d96dcabb13132caa5fe0dc04cd04c445b86c985f5a7e63bea9
+  metadata.gz: ff3a731d3ba7d43faeb6da6573060db09dd0087ceaba056bdbe9d2d6e71223b00ed6452199d895d8d3fcae4a4cd9504708ea82296628d58c6cff11a3a822dbe3
+  data.tar.gz: 79a94b0c5a26173b5d10d39da7a513073be4cf280ad6d47c510d211fc8939cfe7ad2eea45e126dc529174ac34ea96b60b5781f3cd4049bd7593927b69b18ef8a

data/lib/enviroblyd/cli/main.rb

@@ -13,9 +13,13 @@ class Enviroblyd::Cli::Main < Enviroblyd::Base
   API_HOST = ENV.fetch("ENVIROBLYD_API_HOST", "envirobly.com")
   desc "boot", "Get IMDSv2 metadata"
   def boot
-    token = http("http://#{IMDS_HOST}/latest/api/token").body.chomp("")
+    token = http("http://#{IMDS_HOST}/latest/api/token",
+      type: Net::HTTP::Put, headers: { "X-aws-ec2-metadata-token-ttl-seconds" => TOKEN_TTL_SECONDS.to_s }).
+      body.chomp("")
     puts "token: #{token} ."
-    instance_id =  http("http://#{IMDS_HOST}/latest/meta-data/instance-id").body.chomp("")
+    instance_id =  http("http://#{IMDS_HOST}/latest/meta-data/instance-id",
+      headers: { "X-aws-ec2-metadata-token" => token }).
+      body.chomp("")
     puts "instance_id: #{instance_id} ."
 
     response = http("https://#{API_HOST}/api/v1/boots/#{instance_id}", retry_interval: 3, retries: 5, backoff: :exponential)
@@ -35,8 +39,14 @@ class Enviroblyd::Cli::Main < Enviroblyd::Base
 
       yield request if block_given?
 
-      response = http.request(request)
-      if success_codes.include?(response.code.to_i)
+      response =
+        begin
+          http.request(request)
+        rescue Net::OpenTimeout, Net::ReadTimeout, Errno::EHOSTUNREACH, Errno::EHOSTDOWN
+          nil
+        end
+
+      if response && success_codes.include?(response.code.to_i)
         response
       elsif retries <= tries
         $stderr.puts "Retried #{tries} times. Aborting."

data/lib/enviroblyd/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Enviroblyd
-  VERSION = "0.1.1"
+  VERSION = "0.1.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: enviroblyd
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Robert Starsi