ioredis-mock/lib/data.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _typeof = typeof Symbol === "function" && typeof Symbol.iterator === "symbol" ? function (obj) { return typeof obj; } : function (obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; };

exports.default = createData;

var _es6Set = require('es6-set');

var _es6Set2 = _interopRequireDefault(_es6Set);

var _es6Map = require('es6-map');

var _es6Map2 = _interopRequireDefault(_es6Map);

var _lodash = require('lodash');

var _buffer = require('./buffer');

var _buffer2 = _interopRequireDefault(_buffer);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function createData(expiresInstance) {
  var initial = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
  var keyPrefix = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : '';

  var raw = {};

  function createInstance(prefix, expires) {
    return Object.freeze({
      clear: function clear() {
        raw = {};
      },
      delete: function _delete(key) {
        if (expires.has(key)) {
          expires.delete(key);
        }
        delete raw['' + prefix + key];
      },
      get: function get(key) {
        if (expires.has(key) && expires.isExpired(key)) {
          this.delete(key);
        }

        var value = raw['' + prefix + key];

        if (Array.isArray(value)) {
          return value.slice();
        }

        if (Buffer.isBuffer(value)) {
          return (0, _buffer2.default)(value);
        }

        if (value instanceof _es6Set2.default) {
          return new _es6Set2.default(value);
        }

        if (value instanceof _es6Map2.default) {
          return new _es6Map2.default(value);
        }

        if ((typeof value === 'undefined' ? 'undefined' : _typeof(value)) === 'object' && value) {
          return (0, _lodash.assign)({}, value);
        }

        return value;
      },
      has: function has(key) {
        if (expires.has(key) && expires.isExpired(key)) {
          this.delete(key);
        }

        return {}.hasOwnProperty.call(raw, '' + prefix + key);
      },
      keys: function keys() {
        return Object.keys(raw);
      },
      set: function set(key, val) {
        var item = val;

        if (Array.isArray(val)) {
          item = val.slice();
        } else if (Buffer.isBuffer(val)) {
          item = (0, _buffer2.default)(val);
        } else if (val instanceof _es6Set2.default) {
          item = new _es6Set2.default(val);
        } else if (val instanceof _es6Map2.default) {
          item = new _es6Map2.default(val);
        } else if ((typeof val === 'undefined' ? 'undefined' : _typeof(val)) === 'object' && val) {
          item = (0, _lodash.assign)({}, val);
        }

        raw['' + prefix + key] = item;
      },
      withKeyPrefix: function withKeyPrefix(newKeyPrefix) {
        if (newKeyPrefix === prefix) return this;
        return createInstance(newKeyPrefix, expires.withKeyPrefix(newKeyPrefix));
      }
    });
  }

  var data = createInstance(keyPrefix, expiresInstance);

  Object.keys(initial).forEach(function (key) {
    return data.set(key, initial[key]);
  });

  return data;
}