From ba5c6b8d16cd19afda2b79219216a3d4bc1cb446 Mon Sep 17 00:00:00 2001
From: PassTheMayo <jacobgunther12@gmail.com>
Date: Sat, 9 Dec 2017 18:34:00 -0600
Subject: [PATCH] Added RethinkDB storage option & fixed config to use proper
 JSON

---
 README.md                        | 24 +++++++++++++++-
 config.js                        | 47 --------------------------------
 config.json                      | 36 ++++++++++++++++++++++++
 lib/document_stores/rethinkdb.js | 39 ++++++++++++++++++++++++++
 server.js                        |  2 +-
 5 files changed, 99 insertions(+), 49 deletions(-)
 delete mode 100644 config.js
 create mode 100644 config.json
 create mode 100644 lib/document_stores/rethinkdb.js

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 <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
\ 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';