look for .bsp also in the project root

scalameta · Nov 21, 2023 · 8e60ab1 · 8e60ab1
1 parent ede54d8
commit 8e60ab1
Show file tree

Hide file tree

Showing 6 changed files with 52 additions and 31 deletions.
diff --git a/metals/src/main/scala/scala/meta/internal/bsp/BspConnector.scala b/metals/src/main/scala/scala/meta/internal/bsp/BspConnector.scala
@@ -129,6 +129,7 @@ class BspConnector(
             .map(Some(_))
         case ResolvedBspOne(details) =>
           tables.buildServers.chooseServer(details.getName())
+          optSetBuildTool(details.getName())
           bspServers
             .newServer(projectRoot, bspTraceRoot, details, bspStatusOpt)
             .map(Some(_))
@@ -163,6 +164,7 @@ class BspConnector(
                 )
               )
             _ = tables.buildServers.chooseServer(item.getName())
+            _ = optSetBuildTool(item.getName())
             conn <- bspServers.newServer(
               projectRoot,
               bspTraceRoot,
@@ -197,6 +199,13 @@ class BspConnector(
     }
   }
 
+  private def optSetBuildTool(buildServerName: String): Unit =
+    if (
+      buildTools.loadSupported().exists(_.executableName == buildServerName)
+    ) {
+      tables.buildTool.chooseBuildTool(buildServerName)
+    }
+
   private def sbtMetaWorkspaces(root: AbsolutePath): List[AbsolutePath] = {
     def recursive(
         p: AbsolutePath,

diff --git a/metals/src/main/scala/scala/meta/internal/bsp/BspServers.scala b/metals/src/main/scala/scala/meta/internal/bsp/BspServers.scala
@@ -47,6 +47,8 @@ final class BspServers(
     config: MetalsServerConfig,
     userConfig: () => UserConfiguration,
 )(implicit ec: ExecutionContextExecutorService) {
+  private def customProjectRoot =
+    userConfig().getCustomProjectRoot(mainWorkspace)
 
   def resolve(): BspResolvedResult = {
     findAvailableServers() match {
@@ -153,7 +155,7 @@ final class BspServers(
    *  may be a server in the current workspace
    */
   def findAvailableServers(): List[BspConnectionDetails] = {
-    val jsonFiles = findJsonFiles(mainWorkspace)
+    val jsonFiles = findJsonFiles()
     val gson = new Gson()
     for {
       candidate <- jsonFiles
@@ -172,17 +174,16 @@ final class BspServers(
     }
   }
 
-  private def findJsonFiles(
-      projectDirectory: AbsolutePath
-  ): List[AbsolutePath] = {
+  private def findJsonFiles(): List[AbsolutePath] = {
     val buf = List.newBuilder[AbsolutePath]
     def visit(dir: AbsolutePath): Unit =
       dir.list.foreach { p =>
         if (p.extension == "json") {
           buf += p
         }
       }
-    visit(projectDirectory.resolve(".bsp"))
+    visit(mainWorkspace.resolve(".bsp"))
+    customProjectRoot.map(_.resolve(".bsp")).foreach(visit)
     bspGlobalInstallDirectories.foreach(visit)
     buf.result()
   }

diff --git a/metals/src/main/scala/scala/meta/internal/builds/BspOnly.scala b/metals/src/main/scala/scala/meta/internal/builds/BspOnly.scala
@@ -7,16 +7,17 @@ import scala.meta.internal.mtags.MD5
 import scala.meta.io.AbsolutePath
 
 /**
- * Build tool for custom bsp detected in `.bsp/<name>.json`
+ * Build tool for custom bsp detected in `.bsp/<name>.json` or `bspGlobalDirectories`
  */
 case class BspOnly(
     override val executableName: String,
     override val projectRoot: AbsolutePath,
+    pathToBspConfig: AbsolutePath,
 ) extends BuildTool {
   override def digest(workspace: AbsolutePath): Option[String] = {
     val digest = MessageDigest.getInstance("MD5")
     val isSuccess =
-      Digest.digestJson(workspace.resolve(s"$executableName.json"), digest)
+      Digest.digestJson(pathToBspConfig, digest)
     if (isSuccess) Some(MD5.bytesToHex(digest.digest()))
     else None
   }

diff --git a/metals/src/main/scala/scala/meta/internal/builds/BuildTools.scala b/metals/src/main/scala/scala/meta/internal/builds/BuildTools.scala
@@ -55,6 +55,7 @@ final class BuildTools(
   def isBloop: Boolean = bloopProject.isDefined
   def isBsp: Boolean = {
     hasJsonFile(workspace.resolve(".bsp")) ||
+    customProjectRoot.exists(root => hasJsonFile(root.resolve(".bsp"))) ||
     bspGlobalDirectories.exists(hasJsonFile)
   }
   private def hasJsonFile(dir: AbsolutePath): Boolean = {
@@ -123,34 +124,33 @@ final class BuildTools(
   def isBazel: Boolean = bazelProject.isDefined
 
   private def customBsps: List[BspOnly] = {
-    val bspFolder = workspace.resolve(".bsp")
+    val bspFolders =
+      (workspace :: customProjectRoot.toList).distinct
+        .map(_.resolve(".bsp")) ++ bspGlobalDirectories
     val root = customProjectRoot.getOrElse(workspace)
-    if (bspFolder.exists && bspFolder.isDirectory) {
-      bspFolder.toFile
+    for {
+      bspFolder <- bspFolders
+      if (bspFolder.exists && bspFolder.isDirectory)
+      buildTool <- bspFolder.toFile
         .listFiles()
         .collect {
           case file
               if file.isFile() && file.getName().endsWith(".json") &&
                 !knownBsps(file.getName().stripSuffix(".json")) =>
-            BspOnly(file.getName().stripSuffix(".json"), root)
+            BspOnly(
+              file.getName().stripSuffix(".json"),
+              root,
+              AbsolutePath(file.toPath()),
+            )
         }
         .toList
-    } else Nil
+    } yield buildTool
   }
 
   private def knownBsps =
     Set(SbtBuildTool.name, MillBuildTool.name) ++ ScalaCli.names
 
-  private def customProjectRoot =
-    userConfig().customProjectRoot
-      .map(relativePath => workspace.resolve(relativePath.trim()))
-      .filter { projectRoot =>
-        val exists = projectRoot.exists
-        if (!exists) {
-          scribe.error(s"custom project root $projectRoot does not exist")
-        }
-        exists
-      }
+  private def customProjectRoot = userConfig().getCustomProjectRoot(workspace)
 
   private def searchForBuildTool(
       isProjectRoot: AbsolutePath => Boolean

diff --git a/metals/src/main/scala/scala/meta/internal/metals/UserConfiguration.scala b/metals/src/main/scala/scala/meta/internal/metals/UserConfiguration.scala
@@ -68,6 +68,18 @@ case class UserConfiguration(
     ) && this.showInferredType.nonEmpty
     showImplicitArguments || showInferredType || showImplicitConversionsAndClasses
   }
+
+  def getCustomProjectRoot(workspace: AbsolutePath): Option[AbsolutePath] =
+    customProjectRoot
+      .map(relativePath => workspace.resolve(relativePath.trim()))
+      .filter { projectRoot =>
+        val exists = projectRoot.toFile.exists
+        if (!exists) {
+          scribe.error(s"custom project root $projectRoot does not exist")
+        }
+        exists
+      }
+
 }
 
 object UserConfiguration {

diff --git a/tests/unit/src/test/scala/tests/BillLspSuite.scala b/tests/unit/src/test/scala/tests/BillLspSuite.scala
@@ -187,9 +187,7 @@ class BillLspSuite extends BaseLspSuite("bill") {
     testRoundtripCompilation()
   }
 
-  def testSelectServerDialogue(
-      additionalMessages: List[String] = Nil
-  ): Future[Unit] = {
+  def testSelectServerDialogue(): Future[Unit] = {
     // when asked, choose the Bob build tool
     client.selectBspServer = { actions =>
       actions.find(_.getTitle == "Bob").get
@@ -203,11 +201,11 @@ class BillLspSuite extends BaseLspSuite("bill") {
       )
       _ = assertNoDiff(
         client.workspaceMessageRequests,
-        (additionalMessages ++
-          List(
-            Messages.BspSwitch.message,
-            Messages.CheckDoctor.allProjectsMisconfigured,
-          )).mkString("\n"),
+        List(
+          Messages.ChooseBuildTool.message,
+          Messages.BspSwitch.message,
+          Messages.CheckDoctor.allProjectsMisconfigured,
+        ).mkString("\n"),
       )
     } yield ()
   }
@@ -216,7 +214,7 @@ class BillLspSuite extends BaseLspSuite("bill") {
     cleanWorkspace()
     Bill.installWorkspace(workspace.toNIO, "Bill")
     Bill.installWorkspace(workspace.toNIO, "Bob")
-    testSelectServerDialogue(List(Messages.ChooseBuildTool.message))
+    testSelectServerDialogue()
   }
 
   test("mix") {