69 lines
1.2 KiB
JavaScript
69 lines
1.2 KiB
JavaScript
/* @flow */
|
|
|
|
import sourceMap from "source-map";
|
|
import * as t from "babel-types";
|
|
|
|
/**
|
|
* Build a sourcemap.
|
|
*/
|
|
|
|
export default class SourceMap {
|
|
constructor(position, opts, code) {
|
|
this.position = position;
|
|
this.opts = opts;
|
|
|
|
if (opts.sourceMaps) {
|
|
this.map = new sourceMap.SourceMapGenerator({
|
|
file: opts.sourceMapTarget,
|
|
sourceRoot: opts.sourceRoot
|
|
});
|
|
|
|
this.map.setSourceContent(opts.sourceFileName, code);
|
|
} else {
|
|
this.map = null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the sourcemap.
|
|
*/
|
|
|
|
get() {
|
|
let map = this.map;
|
|
if (map) {
|
|
return map.toJSON();
|
|
} else {
|
|
return map;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Mark a node's generated position, and add it to the sourcemap.
|
|
*/
|
|
|
|
mark(node, type) {
|
|
let loc = node.loc;
|
|
if (!loc) return; // no location info
|
|
|
|
let map = this.map;
|
|
if (!map) return; // no source map
|
|
|
|
if (t.isProgram(node) || t.isFile(node)) return; // illegal mapping nodes
|
|
|
|
let position = this.position;
|
|
|
|
let generated = {
|
|
line: position.line,
|
|
column: position.column
|
|
};
|
|
|
|
let original = loc[type];
|
|
|
|
map.addMapping({
|
|
source: this.opts.sourceFileName,
|
|
generated: generated,
|
|
original: original
|
|
});
|
|
}
|
|
}
|