Capability type checking. (#3530)

* Capability type checking.

* Fix.

---------

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 23 insertions, 0 deletions

diff --git a/source/slang/slang-language-server-completion.cpp b/source/slang/slang-language-server-completion.cpp
index 6038a432a..b723e14b8 100644
--- a/source/slang/slang-language-server-completion.cpp
+++ b/source/slang/slang-language-server-completion.cpp
@@ -443,6 +443,11 @@ List<LanguageServerProtocol::CompletionItem> CompletionContext::collectMembersAn
             linkage->contentAssistInfo.completionSuggestions.swizzleBaseType,
             linkage->contentAssistInfo.completionSuggestions.elementCount);
     }
+    else if (linkage->contentAssistInfo.completionSuggestions.scopeKind ==
+        CompletionSuggestions::ScopeKind::Capabilities)
+    {
+        return createCapabilityCandidates();
+    }
     List<LanguageServerProtocol::CompletionItem> result;
     bool useCommitChars = true;
     bool addKeywords = false;
@@ -595,6 +600,24 @@ List<LanguageServerProtocol::CompletionItem> CompletionContext::collectMembersAn
     return result;
 }
 
+List<LanguageServerProtocol::CompletionItem> CompletionContext::createCapabilityCandidates()
+{
+    List<LanguageServerProtocol::CompletionItem> result;
+    List<UnownedStringSlice> names;
+    getCapabilityNames(names);
+    for (auto name : names.getArrayView(1, names.getCount()-1))
+    {
+        if (name.startsWith("_"))
+            continue;
+        LanguageServerProtocol::CompletionItem item;
+        item.data = 0;
+        item.kind = LanguageServerProtocol::kCompletionItemKindEnumMember;
+        item.label = name;
+        result.add(item);
+    }
+    return result;
+}
+
 List<LanguageServerProtocol::CompletionItem> CompletionContext::createSwizzleCandidates(
     Type* type, IntegerLiteralValue elementCount[2])
 {