rules.js 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. /**
  2. * @fileoverview Defines a storage for rules.
  3. * @author Nicholas C. Zakas
  4. */
  5. "use strict";
  6. //------------------------------------------------------------------------------
  7. // Requirements
  8. //------------------------------------------------------------------------------
  9. const lodash = require("lodash");
  10. const ruleReplacements = require("../conf/replacements").rules;
  11. const builtInRules = require("./built-in-rules-index");
  12. //------------------------------------------------------------------------------
  13. // Helpers
  14. //------------------------------------------------------------------------------
  15. /**
  16. * Creates a stub rule that gets used when a rule with a given ID is not found.
  17. * @param {string} ruleId The ID of the missing rule
  18. * @returns {{create: function(RuleContext): Object}} A rule that reports an error at the first location
  19. * in the program. The report has the message `Definition for rule '${ruleId}' was not found` if the rule is unknown,
  20. * or `Rule '${ruleId}' was removed and replaced by: ${replacements.join(", ")}` if the rule is known to have been
  21. * replaced.
  22. */
  23. const createMissingRule = lodash.memoize(ruleId => {
  24. const message = Object.prototype.hasOwnProperty.call(ruleReplacements, ruleId)
  25. ? `Rule '${ruleId}' was removed and replaced by: ${ruleReplacements[ruleId].join(", ")}`
  26. : `Definition for rule '${ruleId}' was not found`;
  27. return {
  28. create: context => ({
  29. Program() {
  30. context.report({
  31. loc: { line: 1, column: 0 },
  32. message
  33. });
  34. }
  35. })
  36. };
  37. });
  38. /**
  39. * Normalizes a rule module to the new-style API
  40. * @param {(Function|{create: Function})} rule A rule object, which can either be a function
  41. * ("old-style") or an object with a `create` method ("new-style")
  42. * @returns {{create: Function}} A new-style rule.
  43. */
  44. function normalizeRule(rule) {
  45. return typeof rule === "function" ? Object.assign({ create: rule }, rule) : rule;
  46. }
  47. //------------------------------------------------------------------------------
  48. // Public Interface
  49. //------------------------------------------------------------------------------
  50. class Rules {
  51. constructor() {
  52. this._rules = Object.create(null);
  53. Object.keys(builtInRules).forEach(ruleId => {
  54. this.define(ruleId, builtInRules[ruleId]);
  55. });
  56. }
  57. /**
  58. * Registers a rule module for rule id in storage.
  59. * @param {string} ruleId Rule id (file name).
  60. * @param {Function} ruleModule Rule handler.
  61. * @returns {void}
  62. */
  63. define(ruleId, ruleModule) {
  64. this._rules[ruleId] = normalizeRule(ruleModule);
  65. }
  66. /**
  67. * Access rule handler by id (file name).
  68. * @param {string} ruleId Rule id (file name).
  69. * @returns {{create: Function, schema: JsonSchema[]}}
  70. * A rule. This is normalized to always have the new-style shape with a `create` method.
  71. */
  72. get(ruleId) {
  73. if (!Object.prototype.hasOwnProperty.call(this._rules, ruleId)) {
  74. return createMissingRule(ruleId);
  75. }
  76. if (typeof this._rules[ruleId] === "string") {
  77. return normalizeRule(require(this._rules[ruleId]));
  78. }
  79. return this._rules[ruleId];
  80. }
  81. /**
  82. * Get an object with all currently loaded rules
  83. * @returns {Map} All loaded rules
  84. */
  85. getAllLoadedRules() {
  86. const allRules = new Map();
  87. Object.keys(this._rules).forEach(name => {
  88. const rule = this.get(name);
  89. allRules.set(name, rule);
  90. });
  91. return allRules;
  92. }
  93. }
  94. module.exports = Rules;