enviroblyd 0.2.2 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8220085dcf938ad62385573bc35fdecd43fc1be67c8c8290809e0e324ffa6b60
-  data.tar.gz: b34fa0008ebbd596a608e71fd4899f6ed6265441acd4539d8191e632324f57e6
+  metadata.gz: 5dbb9c1de2225aad7c7211baf934798dd3a059ce4dabe7efeed4d96b1e804410
+  data.tar.gz: c5eb8ba3ed2880049eb6094c30cdfb8a2514f5c22ba770f5a646027f9368ae7f
 SHA512:
-  metadata.gz: 83c2d244fb58a3f8bc8c94d9325b9fff0f220bcba0df5acadbae99999e0237403a6646a67623fb27eb6891efff98479ba043b8ae23a832673faca852783bea6e
-  data.tar.gz: b0bf6a4b127e324f10ff63556ff8b5922355371305d5c19625b17023b4589a07883fef93bb7fcddc49fd48f539923f3b6d06e2e66795852d0af8466321fccff3
+  metadata.gz: fbb457cd2d63bd5565c1b9ea0ceb96de79ad4def5e4674d691595cbee5706da25b339d96059f20e0779f77d76368688d16d3fe64c3a56eae5a8ace9d1ef2add0
+  data.tar.gz: 2ed04fd30d0522315c2e56fd0a03010eb801c78bc2c17175ad1bc3b744e743dbe105486a1cc465a4de731a6099ced61bb18ada8696e477542f4aff2f4ed58f07

data/lib/enviroblyd/cli/main.rb

@@ -2,6 +2,8 @@
 require "net/http"
 require "uri"
 require "open3"
+require "fileutils"
+require "pathname"
 
 class Enviroblyd::Cli::Main < Enviroblyd::Base
   desc "version", "Show version"
@@ -12,40 +14,51 @@ class Enviroblyd::Cli::Main < Enviroblyd::Base
   TOKEN_TTL_SECONDS = 30
   IMDS_HOST = ENV.fetch("ENVIROBLYD_IMDS_HOST", "169.254.169.254")
   API_HOST = ENV.fetch("ENVIROBLYD_API_HOST", "envirobly.com")
-  desc "boot", "Get IMDSv2 metadata"
+  WORKING_DIR = Pathname.new ENV.fetch("ENVIROBLYD_WORKING_DIR", "/var/envirobly/daemon")
+  INITIALIZED_FILE = WORKING_DIR.join "initialized"
+  desc "boot", "Start the daemon"
   def boot
-    @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 }).
-      body.chomp("")
-    puts "instance_id: #{instance_id} ."
+    # @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 }).
+    #   body.chomp("")
+    # puts "instance_id: #{instance_id} ."
+    #
+    # 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}"
 
-    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
+    if File.exist?(INITIALIZED_FILE)
+      puts "Skipping initialization because #{INITIALIZED_FILE} exists."
+    else
+      init_url = ENV.fetch "ENVIROBLYD_INIT_URL"
+      puts "Init URL: #{init_url}"
+      response = http(init_url, type: Net::HTTP::Put, retry_interval: 3, retries: 10, backoff: :exponential)
+      puts "Init response code: #{response.code}"
+
+      if response.code.to_i == 200
+        FileUtils.mkdir_p WORKING_DIR
+        File.write INITIALIZED_FILE, init_url
       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
 
   private
-    def http(url, type: Net::HTTP::Get, headers: {}, retry_interval: 2, retries: 30, backoff: false, success_codes: 200..299, tries: 1)
+    def http(url, type: Net::HTTP::Get, headers: {}, retry_interval: 2, retries: 30, backoff: false, tries: 1)
+      if retries <= tries
+        $stderr.puts "Retried #{tries} times. Aborting."
+        exit 1
+      end
+
       uri = URI(url)
       http = Net::HTTP.new uri.host, uri.port
       http.use_ssl = true if uri.scheme == "https"
       http.open_timeout = 5
       http.read_timeout = 5
 
-      request = type.new(uri, headers)
+      request = type.new(uri, default_headers.merge(headers))
       # request.content_type = CONTENT_TYPE
 
       yield request if block_given?
@@ -54,30 +67,17 @@ class Enviroblyd::Cli::Main < Enviroblyd::Base
         begin
           http.request(request)
         rescue
-          nil
+          :retry
         end
 
-      if response && success_codes.include?(response.code.to_i)
-        response
-      elsif retries <= tries
-        $stderr.puts "Retried #{tries} times. Aborting."
-        exit 1
-      else
+      # https://developers.cloudflare.com/support/troubleshooting/cloudflare-errors/troubleshooting-cloudflare-1xxx-errors/
+      if response == :retry || (500..599).include?(response.code.to_i)
         sleep_time = (backoff == :exponential) ? (retry_interval * tries) : retry_interval
         $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
+        http(url, type:, retry_interval:, retries:, backoff:, tries: (tries + 1))
       else
-        $stderr.puts "User data does not contain runable script."
+        response
       end
     end
 
@@ -92,4 +92,9 @@ class Enviroblyd::Cli::Main < Enviroblyd::Base
         @exit_code = thread.value.exitstatus
       end
     end
+
+    USER_AGENT = "enviroblyd #{Enviroblyd::VERSION}"
+    def default_headers
+      { "User-Agent" => USER_AGENT }
+    end
 end

data/lib/enviroblyd/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: enviroblyd
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.3.1
 platform: ruby
 authors:
 - Robert Starsi
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-02 00:00:00.000000000 Z
+date: 2024-09-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -84,7 +84,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.14
+rubygems_version: 3.5.18
 signing_key:
 specification_version: 4
 summary: Envirobly instance daemon