chore(transformer): Adjust type signatures in bodyReturnType.ts

Author: martinjlowm

src/transformer/descriptor/method/bodyReturnType.ts

@@ -7,21 +7,21 @@ export function GetReturnTypeFromBodyDescriptor(node: ts.ArrowFunction | ts.Func
   return GetDescriptor(GetReturnNodeFromBody(node), scope);
 }
 
-export function GetReturnNodeFromBody(node: ts.FunctionLikeDeclaration): ts.Node {
-  let returnValue: ts.Node | undefined;
+export function GetReturnNodeFromBody(node: ts.FunctionLikeDeclaration): ts.Expression {
+  let returnValue: ts.Expression | undefined;
 
   const functionBody: ts.ConciseBody | undefined = node.body;
 
   if (functionBody && ts.isBlock(functionBody)) {
-    const returnStatement: ts.ReturnStatement = GetReturnStatement(functionBody);
+    const returnStatement: ts.ReturnStatement | undefined = GetReturnStatement(functionBody);
 
     if (returnStatement) {
       returnValue = returnStatement.expression;
     } else {
       returnValue = GetNullDescriptor();
     }
   } else {
-    returnValue = node.body;
+    returnValue = functionBody;
   }
 
   if (!returnValue) {
@@ -31,6 +31,8 @@ export function GetReturnNodeFromBody(node: ts.FunctionLikeDeclaration): ts.Node
   return returnValue;
 }
 
-function GetReturnStatement(body: ts.FunctionBody): ts.ReturnStatement {
-  return body.statements.find((statement: ts.Statement) => statement.kind === ts.SyntaxKind.ReturnStatement) as ts.ReturnStatement;
+function GetReturnStatement(body: ts.FunctionBody): ts.ReturnStatement | undefined {
+  return body.statements.find(
+    (statement: ts.Statement): statement is ts.ReturnStatement => statement.kind === ts.SyntaxKind.ReturnStatement,
+  );
 }