Spaces:
Running
Running
const path = require('path'); | |
const fs = require('fs'); | |
const Keyv = require('keyv'); | |
const { writeJSON, tryParse } = require('./utils'); | |
const { del } = require('./del'); | |
const cache = { | |
/** | |
* Load a cache identified by the given Id. If the element does not exists, then initialize an empty | |
* cache storage. If specified `cacheDir` will be used as the directory to persist the data to. If omitted | |
* then the cache module directory `./cache` will be used instead | |
* | |
* @method load | |
* @param docId {String} the id of the cache, would also be used as the name of the file cache | |
* @param [cacheDir] {String} directory for the cache entry | |
*/ | |
load: function (docId, cacheDir) { | |
const me = this; | |
me.keyv = new Keyv(); | |
me.__visited = {}; | |
me.__persisted = {}; | |
me._pathToFile = cacheDir ? path.resolve(cacheDir, docId) : path.resolve(__dirname, '../.cache/', docId); | |
if (fs.existsSync(me._pathToFile)) { | |
me._persisted = tryParse(me._pathToFile, {}); | |
} | |
}, | |
get _persisted() { | |
return this.__persisted; | |
}, | |
set _persisted(value) { | |
this.__persisted = value; | |
}, | |
get _visited() { | |
return this.__visited; | |
}, | |
set _visited(value) { | |
this.__visited = value; | |
}, | |
/** | |
* Load the cache from the provided file | |
* @method loadFile | |
* @param {String} pathToFile the path to the file containing the info for the cache | |
*/ | |
loadFile: function (pathToFile) { | |
const me = this; | |
const dir = path.dirname(pathToFile); | |
const fName = path.basename(pathToFile); | |
me.load(fName, dir); | |
}, | |
/** | |
* Returns the entire persisted object | |
* @method all | |
* @returns {*} | |
*/ | |
all: function () { | |
return this._persisted; | |
}, | |
keys: function () { | |
return Object.keys(this._persisted); | |
}, | |
/** | |
* sets a key to a given value | |
* @method setKey | |
* @param key {string} the key to set | |
* @param value {object} the value of the key. Could be any object that can be serialized with JSON.stringify | |
*/ | |
setKey: function (key, value) { | |
this._visited[key] = true; | |
this._persisted[key] = value; | |
}, | |
/** | |
* remove a given key from the cache | |
* @method removeKey | |
* @param key {String} the key to remove from the object | |
*/ | |
removeKey: function (key) { | |
delete this._visited[key]; // esfmt-ignore-line | |
delete this._persisted[key]; // esfmt-ignore-line | |
}, | |
/** | |
* Return the value of the provided key | |
* @method getKey | |
* @param key {String} the name of the key to retrieve | |
* @returns {*} the value from the key | |
*/ | |
getKey: function (key) { | |
this._visited[key] = true; | |
return this._persisted[key]; | |
}, | |
/** | |
* Remove keys that were not accessed/set since the | |
* last time the `prune` method was called. | |
* @method _prune | |
* @private | |
*/ | |
_prune: function () { | |
const me = this; | |
const obj = {}; | |
const keys = Object.keys(me._visited); | |
// no keys visited for either get or set value | |
if (keys.length === 0) { | |
return; | |
} | |
keys.forEach(function (key) { | |
obj[key] = me._persisted[key]; | |
}); | |
me._visited = {}; | |
me._persisted = obj; | |
}, | |
/** | |
* Save the state of the cache identified by the docId to disk | |
* as a JSON structure | |
* @param [noPrune=false] {Boolean} whether to remove from cache the non visited files | |
* @method save | |
*/ | |
save: function (noPrune) { | |
const me = this; | |
!noPrune && me._prune(); | |
writeJSON(me._pathToFile, me._persisted); | |
}, | |
/** | |
* remove the file where the cache is persisted | |
* @method removeCacheFile | |
* @return {Boolean} true or false if the file was successfully deleted | |
*/ | |
removeCacheFile: function () { | |
return del(this._pathToFile); | |
}, | |
/** | |
* Destroy the file cache and cache content. | |
* @method destroy | |
*/ | |
destroy: function () { | |
const me = this; | |
me._visited = {}; | |
me._persisted = {}; | |
me.removeCacheFile(); | |
}, | |
}; | |
module.exports = { | |
/** | |
* Alias for create. Should be considered depreacted. Will be removed in next releases | |
* | |
* @method load | |
* @param docId {String} the id of the cache, would also be used as the name of the file cache | |
* @param [cacheDir] {String} directory for the cache entry | |
* @returns {cache} cache instance | |
*/ | |
load: function (docId, cacheDir) { | |
return this.create(docId, cacheDir); | |
}, | |
/** | |
* Load a cache identified by the given Id. If the element does not exists, then initialize an empty | |
* cache storage. | |
* | |
* @method create | |
* @param docId {String} the id of the cache, would also be used as the name of the file cache | |
* @param [cacheDir] {String} directory for the cache entry | |
* @returns {cache} cache instance | |
*/ | |
create: function (docId, cacheDir) { | |
const obj = Object.create(cache); | |
obj.load(docId, cacheDir); | |
return obj; | |
}, | |
createFromFile: function (filePath) { | |
const obj = Object.create(cache); | |
obj.loadFile(filePath); | |
return obj; | |
}, | |
/** | |
* Clear the cache identified by the given id. Caches stored in a different cache directory can be deleted directly | |
* | |
* @method clearCache | |
* @param docId {String} the id of the cache, would also be used as the name of the file cache | |
* @param cacheDir {String} the directory where the cache file was written | |
* @returns {Boolean} true if the cache folder was deleted. False otherwise | |
*/ | |
clearCacheById: function (docId, cacheDir) { | |
const filePath = cacheDir ? path.resolve(cacheDir, docId) : path.resolve(__dirname, '../.cache/', docId); | |
return del(filePath); | |
}, | |
/** | |
* Remove all cache stored in the cache directory | |
* @method clearAll | |
* @returns {Boolean} true if the cache folder was deleted. False otherwise | |
*/ | |
clearAll: function (cacheDir) { | |
const filePath = cacheDir ? path.resolve(cacheDir) : path.resolve(__dirname, '../.cache/'); | |
return del(filePath); | |
}, | |
}; | |