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/lib/document_stores/rethinkdb.js b/lib/document_stores/rethinkdb.js new file mode 100644 index 0000000..3fa60dc --- /dev/null +++ b/lib/document_stores/rethinkdb.js @@ -0,0 +1,39 @@ +const crypto = require('crypto'); +const rethink = require('rethinkdbdash'); + +var RethinkDBStore = (options) => { + this.client = rethink({ + silent: true, + host: options.host || '127.0.0.1', + port: options.port || 28015, + db: options.db || 'haste', + user: options.user || 'admin', + password: options.password || '' + }); +}; + +RethinkDBStore.md5 = (str) => { + const md5sum = crypto.createHash('md5'); + md5sum.update(str); + return md5sum.digest('hex'); +}; + +RethinkDBStore.prototype.set = (key, data, callback) => { + try { + this.client.table('uploads').insert({ id: RethinkDBStore.md5(key), data: data }).run((error) =? { + if (error) return callback(false); + callback(true); + }); + } catch (err) { + callback(false); + } +}; + +RethinkDBStore.prototype.get = (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;