urls-processor.js 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. var EscapeStore = require('./escape-store');
  2. var reduceUrls = require('../urls/reduce');
  3. var lineBreak = require('os').EOL;
  4. function UrlsProcessor(context, saveWaypoints, keepUrlQuotes) {
  5. this.urls = new EscapeStore('URL');
  6. this.context = context;
  7. this.saveWaypoints = saveWaypoints;
  8. this.keepUrlQuotes = keepUrlQuotes;
  9. }
  10. // Strip urls by replacing them by a special
  11. // marker for further restoring. It's done via string scanning
  12. // instead of regexps to speed up the process.
  13. UrlsProcessor.prototype.escape = function (data) {
  14. var breaksCount;
  15. var lastBreakAt;
  16. var indent;
  17. var saveWaypoints = this.saveWaypoints;
  18. var self = this;
  19. return reduceUrls(data, this.context, function (url, tempData) {
  20. if (saveWaypoints) {
  21. breaksCount = url.split(lineBreak).length - 1;
  22. lastBreakAt = url.lastIndexOf(lineBreak);
  23. indent = lastBreakAt > 0 ?
  24. url.substring(lastBreakAt + lineBreak.length).length :
  25. url.length;
  26. }
  27. var placeholder = self.urls.store(url, saveWaypoints ? [breaksCount, indent] : null);
  28. tempData.push(placeholder);
  29. });
  30. };
  31. function normalize(url, keepUrlQuotes) {
  32. url = url
  33. .replace(/^url/gi, 'url')
  34. .replace(/\\?\n|\\?\r\n/g, '')
  35. .replace(/(\s{2,}|\s)/g, ' ')
  36. .replace(/^url\((['"])? /, 'url($1')
  37. .replace(/ (['"])?\)$/, '$1)');
  38. if (/url\(".*'.*"\)/.test(url) || /url\('.*".*'\)/.test(url))
  39. return url;
  40. if (!keepUrlQuotes && !/^['"].+['"]$/.test(url) && !/url\(.*[\s\(\)].*\)/.test(url) && !/url\(['"]data:[^;]+;charset/.test(url))
  41. url = url.replace(/["']/g, '');
  42. return url;
  43. }
  44. UrlsProcessor.prototype.restore = function (data) {
  45. var tempData = [];
  46. var cursor = 0;
  47. for (; cursor < data.length;) {
  48. var nextMatch = this.urls.nextMatch(data, cursor);
  49. if (nextMatch.start < 0)
  50. break;
  51. tempData.push(data.substring(cursor, nextMatch.start));
  52. var url = normalize(this.urls.restore(nextMatch.match), this.keepUrlQuotes);
  53. tempData.push(url);
  54. cursor = nextMatch.end;
  55. }
  56. return tempData.length > 0 ?
  57. tempData.join('') + data.substring(cursor, data.length) :
  58. data;
  59. };
  60. module.exports = UrlsProcessor;