screenshot-capture/lib/transporter.js

Free URL to Screenshot API - capture crisp and clean snapshots of any website, with any device, any setting and lightning fast. Free for personal use.

'use strict';

var request = require('request');
var _ = require('lodash');

var utils = require('./utils');
var pkg = require('../package.json');


/**
 * Default transporter constructor.
 * Wraps request and callback functions.
 */
function Transporter(options) {
    this.options = options;
}

/**
 * Default user agent.
 */
Transporter.prototype.USER_AGENT = pkg.name + '.' + pkg.version;

/**
 * Configures request options before making a request.
 * @param {object} opts Options to configure.
 * @return {object} Configured options.
 */
Transporter.prototype.configure = function (args) {

    // set transporter user agent
    var headers = this.options.headers || {};
    headers['User-Agent'] = this.USER_AGENT;

    args = utils.extend(
        this.options,
        args,
        {headers: headers}
    );

    return args;
};

/**
 * Makes a request with given options and invokes callback.
 * @param {object} opts Options.
 * @param {Function=} opt_callback Optional callback.
 * @return {Request} Request object
 */
Transporter.prototype.request = function (args, callback) {

    args = this.configure(args);

    var req = request(args, function (err, response, body) {
        if (err) {
            return callback(err);
        }
        callback(null, body, response)
    });
    return req;
};

/**
 * Exports Transporter.
 */
module.exports = Transporter;