index.js

var request = require('request')
  , Parser = require('./lib/parser')
  , split = require('split')
  , multiparty = require('multiparty')
  , ss = require('stream-stream')
  , through = require('through2')
  , querystring = require('querystring')
  , api = 'https://www.googleapis.com'

var Gmail = function (key) {
  if (!key) {
    throw new Error('Access key required')
  }
  this.key = key
}

var retrieveCount = function (key, q, endpoint, opts, next) {
  request({
    url: api + '/gmail/v1/users/me/' + endpoint,
    json: true,
    timeout: opts.timeout,
    qs: {
      q: q,
      fields: 'resultSizeEstimate'
    },
    headers: {
      'Authorization': 'Bearer ' + key
    }
  }, function (err, response, body) {
    if (err) {
      return next(err)
    }
    if (body.error) {
      return next(new Error(body.error.message))
    }
    return next(null, body.resultSizeEstimate)
  })
}

var formQuery = function (query) {
  var formedQuery = '?'
    , params = {}

  Object.keys(query).forEach(function(key) {
    if(query[key]) {
      params[key] = query[key] instanceof Array ?  query[key].join(',') : query[key]
    }
  })

  formedQuery += querystring.stringify(params)

  return formedQuery
}

var retrieve = function (key, q, endpoint, opts) {
  var result = new Parser({objectMode: true})
    , combined = ss()
    , opts = opts || {}
    , i = opts.max
    , partsFound = 0

  var loop = function(page) {
    var reqOpts = {
      url: api + '/gmail/v1/users/me/' + endpoint,
      json: true,
      timeout: opts.timeout,
      qs: {
        q: q,
        maxResults: i < 100 ? i : 100
      },
      headers: {
        'Authorization': 'Bearer ' + key
      }
    }

    var query = formQuery({fields : opts.fields, format : opts.format})

    if (page) reqOpts.qs.pageToken = page
    request(reqOpts, function (err, response, body) {
      if (err) {
        return result.emit('error', err)
      }

      if (body.error) {
        return result.emit('error', new Error(body.error.message))
      }

      result.resultSizeEstimate = body.resultSizeEstimate

      if (!result.resultSizeEstimate) {
        return result.end()
      }
      var messages = body[endpoint].map(function (m) {
        return {
          'Content-Type': 'application/http',
          body: 'GET ' + api + '/gmail/v1/users/me/' + endpoint + '/' + m.id + query + '\n'
        }
      })

      var r = request({
        method: 'POST',
        url: api + '/batch/gmail/v1',
        multipart: messages,
        timeout: opts.timeout,
        headers: {
          'Authorization': 'Bearer ' + key,
          'content-type': 'multipart/mixed'
        }
      })

      r.on('error', function (e) {
        result.emit('error', e)
      })

      r.on('response', function (res) {
        var type = res.headers['content-type']
          , form = new multiparty.Form

        res.headers['content-type'] = type.replace('multipart/mixed', 'multipart/related')
        form.on('part', function (part) {
          partsFound++;
          combined.write(part.pipe(split('\r\n')).pipe(new Parser))
        }).parse(res)
        form.on('close', function () {
          if (opts.max !== partsFound && body.nextPageToken) return loop(body.nextPageToken)
          combined.end()
        })

      })
    })
  }
  loop()

  var counter = through(function(obj, enc, cb) {
    i--
    cb(null, obj)
  })

  return combined.pipe(counter).pipe(result)
}

/*
 * Fetches the number of estimated messages matching the query
 * Invokes callback with err and estimated number
 */
Gmail.prototype.estimatedMessages = function (q, opts, next) {
  if (!next) {
    next = opts
    opts = {}
  }
  return retrieveCount(this.key, q, 'messages', opts, next)
}

/*
 * Fetches email that matches the query. Returns a stream of messages with a max of 100 messages
 * since the batch api sets a limit of 100.
 *
 * e.g. to search an inbox: gmail.messages('label:inbox')
 */
Gmail.prototype.messages = function (q, opts) {
  return retrieve(this.key, q, 'messages', opts)
}

/*
 * Fetches the number of estimated threads matching the query
 * Invokes callback with err and estimated number
 */
Gmail.prototype.estimatedThreads = function (q, opts, next) {
  if (!next) {
    next = opts
    opts = {}
  }
  return retrieveCount(this.key, q, 'threads', opts, next)
}

Gmail.prototype.threads = function (q, opts) {
  return retrieve(this.key, q, 'threads', opts)
}

module.exports = Gmail