discord.js/src/Util/Cache.js

"use strict";

var discrimS = Symbol();
var discrimCacheS = Symbol();

export default class Cache extends Array {
	constructor(discrim, limit) {
		super();
		this[discrimS] = discrim || "id";
		this[discrimCacheS] = {};
	}

	get(key, value) {
		if (typeof key === 'function') {
			var valid = key;
			key = null;
		} else if (key === this[discrimS] && typeof value === "string") {
			return this[discrimCacheS][value] || null;
		} else if (value && value.constructor.name === 'RegExp') {
			var valid = function valid(item) {
				return value.test(item);
			}
		} else if (typeof value !== 'function') {
			var valid = function valid(item) {
				return item == value;
			}
		}

		for (var item of this) {
			if (valid(key == null ? item : item[key])) {
				return item;
			}
		}

		return null;
	}

	has(object) {
		return !!this.get(this[discrimS], object[this[discrimS]]);
	}

	getAll(key, value) {
		var found = new Cache(this[discrimS]);

		if (typeof key === 'function') {
			var valid = key;
			key = null;
		} else if (value && value.constructor.name === 'RegExp') {
			var valid = function valid(item) {
				return value.test(item);
			}
		} else if (typeof value !== 'function') {
			var valid = function valid(item) {
				return item == value;
			}
		}

		for (var item of this) {
			if (valid(key == null ? item : item[key])) {
				found.add(item);
			}
		}

		return found;
	}

	add(data) {
		var cacheKey = this[discrimS] === "id" ? data.id : data[this[discrimS]];
		if (this[discrimCacheS][cacheKey]) {
			return this[discrimCacheS][cacheKey];
		}
		if (this.limit && this.length >= this.limit) {
			this.splice(0, 1);
		}
		this.push(data);
		this[discrimCacheS][cacheKey] = data;
		return data;
	}

	update(old, data) {
		var item = this.get(this[discrimS], old[this[discrimS]]);

		if (item) {
			var index = this.indexOf(item);
			for (var dataIndex in data) {
				if (data.hasOwnProperty(dataIndex)) {
					this[index][dataIndex] = data[dataIndex];
				}
			}
			this[discrimCacheS][data[this[discrimS]]] = this[index];
			return this[index];
		} else {
			return false;
		}
	}

	random() {
		return this[Math.floor(Math.random()*this.length)];
	}

	remove(data) {
		delete this[discrimCacheS][data[this[discrimS]]];
		var index = this.indexOf(data);
		if (~index) {
			this.splice(index, 1);
		} else {
			var item = this.get(this[discrimS], data[this[discrimS]]);
			if (item) {
				this.splice(this.indexOf(item), 1);
			}
		}
		return false;
	}
}