class Sock

# File lib/sock.rb, line 85
def self.connect(host, port, debug = false)
  s = Sock.new(host, port, debug)
  yield s
  s.close
end

# File lib/sock.rb, line 10
def initialize(host, port, debug = false)
  @sock = TCPSocket.new host, port
  @debug = debug
end

# File lib/sock.rb, line 79
def close
  @sock.close
end

# File lib/sock.rb, line 50
def go_interactive
  loop do
    return unless write_out
    return unless read_in
    IO.select([@sock, STDIN], [], [@sock, STDIN])
  end
end

# File lib/sock.rb, line 32
def read(nbytes)
  msg = @sock.recv nbytes
  print msg.cyan if @debug
  msg
end

# File lib/sock.rb, line 38
def readline
  msg = @sock.readline
  print msg.cyan if @debug
  msg
end

# File lib/sock.rb, line 17
def write(str)
  print str.magenta if @debug
  @sock.write(str)
end

# File lib/sock.rb, line 22
def writeline(str)
  send(str + "\n")
end

# File lib/sock.rb, line 69
def read_in(data = nil)
  @sock.write(data) while (data = STDIN.read_nonblock(100)) != ''
  return true
rescue Errno::EAGAIN
  return true
rescue EOFError
  return false
end

# File lib/sock.rb, line 58
def write_out(data = nil)
  STDOUT.write(data) while (data = @sock.read_nonblock(100)) != ''
  return false
rescue Errno::EAGAIN
  return true
rescue EOFError
  @sock.close
  return false
end