haste-server/lib/document_handler.js

156 lines
4.7 KiB
JavaScript
Raw Normal View History

2011-11-18 21:51:38 +01:00
var winston = require('winston');
var Busboy = require('busboy');
2011-11-18 21:51:38 +01:00
// 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;
this.keyGenerator = options.keyGenerator;
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
DocumentHandler.prototype.handleGet = function(request, response, config) {
const key = request.params.id.split('.')[0];
const skipExpire = !!config.documents[key];
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' });
if (request.method === 'HEAD') {
response.end();
} else {
response.end(JSON.stringify({ data: ret, key: key }));
}
2011-11-18 23:12:28 +01:00
}
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.' }));
}
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(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);
};
2011-11-18 22:45:48 +01:00
// Handle adding a new Document
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;
// What to do when done
var onSuccess = function () {
// Check length
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' });
response.end(
JSON.stringify({ message: 'Document exceeds maximum length.' })
);
return;
2011-11-21 16:17:23 +01:00
}
// And then save if we should
_this.chooseKey(function (key) {
_this.store.set(key, buffer, function (res) {
2011-11-22 03:48:09 +01:00
if (res) {
winston.verbose('added document', { key: key });
response.writeHead(200, { 'content-type': 'application/json' });
2011-11-22 03:48:09 +01:00
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
});
};
// If we should, parse a form to grab the data
var ct = request.headers['content-type'];
if (ct && ct.split(';')[0] === 'multipart/form-data') {
var busboy = new Busboy({ headers: request.headers });
busboy.on('field', function (fieldname, val) {
if (fieldname === '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;
});
}
2011-11-18 21:51:38 +01:00
};
// Keep choosing keys until one isn't taken
2011-11-22 03:48:09 +01:00
DocumentHandler.prototype.chooseKey = function(callback) {
var key = this.acceptableKey();
2011-11-22 03:48:09 +01:00
var _this = this;
this.store.get(key, function(ret) {
if (ret) {
_this.chooseKey(callback);
} else {
callback(key);
}
}, true); // Don't bump expirations when key searching
2011-11-22 03:48:09 +01:00
};
DocumentHandler.prototype.acceptableKey = function() {
return this.keyGenerator.createKey(this.keyLength);
};
2011-11-18 21:51:38 +01:00
module.exports = DocumentHandler;