Merge pull request #189 from PassTheMayo/master

Added RethinkDB storage option & fixed config to use proper JSON

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 <john.crepezzi@gmail.com>
@@ -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

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;