fix(42714): do not show inferFunctionReturnType QF on function body

Author: a-tarasyuk

src/services/refactors/inferFunctionReturnType.ts

@@ -59,7 +59,9 @@ namespace ts.refactor.inferFunctionReturnType {
         if (isInJSFile(context.file) || !refactorKindBeginsWith(inferReturnTypeAction.kind, context.kind)) return;
 
         const token = getTokenAtPosition(context.file, context.startPosition);
-        const declaration = findAncestor(token, isConvertibleDeclaration);
+        const declaration = findAncestor(token, n =>
+            isBlock(n) || isParameter(n) || n.parent && isArrowFunction(n.parent) && (n.kind === SyntaxKind.EqualsGreaterThanToken || n.parent.body === n) ? "quit" :
+                isConvertibleDeclaration(n)) as ConvertibleDeclaration | undefined;
         if (!declaration || !declaration.body || declaration.type) {
             return { error: getLocaleSpecificMessage(Diagnostics.Return_type_must_be_inferred_from_a_function) };
         }

tests/cases/fourslash/refactorInferFunctionReturnType16.ts

@@ -0,0 +1,17 @@
+/// <reference path='fourslash.ts' />
+
+////function /*a*/f/*c*/()/*d*//*b*/ /*e*/{
+////    /*g*/return { x: 1, y: 1 };/*h*/
+////}/*f*/
+
+goTo.select("a", "b");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("c", "d");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("e", "f");
+verify.not.refactorAvailable("Infer function return type");
+
+goTo.select("g", "h");
+verify.not.refactorAvailable("Infer function return type");

tests/cases/fourslash/refactorInferFunctionReturnType17.ts

@@ -0,0 +1,19 @@
+/// <reference path='fourslash.ts' />
+
+////class Foo {
+////    /*a*/method/*c*/()/*d*//*b*/ /*e*/{
+////        /*g*/return { x: 1, y: 1 };/*h*/
+////    }/*f*/
+////}
+
+goTo.select("a", "b");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("c", "d");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("e", "f");
+verify.not.refactorAvailable("Infer function return type");
+
+goTo.select("g", "h");
+verify.not.refactorAvailable("Infer function return type");

tests/cases/fourslash/refactorInferFunctionReturnType18.ts

@@ -0,0 +1,17 @@
+/// <reference path='fourslash.ts' />
+
+////const foo = /*a*/function/*c*/()/*d*//*b*/ /*e*/{
+////    /*g*/return { x: 1, y: 1 };/*h*/
+////}/*f*/
+
+goTo.select("a", "b");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("c", "d");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("e", "f");
+verify.not.refactorAvailable("Infer function return type");
+
+goTo.select("g", "h");
+verify.not.refactorAvailable("Infer function return type");

tests/cases/fourslash/refactorInferFunctionReturnType19.ts

@@ -0,0 +1,17 @@
+/// <reference path='fourslash.ts' />
+
+////const foo = /*c*//*a*/()/*b*/ =>/*d*/ /*e*/{
+////    /*g*/return { x: 1, y: 1 };/*h*/
+////}/*f*/
+
+goTo.select("a", "b");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("c", "d");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("e", "f");
+verify.not.refactorAvailable("Infer function return type");
+
+goTo.select("g", "h");
+verify.not.refactorAvailable("Infer function return type");

tests/cases/fourslash/refactorInferFunctionReturnType20.ts

@@ -0,0 +1,14 @@
+/// <reference path='fourslash.ts' />
+
+////function f(/*a*/x/*b*/, /*e*/{ /*c*/y/*d*/ }/*f*/) {
+////    return { x, y: 1 };
+////}
+
+goTo.select("a", "b");
+verify.not.refactorAvailable("Infer function return type");
+
+goTo.select("c", "d");
+verify.not.refactorAvailable("Infer function return type");
+
+goTo.select("e", "f");
+verify.not.refactorAvailable("Infer function return type");

tests/cases/fourslash/refactorInferFunctionReturnType21.ts

@@ -0,0 +1,12 @@
+/// <reference path='fourslash.ts' />
+
+////const fn = /*a*/(x)/*b*/ /*c*/=>/*d*/ /*e*/x/*f*/
+
+goTo.select("a", "b");
+verify.refactorAvailable("Infer function return type");
+
+goTo.select("c", "d");
+verify.not.refactorAvailable("Infer function return type");
+
+goTo.select("e", "f");
+verify.not.refactorAvailable("Infer function return type");

tests/cases/fourslash/refactorKind_rewriteFunction.ts

@@ -7,6 +7,5 @@ verify.refactorKindAvailable("refactor.rewrite",
 [
     "refactor.rewrite.arrow.braces.add",
     "refactor.rewrite.function.named",
-    "refactor.rewrite.function.anonymous",
-    "refactor.rewrite.function.returnType"
+    "refactor.rewrite.function.anonymous"
 ]);