Use selector for rule

Author: jorgezreik

lib/rules/async-server-action.js

@@ -16,26 +16,6 @@ const messages = {
   asyncServerAction: 'Your server action should be async',
 };
 
-/**
- * Detects a `use server` directive in a given AST node
- * @param {ASTNode} node The node to search.
- * @returns {boolean} Whether the node given has a `use server` directive.
- */
-function hasUseServerDirective(node) {
-  if (node.body.type !== 'BlockStatement') return false;
-
-  const functionBody = node.body.body;
-  if (functionBody.length === 0) return false;
-
-  const potentialDirectiveStatement = functionBody[0];
-  if (potentialDirectiveStatement.type !== 'ExpressionStatement') return false;
-
-  const potentialDirectiveExpression = potentialDirectiveStatement.expression;
-  if (potentialDirectiveExpression.type !== 'Literal') return false;
-
-  return potentialDirectiveExpression.value === 'use server';
-}
-
 module.exports = {
   meta: {
     docs: {
@@ -53,31 +33,19 @@ module.exports = {
   },
 
   create(context) {
-    /**
-     * Validates that given AST node is async if it has the `use server` directive
-     * @param {ASTNode} node The node to search.
-     * @returns {void}
-     */
-    function validate(node) {
-      if (hasUseServerDirective(node) && !node.async) {
+    let currentFunction = null;
+
+    return {
+      ':function'(node) {
+        currentFunction = node;
+      },
+      ':function[async=false]>BlockStatement>ExpressionStatement:first-child[expression.value="use server"]'() {
         report(context, messages.asyncServerAction, 'asyncServerAction', {
-          node,
+          node: currentFunction,
           fix(fixer) {
-            return fixer.insertTextBefore(node, 'async ');
+            return fixer.insertTextBefore(currentFunction, 'async ');
           },
         });
-      }
-    }
-
-    return {
-      FunctionDeclaration(node) {
-        validate(node);
-      },
-      FunctionExpression(node) {
-        validate(node);
-      },
-      ArrowFunctionExpression(node) {
-        validate(node);
       },
     };
   },