Contents

require 'celluloid/current'
require 'celluloid/io'

class Eye::Server

  include Celluloid::IO

  attr_reader :socket_path, :server

  def initialize(socket_path)
    @socket_path = socket_path
    @server = begin
      UNIXServer.open(socket_path)
    rescue Errno::EADDRINUSE
      unlink_socket_file
      UNIXServer.open(socket_path)
    end
  end

  def run
    loop { async.handle_connection @server.accept }
  end

  def handle_connection(socket)
    text = socket.read

    begin
      # TODO, remove in 1.0

      payload = Marshal.load(text)
      raise "unknown payload #{payload.inspect}" unless payload.is_a?(Array)
      cmd, *args = payload

    rescue
      # new format
      begin
        sign, msg_size = text[0...8].unpack('N*')
        raise "unknown protocol #{sign}" unless sign == Eye::Client::SIGN
        content = text[8..-1]
        content << socket.read(msg_size - content.length) while content.length < msg_size
        payload = Marshal.load(content)
        cmd = payload[:command]
        args = payload[:args]

      rescue => ex
        error "Failed to read from socket: #{ex.message}"
        return
      end
    end

    response = Eye::Control.command(cmd, *args, {})
    socket.write(Marshal.dump(response))

  rescue Errno::EPIPE
    # client timeouted
    # do nothing

  ensure
    socket.close
  end

  def unlink_socket_file
    File.delete(@socket_path) if @socket_path
  rescue
  end

  finalizer :close_socket

  def close_socket
    @server.close if @server
    unlink_socket_file
  end

end

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
eye-0.10.1.pre	lib/eye/server.rb
eye-0.10.0	lib/eye/server.rb
eye-0.10.0.alpha2	lib/eye/server.rb
eye-0.10.0.alpha	lib/eye/server.rb
eye-0.9.4.nosigar	lib/eye/server.rb
eye-0.9.4	lib/eye/server.rb
eye-0.9.3	lib/eye/server.rb
eye-0.9.2.nosigar	lib/eye/server.rb
eye-0.9.2	lib/eye/server.rb