Rule.js 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. const ChainedMap = require('./ChainedMap');
  2. const ChainedSet = require('./ChainedSet');
  3. const Use = require('./Use');
  4. module.exports = class Rule extends ChainedMap {
  5. constructor(parent, name) {
  6. super(parent);
  7. this.name = name;
  8. this.names = [];
  9. let rule = this;
  10. while (rule instanceof Rule) {
  11. this.names.unshift(rule.name);
  12. rule = rule.parent;
  13. }
  14. this.uses = new ChainedMap(this);
  15. this.include = new ChainedSet(this);
  16. this.exclude = new ChainedSet(this);
  17. this.oneOfs = new ChainedMap(this);
  18. this.extend([
  19. 'enforce',
  20. 'issuer',
  21. 'parser',
  22. 'resource',
  23. 'resourceQuery',
  24. 'sideEffects',
  25. 'test',
  26. 'type',
  27. ]);
  28. }
  29. use(name) {
  30. return this.uses.getOrCompute(name, () => new Use(this, name));
  31. }
  32. oneOf(name) {
  33. return this.oneOfs.getOrCompute(name, () => new Rule(this, name));
  34. }
  35. pre() {
  36. return this.enforce('pre');
  37. }
  38. post() {
  39. return this.enforce('post');
  40. }
  41. toConfig() {
  42. const config = this.clean(
  43. Object.assign(this.entries() || {}, {
  44. include: this.include.values(),
  45. exclude: this.exclude.values(),
  46. oneOf: this.oneOfs.values().map(oneOf => oneOf.toConfig()),
  47. use: this.uses.values().map(use => use.toConfig()),
  48. })
  49. );
  50. Object.defineProperties(config, {
  51. __ruleNames: { value: this.names },
  52. });
  53. return config;
  54. }
  55. merge(obj, omit = []) {
  56. if (!omit.includes('include') && 'include' in obj) {
  57. this.include.merge(obj.include);
  58. }
  59. if (!omit.includes('exclude') && 'exclude' in obj) {
  60. this.exclude.merge(obj.exclude);
  61. }
  62. if (!omit.includes('use') && 'use' in obj) {
  63. Object.keys(obj.use).forEach(name => this.use(name).merge(obj.use[name]));
  64. }
  65. if (!omit.includes('oneOf') && 'oneOf' in obj) {
  66. Object.keys(obj.oneOf).forEach(name =>
  67. this.oneOf(name).merge(obj.oneOf[name])
  68. );
  69. }
  70. if (!omit.includes('test') && 'test' in obj) {
  71. this.test(obj.test instanceof RegExp ? obj.test : new RegExp(obj.test));
  72. }
  73. return super.merge(obj, [
  74. ...omit,
  75. 'include',
  76. 'exclude',
  77. 'use',
  78. 'oneOf',
  79. 'test',
  80. ]);
  81. }
  82. };