enviroblyd 0.1.2 → 0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4662a74b3e3eb17cad2ef6c6f511fc79e4460eb9a40635a8f476e50076b075e7
-  data.tar.gz: 4fba6adca14dd3247decf29333e6452e544c91b9772012c494c1e452cdee74ae
+  metadata.gz: a61c27661f880ff2306805248d6ca4a7eac4f757f337f1380582831b6d85fdce
+  data.tar.gz: 82bbdd550838e7676a94970079a7086eb7c2906de86384b41686556a67e7985e
 SHA512:
-  metadata.gz: 9ea39b7b42bdd486e064ef256524fcb5ca42030bcc3a91d51382436b0f5c035c8da438cc4d5f094aec714319429df255dc717fb74b7240e44cb051d86bd2e987
-  data.tar.gz: eeb5ea402402c421d298da8dd2e641fa8c0c46e39dfd5e27abffe89752aa180df8c5a965461d337ca097446486a27f47e25a265b48142df9018245d7cd43f4cf
+  metadata.gz: 43f1691bbbb5a00f2ac39ff1d991cf5330d62d5be49541f46d312bb70b62e5e5536ff05a5c314b7ca8dc42b868fb6db54866ec1ca27a490e5c7b4429cd972bf1
+  data.tar.gz: 2e4c114de8590593c522b28f5a372fa558f11dbd74fd2a6e9ce32cd9c6d2643d7179ad099082e238a313a1b756571b5fc2f22de97784e02ff988ca4d762f6f95

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
@@ -39,17 +50,46 @@ 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."
         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.2"
+  VERSION = "0.2"
 end

metadata

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