From e9ea799cf2460cf8598d68fcb908fb50ee1dc4cf Mon Sep 17 00:00:00 2001
From: DanielGavin <danielgavin5@hotmail.com>
Date: Sun, 9 Jun 2024 15:54:13 +0200
Subject: Support comp lit inference from return

---
 src/server/analysis.odin | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

(limited to 'src/server')

diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index 7b4d125..b6ae251 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -1853,6 +1853,34 @@ resolve_comp_literal :: proc(
 				value.arg_types[arg_index].type,
 			) or_return
 		}
+	} else if position_context.returns != nil {
+		return_index: int
+
+		if position_context.returns.results == nil {
+			return {}, false
+		}
+
+		for result, i in position_context.returns.results {
+			if position_in_node(result, position_context.position) {
+				return_index = i
+				break
+			}
+		}
+
+		if position_context.function.type == nil {
+			return {}, false
+		}
+
+		if position_context.function.type.results == nil {
+			return {}, false
+		}
+
+		if len(position_context.function.type.results.list) > return_index {
+			symbol = resolve_type_expression(
+				ast_context,
+				position_context.function.type.results.list[return_index].type,
+			) or_return
+		}
 	}
 
 	set_ast_package_set_scoped(ast_context, symbol.pkg)
-- 
cgit v1.2.3