fixed type hint for return expression extracts

HaxeCheckstyle · Nov 25, 2024 · ed57f56 · ed57f56
1 parent 7e16139
commit ed57f56
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/refactor/refactor/extractmethod/CodeGenAsExpression.hx b/src/refactor/refactor/extractmethod/CodeGenAsExpression.hx
@@ -30,6 +30,13 @@ class CodeGenAsExpression extends CodeGenBase {
 					}
 					return Promise.resolve(":" + typeHint.printTypeHint());
 				});
+			case Kwd(KwdReturn):
+				return parentTypeHint().then(function(typeHint):Promise<String> {
+					if (typeHint == null) {
+						return Promise.resolve("");
+					}
+					return Promise.resolve(":" + typeHint.printTypeHint());
+				});
 			default:
 		}
 		return TypingHelper.findTypeWithTyper(context, context.what.fileName, extractData.endToken.pos.max - 1).then(function(typeHint):Promise<String> {