enviroblyd 0.1.3 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c88f9afa8cae744a95f4b3813a6ba92b273fb259b6b58815baadc60a889bc698
-  data.tar.gz: e1e3938c06ffa662ad066cca2b49d3f69500a04178e264721c78faf954523135
+  metadata.gz: e565c96dd7f26d49788503f35f5dc895baa532d1dbae03723d8fb42fad74e013
+  data.tar.gz: 7166e8a78f7091a0b6a3e3dbfa43359224828a0c94430b93d7e7fe7baa6b3619
 SHA512:
-  metadata.gz: ff3a731d3ba7d43faeb6da6573060db09dd0087ceaba056bdbe9d2d6e71223b00ed6452199d895d8d3fcae4a4cd9504708ea82296628d58c6cff11a3a822dbe3
-  data.tar.gz: 79a94b0c5a26173b5d10d39da7a513073be4cf280ad6d47c510d211fc8939cfe7ad2eea45e126dc529174ac34ea96b60b5781f3cd4049bd7593927b69b18ef8a
+  metadata.gz: aa26f07e720624c75d4e7f69de79cc643ffb58f3de56a9c02be818b3152b56caf4798f688f84be8cc223c728c397da46fd7d23a11258bc311cfcabed4fb3827f
+  data.tar.gz: 8efe981de9ecd6844f2323c171109894d6114284a09587b1f9e6d6d79769d8b434388345fa2068efb3543f681771d96a729420c315096def4c82e7214092114e

data/lib/enviroblyd/cli/main.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 require "net/http"
 require "uri"
+require "open3"
 
 class Enviroblyd::Cli::Main < Enviroblyd::Base
   desc "version", "Show version"
@@ -13,15 +14,25 @@ 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",
+    @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",
-      headers: { "X-aws-ec2-metadata-token" => token }).
+    puts "token: #{@token} ."
+    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} ."
 
+    process_user_data
+    unless @exit_code.nil?
+      puts @stdout
+      $stderr.puts @stderr
+      unless @exit_code == 0
+        $stderr.puts "User data script exited with code: #{@exit_code}. Aborting."
+        exit 1
+      end
+    end
+
     response = http("https://#{API_HOST}/api/v1/boots/#{instance_id}", retry_interval: 3, retries: 5, backoff: :exponential)
     puts "/api/v1/boots response code: #{response.code}"
   end
@@ -42,7 +53,7 @@ class Enviroblyd::Cli::Main < Enviroblyd::Base
       response =
         begin
           http.request(request)
-        rescue Net::OpenTimeout, Net::ReadTimeout, Errno::EHOSTUNREACH, Errno::EHOSTDOWN
+        rescue
           nil
         end
 
@@ -53,9 +64,32 @@ class Enviroblyd::Cli::Main < Enviroblyd::Base
         exit 1
       else
         sleep_time = (backoff == :exponential) ? (retry_interval * tries) : retry_interval
-        puts "Try #{uri} in #{sleep_time}s"
+        $stderr.puts "Retry #{uri} in #{sleep_time}s"
         sleep sleep_time
         http(url, type:, retry_interval:, retries:, backoff:, success_codes:, tries: (tries + 1))
       end
     end
+
+    def process_user_data
+      response = http("http://#{IMDS_HOST}/latest/user-data",
+        headers: { "X-aws-ec2-metadata-token" => @token })
+
+      if response.body.start_with?("#!/bin/bash")
+        run response.body
+      else
+        $stderr.puts "User data does not contain runable script."
+      end
+    end
+
+    RUN_TIMEOUT = "5m"
+    def run(script)
+      @stdout = @stderr = @exit_code = nil
+      Open3.popen3("timeout #{RUN_TIMEOUT} /bin/bash") do |stdin, stdout, stderr, thread|
+        stdin.puts script
+        stdin.close
+        @stdout = stdout.read
+        @stderr = stderr.read
+        @exit_code = thread.value.exitstatus
+      end
+    end
 end

data/lib/enviroblyd/version.rb

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

metadata

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