# -*- encoding: utf-8 -*-

require 'socket'
require 'timeout'

require 'coolio'

module EventedBluepill
  module Socket
    class CmdListenerConnection < Coolio::UNIXSocket
      def initialize(socket, application)
        @application = application
        super(socket)
      end

      def on_read(data)
        command, *args = data.strip.split(":")
        response = begin
          @application.send(command, *args)
        rescue Exception => e
          e
        end
        write(Marshal.dump(response))
      end
    end

    TIMEOUT = 10

    extend self

    def client(base_dir, name, &b)
      UNIXSocket.open(socket_path(base_dir, name), &b)
    end

    def client_command(base_dir, name, command)
      client(base_dir, name) do |socket|
        Timeout.timeout(TIMEOUT) do
          socket.puts command
          Marshal.load(socket)
        end
      end
    rescue EOFError, Timeout::Error
      abort("Socket Timeout: Server may not be responding")
    end

    def server(base_dir, name, application)
      socket_path = self.socket_path(base_dir, name)
      begin
        Coolio::UNIXServer.new(socket_path, CmdListenerConnection, application)
      rescue Errno::EADDRINUSE
        # if sock file has been created. test to see if there is a server
        begin
          UNIXSocket.open(socket_path)
        rescue Errno::ECONNREFUSED
          File.delete(socket_path)
          Coolio::UNIXServer.new(socket_path, CmdListenerConnection, application)
        else
          logger.err("Server is already running!")
          exit(7)
        end
      end
    end

    def socket_path(base_dir, name)
      File.join(base_dir, 'socks', name + ".sock")
    end
  end
end