hastebin-ansi/lib/document_handler.js

const winston = require('winston');
const Busboy = require('busboy');

// For handling serving stored documents

const DocumentHandler = function (options) {
    if (!options) {
        options = {};
    }
    this.keyLength = options.keyLength || DocumentHandler.defaultKeyLength;
    this.maxLength = options.maxLength; // none by default
    this.store = options.store;
    this.keyGenerator = options.keyGenerator;
};

DocumentHandler.defaultKeyLength = 10;

// Handle retrieving a document
DocumentHandler.prototype.handleGet = function (request, response, config) {
    const key = request.params.id.split('.')[0];
    const skipExpire = !!config.documents[key];

    this.store.get(key, function (ret) {
        if (ret) {
            winston.verbose('retrieved document', {key: key});
            response.writeHead(200, {'content-type': 'application/json'});
            if (request.method === 'HEAD') {
                response.end();
            } else {
                response.end(JSON.stringify({data: ret, key: key}));
            }
        } else {
            winston.warn('document not found', {key: key});
            response.writeHead(404, {'content-type': 'application/json'});
            if (request.method === 'HEAD') {
                response.end();
            } else {
                response.end(JSON.stringify({message: 'Document not found.'}));
            }
        }
    }, skipExpire);
};

// Handle retrieving the raw version of a document
DocumentHandler.prototype.handleRawGet = function (request, response, config) {
    const key = request.params.id.split('.')[0];
    const skipExpire = !!config.documents[key];

    this.store.get(key, function (ret) {
        if (ret) {
            winston.verbose('retrieved raw document', {key: key});
            response.writeHead(200, {'content-type': 'text/plain; charset=UTF-8'});
            if (request.method === 'HEAD') {
                response.end();
            } else {
                response.end(ret);
            }
        } else {
            winston.warn('raw document not found', {key: key});
            response.writeHead(404, {'content-type': 'application/json'});
            if (request.method === 'HEAD') {
                response.end();
            } else {
                response.end(JSON.stringify({message: 'Document not found.'}));
            }
        }
    }, skipExpire);
};

// Handle adding a new Document
DocumentHandler.prototype.handlePost = function (request, response) {
    const _this = this;
    let buffer = '';
    let cancelled = false;

    // What to do when done
    const onSuccess = function () {
        // Check length
        if (_this.maxLength && buffer.length > _this.maxLength) {
            cancelled = true;
            winston.warn('document >maxLength', {maxLength: _this.maxLength});
            response.writeHead(400, {'content-type': 'application/json'});
            response.end(JSON.stringify({message: 'Document exceeds maximum length.'}));
            return;
        }
        // And then save if we should
        _this.chooseKey(function (key) {
            _this.store.set(key, buffer, function (res) {
                if (res) {
                    winston.verbose('added document', {key: key});
                    response.writeHead(200, {'content-type': 'application/json'});
                    response.end(JSON.stringify({key: key}));
                } else {
                    winston.verbose('error adding document');
                    response.writeHead(500, {'content-type': 'application/json'});
                    response.end(JSON.stringify({message: 'Error adding document.'}));
                }
            });
        });
    };

    // If we should, parse a form to grab the data
    const ct = request.headers['content-type'];
    if (ct && ct.split(';')[0] === 'multipart/form-data') {
        const busboy = new Busboy({headers: request.headers});
        busboy.on('field', function (field_name, val) {
            if (field_name === 'data') {
                buffer = val;
            }
        });
        busboy.on('finish', function () {
            onSuccess();
        });
        request.pipe(busboy);
        // Otherwise, use our own and just grab flat data from POST body
    } else {
        request.on('data', function (data) {
            buffer += data.toString();
        });
        request.on('end', function () {
            if (cancelled) {
                return;
            }
            onSuccess();
        });
        request.on('error', function (error) {
            winston.error('connection error: ' + error.message);
            response.writeHead(500, {'content-type': 'application/json'});
            response.end(JSON.stringify({message: 'Connection error.'}));
            cancelled = true;
        });
    }
};

// Keep choosing keys until one isn't taken
DocumentHandler.prototype.chooseKey = function (callback) {
    const key = this.acceptableKey();
    const _this = this;
    this.store.get(key, function (ret) {
        if (ret) {
            _this.chooseKey(callback);
        } else {
            callback(key);
        }
    }, true); // Don't bump expirations when key searching
};

DocumentHandler.prototype.acceptableKey = function () {
    return this.keyGenerator.createKey(this.keyLength);
};

module.exports = DocumentHandler;