From 63fd34ef37a8ba977e6bfcdcb72cd890aafac2a8 Mon Sep 17 00:00:00 2001
From: DanielGavin <danielgavin5@hotmail.com>
Date: Wed, 17 Mar 2021 17:03:43 +0100
Subject: semantic token fix on package selector, more union completion

---
 src/server/analysis.odin | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'src/server/analysis.odin')

diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index 2e95c3a..c8c1827 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -1164,11 +1164,19 @@ make_symbol_union_from_ast :: proc (ast_context: ^AstContext, v: ast.Union_Type,
 		if ident, ok := variant.derived.(ast.Ident); ok {
 			append(&names, ident.name);
 		}
+
+		else if selector, ok := variant.derived.(ast.Selector_Expr); ok {
+
+			if ident, ok := selector.field.derived.(ast.Ident); ok {
+				append(&names, ident.name);
+			}
+		}
 	}
 
 	symbol.value = index.SymbolUnionValue {
-			names = names[:]
-		};
+			names = names[:],
+			types = v.variants,
+	};
 
 	return symbol;
 }
-- 
cgit v1.2.3