module Celluloid
  module ZMQ
    class Socket
      extend Forwardable
      def_delegator ZMQ, :result_ok?
      # Create a new socket
      def initialize(type)
        @socket = Celluloid::ZMQ.context.socket ::ZMQ.const_get(type.to_s.upcase)
        @linger = 0
      end
      attr_reader :linger

      # Connect to the given 0MQ address
      # Address should be in the form: tcp://1.2.3.4:5678/
      def connect(addr)
        unless result_ok? @socket.connect addr
          fail IOError, "error connecting to #{addr}: #{::ZMQ::Util.error_string}"
        end
        true
      end

      def linger=(value)
        @linger = value || -1

        unless result_ok? @socket.setsockopt(::ZMQ::LINGER, value)
          fail IOError, "couldn't set linger: #{::ZMQ::Util.error_string}"
        end
      end

      def identity=(value)
        unless result_ok? @socket.setsockopt(::ZMQ::IDENTITY, "#{value}")
          fail IOError, "couldn't set identity: #{::ZMQ::Util.error_string}"
        end
        # de @socket.identity = value
      end

      def identity
        # de @socket.identity
        get(::ZMQ::IDENTITY)
      end

      def set(option, value, length = nil)
        unless result_ok? @socket.setsockopt(option, value, length)
          fail IOError, "couldn't set value for option #{option}: #{::ZMQ::Util.error_string}"
        end
      end

      def get(option)
        option_value = []

        unless result_ok? @socket.getsockopt(option, option_value)
          fail IOError, "couldn't get value for option #{option}: #{::ZMQ::Util.error_string}"
        end

        option_value[0]
      end

      # Bind to the given 0MQ address
      # Address should be in the form: tcp://1.2.3.4:5678/
      def bind(addr)
        unless result_ok? @socket.bind(addr)
          fail IOError, "couldn't bind to #{addr}: #{::ZMQ::Util.error_string}"
        end
      end

      # Close the socket
      def close
        @socket.close
      end

      # Hide ffi-rzmq internals
      alias_method :inspect, :to_s
    end
  end
end