WebSocket::Driver client example with Celluloid::IO

wsd.rb

require 'websocket/driver'
require 'socket'
require 'forwardable'
require 'celluloid/io'

class WSDTester
  include Celluloid::IO
  include Celluloid::Logger

  extend Forwardable
  def_delegator :@socket, :write
  def_delegators :@driver, :text, :binary, :close

  attr_reader :driver, :socket

  def initialize addr, port, path
    @addr, @port, @path = addr, port, path
    ip = @addr
    ip = Socket.getaddrinfo(@addr, 'http')[0][3] unless @addr =~ /\d+\.\d+\.\d+\.\d+/
    @socket = TCPSocket.new ip, @port
    @driver = WebSocket::Driver.client self

    @driver.on :open do |e|
      debug "open!"
    end

    @driver.on :message do |e|
      debug "message: #{e.data}"
    end

    @driver.on :close do |e|
      debug "close!"
    end
  end

  def url
    "ws://%s:%d%s" % [@addr, @port, @path]
  end

  # call this with async
  #
  def go
    @driver.start
    # async.ping
    while msg = @socket.readpartial(4096)
      @driver.parse msg
    end
  end

  # def ping
  #   every(30){ @driver.ping }
  # end

end

__END__

wsd = WSDTester.new '127.0.0.1', 4567, '/'
wsd.async.go
wsd.text 'hi there'

sleep 1