feat: implement wrapWithFunctionExpressionFix function

Author: y-hsgw

lib/rules/await-async-queries.ts

@@ -11,6 +11,7 @@ import {
 	isMemberExpression,
 	isPromiseHandled,
 } from '../node-utils';
+import { wrapWithFunctionExpressionFix } from '../utils/wrap-function-expression-fix';
 
 import type { TSESTree } from '@typescript-eslint/utils';
 
@@ -164,20 +165,11 @@ export default createTestingLibraryRule<Options, MessageIds>({
 								);
 							}
 
-							const ruleFixes = [IdentifierNodeFixer];
-							if (!functionExpression.async) {
-								/**
-								 * Mutate the actual node so if other nodes exist in this
-								 * function expression body they don't also try to fix it.
-								 */
-								functionExpression.async = true;
-
-								ruleFixes.push(
-									fixer.insertTextBefore(functionExpression, 'async ')
-								);
-							}
-
-							return ruleFixes;
+							return wrapWithFunctionExpressionFix(
+								fixer,
+								IdentifierNodeFixer,
+								functionExpression
+							);
 						},
 					});
 				}

lib/rules/await-async-utils.ts

@@ -14,6 +14,7 @@ import {
 	isPromiseHandled,
 	isProperty,
 } from '../node-utils';
+import { wrapWithFunctionExpressionFix } from '../utils/wrap-function-expression-fix';
 
 import type { TSESLint, TSESTree } from '@typescript-eslint/utils';
 
@@ -93,26 +94,6 @@ export default createTestingLibraryRule<Options, MessageIds>({
 				}
 			}
 		}
-		function wrapWithFunctionExpressionFix(
-			fixer: TSESLint.RuleFixer,
-			ruleFix: TSESLint.RuleFix,
-			functionExpression:
-				| TSESTree.ArrowFunctionExpression
-				| TSESTree.FunctionDeclaration
-				| TSESTree.FunctionExpression
-				| null
-		) {
-			if (functionExpression && !functionExpression.async) {
-				/**
-				 * Mutate the actual node so if other nodes exist in this
-				 * function expression body they don't also try to fix it.
-				 */
-				functionExpression.async = true;
-
-				return [ruleFix, fixer.insertTextBefore(functionExpression, 'async ')];
-			}
-			return ruleFix;
-		}
 
 		function insertAwaitBeforeNode(
 			fixer: TSESLint.RuleFixer,

lib/utils/wrap-function-expression-fix.ts

@@ -0,0 +1,22 @@
+import type { TSESLint, TSESTree } from '@typescript-eslint/utils';
+
+export const wrapWithFunctionExpressionFix = (
+	fixer: TSESLint.RuleFixer,
+	ruleFix: TSESLint.RuleFix,
+	functionExpression:
+		| TSESTree.ArrowFunctionExpression
+		| TSESTree.FunctionDeclaration
+		| TSESTree.FunctionExpression
+		| null
+) => {
+	if (functionExpression && !functionExpression.async) {
+		/**
+		 * Mutate the actual node so if other nodes exist in this
+		 * function expression body they don't also try to fix it.
+		 */
+		functionExpression.async = true;
+
+		return [ruleFix, fixer.insertTextBefore(functionExpression, 'async ')];
+	}
+	return ruleFix;
+};