/
getCallsite.ts
67 lines (59 loc) · 1.86 KB
/
getCallsite.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
/**
* Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*/
import {TraceMap, originalPositionFor} from '@jridgewell/trace-mapping';
import callsites = require('callsites');
import {readFileSync} from 'graceful-fs';
import type {SourceMapRegistry} from './types';
// Copied from https://github.com/rexxars/sourcemap-decorate-callsites/blob/5b9735a156964973a75dc62fd2c7f0c1975458e8/lib/index.js#L113-L158
const addSourceMapConsumer = (
callsite: callsites.CallSite,
tracer: TraceMap,
) => {
const getLineNumber = callsite.getLineNumber;
const getColumnNumber = callsite.getColumnNumber;
let position: ReturnType<typeof originalPositionFor> | null = null;
function getPosition() {
if (!position) {
position = originalPositionFor(tracer, {
column: getColumnNumber.call(callsite) || -1,
line: getLineNumber.call(callsite) || -1,
});
}
return position;
}
Object.defineProperties(callsite, {
getColumnNumber: {
value() {
return getPosition().column || getColumnNumber.call(callsite);
},
writable: false,
},
getLineNumber: {
value() {
return getPosition().line || getLineNumber.call(callsite);
},
writable: false,
},
});
};
export default function getCallsite(
level: number,
sourceMaps?: SourceMapRegistry | null,
): callsites.CallSite {
const levelAfterThisCall = level + 1;
const stack = callsites()[levelAfterThisCall];
const sourceMapFileName = sourceMaps?.get(stack.getFileName() || '');
if (sourceMapFileName) {
try {
const sourceMap = readFileSync(sourceMapFileName, 'utf8');
addSourceMapConsumer(stack, new TraceMap(sourceMap));
} catch {
// ignore
}
}
return stack;
}