hastebin-ansi/lib/document_stores/file.js

const fs = require('fs');
const crypto = require('crypto');

const winston = require('winston');

// For storing in files
// options[type] = file
// options[path] - Where to store

const FileDocumentStore = function (options) {
    this.basePath = options.path || './data';
    this.expire = options.expire;
};

// Generate md5 of a string
FileDocumentStore.md5 = function (str) {
    const md5sum = crypto.createHash('md5');
    md5sum.update(str);
    return md5sum.digest('hex');
};

// Save data in a file, key as md5 - since we don't know what we could
// be passed here
FileDocumentStore.prototype.set = function (key, data, callback, skipExpire) {
    try {
        const _this = this;
        fs.mkdir(this.basePath, '700', function () {
            const fn = _this.basePath + '/' + FileDocumentStore.md5(key);
            fs.writeFile(fn, data, 'utf8', function (err) {
                if (err) {
                    callback(false);
                } else {
                    callback(true);
                    if (_this.expire && !skipExpire) {
                        winston.warn('file store cannot set expirations on keys');
                    }
                }
            });
        });
    } catch (err) {
        callback(false);
    }
};

// Get data from a file from key
FileDocumentStore.prototype.get = function (key, callback, skipExpire) {
    const _this = this;
    const fn = this.basePath + '/' + FileDocumentStore.md5(key);
    fs.readFile(fn, 'utf8', function (err, data) {
        if (err) {
            callback(false);
        } else {
            callback(data);
            if (_this.expire && !skipExpire) {
                winston.warn('file store cannot set expirations on keys');
            }
        }
    });
};

module.exports = FileDocumentStore;