require "norikra/client/version"

require 'msgpack-rpc-over-http-jruby'

module Norikra
  module RPC
    class ClientError < MessagePack::RPCOverHTTP::RemoteError; end
    class ServerError < MessagePack::RPCOverHTTP::RemoteError; end
  end

  class Client
    RPC_DEFAULT_PORT = 26571
    TIMEOUT_OPTIONS = [:connect_timeout, :send_timeout, :receive_timeout]

    def initialize(host='localhost', port=RPC_DEFAULT_PORT, opts={})
      @client = MessagePack::RPCOverHTTP::Client.new("http://#{host}:#{port}/", encoding: Encoding::UTF_8)

      @client.connect_timeout = opts[:connect_timeout] if opts.has_key?(:connect_timeout) && @client.respond_to?('connect_timeout='.to_sym)
      @client.send_timeout    = opts[:send_timeout]    if opts.has_key?(:send_timeout)    && @client.respond_to?('send_timeout='.to_sym)
      @client.receive_timeout = opts[:receive_timeout] if opts.has_key?(:receive_timeout) && @client.respond_to?('receive_timeout='.to_sym)
    end

    def targets
      @client.call(:targets) #=> {:name => "name", :auto_field => true}
    end

    def open(target, fields=nil, auto_field=true)
      @client.call(:open, target, fields, auto_field)
    end

    def close(target)
      @client.call(:close, target)
    end

    def modify(target, auto_field)
      @client.call(:modify, target, auto_field)
    end

    def queries
      @client.call(:queries)
    end

    def register(query_name, query_group, query_expression)
      @client.call(:register, query_name, query_group, query_expression)
    end

    def deregister(query_name)
      @client.call(:deregister, query_name)
    end

    def fields(target)
      @client.call(:fields, target)
    end

    def reserve(target, field, type)
      @client.call(:reserve, target, field, type)
    end

    def send(target, events)
      @client.call(:send, target, events)
    end

    # [ [time, event], ... ]
    def event(query_name)
      @client.call(:event, query_name)
    end

    # [ [time, event], ... ]
    def see(query_name)
      @client.call(:see, query_name)
    end

    # {'query_name' => [ [time, event], ... ]}
    def sweep(query_group=nil)
      @client.call(:sweep, query_group)
    end

    def logs
      @client.call(:logs)
    end
  end
end