spur-common/src/http/HTTPService.coffee

A Node.JS library of common modules used as a base to most Node.JS applications.

module.exports = (superagent, Promise, _, FormData, HTTPResponseProcessing)->
  Request = superagent.Request

  superagent.globalPlugins = []

  Request::plugin = (plugin)->
    (@_plugins ?= []).push(plugin)
    @

  Request::named = (@name)-> @

  Request::tagged = (@tags = {})-> @

  Request::getDefaultName = ->
    return @url.match(/\/\/(.+)\/?/)[1].replace(/\./g, "_")

  Request::promise = ->
    self = this
    self.name ?= @getDefaultName()
    self.tags ?= {}
    @_plugins = (@_plugins or []).concat(superagent.globalPlugins)

    @_pluginInstances = _.compact _.map @_plugins, (p)->
      p.start(self)

    return new Promise (resolve, reject)=>

      Request::end.call self, (err, res)=>
        self.response = res

        parser = HTTPResponseProcessing.create(@method, @url, res, err).process()
        self.error = parser.error

        if self.error
          reject(self.error)
        else
          resolve(res)

        _.invoke(self._pluginInstances, "end")

  Request::appendFile = (name, contents, params)->
    @_formData ?= new FormData()
    @_formData.append(name, contents, params)
    @

  Request::promiseBody = -> @promise().get("body")
  Request::promiseText = -> @promise().get("text")

  Request::parseBinary = ->
    @parse((res, fn)->
      res.data = ''
      res.setEncoding("binary")
      res.on 'data', (chunk)->
        res.data += chunk.toString("binary")
      res.on 'end', ->
        fn null, new Buffer(res.data, 'binary')
    ).buffer(true)

  Request::setFields = (fields)->
    for k,v of fields
      @field(k,v)
    @

  superagent.setGlobalPlugins = (plugins)->
    if _.isArray(plugins)
      superagent.globalPlugins = _.unique(_.union(superagent.globalPlugins, plugins))
    else
      superagent.addGlobalPlugin(plugins)

  superagent.getGlobalPlugins = ()->
    superagent.globalPlugins

  superagent.addGlobalPlugin = (plugin)->
    superagent.globalPlugins.push(plugin) unless _.contains(superagent.globalPlugins, plugin)

  superagent