loopback-datasource-juggler/lib/connectors/memory.js

var geo = require('../geo');
var utils = require('../utils');

exports.initialize = function initializeSchema(dataSource, callback) {
    dataSource.connector = new Memory();
    dataSource.connector.connect(callback);
};

exports.Memory = Memory;

function Memory(m) {
    if (m) {
        this.isTransaction = true;
        this.cache = m.cache;
        this.ids = m.ids;
        this._models = m._models;
    } else {
        this.isTransaction = false;
        this.cache = {};
        this.ids = {};
        this._models = {};
    }
}

Memory.prototype.connect = function(callback) {
    if (this.isTransaction) {
        this.onTransactionExec = callback;
    } else {
        process.nextTick(callback);
    }
};

Memory.prototype.define = function defineModel(descr) {
    var m = descr.model.modelName;
    this._models[m] = descr;
    this.cache[m] = {};
    this.ids[m] = 1;
};

/**
 * Get the id property name for the given model
 * @param {String} model The model name
 * @returns {String} The id property name
 *
 * @private
 */
Memory.prototype.idName = function(model) {
    return (this.dataSource && this.dataSource.idName && this.dataSource.idName(model)) || 'id';
};

/**
 * Get the id value for the given model
 * @param {String} model The model name
 * @param {Object} data The model instance data
 * @returns {*} The id value
 *
 * @private
 */
Memory.prototype.getIdValue = function(model, data) {
    return data && data[this.idName(model)];
};

/**
 * Set the id value for the given model
 * @param {String} model The model name
 * @param {Object} data The model instance data
 * @param {*} value The id value
 *
 * @private
 */
Memory.prototype.setIdValue = function(model, data, value) {
    if(data) {
        data[this.idName(model)] = value;
    }
};

Memory.prototype.create = function create(model, data, callback) {
    // FIXME: [rfeng] We need to generate unique ids based on the id type
    // FIXME: [rfeng] We don't support composite ids yet
    var currentId = this.ids[model];
    if(currentId === undefined) {
        // First time
        this.ids[model] = 1;
        currentId = 1;
    }
    var id = this.getIdValue(model, data) || currentId;
    if(id > currentId) {
        // If the id is passed in and the value is greater than the current id
        currentId = id;
    }
    this.ids[model] = Number(currentId) + 1;

    var props = this._models[model].properties;
    var idName = this.idName(model);
    id = (props[idName] && props[idName].type && props[idName].type(id)) || id;
    this.setIdValue(model, data, id);
    this.cache[model][id] = JSON.stringify(data);
    process.nextTick(function() {
        callback(null, id);
    });
};

Memory.prototype.updateOrCreate = function (model, data, callback) {
    var self = this;
    this.exists(model, self.getIdValue(model, data), function (err, exists) {
        if (exists) {
            self.save(model, data, callback);
        } else {
            self.create(model, data, function (err, id) {
                self.setIdValue(model, data, id);
                callback(err, data);
            });
        }
    });
};

Memory.prototype.save = function save(model, data, callback) {
    this.cache[model][this.getIdValue(model, data)] = JSON.stringify(data);
    process.nextTick(function () {
        callback(null, data);
    });
};

Memory.prototype.exists = function exists(model, id, callback) {
    process.nextTick(function () {
        callback(null, this.cache[model] && this.cache[model].hasOwnProperty(id));
    }.bind(this));
};

Memory.prototype.find = function find(model, id, callback) {
    var self = this;
    process.nextTick(function () {
        callback(null, id in this.cache[model] && this.fromDb(model, this.cache[model][id]));
    }.bind(this));
};

Memory.prototype.destroy = function destroy(model, id, callback) {
    delete this.cache[model][id];
    process.nextTick(callback);
};

Memory.prototype.fromDb = function(model, data) {
    if (!data) return null;
    data = JSON.parse(data);
    var props = this._models[model].properties;
    Object.keys(data).forEach(function (key) {
        var val = data[key];
        if (typeof val === 'undefined' || val === null) {
            return;
        }
        if (props[key]) {
            switch(props[key].type.name) {
                case 'Date':
                    val = new Date(val.toString().replace(/GMT.*$/, 'GMT'));
                    break;
                case 'Boolean':
                    val = Boolean(val);
                    break;
                case 'Number':
                    val = Number(val);
                    break;
            }
        }
        data[key] = val;
    });
    return data;
};

Memory.prototype.all = function all(model, filter, callback) {
    var self = this;
    var nodes = Object.keys(this.cache[model]).map(function (key) {
        return this.fromDb(model, this.cache[model][key]);
    }.bind(this));

    if (filter) {
        // do we need some sorting?
        if (filter.order) {
            var orders = filter.order;
            if (typeof filter.order === "string") {
                orders = [filter.order];
            }
            orders.forEach(function (key, i) {
                var reverse = 1;
                var m = key.match(/\s+(A|DE)SC$/i);
                if (m) {
                    key = key.replace(/\s+(A|DE)SC/i, '');
                    if (m[1].toLowerCase() === 'de') reverse = -1;
                }
                orders[i] = {"key": key, "reverse": reverse};
            });
            nodes = nodes.sort(sorting.bind(orders));
        }
        
        var nearFilter = geo.nearFilter(filter.where);
        
        // geo sorting
        if(nearFilter) {
            nodes = geo.filter(nodes, nearFilter);
        }
        
        // do we need some filtration?
        if (filter.where) {
            nodes = nodes ? nodes.filter(applyFilter(filter)) : nodes;
        }
        
        // field selection
        if(filter.fields) {
            nodes = nodes.map(utils.selectFields(filter.fields));
        }

        // limit/skip
        filter.skip = filter.skip || 0;
        filter.limit = filter.limit || nodes.length;
        nodes = nodes.slice(filter.skip, filter.skip + filter.limit);
    }

    process.nextTick(function () {
        if (filter && filter.include) {
            self._models[model].model.include(nodes, filter.include, callback);
        } else {
            callback(null, nodes);
        }
    });

    function sorting(a, b) {
        for (var i=0, l=this.length; i<l; i++) {
            if (a[this[i].key] > b[this[i].key]) {
                return 1*this[i].reverse;
            } else  if (a[this[i].key] < b[this[i].key]) {
                return -1*this[i].reverse;
            }
        }
        return 0;
    }
};

function applyFilter(filter) {
    if (typeof filter.where === 'function') {
        return filter.where;
    }
    var keys = Object.keys(filter.where);
    return function (obj) {
        var pass = true;
        keys.forEach(function (key) {
            if (!test(filter.where[key], obj[key])) {
                pass = false;
            }
        });
        return pass;
    }

    function test(example, value) {
        if (typeof value === 'string' && example && example.constructor.name === 'RegExp') {
            return value.match(example);
        }
        if (typeof example === 'undefined') return undefined;
        if (typeof value === 'undefined') return undefined;
        if (typeof example === 'object') {
            // ignore geo near filter
            if(example.near) return true;
            
            if (example.inq) {
                if (!value) return false;
                for (var i = 0; i < example.inq.length; i += 1) {
                    if (example.inq[i] == value) return true;
                }
                return false;
            }
            
            if(isNum(example.gt) && example.gt < value) return true;
            if(isNum(example.gte) && example.gte <= value) return true;
            if(isNum(example.lt) && example.lt > value) return true;
            if(isNum(example.lte) && example.lte >= value) return true;
        }
        // not strict equality
        return (example !== null ? example.toString() : example) == (value !== null ? value.toString() : value);
    }
    
    function isNum(n) {
        return typeof n === 'number';
    }
}

Memory.prototype.destroyAll = function destroyAll(model, where, callback) {
    if(!callback && 'function' === typeof where) {
        callback = where;
        where = undefined;
    }
    var cache = this.cache[model];
    var filter = null;
    if (where) {
        filter = applyFilter({where: where});
    }
    Object.keys(cache).forEach(function (id) {
        if(!filter || filter(this.fromDb(model, cache[id]))) {
            delete cache[id];
        }
    }.bind(this));
    if(!where) {
        this.cache[model] = {};
    }
    process.nextTick(callback);
};

Memory.prototype.count = function count(model, callback, where) {
    var cache = this.cache[model];
    var data = Object.keys(cache);
    if (where) {
        var filter = {where: where};
        data = data.map(function (id) {
          return this.fromDb(model, cache[id]);
        }.bind(this));
        data = data.filter(applyFilter(filter));
    }
    process.nextTick(function () {
        callback(null, data.length);
    });
};

Memory.prototype.updateAttributes = function updateAttributes(model, id, data, cb) {
    if(!id) {
      var err = new Error('You must provide an id when updating attributes!');
      if(cb) {
        return cb(err);
      } else {
        throw err;
      }
    }
  
    this.setIdValue(model, data, id);
    
    var cachedModels = this.cache[model];
    var modelAsString = cachedModels && this.cache[model][id];
    var modelData = modelAsString && JSON.parse(modelAsString);
    
    if(modelData) {
      this.save(model, merge(modelData, data), cb);
    } else {
      cb(new Error('Could not update attributes. Object with id ' + id + ' does not exist!'));
    }
};

Memory.prototype.transaction = function () {
    return new Memory(this);
};

Memory.prototype.exec = function(callback) {
    this.onTransactionExec();
    setTimeout(callback, 50);
};

Memory.prototype.buildNearFilter = function (filter) {
  // noop
}

function merge(base, update) {
    if (!base) return update;
    Object.keys(update).forEach(function (key) {
        base[key] = update[key];
    });
    return base;
}