From 12deb18a07dc28f840555fb6c875d888ab5be13b Mon Sep 17 00:00:00 2001
From: Brad Lewis <22850972+BradLewis@users.noreply.github.com>
Date: Sun, 16 Nov 2025 15:30:54 -0500
Subject: Correct resolving unary exprs immediately after a function call

---
 src/server/analysis.odin | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

(limited to 'src/server')

diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index 45bf0fd..c9a6275 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -1223,13 +1223,22 @@ internal_resolve_type_expression :: proc(ast_context: ^AstContext, node: ^ast.Ex
 	case ^Comp_Lit:
 		return internal_resolve_type_expression(ast_context, v.type, out)
 	case ^Unary_Expr:
+		ok := internal_resolve_type_expression(ast_context, v.expr, out)
 		if v.op.kind == .And {
-			ok := internal_resolve_type_expression(ast_context, v.expr, out)
 			out.pointers += 1
-			return ok
-		} else {
-			return internal_resolve_type_expression(ast_context, v.expr, out)
+		} else if v.op.kind == .Sub || v.op.kind == .Add {
+			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 ^Deref_Expr:
 		ok := internal_resolve_type_expression(ast_context, v.expr, out)
 		out.pointers -= 1
-- 
cgit v1.2.3