Allow implicit return with explicit undefined return type

Author: saschanaz

src/compiler/checker.ts

@@ -35469,8 +35469,8 @@ export function createTypeChecker(host: TypeCheckerHost): TypeChecker {
             const functionFlags = getFunctionFlags(func);
             const type = returnType && unwrapReturnType(returnType, functionFlags);
 
-            // Functions with with an explicitly specified 'void' or 'any' return type don't need any return expressions.
-            if (type && maybeTypeOfKind(type, TypeFlags.Any | TypeFlags.Void)) {
+            // Functions with an explicitly specified 'undefined, 'void' or 'any' return type don't need any return expressions.
+            if (type && maybeTypeOfKind(type, TypeFlags.Undefined | TypeFlags.Void | TypeFlags.Any)) {
                 return;
             }
 

tests/baselines/reference/functionsMissingReturnStatementsAndExpressions.errors.txt

@@ -1,8 +1,8 @@
 tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(1,16): error TS2355: A function whose declared type is neither 'void' nor 'any' must return a value.
 tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(99,17): error TS2355: A function whose declared type is neither 'void' nor 'any' must return a value.
-tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(104,16): error TS2378: A 'get' accessor must return a value.
-tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(126,15): error TS18050: The value 'undefined' cannot be used here.
-tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(127,5): error TS1003: Identifier expected.
+tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(108,16): error TS2378: A 'get' accessor must return a value.
+tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(130,15): error TS18050: The value 'undefined' cannot be used here.
+tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(131,5): error TS1003: Identifier expected.
 
 
 ==== tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts (5 errors) ====
@@ -40,12 +40,12 @@ tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(127,5): e
         return null;
     }
 
-    function f8(): void {
+    function f8(): any {
         // Fine since are typed any.
         return;
     }
 
-    function f9(): void {
+    function f9(): any {
         // Fine since we are typed any and return undefined
         return undefined;
     }
@@ -112,6 +112,10 @@ tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(127,5): e
         // Not okay; union does not contain void or any
     }
 
+    function f22(): undefined {
+        // Okay; implicitly returns undefined
+    }
+    
     class C {
         public get m1() {
                    ~~
@@ -143,4 +147,5 @@ tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts(127,5): e
         }
         ~
 !!! error TS1003: Identifier expected.
-    }
+    }
+    

tests/baselines/reference/functionsMissingReturnStatementsAndExpressions.js

@@ -31,12 +31,12 @@ function f7(): void {
     return null;
 }
 
-function f8(): void {
+function f8(): any {
     // Fine since are typed any.
     return;
 }
 
-function f9(): void {
+function f9(): any {
     // Fine since we are typed any and return undefined
     return undefined;
 }
@@ -101,6 +101,10 @@ function f21(): number | string {
     // Not okay; union does not contain void or any
 }
 
+function f22(): undefined {
+    // Okay; implicitly returns undefined
+}
+
 class C {
     public get m1() {
         // Errors; get accessors must return a value.
@@ -126,7 +130,8 @@ class C {
         throw null;
         throw undefined.
     }
-}
+}
+
 
 //// [functionsMissingReturnStatementsAndExpressions.js]
 function f1() {
@@ -209,6 +214,9 @@ function f20() {
 function f21() {
     // Not okay; union does not contain void or any
 }
+function f22() {
+    // Okay; implicitly returns undefined
+}
 var C = /** @class */ (function () {
     function C() {
     }

tests/baselines/reference/functionsMissingReturnStatementsAndExpressions.symbols

@@ -46,14 +46,14 @@ function f7(): void {
     return null;
 }
 
-function f8(): void {
+function f8(): any {
 >f8 : Symbol(f8, Decl(functionsMissingReturnStatementsAndExpressions.ts, 30, 1))
 
     // Fine since are typed any.
     return;
 }
 
-function f9(): void {
+function f9(): any {
 >f9 : Symbol(f9, Decl(functionsMissingReturnStatementsAndExpressions.ts, 35, 1))
 
     // Fine since we are typed any and return undefined
@@ -152,37 +152,43 @@ function f21(): number | string {
     // Not okay; union does not contain void or any
 }
 
+function f22(): undefined {
+>f22 : Symbol(f22, Decl(functionsMissingReturnStatementsAndExpressions.ts, 100, 1))
+
+    // Okay; implicitly returns undefined
+}
+
 class C {
->C : Symbol(C, Decl(functionsMissingReturnStatementsAndExpressions.ts, 100, 1))
+>C : Symbol(C, Decl(functionsMissingReturnStatementsAndExpressions.ts, 104, 1))
 
     public get m1() {
->m1 : Symbol(C.m1, Decl(functionsMissingReturnStatementsAndExpressions.ts, 102, 9))
+>m1 : Symbol(C.m1, Decl(functionsMissingReturnStatementsAndExpressions.ts, 106, 9))
 
         // Errors; get accessors must return a value.
     }
 
     public get m2() {
->m2 : Symbol(C.m2, Decl(functionsMissingReturnStatementsAndExpressions.ts, 105, 5))
+>m2 : Symbol(C.m2, Decl(functionsMissingReturnStatementsAndExpressions.ts, 109, 5))
 
         // Permissible; returns undefined.
         return;
     }
 
     public get m3() {
->m3 : Symbol(C.m3, Decl(functionsMissingReturnStatementsAndExpressions.ts, 110, 5))
+>m3 : Symbol(C.m3, Decl(functionsMissingReturnStatementsAndExpressions.ts, 114, 5))
 
         return "Okay, because this is a return expression.";
     }
 
     public get m4() {
->m4 : Symbol(C.m4, Decl(functionsMissingReturnStatementsAndExpressions.ts, 114, 5))
+>m4 : Symbol(C.m4, Decl(functionsMissingReturnStatementsAndExpressions.ts, 118, 5))
 
         // Fine since this consists of a single throw statement.
         throw null;
     }
 
     public get m5() {
->m5 : Symbol(C.m5, Decl(functionsMissingReturnStatementsAndExpressions.ts, 119, 5))
+>m5 : Symbol(C.m5, Decl(functionsMissingReturnStatementsAndExpressions.ts, 123, 5))
 
         // Not fine, since we can *only* consist of a single throw statement
         // if no return statements are present but we are a get accessor.
@@ -191,3 +197,4 @@ class C {
 >undefined : Symbol(undefined)
     }
 }
+

tests/baselines/reference/functionsMissingReturnStatementsAndExpressions.types

@@ -48,15 +48,15 @@ function f7(): void {
 >null : null
 }
 
-function f8(): void {
->f8 : () => void
+function f8(): any {
+>f8 : () => any
 
     // Fine since are typed any.
     return;
 }
 
-function f9(): void {
->f9 : () => void
+function f9(): any {
+>f9 : () => any
 
     // Fine since we are typed any and return undefined
     return undefined;
@@ -159,6 +159,12 @@ function f21(): number | string {
     // Not okay; union does not contain void or any
 }
 
+function f22(): undefined {
+>f22 : () => undefined
+
+    // Okay; implicitly returns undefined
+}
+
 class C {
 >C : C
 
@@ -204,3 +210,4 @@ class C {
     }
 > : any
 }
+

tests/cases/compiler/functionsMissingReturnStatementsAndExpressions.ts

@@ -34,12 +34,12 @@ function f7(): void {
     return null;
 }
 
-function f8(): void {
+function f8(): any {
     // Fine since are typed any.
     return;
 }
 
-function f9(): void {
+function f9(): any {
     // Fine since we are typed any and return undefined
     return undefined;
 }
@@ -104,6 +104,10 @@ function f21(): number | string {
     // Not okay; union does not contain void or any
 }
 
+function f22(): undefined {
+    // Okay; implicitly returns undefined
+}
+
 class C {
     public get m1() {
         // Errors; get accessors must return a value.
@@ -129,4 +133,4 @@ class C {
         throw null;
         throw undefined.
     }
-}
+}

tests/cases/fourslash/codeFixCorrectReturnValue6.ts

@@ -4,6 +4,4 @@
 ////     undefined
 //// }
 
-verify.codeFixAvailable([
-    { description: 'Add a return statement' },
-]);
+verify.not.codeFixAvailable();