Improve(ESLint): Support object functions like styled.h1

master-co · Oct 25, 2023 · 89039fe · 89039fe
1 parent 405c579
commit 89039fe
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/packages/eslint-plugin/lib/utils/define-visitors.js b/packages/eslint-plugin/lib/utils/define-visitors.js
@@ -3,7 +3,7 @@ const astUtil = require('./ast')
 function defineVisitors({ context, options, settings, config }, visitNode) {
     const CallExpression = function (node) {
         const calleeStr = astUtil.calleeToString(node.callee)
-        if (settings.functions.findIndex((name) => calleeStr === name) === -1) {
+        if (settings.functions.findIndex((name) => calleeStr.startsWith(name)) === -1) {
             return
         }
         node.arguments.forEach((arg) => {

diff --git a/packages/eslint-plugin/tests/class-matching/react.test.js b/packages/eslint-plugin/tests/class-matching/react.test.js
@@ -28,5 +28,18 @@ new RuleTester({
             `,
             errors: [{ messageId: 'invalidClass' }],
         },
+        {
+            code: `
+                import React from 'react'
+                import { styled } from '@master/css.react'
+
+                const H1 = styled.h1('text-align:cente')
+
+                export default () => (
+                    <H1>Hello World</H1>
+                )
+            `,
+            errors: [{ messageId: 'invalidClass' }],
+        },
     ]
 })