From c4f3cd0efabe8898880ab6adfa43b40f8f216df3 Mon Sep 17 00:00:00 2001
From: Gregory Anders <greg@gpanders.com>
Date: Wed, 10 Aug 2022 16:03:00 -0600
Subject: [PATCH] Only return highlights for symbols in current document

This seems to only occur when highlighting "field access" symbols, e.g.
"bar" in a statement such as `foo.bar`. This is because the
`symbolReferencesInternal` function finds the reference to the field in
the container even when the container is not in the current document.
---
 src/Server.zig | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/src/Server.zig b/src/Server.zig
index bcedadf..c5f5bbb 100644
--- a/src/Server.zig
+++ b/src/Server.zig
@@ -1144,11 +1144,14 @@ fn referencesDefinitionGlobal(
 
     const result: types.ResponseParams = if (highlight) result: {
         var highlights = try std.ArrayList(types.DocumentHighlight).initCapacity(server.arena.allocator(), locs.items.len);
+        const uri = handle.uri();
         for (locs.items) |loc| {
-            highlights.appendAssumeCapacity(.{
-                .range = loc.range,
-                .kind = .Text,
-            });
+            if (std.mem.eql(u8, loc.uri, uri)) {
+                highlights.appendAssumeCapacity(.{
+                    .range = loc.range,
+                    .kind = .Text,
+                });
+            }
         }
         break :result .{ .DocumentHighlight = highlights.items };
     } else .{ .Locations = locs.items };
@@ -1187,11 +1190,14 @@ fn referencesDefinitionFieldAccess(
     );
     const result: types.ResponseParams = if (highlight) result: {
         var highlights = try std.ArrayList(types.DocumentHighlight).initCapacity(server.arena.allocator(), locs.items.len);
+        const uri = handle.uri();
         for (locs.items) |loc| {
-            highlights.appendAssumeCapacity(.{
-                .range = loc.range,
-                .kind = .Text,
-            });
+            if (std.mem.eql(u8, loc.uri, uri)) {
+                highlights.appendAssumeCapacity(.{
+                    .range = loc.range,
+                    .kind = .Text,
+                });
+            }
         }
         break :result .{ .DocumentHighlight = highlights.items };
     } else .{ .Locations = locs.items };
@@ -1218,11 +1224,14 @@ fn referencesDefinitionLabel(
     try references.labelReferences(&server.arena, decl, server.offset_encoding, include_decl, &locs, std.ArrayList(types.Location).append);
     const result: types.ResponseParams = if (highlight) result: {
         var highlights = try std.ArrayList(types.DocumentHighlight).initCapacity(server.arena.allocator(), locs.items.len);
+        const uri = handle.uri();
         for (locs.items) |loc| {
-            highlights.appendAssumeCapacity(.{
-                .range = loc.range,
-                .kind = .Text,
-            });
+            if (std.mem.eql(u8, loc.uri, uri)) {
+                highlights.appendAssumeCapacity(.{
+                    .range = loc.range,
+                    .kind = .Text,
+                });
+            }
         }
         break :result .{ .DocumentHighlight = highlights.items };
     } else .{ .Locations = locs.items };