prism-markup-templating.js 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. (function (Prism) {
  2. /**
  3. * Returns the placeholder for the given language id and index.
  4. *
  5. * @param {string} language
  6. * @param {string|number} index
  7. * @returns {string}
  8. */
  9. function getPlaceholder(language, index) {
  10. return '___' + language.toUpperCase() + index + '___';
  11. }
  12. Object.defineProperties(Prism.languages['markup-templating'] = {}, {
  13. buildPlaceholders: {
  14. /**
  15. * Tokenize all inline templating expressions matching `placeholderPattern`.
  16. *
  17. * If `replaceFilter` is provided, only matches of `placeholderPattern` for which `replaceFilter` returns
  18. * `true` will be replaced.
  19. *
  20. * @param {object} env The environment of the `before-tokenize` hook.
  21. * @param {string} language The language id.
  22. * @param {RegExp} placeholderPattern The matches of this pattern will be replaced by placeholders.
  23. * @param {(match: string) => boolean} [replaceFilter]
  24. */
  25. value: function (env, language, placeholderPattern, replaceFilter) {
  26. if (env.language !== language) {
  27. return;
  28. }
  29. var tokenStack = env.tokenStack = [];
  30. env.code = env.code.replace(placeholderPattern, function (match) {
  31. if (typeof replaceFilter === 'function' && !replaceFilter(match)) {
  32. return match;
  33. }
  34. var i = tokenStack.length;
  35. var placeholder;
  36. // Check for existing strings
  37. while (env.code.indexOf(placeholder = getPlaceholder(language, i)) !== -1)
  38. ++i;
  39. // Create a sparse array
  40. tokenStack[i] = match;
  41. return placeholder;
  42. });
  43. // Switch the grammar to markup
  44. env.grammar = Prism.languages.markup;
  45. }
  46. },
  47. tokenizePlaceholders: {
  48. /**
  49. * Replace placeholders with proper tokens after tokenizing.
  50. *
  51. * @param {object} env The environment of the `after-tokenize` hook.
  52. * @param {string} language The language id.
  53. */
  54. value: function (env, language) {
  55. if (env.language !== language || !env.tokenStack) {
  56. return;
  57. }
  58. // Switch the grammar back
  59. env.grammar = Prism.languages[language];
  60. var j = 0;
  61. var keys = Object.keys(env.tokenStack);
  62. function walkTokens(tokens) {
  63. for (var i = 0; i < tokens.length; i++) {
  64. // all placeholders are replaced already
  65. if (j >= keys.length) {
  66. break;
  67. }
  68. var token = tokens[i];
  69. if (typeof token === 'string' || (token.content && typeof token.content === 'string')) {
  70. var k = keys[j];
  71. var t = env.tokenStack[k];
  72. var s = typeof token === 'string' ? token : token.content;
  73. var placeholder = getPlaceholder(language, k);
  74. var index = s.indexOf(placeholder);
  75. if (index > -1) {
  76. ++j;
  77. var before = s.substring(0, index);
  78. var middle = new Prism.Token(language, Prism.tokenize(t, env.grammar), 'language-' + language, t);
  79. var after = s.substring(index + placeholder.length);
  80. var replacement = [];
  81. if (before) {
  82. replacement.push.apply(replacement, walkTokens([before]));
  83. }
  84. replacement.push(middle);
  85. if (after) {
  86. replacement.push.apply(replacement, walkTokens([after]));
  87. }
  88. if (typeof token === 'string') {
  89. tokens.splice.apply(tokens, [i, 1].concat(replacement));
  90. } else {
  91. token.content = replacement;
  92. }
  93. }
  94. } else if (token.content /* && typeof token.content !== 'string' */) {
  95. walkTokens(token.content);
  96. }
  97. }
  98. return tokens;
  99. }
  100. walkTokens(env.tokens);
  101. }
  102. }
  103. });
  104. }(Prism));