ChatServer.rb

#!/usr/bin/env ruby
require 'gserver'
class ChatServer < GServer
  def initialize(*args)
    super(*args)
    # Keep an overall record of the client IDs allocated
    # and the lines of chat
    @@client_id = 0
    @@chat = []
  end
  
  def serve(io)
    # Increment the client ID so each client gets a unique ID
    @@client_id += 1
    my_client_id = @@client_id
    my_position = @@chat.size
    
    io.puts("Welcome to the chat, client #{@@client_id}")
    
    # Leave a message on the chat queue to signify this client
    # has joined the chat
    @@chat << [my_client_id, "<joins the chat>"]
    
    loop do
      # Every 5 seconds check to see if we are receiving any data
      if IO.select([io], nil, nil, 0)
        # If so, retrieve the data and process it...
        line = io.gets
        
        # If the user say 'quit', disconnect them
        if line =~ /quit/
          @@chat << [my_client_id, "<leave the chat>"]
          #puts "#{@@chat[size - 1][0]} says: #{@@chat[size - 1][1]}"
          puts "#{@@chat[@@chat.size - 1][0]} said: #{@@chat[@@chat.size - 1][1]}"
          break
        end
        
        # Shut down the server if we hear 'shutdown'
        if line =~ /shutdown/
          puts "Server has been shutdown!"
          self.stop
        end
        
        # Add the client's text to the chat array along with the
        # client's ID
        @@chat << [my_client_id, line]
      else
        # No data, so print any new lines from the chat stream
        @@chat[my_position..(@@chat.size - 1)].each_with_index do |line, index|
          io.puts("#{line[0]} says: #{line[1]}")
        end
        
        # Move the position to one past the end of the array
        my_position = @@chat.size
      end
    end
    puts "Serve Method is stopped!"
  end
end

server = ChatServer.new(1234)
server.start

loop do
  break if server.stopped?
end