prism-normalize-whitespace.js 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194
  1. (function() {
  2. var assign = Object.assign || function (obj1, obj2) {
  3. for (var name in obj2) {
  4. if (obj2.hasOwnProperty(name))
  5. obj1[name] = obj2[name];
  6. }
  7. return obj1;
  8. }
  9. function NormalizeWhitespace(defaults) {
  10. this.defaults = assign({}, defaults);
  11. }
  12. function toCamelCase(value) {
  13. return value.replace(/-(\w)/g, function(match, firstChar) {
  14. return firstChar.toUpperCase();
  15. });
  16. }
  17. function tabLen(str) {
  18. var res = 0;
  19. for (var i = 0; i < str.length; ++i) {
  20. if (str.charCodeAt(i) == '\t'.charCodeAt(0))
  21. res += 3;
  22. }
  23. return str.length + res;
  24. }
  25. NormalizeWhitespace.prototype = {
  26. setDefaults: function (defaults) {
  27. this.defaults = assign(this.defaults, defaults);
  28. },
  29. normalize: function (input, settings) {
  30. settings = assign(this.defaults, settings);
  31. for (var name in settings) {
  32. var methodName = toCamelCase(name);
  33. if (name !== "normalize" && methodName !== 'setDefaults' &&
  34. settings[name] && this[methodName]) {
  35. input = this[methodName].call(this, input, settings[name]);
  36. }
  37. }
  38. return input;
  39. },
  40. /*
  41. * Normalization methods
  42. */
  43. leftTrim: function (input) {
  44. return input.replace(/^\s+/, '');
  45. },
  46. rightTrim: function (input) {
  47. return input.replace(/\s+$/, '');
  48. },
  49. tabsToSpaces: function (input, spaces) {
  50. spaces = spaces|0 || 4;
  51. return input.replace(/\t/g, new Array(++spaces).join(' '));
  52. },
  53. spacesToTabs: function (input, spaces) {
  54. spaces = spaces|0 || 4;
  55. return input.replace(RegExp(' {' + spaces + '}', 'g'), '\t');
  56. },
  57. removeTrailing: function (input) {
  58. return input.replace(/\s*?$/gm, '');
  59. },
  60. // Support for deprecated plugin remove-initial-line-feed
  61. removeInitialLineFeed: function (input) {
  62. return input.replace(/^(?:\r?\n|\r)/, '');
  63. },
  64. removeIndent: function (input) {
  65. var indents = input.match(/^[^\S\n\r]*(?=\S)/gm);
  66. if (!indents || !indents[0].length)
  67. return input;
  68. indents.sort(function(a, b){return a.length - b.length; });
  69. if (!indents[0].length)
  70. return input;
  71. return input.replace(RegExp('^' + indents[0], 'gm'), '');
  72. },
  73. indent: function (input, tabs) {
  74. return input.replace(/^[^\S\n\r]*(?=\S)/gm, new Array(++tabs).join('\t') + '$&');
  75. },
  76. breakLines: function (input, characters) {
  77. characters = (characters === true) ? 80 : characters|0 || 80;
  78. var lines = input.split('\n');
  79. for (var i = 0; i < lines.length; ++i) {
  80. if (tabLen(lines[i]) <= characters)
  81. continue;
  82. var line = lines[i].split(/(\s+)/g),
  83. len = 0;
  84. for (var j = 0; j < line.length; ++j) {
  85. var tl = tabLen(line[j]);
  86. len += tl;
  87. if (len > characters) {
  88. line[j] = '\n' + line[j];
  89. len = tl;
  90. }
  91. }
  92. lines[i] = line.join('');
  93. }
  94. return lines.join('\n');
  95. }
  96. };
  97. // Support node modules
  98. if (typeof module !== 'undefined' && module.exports) {
  99. module.exports = NormalizeWhitespace;
  100. }
  101. // Exit if prism is not loaded
  102. if (typeof Prism === 'undefined') {
  103. return;
  104. }
  105. Prism.plugins.NormalizeWhitespace = new NormalizeWhitespace({
  106. 'remove-trailing': true,
  107. 'remove-indent': true,
  108. 'left-trim': true,
  109. 'right-trim': true,
  110. /*'break-lines': 80,
  111. 'indent': 2,
  112. 'remove-initial-line-feed': false,
  113. 'tabs-to-spaces': 4,
  114. 'spaces-to-tabs': 4*/
  115. });
  116. Prism.hooks.add('before-sanity-check', function (env) {
  117. var Normalizer = Prism.plugins.NormalizeWhitespace;
  118. // Check settings
  119. if (env.settings && env.settings['whitespace-normalization'] === false) {
  120. return;
  121. }
  122. // Check classes
  123. if (!Prism.util.isActive(env.element, 'whitespace-normalization', true)) {
  124. return;
  125. }
  126. // Simple mode if there is no env.element
  127. if ((!env.element || !env.element.parentNode) && env.code) {
  128. env.code = Normalizer.normalize(env.code, env.settings);
  129. return;
  130. }
  131. // Normal mode
  132. var pre = env.element.parentNode;
  133. if (!env.code || !pre || pre.nodeName.toLowerCase() !== 'pre') {
  134. return;
  135. }
  136. var children = pre.childNodes,
  137. before = '',
  138. after = '',
  139. codeFound = false;
  140. // Move surrounding whitespace from the <pre> tag into the <code> tag
  141. for (var i = 0; i < children.length; ++i) {
  142. var node = children[i];
  143. if (node == env.element) {
  144. codeFound = true;
  145. } else if (node.nodeName === "#text") {
  146. if (codeFound) {
  147. after += node.nodeValue;
  148. } else {
  149. before += node.nodeValue;
  150. }
  151. pre.removeChild(node);
  152. --i;
  153. }
  154. }
  155. if (!env.element.children.length || !Prism.plugins.KeepMarkup) {
  156. env.code = before + env.code + after;
  157. env.code = Normalizer.normalize(env.code, env.settings);
  158. } else {
  159. // Preserve markup for keep-markup plugin
  160. var html = before + env.element.innerHTML + after;
  161. env.element.innerHTML = Normalizer.normalize(html, env.settings);
  162. env.code = env.element.textContent;
  163. }
  164. });
  165. }());