enviroblyd 0.4.2 → 0.4.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 28e33b71b8d370a6ea6239748afe9e0bbb6023325cd3f46ae070e2e8433531f3
-  data.tar.gz: 3485336ad0bff70f3bb983d9487ad650c8b18f5afce4a45e2ec4a2ef167815f8
+  metadata.gz: 45118408246d815ecd5132ae18a22cc3fa33f4106b1a825984285c58d1c0129e
+  data.tar.gz: a948c9c4ffd8f1bbda57a92d3acf63b4abdf653d737fa36ddb4f3c7775301244
 SHA512:
-  metadata.gz: 3c303fba4a8693104fe31b69fd6af04a8d4c1899a51c9c8f949bec21dd8f378fefcc9fb81161db793b4344f9eceffe3d160e0e0bb04709e8dcbc9c555ce7f163
-  data.tar.gz: dd2b1d9035182ec8a18ed9f3b2cff027afe017f7c37f3b2d31b142b1ae92a783461c716d7804d2daf4a109bad334eae3ce47d09f9dc0efdc0574ffce2583a707
+  metadata.gz: c0016f127f30a58c53052664526b1b7a5da5cd28c7a539e7e36a559c2f0ab530d9bf48ab1340986e671547c89d2c38e03d901264340c4f7825b15edf611d753c
+  data.tar.gz: 57cf14b9e65ccaeb7a84a2681488815fe7ba0f7332edb3186dfdddf66bc6ec80e3a35b6287e48846e439e833ea578963ced5c1693a6856e7db394a5fabf94c5c

data/lib/enviroblyd/command.rb

@@ -6,8 +6,13 @@ class Enviroblyd::Command
   DEFAULT_TIMEOUT_SECONDS = 5 * 60
   DEFAULT_RUNTIME = "/bin/bash"
 
+  def self.run(params)
+    Thread.new do
+      new(params).run
+    end
+  end
+
   def initialize(params)
-    @web = Enviroblyd::Web.new
     @url = params.fetch "url"
     @script = params.fetch "script"
     @runtime = params.fetch "runtime", DEFAULT_RUNTIME
@@ -17,7 +22,6 @@ class Enviroblyd::Command
 
   def run
     puts "Command #{@url} starting"
-
     Open3.popen3("timeout #{@timeout} #{@runtime}") do |stdin, stdout, stderr, thread|
       stdin.puts @script
       stdin.close
@@ -25,11 +29,10 @@ class Enviroblyd::Command
       @stderr = stderr.read
       @exit_code = thread.value.exitstatus
     end
-
     puts "Command #{@url} exited with #{@exit_code}"
     $stdout.flush
 
-    @web.http(@url, type: Net::HTTP::Put, params: to_complete_params)
+    Enviroblyd::Web.http(@url, type: Net::HTTP::Put, params: to_complete_params)
   end
 
   private

data/lib/enviroblyd/daemon.rb

@@ -8,42 +8,36 @@ class Enviroblyd::Daemon
   LISTEN_PORT = ENV.fetch("ENVIROBLYD_PORT", 63106).to_i
 
   def self.start
-    web = Enviroblyd::Web.new
-    web.register
-
-    daemon = new
-    daemon.listen
+    imds = Enviroblyd::IMDS.new
+    host = imds.private_ipv4
+    new.listen(host) do
+      Enviroblyd::Web.register
+    end
   end
 
-  def listen
-    server = TCPServer.new LISTEN_PORT
-    puts "Listening on port #{LISTEN_PORT}"
+  def listen(host)
+    server = TCPServer.new host, LISTEN_PORT
+    puts "Listening on #{host}:#{LISTEN_PORT}"
+
+    yield
 
     loop do
       Thread.start(server.accept) do |client|
-        message = client.recv(MAX_MESSAGE_SIZE)
-
         params =
           begin
-            JSON.parse message
+            JSON.parse client.recv(MAX_MESSAGE_SIZE)
           rescue
-            nil
+            :invalid_json
           end
 
-        if params.nil?
+        if params == :invalid_json
           client.puts "Error parsing JSON"
         else
-          puts "Received valid JSON:"
-          puts params
           client.puts "OK"
         end
 
-        # TODO: Handle Broken pipe (Errno::EPIPE) (client closing connection before we write back)
         client.close
-
-        Thread.new do
-          Enviroblyd::Command.new(params).run
-        end
+        Enviroblyd::Command.run(params)
       end
     end
   end

data/lib/enviroblyd/imds.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+class Enviroblyd::IMDS
+  TOKEN_TTL_SECONDS = 30
+  IMDS_HOST = ENV.fetch("ENVIROBLYD_IMDS_HOST", "169.254.169.254")
+
+  def initialize
+    @token = Enviroblyd::Web.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.strip
+  end
+
+  def get(key)
+    Enviroblyd::Web.http("http://#{IMDS_HOST}/latest/meta-data/#{key}",
+      headers: { "X-aws-ec2-metadata-token" => @token }).
+      body.strip
+  end
+
+  def private_ipv4
+    get "local-ipv4"
+  end
+end

data/lib/enviroblyd/version.rb

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

data/lib/enviroblyd/web.rb

@@ -10,63 +10,63 @@ require "json"
 class Enviroblyd::Web
   USER_AGENT = "enviroblyd #{Enviroblyd::VERSION}"
   CONTENT_TYPE = "application/json"
-  # TOKEN_TTL_SECONDS = 30
-  # IMDS_HOST = ENV.fetch("ENVIROBLYD_IMDS_HOST", "169.254.169.254")
   API_HOST = ENV.fetch("ENVIROBLYD_API_HOST", "envirobly.com")
   WORKING_DIR = Pathname.new ENV.fetch("ENVIROBLYD_WORKING_DIR", "/var/envirobly/daemon")
   INITIALIZED_FILE = WORKING_DIR.join "initialized"
 
-  def register
-    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)
-      puts "Init response code: #{response.code}"
+  class << self
+    def register
+      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)
+        puts "Init response code: #{response.code}"
 
-      if response.code.to_i == 200
-        FileUtils.mkdir_p WORKING_DIR
-        File.write INITIALIZED_FILE, init_url
+        if response.code.to_i == 200
+          FileUtils.mkdir_p WORKING_DIR
+          File.write INITIALIZED_FILE, init_url
+        end
       end
     end
-  end
 
-  def http(url, type: Net::HTTP::Get, params: nil, headers: {}, retry_interval: 3, retries: 10, backoff: :exponential, tries: 1)
-    if retries <= tries
-      $stderr.puts "Retried #{tries} times. Aborting."
-      exit 1
-    end
+    def http(url, type: Net::HTTP::Get, params: nil, headers: {}, retry_interval: 3, retries: 10, backoff: :exponential, 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 = 10
-    http.read_timeout = 10
+      uri = URI(url)
+      http = Net::HTTP.new uri.host, uri.port
+      http.use_ssl = true if uri.scheme == "https"
+      http.open_timeout = 10
+      http.read_timeout = 10
 
-    request = type.new(uri, default_headers.merge(headers))
-    request.content_type = CONTENT_TYPE
-    request.body = JSON.dump(params) unless params.nil?
+      request = type.new(uri, default_headers.merge(headers))
+      request.content_type = CONTENT_TYPE
+      request.body = JSON.dump(params) unless params.nil?
 
-    response =
-      begin
-        http.request(request)
-      rescue
-        :retry
-      end
+      response =
+        begin
+          http.request(request)
+        rescue
+          :retry
+        end
 
-    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:, params:, retry_interval:, retries:, backoff:, tries: (tries + 1))
-    else
-      response
+      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:, params:, retry_interval:, retries:, backoff:, tries: (tries + 1))
+      else
+        response
+      end
     end
-  end
 
-  private
-    def default_headers
-      { "User-Agent" => USER_AGENT }
-    end
+    private
+      def default_headers
+        { "User-Agent" => USER_AGENT }
+      end
+  end
 end

data/lib/enviroblyd.rb

@@ -8,5 +8,8 @@ end
 require "zeitwerk"
 
 loader = Zeitwerk::Loader.for_gem
+loader.inflector.inflect(
+  "imds" => "IMDS"
+)
 loader.setup
 loader.eager_load

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: enviroblyd
 version: !ruby/object:Gem::Version
-  version: 0.4.2
+  version: 0.4.4
 platform: ruby
 authors:
 - Robert Starsi
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-04 00:00:00.000000000 Z
+date: 2024-09-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -68,6 +68,7 @@ files:
 - lib/enviroblyd/cli/main.rb
 - lib/enviroblyd/command.rb
 - lib/enviroblyd/daemon.rb
+- lib/enviroblyd/imds.rb
 - lib/enviroblyd/version.rb
 - lib/enviroblyd/web.rb
 homepage: https://github.com/envirobly/enviroblyd