feat: add auto-fix for no-wait-for-side-effects

Author: y-hsgw

lib/rules/no-wait-for-side-effects.ts

@@ -42,6 +42,8 @@ export default createTestingLibraryRule<Options, MessageIds>({
 	},
 	defaultOptions: [],
 	create(context, _, helpers) {
+		const sourceCode = getSourceCode(context);
+
 		function isCallerWaitFor(
 			node:
 				| TSESTree.AssignmentExpression
@@ -155,8 +157,6 @@ export default createTestingLibraryRule<Options, MessageIds>({
 				}
 				return false;
 			});
-
-			return false;
 		}
 
 		function getSideEffectNodes(
@@ -195,7 +195,17 @@ export default createTestingLibraryRule<Options, MessageIds>({
 			}) as TSESTree.ExpressionStatement[];
 		}
 
-		function reportSideEffects(node: TSESTree.BlockStatement) {
+		function reportSideEffects(
+			node: TSESTree.BlockStatement & {
+				parent: (
+					| TSESTree.ArrowFunctionExpression
+					| TSESTree.FunctionDeclaration
+					| TSESTree.FunctionExpression
+				) & {
+					parent: TSESTree.CallExpression;
+				};
+			}
+		) {
 			if (!isCallerWaitFor(node)) {
 				return;
 			}
@@ -208,6 +218,38 @@ export default createTestingLibraryRule<Options, MessageIds>({
 				context.report({
 					node: sideEffectNode,
 					messageId: 'noSideEffectsWaitFor',
+					fix(fixer) {
+						const { parent: callExpressionNode } = node.parent;
+						const targetNode = isAwaitExpression(callExpressionNode.parent)
+							? callExpressionNode.parent
+							: callExpressionNode;
+						const lines = sourceCode.getText().split('\n');
+
+						const line = lines[targetNode.loc.start.line - 1];
+
+						const indent = line.match(/^\s*/)?.[0] ?? '';
+						const lineStart = sourceCode.getIndexFromLoc({
+							line: sideEffectNode.loc.start.line,
+							column: 0,
+						});
+						const lineEnd = sourceCode.getIndexFromLoc({
+							line: sideEffectNode.loc.end.line + 1,
+							column: 0,
+						});
+
+						const sideEffectLines = lines.slice(
+							sideEffectNode.loc.start.line - 1,
+							sideEffectNode.loc.end.line
+						);
+						const sideEffectNodeText = sideEffectLines.join('\n');
+						return [
+							fixer.insertTextBefore(
+								targetNode,
+								sideEffectNodeText.trimStart() + '\n' + indent
+							),
+							fixer.removeRange([lineStart, lineEnd]),
+						];
+					},
 				})
 			);
 		}
@@ -260,7 +302,7 @@ export default createTestingLibraryRule<Options, MessageIds>({
 					const targetNode = isAwaitExpression(callExpressionNode.parent)
 						? callExpressionNode.parent
 						: callExpressionNode;
-					const sourceCode = getSourceCode(context);
+
 					return fixer.replaceText(targetNode, sourceCode.getText(node));
 				},
 			});