module FFI
  module HiredisVip
    class Set
      def initialize(client)
        @client = client
      end

      def psetex(key, value, expiry)
        set(key, value, :px => expiry)
      end

      def set(key, value, options = {})
        reply = nil
        command = "SET %b %b"
        value = value.to_s
        command_args = [ :pointer, key, :size_t, key.size, :pointer, value, :size_t, value.size ]

        if options[:ex]
          expiry = "#{options[:ex]}"
          command << " EX %b"
          command_args << :string << expiry << :size_t << expiry.size
        end

        if options[:px]
          px_expiry = "#{options[:px]}"
          command << " PX %b"
          command_args << :string << px_expiry << :size_t << px_expiry.size
        end
        
        command << " NX" if options[:nx]
        command << " XX" if options[:xx]

        synchronize do |connection|
          reply = @client.execute_command(connection, command, *command_args)
        end

        return nil if reply.nil? || reply.null?

        case reply[:type] 
        when :REDIS_REPLY_STRING
          reply[:str].dup
        when :REDIS_REPLY_STATUS
          reply[:str].dup
        when :REDIS_REPLY_NIL
          nil
        else
          ""
        end
      ensure
        ::FFI::HiredisVip::Core.freeReplyObject(reply.pointer) if reply
      end

      def setex(key, value, expiry)
        set(key, value, :ex => expiry)
      end

      def setnx(key, value)
        set(key, value, :nx => true)
      end

      private

      def synchronize
        @client.synchronize do |connection|
          yield(connection)
        end
      end

    end # class Set
  end # module HiredisVip
end # module FFI