Correctly resolve poly proc returns with #soa arrays

1 files changed, 15 insertions, 0 deletions

diff --git a/src/server/symbol.odin b/src/server/symbol.odin
index 6cf24df..5181ca3 100644
--- a/src/server/symbol.odin
+++ b/src/server/symbol.odin
@@ -821,11 +821,21 @@ symbol_to_expr :: proc(symbol: Symbol, file: string, allocator := context.temp_a
 	case SymbolDynamicArrayValue:
 		type := new_type(ast.Dynamic_Array_Type, pos, end, allocator)
 		type.elem = v.expr
+		if .Soa in symbol.flags {
+			directive := new_type(ast.Basic_Directive, pos, end, allocator)
+			directive.name = "soa"
+			type.tag = directive
+		}
 		return type
 	case SymbolFixedArrayValue:
 		type := new_type(ast.Array_Type, pos, end, allocator)
 		type.elem = v.expr
 		type.len = v.len
+		if .Soa in symbol.flags {
+			directive := new_type(ast.Basic_Directive, pos, end, allocator)
+			directive.name = "soa"
+			type.tag = directive
+		}
 		return type
 	case SymbolMapValue:
 		type := new_type(ast.Map_Type, pos, end, allocator)
@@ -837,6 +847,11 @@ symbol_to_expr :: proc(symbol: Symbol, file: string, allocator := context.temp_a
 	case SymbolSliceValue:
 		type := new_type(ast.Array_Type, pos, end, allocator)
 		type.elem = v.expr
+		if .Soa in symbol.flags {
+			directive := new_type(ast.Basic_Directive, pos, end, allocator)
+			directive.name = "soa"
+			type.tag = directive
+		}
 		return type
 	case SymbolStructValue:
 		type := new_type(ast.Struct_Type, pos, end, allocator)