fix(vscode): more robustly handle VSCode document URI (#1376)

zenstackhq · Apr 25, 2024 · 5cc8f18 · 5cc8f18
1 parent da0bbd1
commit 5cc8f18
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/packages/schema/src/utils/ast-utils.ts b/packages/schema/src/utils/ast-utils.ts
@@ -181,15 +181,19 @@ function resolveTransitiveImportsInternal(
     documents: LangiumDocuments,
     model: Model,
     initialModel = model,
-    visited: Set<URI> = new Set(),
+    visited: Set<string> = new Set(),
     models: Set<Model> = new Set()
 ): Model[] {
     const doc = getDocument(model);
-    if (initialModel !== model) {
+    const initialDoc = getDocument(initialModel);
+
+    if (initialDoc.uri.fsPath.toLowerCase() !== doc.uri.fsPath.toLowerCase()) {
         models.add(model);
     }
-    if (!visited.has(doc.uri)) {
-        visited.add(doc.uri);
+
+    const normalizedPath = doc.uri.fsPath.toLowerCase();
+    if (!visited.has(normalizedPath)) {
+        visited.add(normalizedPath);
         for (const imp of model.imports) {
             const importedModel = resolveImport(documents, imp);
             if (importedModel) {