Allow restricting classes by regexp in vue/no-restricted-class

docs/rules/no-restricted-class.md

@@ -20,7 +20,7 @@ in the rule configuration.
 
 ```json
 {
-  "vue/no-restricted-class": ["error", "forbidden", "forbidden-two", "forbidden-three"]
+  "vue/no-restricted-class": ["error", "forbidden", "forbidden-two", "forbidden-three", "/^for(bidden|gotten)/"]
 }
 ```
 

lib/rules/no-restricted-class.js

@@ -5,16 +5,25 @@
 'use strict'
 
 const utils = require('../utils')
+const regexp = require('../utils/regexp')
 
 /**
  * Report a forbidden class
  * @param {string} className
  * @param {*} node
  * @param {RuleContext} context
- * @param {Set<string>} forbiddenClasses
+ * @param {Array<string>} forbiddenClasses
  */
 const reportForbiddenClass = (className, node, context, forbiddenClasses) => {
-  if (forbiddenClasses.has(className)) {
+  const isForbidden = forbiddenClasses.some((cl) => {
+    if (regexp.isRegExp(cl)) {
+      const re = regexp.toRegExp(cl)
+      return re.test(className)
+    } else {
+      return className.includes(cl)
+    }
+  })
+  if (isForbidden) {
     const loc = node.value ? node.value.loc : node.loc
     context.report({
       node,
@@ -112,7 +121,7 @@ module.exports = {
 
   /** @param {RuleContext} context */
   create(context) {
-    const forbiddenClasses = new Set(context.options || [])
+    const forbiddenClasses = context.options || []
 
     return utils.defineTemplateBodyVisitor(context, {
       /**

tests/lib/rules/no-restricted-class.js

@@ -30,6 +30,10 @@ ruleTester.run('no-restricted-class', rule, {
     {
       code: `<template><div :class="'' + {forbidden: true}">Content</div></template>`,
       options: ['forbidden']
+    },
+    {
+      code: `<template><div class="allowed"">Content</div></template>`,
+      options: ['/^for(bidden|gotten)/']
     }
   ],
 
@@ -113,6 +117,16 @@ ruleTester.run('no-restricted-class', rule, {
         }
       ],
       options: ['forbidden']
+    },
+    {
+      code: `<template><div class="forbidden allowed" /></template>`,
+      errors: [
+        {
+          message: "'forbidden' class is not allowed.",
+          type: 'VAttribute'
+        }
+      ],
+      options: ['/^for(bidden|gotten)/']
     }
   ]
 })