Fix: Exclude generator functions from convert-to-class suggestions

src/services/suggestionDiagnostics.ts

@@ -291,6 +291,11 @@ function getKeyFromNode(exp: FunctionLikeDeclaration) {
 
 function canBeConvertedToClass(node: Node, checker: TypeChecker): boolean {
     if (isFunctionExpression(node)) {
+        // Generator functions cannot be converted to classes
+        if (getFunctionFlags(node) & FunctionFlags.Generator) {
+            return false;
+        }
+
         if (isVariableDeclaration(node.parent) && node.symbol.members?.size) {
             return true;
         }
@@ -300,6 +305,11 @@ function canBeConvertedToClass(node: Node, checker: TypeChecker): boolean {
     }
 
     if (isFunctionDeclaration(node)) {
+        // Generator functions cannot be converted to classes
+        if (getFunctionFlags(node) & FunctionFlags.Generator) {
+            return false;
+        }
+
         return !!node.symbol.members?.size;
     }
 

tests/cases/fourslash/convertFunctionToEs6Class_generatorExpression_falsePositive.ts

@@ -0,0 +1,10 @@
+/// <reference path='fourslash.ts' />
+
+// @allowJs: true
+// @Filename: /a.js
+////const gen = function*() {};
+////gen.prototype.next = gen.prototype.next;
+////gen.prototype.return = gen.prototype.return;
+
+// Generator function expressions should not trigger the convert-to-class suggestion
+verify.getSuggestionDiagnostics([]);

tests/cases/fourslash/convertFunctionToEs6Class_generator_falsePositive.ts

@@ -0,0 +1,10 @@
+/// <reference path='fourslash.ts' />
+
+// @allowJs: true
+// @Filename: /a.js
+////function* gen() {}
+////gen.prototype.next = gen.prototype.next;
+////gen.prototype.return = gen.prototype.return;
+
+// Generator functions should not trigger the convert-to-class suggestion
+verify.getSuggestionDiagnostics([]);

tests/cases/fourslash/convertFunctionToEs6Class_regularVsGenerator.ts

@@ -0,0 +1,17 @@
+/// <reference path='fourslash.ts' />
+
+// @allowJs: true
+// @Filename: /a.js
+////function [|regular|]() {}
+////regular.prototype.method = function() { this.x = 1; };
+////
+////function* gen() {}
+////gen.prototype.next = gen.prototype.next;
+////gen.prototype.return = gen.prototype.return;
+
+// Regular constructor functions should still trigger the convert-to-class suggestion
+// but generator functions should not
+verify.getSuggestionDiagnostics([{
+    message: "This constructor function may be converted to a class declaration.",
+    code: 80002,
+}]);