index.js 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.codeFrameColumns = codeFrameColumns;
  6. exports.default = _default;
  7. function _highlight() {
  8. var data = _interopRequireWildcard(require("@babel/highlight"));
  9. _highlight = function _highlight() {
  10. return data;
  11. };
  12. return data;
  13. }
  14. function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
  15. var deprecationWarningShown = false;
  16. function getDefs(chalk) {
  17. return {
  18. gutter: chalk.grey,
  19. marker: chalk.red.bold,
  20. message: chalk.red.bold
  21. };
  22. }
  23. var NEWLINE = /\r\n|[\n\r\u2028\u2029]/;
  24. function getMarkerLines(loc, source, opts) {
  25. var startLoc = Object.assign({}, {
  26. column: 0,
  27. line: -1
  28. }, loc.start);
  29. var endLoc = Object.assign({}, startLoc, loc.end);
  30. var _ref = opts || {},
  31. _ref$linesAbove = _ref.linesAbove,
  32. linesAbove = _ref$linesAbove === void 0 ? 2 : _ref$linesAbove,
  33. _ref$linesBelow = _ref.linesBelow,
  34. linesBelow = _ref$linesBelow === void 0 ? 3 : _ref$linesBelow;
  35. var startLine = startLoc.line;
  36. var startColumn = startLoc.column;
  37. var endLine = endLoc.line;
  38. var endColumn = endLoc.column;
  39. var start = Math.max(startLine - (linesAbove + 1), 0);
  40. var end = Math.min(source.length, endLine + linesBelow);
  41. if (startLine === -1) {
  42. start = 0;
  43. }
  44. if (endLine === -1) {
  45. end = source.length;
  46. }
  47. var lineDiff = endLine - startLine;
  48. var markerLines = {};
  49. if (lineDiff) {
  50. for (var i = 0; i <= lineDiff; i++) {
  51. var lineNumber = i + startLine;
  52. if (!startColumn) {
  53. markerLines[lineNumber] = true;
  54. } else if (i === 0) {
  55. var sourceLength = source[lineNumber - 1].length;
  56. markerLines[lineNumber] = [startColumn, sourceLength - startColumn];
  57. } else if (i === lineDiff) {
  58. markerLines[lineNumber] = [0, endColumn];
  59. } else {
  60. var _sourceLength = source[lineNumber - i].length;
  61. markerLines[lineNumber] = [0, _sourceLength];
  62. }
  63. }
  64. } else {
  65. if (startColumn === endColumn) {
  66. if (startColumn) {
  67. markerLines[startLine] = [startColumn, 0];
  68. } else {
  69. markerLines[startLine] = true;
  70. }
  71. } else {
  72. markerLines[startLine] = [startColumn, endColumn - startColumn];
  73. }
  74. }
  75. return {
  76. start: start,
  77. end: end,
  78. markerLines: markerLines
  79. };
  80. }
  81. function codeFrameColumns(rawLines, loc, opts) {
  82. if (opts === void 0) {
  83. opts = {};
  84. }
  85. var highlighted = (opts.highlightCode || opts.forceColor) && (0, _highlight().shouldHighlight)(opts);
  86. var chalk = (0, _highlight().getChalk)(opts);
  87. var defs = getDefs(chalk);
  88. var maybeHighlight = function maybeHighlight(chalkFn, string) {
  89. return highlighted ? chalkFn(string) : string;
  90. };
  91. if (highlighted) rawLines = (0, _highlight().default)(rawLines, opts);
  92. var lines = rawLines.split(NEWLINE);
  93. var _getMarkerLines = getMarkerLines(loc, lines, opts),
  94. start = _getMarkerLines.start,
  95. end = _getMarkerLines.end,
  96. markerLines = _getMarkerLines.markerLines;
  97. var hasColumns = loc.start && typeof loc.start.column === "number";
  98. var numberMaxWidth = String(end).length;
  99. var frame = lines.slice(start, end).map(function (line, index) {
  100. var number = start + 1 + index;
  101. var paddedNumber = (" " + number).slice(-numberMaxWidth);
  102. var gutter = " " + paddedNumber + " | ";
  103. var hasMarker = markerLines[number];
  104. var lastMarkerLine = !markerLines[number + 1];
  105. if (hasMarker) {
  106. var markerLine = "";
  107. if (Array.isArray(hasMarker)) {
  108. var markerSpacing = line.slice(0, Math.max(hasMarker[0] - 1, 0)).replace(/[^\t]/g, " ");
  109. var numberOfMarkers = hasMarker[1] || 1;
  110. markerLine = ["\n ", maybeHighlight(defs.gutter, gutter.replace(/\d/g, " ")), markerSpacing, maybeHighlight(defs.marker, "^").repeat(numberOfMarkers)].join("");
  111. if (lastMarkerLine && opts.message) {
  112. markerLine += " " + maybeHighlight(defs.message, opts.message);
  113. }
  114. }
  115. return [maybeHighlight(defs.marker, ">"), maybeHighlight(defs.gutter, gutter), line, markerLine].join("");
  116. } else {
  117. return " " + maybeHighlight(defs.gutter, gutter) + line;
  118. }
  119. }).join("\n");
  120. if (opts.message && !hasColumns) {
  121. frame = "" + " ".repeat(numberMaxWidth + 1) + opts.message + "\n" + frame;
  122. }
  123. if (highlighted) {
  124. return chalk.reset(frame);
  125. } else {
  126. return frame;
  127. }
  128. }
  129. function _default(rawLines, lineNumber, colNumber, opts) {
  130. if (opts === void 0) {
  131. opts = {};
  132. }
  133. if (!deprecationWarningShown) {
  134. deprecationWarningShown = true;
  135. var message = "Passing lineNumber and colNumber is deprecated to @babel/code-frame. Please use `codeFrameColumns`.";
  136. if (process.emitWarning) {
  137. process.emitWarning(message, "DeprecationWarning");
  138. } else {
  139. var deprecationError = new Error(message);
  140. deprecationError.name = "DeprecationWarning";
  141. console.warn(new Error(message));
  142. }
  143. }
  144. colNumber = Math.max(colNumber, 0);
  145. var location = {
  146. start: {
  147. column: colNumber,
  148. line: lineNumber
  149. }
  150. };
  151. return codeFrameColumns(rawLines, location, opts);
  152. }