diff --git a/README.md b/README.md index 23c3ca1..1a112c4 100644 --- a/README.md +++ b/README.md @@ -174,6 +174,28 @@ forward on GETs. All of which are optional except `type` with very logical default values. +### RethinkDB + +To use the RethinkDB storage system, you must install the `rethinkdbdash` package via npm + +`npm install rethinkdbdash` + +Once you've done that, your config section should look like this: + +``` json +{ + "type": "rethinkdb", + "host": "127.0.0.1", + "port": 28015, + "db": "haste" +} +``` + +In order for this to work, the database must be pre-created before the script is ran. +Also, you must create an `uploads` table, which will store all the data for uploads. + +You can optionally add the `user` and `password` properties to use a user system. + ## Author John Crepezzi @@ -206,4 +228,4 @@ SOFTWARE * jQuery: MIT/GPL license * highlight.js: Copyright © 2006, Ivan Sagalaev -* highlightjs-coffeescript: WTFPL - Copyright © 2011, Dmytrii Nagirniak +* highlightjs-coffeescript: WTFPL - Copyright © 2011, Dmytrii Nagirniak \ No newline at end of file diff --git a/config.js b/config.js deleted file mode 100644 index 52ad004..0000000 --- a/config.js +++ /dev/null @@ -1,47 +0,0 @@ -{ - - "host": "0.0.0.0", - "port": 7777, - - "keyLength": 10, - - "maxLength": 400000, - - "staticMaxAge": 86400, - - "recompressStaticAssets": true, - - "logging": [ - { - "level": "verbose", - "type": "Console", - "colorize": true - } - ], - - "keyGenerator": { - "type": "phonetic" - }, - - "rateLimits": { - "categories": { - "normal": { - "totalRequests": 500, - "every": 60000 - } - } - }, - - "storage": { - "type": "redis", - "host": "0.0.0.0", - "port": 6379, - "db": 2, - "expire": 2592000 - }, - - "documents": { - "about": "./about.md" - } - -} diff --git a/config.json b/config.json new file mode 100644 index 0000000..6a23c5e --- /dev/null +++ b/config.json @@ -0,0 +1,36 @@ +{ + "host": "0.0.0.0", + "port": 7777, + "keyLength": 10, + "maxLength": 400000, + "staticMaxAge": 86400, + "recompressStaticAssets": true, + "logging": [ + { + "level": "verbose", + "type": "Console", + "colorize": true + } + ], + "keyGenerator": { + "type": "phonetic" + }, + "rateLimits": { + "categories": { + "normal": { + "totalRequests": 500, + "every": 60000 + } + } + }, + "storage": { + "type": "redis", + "host": "0.0.0.0", + "port": 6379, + "db": 2, + "expire": 2592000 + }, + "documents": { + "about": "./about.md" + } +} \ No newline at end of file diff --git a/lib/document_stores/rethinkdb.js b/lib/document_stores/rethinkdb.js new file mode 100644 index 0000000..fdfadff --- /dev/null +++ b/lib/document_stores/rethinkdb.js @@ -0,0 +1,39 @@ +var crypto = require('crypto'); +var rethink = require('rethinkdbdash'); + +var RethinkDBStore = function (options) { + this._options = options; + this._options.silent = true; + this._options.host = options.host || '127.0.0.1'; + this._options.port = options.port || 28015; + this._options.db = options.db || 'haste'; + this._options.user = options.user || 'admin'; + this._options.password = options.password || ''; + this.client = rethink(this._options); +}; + +RethinkDBStore.md5 = function (str) { + var md5sum = crypto.createHash('md5'); + md5sum.update(str); + return md5sum.digest('hex'); +}; + +RethinkDBStore.prototype.set = function (key, data, callback) { + try { + this.client.table('uploads').insert({ id: RethinkDBStore.md5(key), data: data }).run(function (error) { + if (error) return callback(false); + callback(true); + }); + } catch (err) { + callback(false); + } +}; + +RethinkDBStore.prototype.get = function (key, callback) { + this.client.table('uploads').get(RethinkDBStore.md5(key)).run((error, result) => { + if (error || !result) return callback(false); + callback(result.data); + }); +}; + +module.exports = RethinkDBStore; diff --git a/server.js b/server.js index 57bd3da..65d844b 100644 --- a/server.js +++ b/server.js @@ -11,7 +11,7 @@ var connect_rate_limit = require('connect-ratelimit'); var DocumentHandler = require('./lib/document_handler'); // Load the configuration and set some defaults -var config = JSON.parse(fs.readFileSync('./config.js', 'utf8')); +var config = require('./config.json'); config.port = process.env.PORT || config.port || 7777; config.host = process.env.HOST || config.host || 'localhost';