Merge pull request #974 from BradLewis/fix/completion-poly-type-narrowing

Add completion for poly type narrowing

2 files changed, 19 insertions, 0 deletions

diff --git a/src/server/completion.odin b/src/server/completion.odin
index 29ae7bb..732a940 100644
--- a/src/server/completion.odin
+++ b/src/server/completion.odin
@@ -1553,6 +1553,12 @@ get_identifier_completion :: proc(
 	if position_context.identifier != nil {
 		if ident, ok := position_context.identifier.derived.(^ast.Ident); ok {
 			lookup_name = ident.name
+		} else if poly, ok := position_context.identifier.derived.(^ast.Poly_Type); ok {
+			if poly.specialization != nil {
+				if ident, ok := poly.specialization.derived.(^ast.Ident); ok {
+					lookup_name = ident.name
+				}
+			}
 		}
 	}
 
diff --git a/tests/completions_test.odin b/tests/completions_test.odin
index a19ab22..d6af935 100644
--- a/tests/completions_test.odin
+++ b/tests/completions_test.odin
@@ -4599,3 +4599,16 @@ ast_completion_handle_matching_from_overloaded_proc :: proc(t: ^testing.T) {
 	}
 	test.expect_completion_insert_text(t, &source, "", {"&foo"})
 }
+
+@(test)
+ast_completion_poly_proc_narrow_type :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+
+		Foo :: struct {x, y: i32}
+
+		foo :: proc (a: $T/F{*}) {}
+		`,
+	}
+	test.expect_completion_docs(t, &source, "", {"test.Foo: struct {..}"})
+}