index.js 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137
  1. "use strict";
  2. exports.__esModule = true;
  3. exports.default = function (rawLines, lineNumber, colNumber) {
  4. var opts = arguments.length <= 3 || arguments[3] === undefined ? {} : arguments[3];
  5. colNumber = Math.max(colNumber, 0);
  6. var highlighted = opts.highlightCode && _chalk2.default.supportsColor;
  7. if (highlighted) rawLines = highlight(rawLines);
  8. var lines = rawLines.split(NEWLINE);
  9. var start = Math.max(lineNumber - 3, 0);
  10. var end = Math.min(lines.length, lineNumber + 3);
  11. if (!lineNumber && !colNumber) {
  12. start = 0;
  13. end = lines.length;
  14. }
  15. var numberMaxWidth = String(end).length;
  16. var frame = lines.slice(start, end).map(function (line, index) {
  17. var number = start + 1 + index;
  18. var paddedNumber = (" " + number).slice(-numberMaxWidth);
  19. var gutter = " " + paddedNumber + " | ";
  20. if (number === lineNumber) {
  21. var markerLine = "";
  22. if (colNumber) {
  23. var markerSpacing = line.slice(0, colNumber - 1).replace(/[^\t]/g, " ");
  24. markerLine = "\n " + gutter.replace(/\d/g, " ") + markerSpacing + "^";
  25. }
  26. return ">" + gutter + line + markerLine;
  27. } else {
  28. return " " + gutter + line;
  29. }
  30. }).join("\n");
  31. if (highlighted) {
  32. return _chalk2.default.reset(frame);
  33. } else {
  34. return frame;
  35. }
  36. };
  37. var _jsTokens = require("js-tokens");
  38. var _jsTokens2 = _interopRequireDefault(_jsTokens);
  39. var _esutils = require("esutils");
  40. var _esutils2 = _interopRequireDefault(_esutils);
  41. var _chalk = require("chalk");
  42. var _chalk2 = _interopRequireDefault(_chalk);
  43. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  44. /**
  45. * Chalk styles for token types.
  46. */
  47. var defs = {
  48. string: _chalk2.default.red,
  49. punctuator: _chalk2.default.bold,
  50. curly: _chalk2.default.green,
  51. parens: _chalk2.default.blue.bold,
  52. square: _chalk2.default.yellow,
  53. keyword: _chalk2.default.cyan,
  54. number: _chalk2.default.magenta,
  55. regex: _chalk2.default.magenta,
  56. comment: _chalk2.default.grey,
  57. invalid: _chalk2.default.inverse
  58. };
  59. /**
  60. * RegExp to test for newlines in terminal.
  61. */
  62. var NEWLINE = /\r\n|[\n\r\u2028\u2029]/;
  63. /**
  64. * Get the type of token, specifying punctuator type.
  65. */
  66. function getTokenType(match) {
  67. var token = _jsTokens2.default.matchToToken(match);
  68. if (token.type === "name" && _esutils2.default.keyword.isReservedWordES6(token.value)) {
  69. return "keyword";
  70. }
  71. if (token.type === "punctuator") {
  72. switch (token.value) {
  73. case "{":
  74. case "}":
  75. return "curly";
  76. case "(":
  77. case ")":
  78. return "parens";
  79. case "[":
  80. case "]":
  81. return "square";
  82. }
  83. }
  84. return token.type;
  85. }
  86. /**
  87. * Highlight `text`.
  88. */
  89. function highlight(text) {
  90. return text.replace(_jsTokens2.default, function () {
  91. for (var _len = arguments.length, args = Array(_len), _key = 0; _key < _len; _key++) {
  92. args[_key] = arguments[_key];
  93. }
  94. var type = getTokenType(args);
  95. var colorize = defs[type];
  96. if (colorize) {
  97. return args[0].split(NEWLINE).map(function (str) {
  98. return colorize(str);
  99. }).join("\n");
  100. } else {
  101. return args[0];
  102. }
  103. });
  104. }
  105. /**
  106. * Create a code frame, adding line numbers, code highlighting, and pointing to a given position.
  107. */
  108. module.exports = exports["default"];