Projekt

Obecné

Profil

Stáhnout (8.81 KB) Statistiky
| Větev: | Revize:
1 3a515b92 cagy
/*
2
	MIT License http://www.opensource.org/licenses/mit-license.php
3
	Author Tobias Koppers @sokra
4
*/
5
"use strict";
6
7
var Source = require("./Source");
8
var SourceNode = require("source-map").SourceNode;
9
10
class Replacement {
11
	constructor(start, end, content, insertIndex, name) {
12
		this.start = start;
13
		this.end = end;
14
		this.content = content;
15
		this.insertIndex = insertIndex;
16
		this.name = name;
17
	}
18
}
19
20
class ReplaceSource extends Source {
21
	constructor(source, name) {
22
		super();
23
		this._source = source;
24
		this._name = name;
25
		/** @type {Replacement[]} */
26
		this.replacements = [];
27
	}
28
29
	replace(start, end, newValue, name) {
30
		if(typeof newValue !== "string")
31
			throw new Error("insertion must be a string, but is a " + typeof newValue);
32
		this.replacements.push(new Replacement(start, end, newValue, this.replacements.length, name));
33
	}
34
35
	insert(pos, newValue, name) {
36
		if(typeof newValue !== "string")
37
			throw new Error("insertion must be a string, but is a " + typeof newValue + ": " + newValue);
38
		this.replacements.push(new Replacement(pos, pos - 1, newValue, this.replacements.length, name));
39
	}
40
41
	source(options) {
42
		return this._replaceString(this._source.source());
43
	}
44
45
	original() {
46
		return this._source;
47
	}
48
49
	_sortReplacements() {
50
		this.replacements.sort(function(a, b) {
51
			var diff = b.end - a.end;
52
			if(diff !== 0)
53
				return diff;
54
			diff = b.start - a.start;
55
			if(diff !== 0)
56
				return diff;
57
			return b.insertIndex - a.insertIndex;
58
		});
59
	}
60
61
	_replaceString(str) {
62
		if(typeof str !== "string")
63
			throw new Error("str must be a string, but is a " + typeof str + ": " + str);
64
		this._sortReplacements();
65
		var result = [str];
66
		this.replacements.forEach(function(repl) {
67
			var remSource = result.pop();
68
			var splitted1 = this._splitString(remSource, Math.floor(repl.end + 1));
69
			var splitted2 = this._splitString(splitted1[0], Math.floor(repl.start));
70
			result.push(splitted1[1], repl.content, splitted2[0]);
71
		}, this);
72
73
		// write out result array in reverse order
74
		let resultStr = "";
75
		for(let i = result.length - 1; i >= 0; --i) {
76
			resultStr += result[i];
77
		}
78
		return resultStr;
79
	}
80
81
	node(options) {
82
		var node = this._source.node(options);
83
		if(this.replacements.length === 0) {
84
			return node;
85
		}
86
		this._sortReplacements();
87
		var replace = new ReplacementEnumerator(this.replacements);
88
		var output = [];
89
		var position = 0;
90
		var sources = Object.create(null);
91
		var sourcesInLines = Object.create(null);
92
93
		// We build a new list of SourceNodes in "output"
94
		// from the original mapping data
95
96
		var result = new SourceNode();
97
98
		// We need to add source contents manually
99
		// because "walk" will not handle it
100
		node.walkSourceContents(function(sourceFile, sourceContent) {
101
			result.setSourceContent(sourceFile, sourceContent);
102
			sources["$" + sourceFile] = sourceContent;
103
		});
104
105
		var replaceInStringNode = this._replaceInStringNode.bind(this, output, replace, function getOriginalSource(mapping) {
106
			var key = "$" + mapping.source;
107
			var lines = sourcesInLines[key];
108
			if(!lines) {
109
				var source = sources[key];
110
				if(!source) return null;
111
				lines = source.split("\n").map(function(line) {
112
					return line + "\n";
113
				});
114
				sourcesInLines[key] = lines;
115
			}
116
			// line is 1-based
117
			if(mapping.line > lines.length) return null;
118
			var line = lines[mapping.line - 1];
119
			return line.substr(mapping.column);
120
		});
121
122
		node.walk(function(chunk, mapping) {
123
			position = replaceInStringNode(chunk, position, mapping);
124
		});
125
126
		// If any replacements occur after the end of the original file, then we append them
127
		// directly to the end of the output
128
		var remaining = replace.footer();
129
		if(remaining) {
130
			output.push(remaining);
131
		}
132
133
		result.add(output);
134
135
		return result;
136
	}
137
138
	listMap(options) {
139
		this._sortReplacements();
140
		var map = this._source.listMap(options);
141
		var currentIndex = 0;
142
		var replacements = this.replacements;
143
		var idxReplacement = replacements.length - 1;
144
		var removeChars = 0;
145
		map = map.mapGeneratedCode(function(str) {
146
			var newCurrentIndex = currentIndex + str.length;
147
			if(removeChars > str.length) {
148
				removeChars -= str.length;
149
				str = "";
150
			} else {
151
				if(removeChars > 0) {
152
					str = str.substr(removeChars);
153
					currentIndex += removeChars;
154
					removeChars = 0;
155
				}
156
				var finalStr = "";
157
				while(idxReplacement >= 0 && replacements[idxReplacement].start < newCurrentIndex) {
158
					var repl = replacements[idxReplacement];
159
					var start = Math.floor(repl.start);
160
					var end = Math.floor(repl.end + 1);
161
					var before = str.substr(0, Math.max(0, start - currentIndex));
162
					if(end <= newCurrentIndex) {
163
						var after = str.substr(Math.max(0, end - currentIndex));
164
						finalStr += before + repl.content;
165
						str = after;
166
						currentIndex = Math.max(currentIndex, end);
167
					} else {
168
						finalStr += before + repl.content;
169
						str = "";
170
						removeChars = end - newCurrentIndex;
171
					}
172
					idxReplacement--;
173
				}
174
				str = finalStr + str;
175
			}
176
			currentIndex = newCurrentIndex;
177
			return str;
178
		});
179
		var extraCode = "";
180
		while(idxReplacement >= 0) {
181
			extraCode += replacements[idxReplacement].content;
182
			idxReplacement--;
183
		}
184
		if(extraCode) {
185
			map.add(extraCode);
186
		}
187
		return map;
188
	}
189
190
	_splitString(str, position) {
191
		return position <= 0 ? ["", str] : [str.substr(0, position), str.substr(position)];
192
	}
193
194
	_replaceInStringNode(output, replace, getOriginalSource, node, position, mapping) {
195
		var original = undefined;
196
197
		do {
198
			var splitPosition = replace.position - position;
199
			// If multiple replaces occur in the same location then the splitPosition may be
200
			// before the current position for the subsequent splits. Ensure it is >= 0
201
			if(splitPosition < 0) {
202
				splitPosition = 0;
203
			}
204
			if(splitPosition >= node.length || replace.done) {
205
				if(replace.emit) {
206
					var nodeEnd = new SourceNode(
207
						mapping.line,
208
						mapping.column,
209
						mapping.source,
210
						node,
211
						mapping.name
212
					);
213
					output.push(nodeEnd);
214
				}
215
				return position + node.length;
216
			}
217
218
			var originalColumn = mapping.column;
219
220
			// Try to figure out if generated code matches original code of this segement
221
			// If this is the case we assume that it's allowed to move mapping.column
222
			// Because getOriginalSource can be expensive we only do it when neccessary
223
224
			var nodePart;
225
			if(splitPosition > 0) {
226
				nodePart = node.slice(0, splitPosition);
227
				if(original === undefined) {
228
					original = getOriginalSource(mapping);
229
				}
230
				if(original && original.length >= splitPosition && original.startsWith(nodePart)) {
231
					mapping.column += splitPosition;
232
					original = original.substr(splitPosition);
233
				}
234
			}
235
236
			var emit = replace.next();
237
			if(!emit) {
238
				// Stop emitting when we have found the beginning of the string to replace.
239
				// Emit the part of the string before splitPosition
240
				if(splitPosition > 0) {
241
					var nodeStart = new SourceNode(
242
						mapping.line,
243
						originalColumn,
244
						mapping.source,
245
						nodePart,
246
						mapping.name
247
					);
248
					output.push(nodeStart);
249
				}
250
251
				// Emit the replacement value
252
				if(replace.value) {
253
					output.push(new SourceNode(
254
						mapping.line,
255
						mapping.column,
256
						mapping.source,
257
						replace.value,
258
						mapping.name || replace.name
259
					));
260
				}
261
			}
262
263
			// Recurse with remainder of the string as there may be multiple replaces within a single node
264
			node = node.substr(splitPosition);
265
			position += splitPosition;
266
		} while (true);
267
	}
268
}
269
270
class ReplacementEnumerator {
271
	/**
272
	 * @param {Replacement[]} replacements list of replacements
273
	 */
274
	constructor(replacements) {
275
		this.replacements = replacements || [];
276
		this.index = this.replacements.length;
277
		this.done = false;
278
		this.emit = false;
279
		// Set initial start position
280
		this.next();
281
	}
282
283
	next() {
284
		if(this.done)
285
			return true;
286
		if(this.emit) {
287
			// Start point found. stop emitting. set position to find end
288
			var repl = this.replacements[this.index];
289
			var end = Math.floor(repl.end + 1);
290
			this.position = end;
291
			this.value = repl.content;
292
			this.name = repl.name;
293
		} else {
294
			// End point found. start emitting. set position to find next start
295
			this.index--;
296
			if(this.index < 0) {
297
				this.done = true;
298
			} else {
299
				var nextRepl = this.replacements[this.index];
300
				var start = Math.floor(nextRepl.start);
301
				this.position = start;
302
			}
303
		}
304
		if(this.position < 0)
305
			this.position = 0;
306
		this.emit = !this.emit;
307
		return this.emit;
308
	}
309
310
	footer() {
311
		if(!this.done && !this.emit)
312
			this.next(); // If we finished _replaceInNode mid emit we advance to next entry
313
		if(this.done) {
314
			return [];
315
		} else {
316
			var resultStr = "";
317
			for(var i = this.index; i >= 0; i--) {
318
				var repl = this.replacements[i];
319
				// this doesn't need to handle repl.name, because in SourceMaps generated code
320
				// without pointer to original source can't have a name
321
				resultStr += repl.content;
322
			}
323
			return resultStr;
324
		}
325
	}
326
}
327
328
require("./SourceAndMapMixin")(ReplaceSource.prototype);
329
330
module.exports = ReplaceSource;