enviroblyd 0.4.4 → 0.4.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 45118408246d815ecd5132ae18a22cc3fa33f4106b1a825984285c58d1c0129e
-  data.tar.gz: a948c9c4ffd8f1bbda57a92d3acf63b4abdf653d737fa36ddb4f3c7775301244
+  metadata.gz: a8467f960365419e9f02f14d3d3186a511787dff28798fd8883ac328a2443e9c
+  data.tar.gz: 2f39dbca93e49cf69ab2bb9aaed76fe89ed2ebd7867baebba640c8ea665082ae
 SHA512:
-  metadata.gz: c0016f127f30a58c53052664526b1b7a5da5cd28c7a539e7e36a559c2f0ab530d9bf48ab1340986e671547c89d2c38e03d901264340c4f7825b15edf611d753c
-  data.tar.gz: 57cf14b9e65ccaeb7a84a2681488815fe7ba0f7332edb3186dfdddf66bc6ec80e3a35b6287e48846e439e833ea578963ced5c1693a6856e7db394a5fabf94c5c
+  metadata.gz: dc3ea42ebc9125f88ec1828b4ebdf9032ba2482d1db359860b9b7c117ae0cb12eb57369531bd760cb54b2563f64c63e3afb89a304a574c23fa19bc7ff4334214
+  data.tar.gz: 3c9baed9045d458595627b186f5be800abe5da2df36beea47e22fd478bc28e25e984814de9731118040b4a65f16fabea10923d06daaa0f419639ace81f0be67f

data/lib/enviroblyd/command.rb

@@ -1,23 +1,26 @@
 # frozen_string_literal: true
 
 require "open3"
+require "json"
 
 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)
+  def initialize(message)
+    params = parse_message message
     @url = params.fetch "url"
     @script = params.fetch "script"
     @runtime = params.fetch "runtime", DEFAULT_RUNTIME
     @timeout = params.fetch "timeout", DEFAULT_TIMEOUT_SECONDS
     @stdout = @stderr = @exit_code = nil
+    @valid = true
+  rescue NoMethodError, KeyError
+    @valid = false
+  end
+
+  def valid?
+    @valid
   end
 
   def run
@@ -45,4 +48,10 @@ class Enviroblyd::Command
         }
       }
     end
+
+    def parse_message(message)
+      JSON.parse message
+    rescue
+      nil
+    end
 end

data/lib/enviroblyd/daemon.rb

@@ -1,43 +1,47 @@
 # frozen_string_literal: true
 
 require "socket"
-require "json"
 
 class Enviroblyd::Daemon
   MAX_MESSAGE_SIZE = 6000 # bytes
-  LISTEN_PORT = ENV.fetch("ENVIROBLYD_PORT", 63106).to_i
+  PORT = ENV.fetch("ENVIROBLYD_PORT", 63106).to_i
 
   def self.start
     imds = Enviroblyd::IMDS.new
     host = imds.private_ipv4
-    new.listen(host) do
-      Enviroblyd::Web.register
-    end
+    daemon = new(host)
+    daemon.listen
   end
 
-  def listen(host)
-    server = TCPServer.new host, LISTEN_PORT
-    puts "Listening on #{host}:#{LISTEN_PORT}"
+  def initialize(host)
+    @host = host
+  end
 
-    yield
+  def listen
+    server = TCPServer.new @host, PORT
+    puts "Listening on #{@host}:#{PORT}"
+    Enviroblyd::Web.register
 
     loop do
       Thread.start(server.accept) do |client|
-        params =
-          begin
-            JSON.parse client.recv(MAX_MESSAGE_SIZE)
-          rescue
-            :invalid_json
-          end
-
-        if params == :invalid_json
-          client.puts "Error parsing JSON"
-        else
-          client.puts "OK"
+        message = client.recv(MAX_MESSAGE_SIZE)
+        command = Enviroblyd::Command.new message
+
+        unless command.valid?
+          $stderr.puts "Invalid message received: #{message}"
+          client.puts "Invalid message"
+          next
         end
 
+        client.puts "OK"
+        client.close
+        command.run
+      ensure
         client.close
-        Enviroblyd::Command.run(params)
+        command = nil
+        message = nil
+        client = nil
+        GC.start
       end
     end
   end

data/lib/enviroblyd/imds.rb

@@ -6,7 +6,8 @@ class Enviroblyd::IMDS
 
   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 }).
+      type: Net::HTTP::Put,
+      headers: { "X-aws-ec2-metadata-token-ttl-seconds" => TOKEN_TTL_SECONDS.to_s }).
       body.strip
   end
 

data/lib/enviroblyd/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: enviroblyd
 version: !ruby/object:Gem::Version
-  version: 0.4.4
+  version: 0.4.6
 platform: ruby
 authors:
 - Robert Starsi
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-05 00:00:00.000000000 Z
+date: 2024-09-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor