Fix findWorkspace function

Part of #88

Author: JohnnyMorganz

CHANGELOG.md

@@ -22,6 +22,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 - Fixed crash when hovering over local in incomplete syntax tree
 - Fixed language server not working for newly created files not yet stored on disk
 - Luau LSP will now activate if you run an LSP command
+- Fixed finding the incorrect workspace folder to analyze with in a multi-workspace environment
 
 ## [1.7.1] - 2022-07-17
 

src/LanguageServer.cpp

@@ -21,13 +21,32 @@ using ClientPtr = std::shared_ptr<Client>;
 /// If no workspace is found, the file is attached to the null workspace
 WorkspaceFolderPtr LanguageServer::findWorkspace(const lsp::DocumentUri file)
 {
+    WorkspaceFolderPtr bestWorkspace = nullptr;
+    size_t length = 0;
+    auto checkStr = file.toString();
+
+
     for (auto& workspace : workspaceFolders)
     {
-        if (workspace->isInWorkspace(file))
+        if (file == workspace->rootUri)
+            return workspace;
+
+        // Check if the root uri is a prefix of the file
+        auto prefixStr = workspace->rootUri.toString();
+        auto size = prefixStr.size();
+        if (size < length)
+            continue;
+
+        if (checkStr.compare(0, size, prefixStr) == 0)
         {
-            return workspace; // TODO: should we return early here? maybe a better match comes along?
+            bestWorkspace = workspace;
+            length = size;
         }
     }
+
+    if (bestWorkspace)
+        return bestWorkspace;
+
     client->sendTrace("cannot find workspace for " + file.toString());
     return nullWorkspace;
 }