prism-normalize-whitespace.js 4.7 KB

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