index.js 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. /**
  2. * The MIT License (MIT)
  3. * Copyright (c) 2017-present Dmitry Soshnikov <dmitry.soshnikov@gmail.com>
  4. */
  5. 'use strict';
  6. var clone = require('../utils/clone');
  7. var parser = require('../parser');
  8. var transform = require('../transform');
  9. var optimizationTransforms = require('./transforms');
  10. module.exports = {
  11. /**
  12. * Optimizer transforms a regular expression into an optimized version,
  13. * replacing some sub-expressions with their idiomatic patterns.
  14. *
  15. * @param string | RegExp | AST - a regexp to optimize.
  16. *
  17. * @return TransformResult - an optimized regexp.
  18. *
  19. * Example:
  20. *
  21. * /[a-zA-Z_0-9][a-zA-Z_0-9]*\e{1,}/
  22. *
  23. * Optimized to:
  24. *
  25. * /\w+e+/
  26. */
  27. optimize: function optimize(regexp) {
  28. var transformsWhitelist = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
  29. var transformToApply = transformsWhitelist.length > 0 ? transformsWhitelist : Object.keys(optimizationTransforms);
  30. var ast = regexp;
  31. if (regexp instanceof RegExp) {
  32. regexp = '' + regexp;
  33. }
  34. if (typeof regexp === 'string') {
  35. ast = parser.parse(regexp);
  36. }
  37. var result = new transform.TransformResult(ast);
  38. var prevResultString = void 0;
  39. do {
  40. // Get a copy of the current state here so
  41. // we can compare it with the state at the
  42. // end of the loop.
  43. prevResultString = result.toString();
  44. ast = clone(result.getAST());
  45. transformToApply.forEach(function (transformName) {
  46. if (!optimizationTransforms.hasOwnProperty(transformName)) {
  47. throw new Error('Unknown optimization-transform: ' + transformName + '. ' + 'Available transforms are: ' + Object.keys(optimizationTransforms).join(', '));
  48. }
  49. var transformer = optimizationTransforms[transformName];
  50. // Don't override result just yet since we
  51. // might want to rollback the transform
  52. var newResult = transform.transform(ast, transformer);
  53. if (newResult.toString() !== result.toString()) {
  54. if (newResult.toString().length <= result.toString().length) {
  55. result = newResult;
  56. } else {
  57. // Result has changed but is not shorter:
  58. // restore ast to its previous state.
  59. ast = clone(result.getAST());
  60. }
  61. }
  62. });
  63. // Keep running the optimizer until it stops
  64. // making any change to the regexp.
  65. } while (result.toString() !== prevResultString);
  66. return result;
  67. }
  68. };