| 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); |
| }, |
| }; |