forked from projectmoon/tenebrous-dicebot
353 lines
9.7 KiB
JavaScript
353 lines
9.7 KiB
JavaScript
/*
|
|
MIT License http://www.opensource.org/licenses/mit-license.php
|
|
Author Tobias Koppers @sokra
|
|
*/
|
|
"use strict";
|
|
|
|
const Source = require("./Source");
|
|
const { SourceMapConsumer, SourceNode } = require("source-map");
|
|
const { SourceListMap, fromStringWithSourceMap } = require("source-list-map");
|
|
|
|
const LIST_MAP_OPTIONS = { columns: false };
|
|
|
|
const mapToBufferedMap = map => {
|
|
if (typeof map !== "object" || !map) return map;
|
|
const bufferedMap = Object.assign({}, map);
|
|
if (map.mappings) {
|
|
bufferedMap.mappings = Buffer.from(map.mappings, "utf-8");
|
|
}
|
|
if (map.sourcesContent) {
|
|
bufferedMap.sourcesContent = map.sourcesContent.map(
|
|
str => str && Buffer.from(str, "utf-8")
|
|
);
|
|
}
|
|
return bufferedMap;
|
|
};
|
|
|
|
const bufferedMapToMap = bufferedMap => {
|
|
if (typeof bufferedMap !== "object" || !bufferedMap) return bufferedMap;
|
|
const map = Object.assign({}, bufferedMap);
|
|
if (bufferedMap.mappings) {
|
|
map.mappings = bufferedMap.mappings.toString("utf-8");
|
|
}
|
|
if (bufferedMap.sourcesContent) {
|
|
map.sourcesContent = bufferedMap.sourcesContent.map(
|
|
buffer => buffer && buffer.toString("utf-8")
|
|
);
|
|
}
|
|
return map;
|
|
};
|
|
|
|
const sourceAndMapToNode = (source, map) => {
|
|
if (map) {
|
|
return SourceNode.fromStringWithSourceMap(
|
|
source,
|
|
new SourceMapConsumer(map)
|
|
);
|
|
} else {
|
|
return new SourceNode(null, null, null, source);
|
|
}
|
|
};
|
|
|
|
const sourceAndMapToListMap = (source, map) => {
|
|
if (map) {
|
|
return fromStringWithSourceMap(source, map);
|
|
} else {
|
|
return new SourceListMap(source);
|
|
}
|
|
};
|
|
|
|
class CachedSource extends Source {
|
|
constructor(source, cachedData) {
|
|
super();
|
|
this._source = source;
|
|
this._cachedSourceType = cachedData ? cachedData.source : undefined;
|
|
this._cachedSource = undefined;
|
|
this._cachedBuffer = cachedData ? cachedData.buffer : undefined;
|
|
this._cachedSize = cachedData ? cachedData.size : undefined;
|
|
this._cachedMaps = cachedData ? cachedData.maps : new Map();
|
|
this._cachedHashUpdate = cachedData ? cachedData.hash : undefined;
|
|
}
|
|
|
|
getCachedData() {
|
|
const bufferedMaps = new Map();
|
|
for (const pair of this._cachedMaps) {
|
|
let cacheEntry = pair[1];
|
|
if (cacheEntry.bufferedMap === undefined) {
|
|
cacheEntry.bufferedMap = mapToBufferedMap(
|
|
this._getMapFromCacheEntry(cacheEntry)
|
|
);
|
|
}
|
|
bufferedMaps.set(pair[0], {
|
|
map: undefined,
|
|
node: undefined,
|
|
listMap: undefined,
|
|
bufferedMap: cacheEntry.bufferedMap
|
|
});
|
|
}
|
|
// We don't want to cache strings
|
|
// So if we have a caches sources
|
|
// create a buffer from it and only store
|
|
// if it was a Buffer or string
|
|
if (this._cachedSource) {
|
|
this.buffer();
|
|
}
|
|
return {
|
|
buffer: this._cachedBuffer,
|
|
source:
|
|
this._cachedSourceType !== undefined
|
|
? this._cachedSourceType
|
|
: typeof this._cachedSource === "string"
|
|
? true
|
|
: Buffer.isBuffer(this._cachedSource)
|
|
? false
|
|
: undefined,
|
|
size: this._cachedSize,
|
|
maps: bufferedMaps,
|
|
hash: this._cachedHashUpdate
|
|
};
|
|
}
|
|
|
|
originalLazy() {
|
|
return this._source;
|
|
}
|
|
|
|
original() {
|
|
if (typeof this._source === "function") this._source = this._source();
|
|
return this._source;
|
|
}
|
|
|
|
source() {
|
|
const source = this._getCachedSource();
|
|
if (source !== undefined) return source;
|
|
return (this._cachedSource = this.original().source());
|
|
}
|
|
|
|
_getMapFromCacheEntry(cacheEntry) {
|
|
if (cacheEntry.map !== undefined) {
|
|
return cacheEntry.map;
|
|
} else if (cacheEntry.bufferedMap !== undefined) {
|
|
return (cacheEntry.map = bufferedMapToMap(cacheEntry.bufferedMap));
|
|
} else if (cacheEntry.node !== undefined) {
|
|
const result = cacheEntry.node.toStringWithSourceMap({
|
|
file: "x"
|
|
});
|
|
if (this._cachedSource === undefined) this._cachedSource = result.code;
|
|
return (cacheEntry.map = result.map.toJSON());
|
|
} else if (cacheEntry.listMap !== undefined) {
|
|
const result = cacheEntry.listMap.toStringWithSourceMap({
|
|
file: "x"
|
|
});
|
|
if (this._cachedSource === undefined) this._cachedSource = result.source;
|
|
return (cacheEntry.map = result.map);
|
|
}
|
|
}
|
|
|
|
_getCachedSource() {
|
|
if (this._cachedSource !== undefined) return this._cachedSource;
|
|
if (this._cachedBuffer && this._cachedSourceType !== undefined) {
|
|
return (this._cachedSource = this._cachedSourceType
|
|
? this._cachedBuffer.toString("utf-8")
|
|
: this._cachedBuffer);
|
|
}
|
|
for (const cacheEntry of this._cachedMaps.values()) {
|
|
if (cacheEntry.node !== undefined) {
|
|
return (this._cachedSource = cacheEntry.node.toString());
|
|
}
|
|
if (cacheEntry.listMap !== undefined) {
|
|
return (this._cachedSource = cacheEntry.listMap.toString());
|
|
}
|
|
}
|
|
}
|
|
|
|
buffer() {
|
|
if (this._cachedBuffer !== undefined) return this._cachedBuffer;
|
|
if (this._cachedSource !== undefined) {
|
|
if (Buffer.isBuffer(this._cachedSource)) {
|
|
return (this._cachedBuffer = this._cachedSource);
|
|
}
|
|
return (this._cachedBuffer = Buffer.from(this._cachedSource, "utf-8"));
|
|
}
|
|
if (typeof this.original().buffer === "function") {
|
|
return (this._cachedBuffer = this.original().buffer());
|
|
}
|
|
const bufferOrString = this.source();
|
|
if (Buffer.isBuffer(bufferOrString)) {
|
|
return (this._cachedBuffer = bufferOrString);
|
|
}
|
|
return (this._cachedBuffer = Buffer.from(bufferOrString, "utf-8"));
|
|
}
|
|
|
|
size() {
|
|
if (this._cachedSize !== undefined) return this._cachedSize;
|
|
if (this._cachedBuffer !== undefined) {
|
|
return (this._cachedSize = this._cachedBuffer.length);
|
|
}
|
|
const source = this._getCachedSource();
|
|
if (source !== undefined) {
|
|
return (this._cachedSize = Buffer.byteLength(source));
|
|
}
|
|
return (this._cachedSize = this.original().size());
|
|
}
|
|
|
|
sourceAndMap(options) {
|
|
const key = options ? JSON.stringify(options) : "{}";
|
|
const cacheEntry = this._cachedMaps.get(key);
|
|
// Look for a cached map
|
|
if (cacheEntry !== undefined) {
|
|
// We have a cached map in some representation
|
|
const map = this._getMapFromCacheEntry(cacheEntry);
|
|
// Either get the cached source or compute it
|
|
return { source: this.source(), map };
|
|
}
|
|
// Look for a cached source
|
|
let source = this._getCachedSource();
|
|
// Compute the map
|
|
let map;
|
|
if (source !== undefined) {
|
|
map = this.original().map(options);
|
|
} else {
|
|
// Compute the source and map together.
|
|
const sourceAndMap = this.original().sourceAndMap(options);
|
|
source = sourceAndMap.source;
|
|
map = sourceAndMap.map;
|
|
this._cachedSource = source;
|
|
}
|
|
this._cachedMaps.set(key, {
|
|
map,
|
|
node: undefined,
|
|
listMap: undefined,
|
|
bufferedMap: undefined
|
|
});
|
|
return { source, map };
|
|
}
|
|
|
|
node(options) {
|
|
const key = options ? JSON.stringify(options) : "{}";
|
|
let cacheEntry = this._cachedMaps.get(key);
|
|
// Check cache
|
|
if (cacheEntry !== undefined) {
|
|
// Directly cached
|
|
if (cacheEntry.node) return cacheEntry.node;
|
|
// Construct from cached map
|
|
const map = this._getMapFromCacheEntry(cacheEntry);
|
|
const source = this.source();
|
|
const node = sourceAndMapToNode(source, map);
|
|
cacheEntry.node = node;
|
|
return node;
|
|
}
|
|
let node;
|
|
const original = this.original();
|
|
if (typeof original.node === "function") {
|
|
node = original.node(options);
|
|
this._cachedMaps.set(key, {
|
|
map: undefined,
|
|
node,
|
|
listMap: undefined,
|
|
bufferedMap: undefined
|
|
});
|
|
} else {
|
|
const sourceAndMap = this.sourceAndMap(options);
|
|
node = sourceAndMapToNode(sourceAndMap.source, sourceAndMap.map);
|
|
this._cachedMaps.get(key).node = node;
|
|
}
|
|
return node;
|
|
}
|
|
|
|
listMap(options) {
|
|
let key;
|
|
// Enforce options must include columns: false
|
|
if (!options) {
|
|
key = '{"columns":false}';
|
|
options = LIST_MAP_OPTIONS;
|
|
} else {
|
|
if (options.columns !== false) {
|
|
options = Object.assign({}, options, LIST_MAP_OPTIONS);
|
|
}
|
|
key = JSON.stringify(options);
|
|
}
|
|
// Check cache
|
|
let cacheEntry = this._cachedMaps.get(key);
|
|
if (cacheEntry !== undefined) {
|
|
// Directly cached
|
|
if (cacheEntry.listMap) return cacheEntry.listMap;
|
|
// Construct from cached map
|
|
const map = this._getMapFromCacheEntry(cacheEntry);
|
|
const source = this.source();
|
|
const listMap = sourceAndMapToListMap(source, map);
|
|
cacheEntry.listMap = listMap;
|
|
return listMap;
|
|
}
|
|
let listMap;
|
|
const original = this.original();
|
|
if (typeof original.listMap === "function") {
|
|
listMap = original.listMap(options);
|
|
this._cachedMaps.set(key, {
|
|
map: undefined,
|
|
node: undefined,
|
|
listMap,
|
|
bufferedMap: undefined
|
|
});
|
|
} else {
|
|
const sourceAndMap = this.sourceAndMap(options);
|
|
listMap = sourceAndMapToListMap(sourceAndMap.source, sourceAndMap.map);
|
|
this._cachedMaps.get(key).listMap = listMap;
|
|
}
|
|
return listMap;
|
|
}
|
|
|
|
map(options) {
|
|
const key = options ? JSON.stringify(options) : "{}";
|
|
const cacheEntry = this._cachedMaps.get(key);
|
|
if (cacheEntry !== undefined) {
|
|
return this._getMapFromCacheEntry(cacheEntry);
|
|
}
|
|
const map = this.original().map(options);
|
|
this._cachedMaps.set(key, {
|
|
map,
|
|
node: undefined,
|
|
listMap: undefined,
|
|
bufferedMap: undefined
|
|
});
|
|
return map;
|
|
}
|
|
|
|
updateHash(hash) {
|
|
if (this._cachedHashUpdate !== undefined) {
|
|
for (const item of this._cachedHashUpdate) hash.update(item);
|
|
return;
|
|
}
|
|
const update = [];
|
|
let currentString = undefined;
|
|
const tracker = {
|
|
update: item => {
|
|
if (typeof item === "string" && item.length < 10240) {
|
|
if (currentString === undefined) {
|
|
currentString = item;
|
|
} else {
|
|
currentString += item;
|
|
if (currentString > 102400) {
|
|
update.push(Buffer.from(currentString));
|
|
currentString = undefined;
|
|
}
|
|
}
|
|
} else {
|
|
if (currentString !== undefined) {
|
|
update.push(Buffer.from(currentString));
|
|
currentString = undefined;
|
|
}
|
|
update.push(item);
|
|
}
|
|
}
|
|
};
|
|
this.original().updateHash(tracker);
|
|
if (currentString !== undefined) {
|
|
update.push(Buffer.from(currentString));
|
|
}
|
|
for (const item of update) hash.update(item);
|
|
this._cachedHashUpdate = update;
|
|
}
|
|
}
|
|
|
|
module.exports = CachedSource;
|