fractor 0.1.3 → 0.1.6

data/examples/continuous_chat_server/chat_client.rb

@@ -0,0 +1,303 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "socket"
+require "json"
+require "fileutils"
+
+module ContinuousChat
+  # Simple Chat Client using Ruby's standard socket library
+  class ChatClient
+    def initialize(username, server_host = "localhost", server_port = 3000,
+log_file_path = nil)
+      @username = username
+      @server_host = server_host
+      @server_port = server_port
+      @running = true
+
+      # Set up logging
+      @log_file_path = log_file_path || "logs/client_#{username}_messages.log"
+      FileUtils.mkdir_p(File.dirname(@log_file_path))
+      @log_file = File.open(@log_file_path, "w")
+
+      log_message("Client initialized for #{username}, connecting to #{server_host}:#{server_port}")
+    end
+
+    def connect
+      puts "Connecting to server at #{@server_host}:#{@server_port}..."
+      log_message("Connecting to server at #{@server_host}:#{@server_port}")
+
+      begin
+        @socket = TCPSocket.new(@server_host, @server_port)
+
+        # Send join message
+        join_data = {
+          type: "join",
+          data: {
+            username: @username,
+          },
+          timestamp: Time.now.to_i,
+        }
+        @socket.puts(JSON.generate(join_data))
+        log_message("Sent join message: #{join_data}")
+
+        puts "Connected to chat server!"
+        log_message("Connected to chat server")
+
+        true
+      rescue StandardError => e
+        puts "Failed to connect: #{e.message}"
+        log_message("Failed to connect: #{e.message}")
+        false
+      end
+    end
+
+    def start
+      return false unless @socket
+
+      # Main event loop that handles both sending and receiving messages
+      # without creating separate threads
+      main_event_loop
+
+      true
+    end
+
+    # Main event loop that handles both user input and server messages
+    def main_event_loop
+      log_message("Starting main event loop")
+
+      # Print initial prompt
+      print "(#{@username})> "
+      $stdout.flush
+
+      while @running
+        # Use IO.select to wait for input from either STDIN or the socket
+        # This is non-blocking and allows us to handle both in a single loop
+        readable, = IO.select([@socket, $stdin], nil, nil, 0.1)
+
+        next unless readable # Nothing to process this iteration
+
+        readable.each do |io|
+          if io == $stdin
+            # Handle user input
+            handle_user_input
+          elsif io == @socket
+            # Handle server message
+            handle_server_message
+          end
+        end
+      end
+    rescue Interrupt
+      log_message("Client interrupted")
+      @running = false
+    rescue StandardError => e
+      log_message("Error in main event loop: #{e.message}")
+      @running = false
+    ensure
+      disconnect
+    end
+
+    # Handle user input (non-blocking)
+    def handle_user_input
+      text = $stdin.gets&.chomp
+      return unless text
+
+      # Check if client wants to quit
+      if text == "/quit" || text.nil?
+        @running = false
+        return
+      end
+
+      # Create message packet
+      message_data = {
+        type: "message",
+        data: {
+          content: text,
+          recipient: "all", # Default to broadcast
+        },
+        timestamp: Time.now.to_i,
+      }
+
+      # Send to server
+      @socket.puts(JSON.generate(message_data))
+      log_message("Sent message: #{text}")
+
+      # Print prompt for next input
+      print "(#{@username})> "
+      $stdout.flush
+    rescue StandardError => e
+      log_message("Error handling user input: #{e.message}")
+      @running = false
+    end
+
+    # Handle server message (non-blocking)
+    def handle_server_message
+      line = @socket.gets&.chomp
+
+      if line.nil?
+        # Server closed the connection
+        log_message("Connection to server lost")
+        @running = false
+        return
+      end
+
+      # Parse and handle the message
+      message = JSON.parse(line)
+      log_message("Received: #{line}")
+
+      # Display formatted message based on type
+      case message["type"]
+      when "broadcast"
+        puts "\r#{message['data']['from']}: #{message['data']['content']}"
+      when "direct_message"
+        puts "\r[DM] #{message['data']['from']}: #{message['data']['content']}"
+      when "server_message"
+        puts "\r[Server] #{message['data']['message']}"
+      when "user_list"
+        puts "\r[Server] Users online: #{message['data']['users'].join(', ')}"
+      when "error"
+        puts "\r[Error] #{message['data']['message']}"
+      end
+
+      # Reprint the prompt
+      print "(#{@username})> "
+      $stdout.flush
+    rescue StandardError => e
+      log_message("Error handling server message: #{e.message}")
+      # Don't immediately break for connection errors, may be temporary
+      # Just log and continue, IO.select will catch closed connections
+    end
+
+    def run_with_messages(messages, _delay_between_messages = 1)
+      return false unless @socket && @running
+
+      log_message("Running with #{messages.size} predefined messages")
+      puts "Sending #{messages.size} predefined messages"
+
+      # Send all messages in a non-blocking way
+      batch_send_messages(messages)
+
+      log_message("Finished sending all predefined messages")
+
+      # Start the event loop to receive responses
+      main_event_loop
+
+      true
+    end
+
+    # Helper to send a batch of messages without blocking
+    def batch_send_messages(messages)
+      messages.each_with_index do |msg, index|
+        content = msg[:content]
+        recipient = msg[:recipient] || "all"
+
+        log_message("Sending message #{index + 1}: '#{content}' to #{recipient}")
+
+        message_data = {
+          type: "message",
+          data: {
+            content: content,
+            recipient: recipient,
+          },
+          timestamp: Time.now.to_i,
+        }
+
+        @socket.puts(JSON.generate(message_data))
+        log_message("Sent message to #{recipient}: #{content}")
+
+        # Small delay between messages for stability
+        sleep(0.1)
+      end
+    end
+
+    def disconnect
+      return unless @running
+
+      @running = false
+      log_message("Disconnecting from server")
+
+      if @socket && !@socket.closed?
+        # Send leave message
+        leave_data = {
+          type: "leave",
+          data: {
+            username: @username,
+          },
+          timestamp: Time.now.to_i,
+        }
+        @socket.puts(JSON.generate(leave_data))
+        log_message("Sent leave message")
+
+        @socket.close
+      end
+
+      @log_file&.close
+
+      puts "Disconnected from server."
+      true
+    end
+
+    private
+
+    def log_message(message)
+      timestamp = Time.now.strftime("%Y-%m-%d %H:%M:%S.%L")
+      log_entry = "[#{timestamp}] #{message}"
+
+      @log_file.puts(log_entry)
+      @log_file.flush # Ensure it's written immediately
+    end
+  end
+end
+
+# When run directly, start the client
+if __FILE__ == $PROGRAM_NAME
+  require "fileutils"
+  require "json"
+
+  puts "Chat Client"
+  puts "==========="
+  puts "This is the chat client that connects to the chat server."
+  puts "All messages are logged to a file for later analysis."
+  puts
+
+  # Get username from command line or prompt
+  username = ARGV[0]
+
+  unless username
+    print "Enter your username: "
+    username = gets.chomp
+  end
+
+  # Get port from command line or use default
+  port = ARGV[1]&.to_i || 3000
+  log_file = ARGV[2] || "logs/client_#{username}_messages.log"
+
+  # Check for messages file
+  messages_file = "logs/client_#{username}_send_messages.json"
+
+  # Create and run the client
+  client = ContinuousChat::ChatClient.new(username, "localhost", port, log_file)
+
+  if client.connect
+    begin
+      # Load and send messages if the file exists
+      if File.exist?(messages_file)
+        puts "Loading messages from #{messages_file}"
+        messages = JSON.parse(File.read(messages_file), symbolize_names: true)
+        puts "Loaded #{messages.size} messages"
+
+        # Send the messages
+        client.run_with_messages(messages)
+      end
+
+      # Start the client
+      client.start
+    rescue Interrupt
+      puts "\nClient interrupted."
+    ensure
+      client.disconnect
+    end
+  end
+
+  puts "Chat client exited."
+end

data/examples/continuous_chat_server/chat_server.rb

@@ -0,0 +1,359 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "socket"
+require "json"
+require "time"
+require "fileutils"
+require_relative "../continuous_chat_common/message_protocol"
+
+# Simple Chat Server using Ruby's standard socket library
+# Based on the approach from https://dev.to/aurelieverrot/create-a-chat-in-the-command-line-with-ruby-2po9
+# but modified to use JSON for message passing.
+puts "Starting chat server..."
+
+# Parse command line args
+port = ARGV[0]&.to_i || 3000
+log_file_path = ARGV[1] || "logs/server_messages.log"
+
+# Create logs directory if it doesn't exist
+FileUtils.mkdir_p(File.dirname(log_file_path))
+log_file = File.open(log_file_path, "w")
+
+def log_message(message, log_file)
+  timestamp = Time.now.strftime("%Y-%m-%d %H:%M:%S.%L")
+  log_entry = "[#{timestamp}] #{message}"
+
+  # Check if log file is still open before writing
+  if log_file && !log_file.closed?
+    log_file.puts(log_entry)
+    log_file.flush # Ensure it's written immediately
+  end
+
+  # Also print to console for debugging
+  puts log_entry
+end
+
+# Create the server socket
+server = TCPServer.new("0.0.0.0", port)
+log_message("Server started on port #{port}", log_file)
+puts "Server bound to port #{port}"
+
+# Array to store connected clients
+clients = {}
+
+# Broadcast a message to all clients
+def announce_to_everyone(clients, message, log_file)
+  log_message("Broadcasting: #{message}", log_file)
+
+  # Convert to JSON if it's not already a string
+  message_json = message.is_a?(String) ? message : JSON.generate(message)
+
+  # Create a copy of the clients hash to avoid modification during iteration
+  clients_copy = clients.dup
+
+  clients_copy.each do |username, client|
+    client.puts(message_json)
+  rescue StandardError => e
+    log_message("Error broadcasting to #{username}: #{e.message}", log_file)
+    # We don't disconnect here, that's handled in the client thread
+  end
+end
+
+# Handle a new client joining the chat
+def handle_new_client(clients, client, log_file)
+  line = client.gets&.chomp
+  return client.close unless line
+
+  log_message("Received join message: #{line}", log_file)
+
+  # Parse the join message
+  message = JSON.parse(line)
+
+  if message["type"] == "join" && message["data"] && message["data"]["username"]
+    username = message["data"]["username"]
+
+    # Store the client with username as key
+    clients[username] = client
+
+    # Send welcome message to the client
+    welcome_msg = {
+      type: "server_message",
+      data: {
+        message: "Hello #{username}! Connected clients: #{clients.count}",
+      },
+      timestamp: Time.now.to_i,
+    }
+    client.puts(JSON.generate(welcome_msg))
+
+    # Broadcast to all clients that a new user joined
+    join_msg = {
+      type: "server_message",
+      data: {
+        message: "#{username} joined the chat!",
+      },
+      timestamp: Time.now.to_i,
+    }
+    announce_to_everyone(clients, join_msg, log_file)
+
+    # Broadcast the updated user list
+    user_list_msg = {
+      type: "user_list",
+      data: {
+        users: clients.keys,
+      },
+      timestamp: Time.now.to_i,
+    }
+    announce_to_everyone(clients, user_list_msg, log_file)
+
+    true
+  else
+    # Invalid join message
+    error_msg = {
+      type: "error",
+      data: {
+        message: "First message must be a valid join command",
+      },
+      timestamp: Time.now.to_i,
+    }
+    client.puts(JSON.generate(error_msg))
+    client.close
+    false
+  end
+rescue JSON::ParserError => e
+  log_message("Error parsing initial join message: #{e.message}", log_file)
+  client.puts(JSON.generate({
+                              type: "error",
+                              data: {
+                                message: "Invalid JSON format in join message",
+                              },
+                              timestamp: Time.now.to_i,
+                            }))
+  client.close
+  false
+rescue StandardError => e
+  log_message("Error handling new client: #{e.message}", log_file)
+  client.close
+  false
+end
+
+# Process a message from an existing client
+def process_client_message(clients, client, username, log_file)
+  # Read a message from the client (non-blocking with timeout)
+  readable, = IO.select([client], nil, nil, 0)
+  return true unless readable # No data to read yet
+
+  line = client.gets&.chomp
+  return false unless line # Client disconnected
+
+  begin
+    message = JSON.parse(line)
+    log_message("Received from #{username}: #{line}", log_file)
+
+    case message["type"]
+    when "message"
+      content = message["data"]["content"]
+      recipient = message["data"]["recipient"] || "all"
+
+      if content.start_with?("/")
+        # Handle commands
+        case content
+        when "/list"
+          list_msg = {
+            type: "user_list",
+            data: {
+              users: clients.keys,
+            },
+            timestamp: Time.now.to_i,
+          }
+          client.puts(JSON.generate(list_msg))
+        else
+          # Unknown command
+          client.puts(JSON.generate({
+                                      type: "error",
+                                      data: {
+                                        message: "Unknown command: #{content}",
+                                      },
+                                      timestamp: Time.now.to_i,
+                                    }))
+        end
+      elsif recipient == "all"
+        # Broadcast message
+        broadcast_msg = {
+          type: "broadcast",
+          data: {
+            from: username,
+            content: content,
+          },
+          timestamp: Time.now.to_i,
+        }
+        announce_to_everyone(clients, broadcast_msg, log_file)
+      elsif clients[recipient]
+        # Direct message
+        dm_msg = {
+          type: "direct_message",
+          data: {
+            from: username,
+            content: content,
+          },
+          timestamp: Time.now.to_i,
+        }
+        clients[recipient].puts(JSON.generate(dm_msg))
+        # Also send to sender if not the same person
+        client.puts(JSON.generate(dm_msg)) if username != recipient
+      else
+        # Recipient not found
+        client.puts(JSON.generate({
+                                    type: "error",
+                                    data: {
+                                      message: "User #{recipient} not found",
+                                    },
+                                    timestamp: Time.now.to_i,
+                                  }))
+      end
+    when "leave"
+      # Client wants to leave
+      return false
+    end
+  rescue JSON::ParserError => e
+    log_message("Error parsing JSON from #{username}: #{e.message}", log_file)
+    client.puts(JSON.generate({
+                                type: "error",
+                                data: {
+                                  message: "Invalid JSON format",
+                                },
+                                timestamp: Time.now.to_i,
+                              }))
+  rescue StandardError => e
+    log_message("Error processing message from #{username}: #{e.message}",
+                log_file)
+    return false
+  end
+
+  true # Client still connected
+end
+
+# Handle client disconnection
+def handle_client_disconnect(clients, client, log_file)
+  username = clients.key(client)
+  return unless username
+
+  # Remove from clients list
+  clients.delete(username)
+  log_message("Client disconnected: #{username}", log_file)
+
+  # Notify everyone
+  leave_msg = {
+    type: "server_message",
+    data: {
+      message: "#{username} left the chat.",
+    },
+    timestamp: Time.now.to_i,
+  }
+  announce_to_everyone(clients, leave_msg, log_file)
+
+  # Update user list
+  user_list_msg = {
+    type: "user_list",
+    data: {
+      users: clients.keys,
+    },
+    timestamp: Time.now.to_i,
+  }
+  announce_to_everyone(clients, user_list_msg, log_file)
+
+  # Close the client socket
+  begin
+    client.close
+  rescue StandardError
+    nil
+  end
+end
+
+# Main server loop
+begin
+  log_message("Server ready to accept connections", log_file)
+
+  # Add the server socket to the list of sockets to monitor
+  sockets = [server]
+
+  # Main event loop
+  loop do
+    # Use IO.select to check which sockets have data to read
+    # This is non-blocking and allows us to handle multiple clients sequentially
+    readable, _, errored = IO.select(sockets, [], sockets, 0.1)
+
+    # Handle errors first
+    if errored && !errored.empty?
+      errored.each do |socket|
+        raise "Server socket error" if socket == server
+
+        # Server socket error - critical
+
+        # Client socket error
+        username = clients.key(socket)
+        log_message("Error on client socket: #{username || 'unknown'}",
+                    log_file)
+        handle_client_disconnect(clients, socket, log_file)
+        sockets.delete(socket)
+      end
+    end
+
+    # Nothing to process this iteration
+    next unless readable
+
+    # Process readable sockets
+    readable.each do |socket|
+      if socket == server
+        # New client connection
+        begin
+          client = server.accept
+          log_message(
+            "New client connection from #{client.peeraddr[2]}:#{client.peeraddr[1]}", log_file
+          )
+
+          # Add the client socket to our monitoring list
+          sockets << client
+
+          # We'll process the initial join message in the next iteration
+        rescue StandardError => e
+          log_message("Error accepting client: #{e.message}", log_file)
+        end
+      elsif clients.key(socket)
+        # Existing client sent data
+        username = clients.key(socket)
+
+        # Process message, remove client if it disconnected
+        unless process_client_message(clients, socket, username, log_file)
+          handle_client_disconnect(clients, socket, log_file)
+          sockets.delete(socket)
+        end
+      else
+        # This is a new client that needs to send their join message
+        unless handle_new_client(clients, socket, log_file)
+          # Join failed, remove from sockets
+          sockets.delete(socket)
+        end
+      end
+    end
+  end
+rescue Interrupt
+  log_message("Server interrupted, shutting down...", log_file)
+rescue StandardError => e
+  log_message("Server error: #{e.message}", log_file)
+ensure
+  # Close all client connections
+  clients.each_value do |client|
+    client.close
+  rescue StandardError
+    # Ignore errors when closing
+  end
+
+  # Close the server socket
+  server&.close
+
+  # Close the log file
+  log_file&.close
+
+  log_message("Server stopped", log_file)
+end