Merge pull request #1189 from BradLewis/fix/unary-expr-function-call

Fix/unary expr function call

2 files changed, 41 insertions, 2 deletions

diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index c9a6275..723ec85 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -1226,7 +1226,7 @@ internal_resolve_type_expression :: proc(ast_context: ^AstContext, node: ^ast.Ex
 		ok := internal_resolve_type_expression(ast_context, v.expr, out)
 		if v.op.kind == .And {
 			out.pointers += 1
-		} else if v.op.kind == .Sub || v.op.kind == .Add {
+		} else if v.op.kind == .Sub || v.op.kind == .Add || v.op.kind == .Not || v.op.kind == .Xor {
 			if value, ok := out.value.(SymbolProcedureValue); ok {
 				if len(value.return_types) > 0 {
 					type := value.return_types[0].type
@@ -1244,7 +1244,18 @@ internal_resolve_type_expression :: proc(ast_context: ^AstContext, node: ^ast.Ex
 		out.pointers -= 1
 		return ok
 	case ^Paren_Expr:
-		return internal_resolve_type_expression(ast_context, v.expr, out)
+		ok = internal_resolve_type_expression(ast_context, v.expr, out)
+		if value, ok := out.value.(SymbolProcedureValue); ok {
+			if len(value.return_types) > 0 {
+				type := value.return_types[0].type
+				if type == nil {
+					type = value.return_types[0].default_value
+				}
+				ok = internal_resolve_type_expression(ast_context, type, out)
+				return ok
+			}
+		}
+		return ok
 	case ^Slice_Expr:
 		out^, ok = resolve_slice_expression(ast_context, v, v.expr)
 		return ok
diff --git a/tests/hover_test.odin b/tests/hover_test.odin
index 8d5f461..282ffb7 100644
--- a/tests/hover_test.odin
+++ b/tests/hover_test.odin
@@ -5675,6 +5675,34 @@ ast_hover_unary_overload_function_call :: proc(t: ^testing.T) {
 	}
 	test.expect_hover(t, &source, "test.bar: int")
 }
+
+@(test)
+ast_hover_negate_function_call :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+		foo :: proc() -> bool {}
+
+		main :: proc() {
+			b{*}ar := !foo()
+		}
+		`,
+	}
+	test.expect_hover(t, &source, "test.bar: bool")
+}
+
+@(test)
+ast_hover_function_call_with_parens :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+		foo :: proc() -> bool {}
+
+		main :: proc() {
+			b{*}ar := (foo())
+		}
+		`,
+	}
+	test.expect_hover(t, &source, "test.bar: bool")
+}
 /*
 
 Waiting for odin fix