_ = require('underscore')

apiClient = require('./api_client')

class Client
  constructor: (options) ->
    @api = new apiClient.APIClient(options)

  queue: (name) ->
    new Client(_.extend({}, @api.options, {queue_name: name}))

  create_queue: (queue_name, options, cb) ->
    @api.queuesCreate(queue_name, {queue: options}, (error, body) ->
      cb? cb(error, body)
    )

  update_queue: (queue_name, options, cb) ->
    @api.queuesUpdate(queue_name, {queue: options}, (error, body) ->
      cb? cb(error, body)
    )

  queues: (options, cb) ->
    @api.queuesList(options, (error, body) ->
      cb(error, body)
    )

  info: (cb) ->
    @api.queuesGet(@api.options.queue_name, (error, body) ->
      cb(error, body)
    )

  clear: (cb) ->
    @api.queuesClear(@api.options.queue_name, (error, body) ->
      cb? cb(error, body)
    )

  update: (options, cb) ->
    @api.queuesUpdate(@api.options.queue_name, {queue: options}, (error, body) ->
      cb? cb(error, body)
    )

  add_subscribers: (subscribers, cb) ->
    unless subscribers instanceof Array
      subscribers = [subscribers]

    @api.queuesAddSubscribers(
      @api.options.queue_name,
      { subscribers: subscribers },
    (error, body) ->
      cb? cb(error, body)
    )

  rm_subscribers: (subscribers, cb) ->
    unless subscribers instanceof Array
      subscribers = [subscribers]

    @api.queuesRemoveSubscribers(
      @api.options.queue_name,
      { subscribers: subscribers },
    (error, body) ->
      cb? cb(error, body)
    )

  rpl_subscribers: (subscribers, cb) ->
    unless subscribers instanceof Array
      subscribers = [subscribers]

    @api.queuesReplaceSubscribers(
      @api.options.queue_name,
      {subscribers: subscribers},
      (error, body) ->
        cb? cb(error, body)
    )

  del_queue: (cb) ->
    @api.queuesDelete(@api.options.queue_name, (error, body) ->
      cb? cb(error, body)
    )

  post: (messages, cb) ->
    unless messages instanceof Array
      messages = [messages]

    messages = _.map(messages, (message) ->
      if typeof(message) == 'string' then {body: message} else message
    )

    @api.messagesPost(@api.options.queue_name, messages, (error, body) ->
      if cb
        if not error?
          cb(error, if messages.length == 1 then body.ids[0] else body.ids)
        else
          cb(error, body)
    )

  get: (options, cb) ->
    @get_n(options, (error, body) ->
      if not error?
        cb(error, if (not options.n?) or options.n == 1 then body[0] else body)
      else
        cb(error, body)
    )

  reserve: (options, cb) ->
    @get_n(options, (error, body) ->
      if not error?
        cb(error, if (not options.n?) or options.n == 1 then body[0] else body)
      else
        cb(error, body)
    )


  get_n: (options, cb) ->
    @api.messagesGet(@api.options.queue_name, options, (error, body) ->
      if not error?
        cb(error, body.messages or [])
      else
        cb(error, body)
    )

  peek: (options, cb) ->
    @peek_n(options, (error, body) ->
      if not error?
        cb(error, if (not options.n?) or options.n == 1 then body[0] else body)
      else
        cb(error, body)
    )

  peek_n: (options, cb) ->
    @api.messagesPeek(@api.options.queue_name, options, (error, body) ->
      if not error?
        cb(error, body.messages)
      else
        cb(error, body)
    )

  del: (message_id, options, cb) ->
    @api.messagesDelete(@api.options.queue_name, message_id, options, (error, body) ->
      cb? cb(error, body)
    )

  del_multiple: (options, cb) ->
    ids = prepareIdsToRemove(options)
    @api.messagesMultipleDelete(@api.options.queue_name, ids, (error, body) ->
      cb? cb(error, body)
    )

  msg_get: (message_id, cb) ->
    @api.messagesGetById(@api.options.queue_name, message_id, (error, body) ->
      cb(error, body)
    )

  msg_touch: (message_id, reservation_id, options, cb) ->
    @api.messageTouch(@api.options.queue_name, message_id, reservation_id, options, (error, body) ->
      cb(error, body)
    )

  msg_release: (message_id, reservation_id, options, cb) ->
    @api.messageRelease(@api.options.queue_name, message_id, reservation_id, options, (error, body) ->
      cb? cb(error, body)
    )

  msg_push_statuses: (message_id, cb) ->
    @api.messagesPushStatuses(@api.options.queue_name, message_id, (error, body) ->
      cb(error, body)
    )

  del_msg_push_status: (message_id, reservation_id, subscriber_name, cb) ->
    @del message_id, {reservation_id: reservation_id, subscriber_name: subscriber_name}, cb

  prepareIdsToRemove = (options) ->
    body = {}
    if options["ids"]
      body["ids"] = _.map(options["ids"], (val) -> {id: val})
    else if options["reservation_ids"]
      body["ids"] = _.map(options["reservation_ids"], (val) -> {id: val.id, reservation_id: val.reservation_id})
    body

module.exports.Client = Client