Commit ba5c6b8d authored by PassTheMayo's avatar PassTheMayo
Browse files

Added RethinkDB storage option & fixed config to use proper JSON

parent cfef5882
Loading
Loading
Loading
Loading
+23 −1
Original line number Diff line number Diff line
@@ -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>

config.js

deleted100644 → 0
+0 −47
Original line number Diff line number Diff line
{

  "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"
  }

}

config.json

0 → 100644
+36 −0
Original line number Diff line number Diff line
{
	"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
+39 −0
Original line number Diff line number Diff line
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;
+1 −1
Original line number Diff line number Diff line
@@ -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';