Run command in the project directory, if available

Fixes #57.

Author: liff

CHANGELOG.md

@@ -4,6 +4,8 @@
 
 ## [Unreleased]
 
+- Attempt to use the project directory as the working directory when running the command.
+
 ## [0.1.0] - 2024-03-18
 
 - Extend compatibility to 2024.1

README.md

@@ -7,7 +7,8 @@
 <!-- Plugin description -->
 Use an external command to acquire database connection credentials.
 
-Executes an external command and extracts database and username from its output.
+Executes an external command and extracts database and username from its output. The command is executed in the
+project’s root directory, if that can be resolved.
 
 The plugin will first attempt to extract the information from output from a [Vault](https://www.vaultproject.io/)-like JSON with the following format:
 

src/main/kotlin/com/github/liff/dbexternalcommandauth/Acquire.kt

@@ -4,13 +4,20 @@ import com.intellij.credentialStore.Credentials
 import com.intellij.util.io.awaitExit
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.withContext
+import java.io.File
 
 suspend fun acquire(
     command: String,
+    directory: File?,
     parse: (String) -> Credentials? = ::parse,
 ): Credentials? =
     withContext(Dispatchers.IO) {
-        val proc = wrap(command).start()
+        val proc = wrap(command)
+            .apply {
+                if (directory != null)
+                    directory(directory)
+            }
+            .start()
 
         proc.outputStream.close()
 

src/main/kotlin/com/github/liff/dbexternalcommandauth/CommandAuthProvider.kt

@@ -9,6 +9,7 @@ import com.intellij.database.dataSource.DatabaseCredentialsAuthProvider.Companio
 import com.intellij.database.dataSource.ui.AuthWidgetBuilder
 import com.intellij.database.view.DatabaseCoreUiService
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.project.guessProjectDir
 
 class CommandAuthProvider : DatabaseAuthProvider {
     override suspend fun interceptConnection(
@@ -19,7 +20,15 @@ class CommandAuthProvider : DatabaseAuthProvider {
             proto.connectionPoint.getAdditionalProperty("command")
                 ?: error(MyBundle.message("no-command"))
 
-        val credentials = acquire(command)
+        val directory = proto.project.guessProjectDir()?.let {
+            try {
+                it.toNioPath().toFile()
+            } catch (_: UnsupportedOperationException) {
+                null
+            }
+        }
+
+        val credentials = acquire(command = command, directory = directory)
 
         applyCredentials(
             proto = proto,