120 lines
3.9 KiB
JavaScript
120 lines
3.9 KiB
JavaScript
var winston = require('winston');
|
|
|
|
// For handling serving stored documents
|
|
|
|
var 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(key, callback, response, skipExpire) {
|
|
this.store.get(key, function(ret) {
|
|
if (ret) {
|
|
winston.verbose('retrieved document', { key: key });
|
|
var responseData = JSON.stringify({ data: ret, key: key });
|
|
if (callback) {
|
|
if (callback.match(/^[a-z0-9]+$/i)) {
|
|
response.writeHead(200, { 'content-type': 'application/javascript' });
|
|
response.end(callback + '(' + responseData + ');');
|
|
} else {
|
|
response.writeHead(400, { 'content-type': 'application/json' });
|
|
response.end(JSON.stringify({ message: 'invalid callback function name' }));
|
|
}
|
|
} else {
|
|
response.writeHead(200, { 'content-type': 'application/json' });
|
|
response.end(responseData);
|
|
}
|
|
}
|
|
else {
|
|
winston.warn('document not found', { key: key });
|
|
response.writeHead(404, { 'content-type': 'application/json' });
|
|
response.end(JSON.stringify({ message: 'Document not found.' }));
|
|
}
|
|
}, skipExpire);
|
|
};
|
|
|
|
// Handle retrieving the raw version of a document
|
|
DocumentHandler.prototype.handleRawGet = function(key, response, skipExpire) {
|
|
this.store.get(key, function(ret) {
|
|
if (ret) {
|
|
winston.verbose('retrieved raw document', { key: key });
|
|
response.writeHead(200, { 'content-type': 'text/plain' });
|
|
response.end(ret);
|
|
}
|
|
else {
|
|
winston.warn('raw document not found', { key: key });
|
|
response.writeHead(404, { 'content-type': 'application/json' });
|
|
response.end(JSON.stringify({ message: 'Document not found.' }));
|
|
}
|
|
}, skipExpire);
|
|
};
|
|
|
|
// Handle adding a new Document
|
|
DocumentHandler.prototype.handlePost = function(request, response) {
|
|
var _this = this;
|
|
var buffer = '';
|
|
var cancelled = false;
|
|
request.on('data', function(data) {
|
|
if (!buffer) {
|
|
response.writeHead(200, { 'content-type': 'application/json' });
|
|
}
|
|
buffer += data.toString();
|
|
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.' })
|
|
);
|
|
}
|
|
});
|
|
request.on('end', function(end) {
|
|
if (cancelled) return;
|
|
_this.chooseKey(function(key) {
|
|
_this.store.set(key, buffer, function(res) {
|
|
if (res) {
|
|
winston.verbose('added document', { key: key });
|
|
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.' }));
|
|
}
|
|
});
|
|
});
|
|
});
|
|
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.' }));
|
|
});
|
|
};
|
|
|
|
// Keep choosing keys until one isn't taken
|
|
DocumentHandler.prototype.chooseKey = function(callback) {
|
|
var key = this.acceptableKey();
|
|
var _this = this;
|
|
this.store.get(key, function(ret) {
|
|
if (ret) {
|
|
_this.chooseKey(callback);
|
|
} else {
|
|
callback(key);
|
|
}
|
|
});
|
|
};
|
|
|
|
DocumentHandler.prototype.acceptableKey = function() {
|
|
return this.keyGenerator.createKey(this.keyLength);
|
|
};
|
|
|
|
module.exports = DocumentHandler;
|