only report workspace symbols on files inside a workspace folder

Author: Techatrix

src/DocumentStore.zig

@@ -1077,16 +1077,28 @@ fn invalidateBuildFileWorker(self: *DocumentStore, build_file: *BuildFile) void
     }
 }
 
-pub fn loadTrigramStores(store: *DocumentStore) error{OutOfMemory}![]*DocumentStore.Handle {
+pub fn loadTrigramStores(
+    store: *DocumentStore,
+    filter_paths: []const []const u8,
+) error{OutOfMemory}![]*DocumentStore.Handle {
     const tracy_zone = tracy.trace(@src());
     defer tracy_zone.end();
 
     var handles: std.ArrayListUnmanaged(*DocumentStore.Handle) = try .initCapacity(store.allocator, store.handles.count());
     errdefer handles.deinit(store.allocator);
 
     for (store.handles.values()) |handle| {
-        // TODO check if the handle is in a workspace folder instead
-        if (isInStd(handle.uri)) continue;
+        if (URI.toFsPath(store.allocator, handle.uri)) |path| {
+            defer store.allocator.free(path);
+            for (filter_paths) |filter_path| {
+                if (std.mem.startsWith(u8, path, filter_path)) break;
+            } else break;
+        } else |err| switch (err) {
+            error.OutOfMemory => return error.OutOfMemory,
+            else => {
+                // The URI is either invalid or not a `file` scheme. Either way, we should include it.
+            },
+        }
         handles.appendAssumeCapacity(handle);
     }
 

src/features/workspace_symbols.zig

@@ -9,11 +9,22 @@ const DocumentStore = @import("../DocumentStore.zig");
 const offsets = @import("../offsets.zig");
 const Server = @import("../Server.zig");
 const TrigramStore = @import("../TrigramStore.zig");
+const URI = @import("../uri.zig");
 
 pub fn handler(server: *Server, arena: std.mem.Allocator, request: types.WorkspaceSymbolParams) error{OutOfMemory}!lsp.ResultType("workspace/symbol") {
     if (request.query.len < 3) return null;
 
-    const handles = try server.document_store.loadTrigramStores();
+    var workspace_paths: std.ArrayList([]const u8) = try .initCapacity(arena, server.workspaces.items.len);
+    for (server.workspaces.items) |workspace| {
+        const path = URI.toFsPath(arena, workspace.uri) catch |err| switch (err) {
+            error.OutOfMemory => return error.OutOfMemory,
+            error.UnsupportedScheme => continue,
+            else => continue,
+        };
+        workspace_paths.appendAssumeCapacity(path);
+    }
+
+    const handles = try server.document_store.loadTrigramStores(workspace_paths.items);
     defer server.document_store.allocator.free(handles);
 
     var symbols: std.ArrayListUnmanaged(lsp.types.WorkspaceSymbol) = .empty;