/*
	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;
