process_bot 0.1.18 → 0.1.20

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 00371cf3c9ed3fa904d5d3a8655be99178ce77a729ab2aad459572fc63bb4779
-  data.tar.gz: db9145067629c7872c810c03e1e97caa6c626837ec26d07fa6ceeebf1a623bdd
+  metadata.gz: 667ddcc3026131cd08e43c08b0d21fa797b563dcc9ba577bb6c02feb77ed4d78
+  data.tar.gz: b37c21f69c8bf609f7889f994156bb18889f88e9ea53e00d84ad31d3bc791710
 SHA512:
-  metadata.gz: 89a0dde6cbadf510efb51b07f690ec90ca03af46ba06c9a1eca1f75a86e718bf23d263207aedd5a2763463e9470d9655d81487d5073f24a077edd1798acb2691
-  data.tar.gz: 27a95d5f8911f704c15faf58339c0173ca6cc01a3d7e8caa70743d6f3eadbbb9a44122b2ece7e74feaf4b7b6e7a03b1d7809e22cfa26efb4eb9967972f6846ff
+  metadata.gz: 4958033faa3e693c056fbf3e38f608bcbf98a6be923bd98cb895956c02d56ca423428835a64f5be9ec6c77d6a0f5a81d2498db2ee20ba2a52c8612a50a211c5f
+  data.tar.gz: f4428d73807c72cc8037a3d1e0631077f02684cb0bc0080b71b6499cd8dc7daeea514beb97a153b8618bd92ec475d208ee1f01d663b09bd783e29db316fecdca

data/AGENTS.md

@@ -13,3 +13,4 @@
 - Added `graceful_no_wait` command and Capistrano task for non-blocking graceful shutdowns.
 - Always add or update tests for new/changed functionality, and run them.
 - Added coverage for graceful_no_wait and Capistrano wait defaults.
+- Bumped version to 0.1.20 for log streaming updates.

data/CHANGELOG.md

@@ -1,5 +1,8 @@
 ## [Unreleased]
 - Stop accepting new control commands during shutdown so in-flight responses complete reliably.
+- Stream ProcessBot logs to connected control clients for Capistrano output.
+- Sanitize broadcast log output to keep JSON encoding safe.
+- Bump version to 0.1.20.
 
 ## [0.1.0] - 2022-04-03
 

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    process_bot (0.1.18)
+    process_bot (0.1.20)
       knjrbfw (>= 0.0.116)
       pry
       rake

data/lib/process_bot/client_socket.rb

@@ -1,3 +1,4 @@
+require "json"
 require "socket"
 
 class ProcessBot::ClientSocket
@@ -26,23 +27,39 @@ class ProcessBot::ClientSocket
     logger.logs "Sending: #{data}"
     begin
       client.puts(JSON.generate(data))
-      response_raw = client.gets
-    rescue Errno::ECONNRESET, Errno::EPIPE
-      return :nil
-    end
+      loop do
+        response_raw = client.gets
+        return :nil if response_raw.nil?
 
-    # Happens if process is interrupted
-    return :nil if response_raw.nil?
+        response = JSON.parse(response_raw)
 
-    response = JSON.parse(response_raw)
+        case response.fetch("type")
+        when "log"
+          write_log_output(response)
+        when "success"
+          return :success
+        when "error"
+          error = RuntimeError.new("Command raised an error: #{response.fetch("message")}")
+          error.set_backtrace(response.fetch("backtrace") + Thread.current.backtrace)
 
-    return :success if response.fetch("type") == "success"
+          raise error
+        else
+          raise "Unknown response type: #{response.fetch("type")}"
+        end
+      end
+    rescue Errno::ECONNRESET, Errno::EPIPE
+      :nil
+    end
+  end
 
-    if response.fetch("type") == "error"
-      error = RuntimeError.new("Command raised an error: #{response.fetch("message")}")
-      error.set_backtrace(response.fetch("backtrace") + Thread.current.backtrace)
+  def write_log_output(response)
+    output = response["output"].to_s
+    stream = response.fetch("stream", "stdout")
 
-      raise error
+    if stream == "stderr"
+      $stderr.print output
+    else
+      $stdout.print output
     end
   end
 end

data/lib/process_bot/control_socket.rb

@@ -3,12 +3,14 @@ require "json"
 require "knjrbfw"
 
 class ProcessBot::ControlSocket
-  attr_reader :options, :port, :process, :server
+  attr_reader :clients, :clients_mutex, :options, :port, :process, :server
 
   def initialize(options:, process:)
     @options = options
     @process = process
     @port = options.fetch(:port).to_i
+    @clients = []
+    @clients_mutex = Mutex.new
   end
 
   def logger
@@ -20,6 +22,9 @@ class ProcessBot::ControlSocket
     run_client_loop
     logger.logs "TCPServer started"
     options.events.call(:on_socket_opened, port: @port)
+    options.events.connect(:on_log) do |event_name, output:, type:|
+      broadcast_log(event_name, output: output, type: type)
+    end
   end
 
   def start_tcp_server
@@ -70,6 +75,8 @@ class ProcessBot::ControlSocket
   end
 
   def handle_client(client) # rubocop:disable Metrics/AbcSize
+    add_client(client)
+
     loop do
       data = client.gets
       break if data.nil? # Client disconnected
@@ -105,6 +112,42 @@ class ProcessBot::ControlSocket
         client.puts(JSON.generate(type: "error", message: "Unknown command: #{command_type}", backtrace: Thread.current.backtrace))
       end
     end
+  ensure
+    remove_client(client)
+    client.close unless client.closed?
+  end
+
+  def broadcast_log(_event_name, output:, type:)
+    safe_output = normalize_output(output)
+    payload = JSON.generate(type: "log", stream: type.to_s, output: safe_output)
+
+    clients_snapshot.each do |client|
+      client.puts(payload)
+    rescue IOError, Errno::EPIPE, Errno::ECONNRESET
+      remove_client(client)
+    end
+  end
+
+  def add_client(client)
+    clients_mutex.synchronize do
+      clients << client
+    end
+  end
+
+  def remove_client(client)
+    clients_mutex.synchronize do
+      clients.delete(client)
+    end
+  end
+
+  def clients_snapshot
+    clients_mutex.synchronize do
+      clients.dup
+    end
+  end
+
+  def normalize_output(output)
+    output.to_s.encode("UTF-8", invalid: :replace, undef: :replace, replace: "?")
   end
 
   def symbolize_keys(hash)

data/lib/process_bot/logger.rb

@@ -11,6 +11,7 @@ class ProcessBot::Logger
 
   def log(output, type: :stdout)
     write_output(output, type)
+    broadcast_log(output, type)
 
     return unless log_to_file?
     return if type == :debug && !debug_enabled?
@@ -54,6 +55,25 @@ class ProcessBot::Logger
     fp_log.flush
   end
 
+  def broadcast_log(output, type)
+    return unless should_broadcast?(type)
+
+    options.events.call(:on_log, output: output, type: type)
+  end
+
+  def should_broadcast?(type)
+    case type
+    when :stdout, :stderr
+      true
+    when :info
+      logging_enabled?
+    when :debug
+      debug_enabled?
+    else
+      false
+    end
+  end
+
   def debug_enabled?
     truthy_option?(:debug)
   end

data/lib/process_bot/options.rb

@@ -24,6 +24,7 @@ class ProcessBot::Options
       require "knjrbfw"
 
       event_handler = ::Knj::Event_handler.new
+      event_handler.add_event(name: :on_log)
       event_handler.add_event(name: :on_process_started)
       event_handler.add_event(name: :on_socket_opened)
       event_handler

data/lib/process_bot/version.rb

@@ -1,3 +1,3 @@
 module ProcessBot
-  VERSION = "0.1.18".freeze
+  VERSION = "0.1.20".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: process_bot
 version: !ruby/object:Gem::Version
-  version: 0.1.18
+  version: 0.1.20
 platform: ruby
 authors:
 - kaspernj
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-01-08 00:00:00.000000000 Z
+date: 2026-01-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: knjrbfw