77 lines
1.8 KiB
JavaScript
77 lines
1.8 KiB
JavaScript
|
/*
|
||
|
|
MIT License http://www.opensource.org/licenses/mit-license.php
|
||
|
|
Author Tobias Koppers @sokra
|
||
|
|
*/
|
||
|
|
"use strict";
|
||
|
|
|
||
|
|
var Source = require("./Source");
|
||
|
|
var SourceNode = require("source-map").SourceNode;
|
||
|
|
|
||
|
|
var REPLACE_REGEX = /\n(?=.|\s)/g;
|
||
|
|
|
||
|
|
function cloneAndPrefix(node, prefix, append) {
|
||
|
|
if(typeof node === "string") {
|
||
|
|
var result = node.replace(REPLACE_REGEX, "\n" + prefix);
|
||
|
|
if(append.length > 0) result = append.pop() + result;
|
||
|
|
if(/\n$/.test(node)) append.push(prefix);
|
||
|
|
return result;
|
||
|
|
} else {
|
||
|
|
var newNode = new SourceNode(
|
||
|
|
node.line,
|
||
|
|
node.column,
|
||
|
|
node.source,
|
||
|
|
node.children.map(function(node) {
|
||
|
|
return cloneAndPrefix(node, prefix, append);
|
||
|
|
}),
|
||
|
|
node.name
|
||
|
|
);
|
||
|
|
newNode.sourceContents = node.sourceContents;
|
||
|
|
return newNode;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
class PrefixSource extends Source {
|
||
|
|
constructor(prefix, source) {
|
||
|
|
super();
|
||
|
|
this._source = source;
|
||
|
|
this._prefix = prefix;
|
||
|
|
}
|
||
|
|
|
||
|
|
source() {
|
||
|
|
var node = typeof this._source === "string" ? this._source : this._source.source();
|
||
|
|
var prefix = this._prefix;
|
||
|
|
return prefix + node.replace(REPLACE_REGEX, "\n" + prefix);
|
||
|
|
}
|
||
|
|
|
||
|
|
node(options) {
|
||
|
|
var node = this._source.node(options);
|
||
|
|
var append = [this._prefix];
|
||
|
|
return new SourceNode(null, null, null, [
|
||
|
|
cloneAndPrefix(node, this._prefix, append)
|
||
|
|
]);
|
||
|
|
}
|
||
|
|
|
||
|
|
listMap(options) {
|
||
|
|
var prefix = this._prefix;
|
||
|
|
var map = this._source.listMap(options);
|
||
|
|
return map.mapGeneratedCode(function(code) {
|
||
|
|
return prefix + code.replace(REPLACE_REGEX, "\n" + prefix);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
updateHash(hash) {
|
||
|
|
if(typeof this._source === "string")
|
||
|
|
hash.update(this._source);
|
||
|
|
else
|
||
|
|
this._source.updateHash(hash);
|
||
|
|
if(typeof this._prefix === "string")
|
||
|
|
hash.update(this._prefix);
|
||
|
|
else
|
||
|
|
this._prefix.updateHash(hash);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
require("./SourceAndMapMixin")(PrefixSource.prototype);
|
||
|
|
|
||
|
|
module.exports = PrefixSource;
|