Correctly filter used unions when ptr

1 files changed, 14 insertions, 4 deletions

diff --git a/src/server/completion.odin b/src/server/completion.odin
index 13a589d..a84631f 100644
--- a/src/server/completion.odin
+++ b/src/server/completion.odin
@@ -1581,6 +1581,18 @@ search_for_packages :: proc(fullpath: string) -> []string {
 	return packages[:]
 }
 
+get_used_switch_name :: proc(node: ^ast.Expr) -> (string, bool) {
+	#partial switch n in node.derived {
+	case ^ast.Ident:
+		return n.name, true
+	case ^ast.Selector_Expr:
+		return n.field.name, true
+	case ^ast.Pointer_Type:
+		return get_used_switch_name(n.elem)
+	}
+	return "", false
+}
+
 get_type_switch_completion :: proc(
 	ast_context: ^AstContext,
 	position_context: ^DocumentPositionContext,
@@ -1595,10 +1607,8 @@ get_type_switch_completion :: proc(
 		for stmt in block.stmts {
 			if case_clause, ok := stmt.derived.(^ast.Case_Clause); ok {
 				for name in case_clause.list {
-					if ident, ok := name.derived.(^ast.Ident); ok {
-						used_unions[ident.name] = true
-					} else if selector, ok := name.derived.(^ast.Selector_Expr); ok {
-						used_unions[selector.field.name] = true
+					if n, ok := get_used_switch_name(name); ok {
+						used_unions[n] = true
 					}
 				}
 			}