From cbdd09b0ef47234e42a3f6d9133381c3d11f4b40 Mon Sep 17 00:00:00 2001
From: Brad Lewis <22850972+BradLewis@users.noreply.github.com>
Date: Tue, 10 Feb 2026 14:50:36 +1100
Subject: Correctly resolve poly types from nested function calls that contain
 more than one return value

---
 src/server/generics.odin | 19 ++++++++++++++++++-
 tests/hover_test.odin    | 20 ++++++++++++++++++++
 2 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/src/server/generics.odin b/src/server/generics.odin
index 82f71c1..2f2cbfa 100644
--- a/src/server/generics.odin
+++ b/src/server/generics.odin
@@ -503,6 +503,18 @@ resolve_generic_function_ast :: proc(
 	return resolve_generic_function_symbol(ast_context, params, results, proc_lit.inlining, proc_symbol)
 }
 
+get_proc_return_value_count :: proc(fields: []^ast.Field) -> int {
+	total := 0
+	for field in fields {
+		if len(field.names) == 0 {
+			total += 1
+		} else {
+			total += len(field.names)
+		}
+	}
+
+	return total
+}
 
 resolve_generic_function_symbol :: proc(
 	ast_context: ^AstContext,
@@ -524,6 +536,8 @@ resolve_generic_function_symbol :: proc(
 
 	i := 0
 	count_required_params := 0
+	// Total number of args passed in the call when expanded to include functions that may return multiple values
+	call_arg_count := 0
 
 	for param in params {
 		if param.default_value == nil {
@@ -560,6 +574,7 @@ resolve_generic_function_symbol :: proc(
 
 				//If we have a function call, we should instead look at the return value: bar(foo(123))
 				if symbol_value, ok := symbol.value.(SymbolProcedureValue); ok && len(symbol_value.return_types) > 0 {
+					call_arg_count += get_proc_return_value_count(symbol_value.return_types)
 					if _, ok := call_expr.args[i].derived.(^ast.Call_Expr); ok {
 						if symbol_value.return_types[0].type != nil {
 							if symbol, ok = resolve_type_expression(ast_context, symbol_value.return_types[0].type);
@@ -575,6 +590,8 @@ resolve_generic_function_symbol :: proc(
 							}
 						}
 					}
+				} else {
+					call_arg_count += 1
 				}
 
 				// We set the offset so we can find it as a local if it's based on the type of a local var
@@ -600,7 +617,7 @@ resolve_generic_function_symbol :: proc(
 		find_and_replace_poly_type(v, &poly_map)
 	}
 
-	if count_required_params > len(call_expr.args) || count_required_params == 0 || len(call_expr.args) == 0 {
+	if count_required_params > call_arg_count || count_required_params == 0 || call_arg_count == 0 {
 		return {}, false
 	}
 
diff --git a/tests/hover_test.odin b/tests/hover_test.odin
index df65f3a..5dc6c91 100644
--- a/tests/hover_test.odin
+++ b/tests/hover_test.odin
@@ -6055,6 +6055,26 @@ ast_hover_union_multiple_poly :: proc(t: ^testing.T) {
 	}
 	test.expect_hover(t, &source, "test.Bazz :: union(T) {\n\tFoo(T),\n\tBar,\n}")
 }
+
+@(test)
+ast_hover_poly_proc_passthrough :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+		make :: proc() -> (int, bool) {
+			return 1, true
+		}
+
+		confirm_bool_one :: #force_inline proc(v: $T, ok: $B) -> (T, bool) {
+			return v, bool(ok)
+		}
+
+		main :: proc() {
+			v{*}alue, ok := confirm_bool_one(make())
+		}
+		`,
+	}
+	test.expect_hover(t, &source, "test.value: int")
+}
 /*
 
 Waiting for odin fix
-- 
cgit v1.2.3