Add documentation to union variants

1 files changed, 10 insertions, 3 deletions

diff --git a/src/server/collector.odin b/src/server/collector.odin
index 37ba74c..c06879d 100644
--- a/src/server/collector.odin
+++ b/src/server/collector.odin
@@ -247,6 +247,7 @@ collect_union_fields :: proc(
 	collection: ^SymbolCollection,
 	union_type: ast.Union_Type,
 	package_map: map[string]string,
+	file: ast.File,
 ) -> SymbolUnionValue {
 	types := make([dynamic]^ast.Expr, 0, collection.allocator)
 
@@ -256,9 +257,15 @@ collect_union_fields :: proc(
 		append(&types, cloned)
 	}
 
+	temp_docs, temp_comments := get_field_docs_and_comments(file, union_type.variants, context.temp_allocator)
+	docs := clone_dynamic_array(temp_docs, collection.allocator, &collection.unique_strings)
+	comments := clone_dynamic_array(temp_comments, collection.allocator, &collection.unique_strings)
+
 	value := SymbolUnionValue {
-		types = types[:],
-		poly  = cast(^ast.Field_List)clone_type(union_type.poly_params, collection.allocator, &collection.unique_strings),
+		types    = types[:],
+		poly     = cast(^ast.Field_List)clone_type(union_type.poly_params, collection.allocator, &collection.unique_strings),
+		comments = comments[:],
+		docs     = docs[:],
 	}
 
 	return value
@@ -567,7 +574,7 @@ collect_symbols :: proc(collection: ^SymbolCollection, file: ast.File, uri: stri
 		case ^ast.Union_Type:
 			token = v^
 			token_type = .Union
-			symbol.value = collect_union_fields(collection, v^, package_map)
+			symbol.value = collect_union_fields(collection, v^, package_map, file)
 			symbol.signature = "union"
 		case ^ast.Bit_Set_Type:
 			token = v^