haste-server/lib/document_handler.js

112 lines
3.6 KiB
JavaScript
Raw Normal View History

2011-11-18 21:51:38 +01:00
var winston = require('winston');
// For handling serving stored documents
var DocumentHandler = function(options) {
2011-11-28 22:46:59 +01:00
if (!options) {
options = {};
2011-11-18 22:57:23 +01:00
}
2011-11-28 22:46:59 +01:00
this.keyLength = options.keyLength || DocumentHandler.defaultKeyLength;
this.maxLength = options.maxLength; // none by default
this.store = options.store;
2011-11-18 21:51:38 +01:00
};
2011-11-28 22:46:59 +01:00
DocumentHandler.defaultKeyLength = 10;
2011-11-18 22:45:48 +01:00
// Handle retrieving a document
2011-11-28 07:13:14 +01:00
DocumentHandler.prototype.handleGet = function(key, response, skipExpire) {
2011-11-18 23:54:57 +01:00
this.store.get(key, function(ret) {
2011-11-18 23:12:28 +01:00
if (ret) {
winston.verbose('retrieved document', { key: key });
response.writeHead(200, { 'content-type': 'application/json' });
response.end(JSON.stringify({ data: ret, key: key }));
}
else {
winston.warn('document not found', { key: key });
response.writeHead(404, { 'content-type': 'application/json' });
2011-12-19 18:44:12 +01:00
response.end(JSON.stringify({ message: 'Document not found.' }));
2011-11-18 23:12:28 +01:00
}
2011-11-28 07:13:14 +01:00
}, skipExpire);
2011-11-18 21:51:38 +01:00
};
// 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' });
2011-12-19 18:44:12 +01:00
response.end(JSON.stringify({ message: 'Document not found.' }));
}
}, skipExpire);
};
2011-11-18 22:45:48 +01:00
// Handle adding a new Document
2011-11-18 21:51:38 +01:00
DocumentHandler.prototype.handlePost = function(request, response) {
2011-11-18 23:54:57 +01:00
var _this = this;
2011-11-18 23:12:28 +01:00
var buffer = '';
var cancelled = false;
2011-11-18 21:51:38 +01:00
request.on('data', function(data) {
2011-11-18 23:12:28 +01:00
if (!buffer) {
2011-11-18 22:28:09 +01:00
response.writeHead(200, { 'content-type': 'application/json' });
2011-11-18 21:51:38 +01:00
}
2011-11-18 23:12:28 +01:00
buffer += data.toString();
2011-11-21 16:17:23 +01:00
if (_this.maxLength && buffer.length > _this.maxLength) {
cancelled = true;
2011-11-22 03:48:09 +01:00
winston.warn('document >maxLength', { maxLength: _this.maxLength });
2011-11-21 16:17:23 +01:00
response.writeHead(400, { 'content-type': 'application/json' });
2011-12-19 18:44:12 +01:00
response.end(JSON.stringify({ message: 'Document exceeds maximum length.' }));
2011-11-21 16:17:23 +01:00
}
2011-11-18 21:51:38 +01:00
});
request.on('end', function(end) {
if (cancelled) return;
2011-11-22 03:48:09 +01:00
_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' });
2011-12-19 18:44:12 +01:00
response.end(JSON.stringify({ message: 'Error adding document.' }));
2011-11-22 03:48:09 +01:00
}
});
2011-11-18 23:12:28 +01:00
});
2011-11-18 21:51:38 +01:00
});
request.on('error', function(error) {
2011-11-18 22:28:09 +01:00
winston.error('connection error: ' + error.message);
response.writeHead(500, { 'content-type': 'application/json' });
2011-12-19 18:44:12 +01:00
response.end(JSON.stringify({ message: 'Connection error.' }));
2011-11-18 21:51:38 +01:00
});
};
2011-11-22 03:48:09 +01:00
// Get a random key that hasn't been already used
DocumentHandler.prototype.chooseKey = function(callback) {
var key = this.randomKey();
var _this = this;
this.store.get(key, function(ret) {
if (ret) {
_this.chooseKey(callback);
} else {
callback(key);
}
});
};
2011-11-18 22:45:48 +01:00
// Generate a random key
DocumentHandler.prototype.randomKey = function() {
var text = '';
var keyspace = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
for (var i = 0; i < this.keyLength; i++) {
2011-11-18 22:45:48 +01:00
text += keyspace.charAt(Math.floor(Math.random() * keyspace.length));
}
return text;
};
2011-11-18 22:22:00 +01:00
2011-11-18 21:51:38 +01:00
module.exports = DocumentHandler;