Enhance no-array-for-each to handle optional chaining

docs/rules/no-array-for-each.md

@@ -21,6 +21,12 @@ array.forEach(element => {
 });
 ```
 
+```js
+array?.forEach(element => {
+	bar(element);
+});
+```
+
 ```js
 array.forEach((element, index) => {
 	bar(element, index);

rules/no-array-for-each.js

@@ -73,6 +73,7 @@ function getFixFunction(callExpression, functionInfo, context) {
 	const parameters = callback.params;
 	const array = callExpression.callee.object;
 	const {returnStatements} = functionInfo.get(callback);
+	const isOptionalChaining = callExpression.callee.optional;
 
 	const getForOfLoopHeadText = () => {
 		const [elementText, indexText] = parameters.map(parameter => sourceCode.getText(parameter));
@@ -228,7 +229,7 @@ function getFixFunction(callExpression, functionInfo, context) {
 			yield * replaceReturnStatement(returnStatement, fixer);
 		}
 
-		const expressionStatementLastToken = sourceCode.getLastToken(callExpression.parent);
+		const expressionStatementLastToken = sourceCode.getLastToken(isOptionalChaining ? callExpression.parent.parent : callExpression.parent);
 		// Remove semicolon if it's not needed anymore
 		// foo.forEach(bar => {});
 		//                       ^
@@ -238,6 +239,10 @@ function getFixFunction(callExpression, functionInfo, context) {
 
 		yield * fixSpaceAroundKeyword(fixer, callExpression.parent, sourceCode);
 
+		if (isOptionalChaining) {
+			yield fixer.insertTextBefore(callExpression, `if (${callExpression.callee.object.name}) `);
+		}
+
 		// Prevent possible variable conflicts
 		yield * extendFixRange(fixer, callExpression.parent.range);
 	};
@@ -314,14 +319,7 @@ function isFixable(callExpression, {scope, functionInfo, allIdentifiers, context
 	}
 
 	// Check `CallExpression.parent`
-	if (callExpression.parent.type !== 'ExpressionStatement') {
-		return false;
-	}
-
-	// Check `CallExpression.callee`
-	// Because of `ChainExpression` wrapper, `foo?.forEach()` is already failed on previous check keep this just for safety
-	/* c8 ignore next 3 */
-	if (callExpression.callee.optional) {
+	if (callExpression.parent.type !== 'ExpressionStatement' && callExpression.parent.type !== 'ChainExpression') {
 		return false;
 	}
 

test/no-array-for-each.mjs

@@ -25,7 +25,6 @@ test.snapshot({
 		'foo.forEach(element => bar(element), thisArgument)',
 		'foo.forEach()',
 		'const baz = foo.forEach(element => bar(element))',
-		'foo?.forEach(element => bar(element))',
 		'foo.forEach(bar)',
 		'foo.forEach(async function(element) {})',
 		'foo.forEach(function * (element) {})',
@@ -439,6 +438,100 @@ test({
 				sourceType: 'script',
 			},
 		},
+		{
+			code: outdent`
+				foo?.forEach(function(element) {
+					delete element;
+					console.log(element)
+				});
+			`,
+			output: outdent`
+				if (foo) for (const element of foo) {
+					delete element;
+					console.log(element)
+				}
+			`,
+			errors: 1,
+			parserOptions: {
+				sourceType: 'script',
+			},
+		},
+		{
+			code: outdent`
+				foo?.forEach(element => {
+					delete element;
+					console.log(element)
+				});
+			`,
+			output: outdent`
+				if (foo) for (const element of foo) {
+					delete element;
+					console.log(element)
+				}
+			`,
+			errors: 1,
+			parserOptions: {
+				sourceType: 'script',
+			},
+		},
+		{
+			code: outdent`
+				foo?.forEach(element => console.log(element));
+			`,
+			output: outdent`
+				if (foo) for (const element of foo)  console.log(element);
+			`,
+			errors: 1,
+			parserOptions: {
+				sourceType: 'script',
+			},
+		},
+		{
+			code: outdent`
+				foo?.forEach((element) => console.log(element));
+			`,
+			output: outdent`
+				if (foo) for (const element of foo)  console.log(element);
+			`,
+			errors: 1,
+			parserOptions: {
+				sourceType: 'script',
+			},
+		},
+		{
+			code: outdent`
+				foo?.forEach(element => { console.log(element) });
+			`,
+			output: outdent`
+				if (foo) for (const element of foo) { console.log(element) }
+			`,
+			errors: 1,
+			parserOptions: {
+				sourceType: 'script',
+			},
+		},
+		{
+			code: outdent`
+				function a() {
+					foo?.forEach(function(element) {
+						delete element;
+						console.log(element)
+					});
+				}
+			`,
+			output: outdent`
+				function a() {
+					if (foo) for (const element of foo) {
+						delete element;
+						console.log(element)
+					}
+				}
+			`,
+			errors: 1,
+			parserOptions: {
+				sourceType: 'script',
+			},
+		},
 		{
 			code: 'foo.forEach(function(element, element) {})',
 			errors: 1,