fix jetbrains

Author: sxlijin

engine/language_server/src/server/api/requests/execute_command.rs

@@ -75,6 +75,7 @@ impl SyncRequestHandler for ExecuteCommand {
             return Ok(None);
         }
 
+        tracing::info!("Executing command: {:?}", params);
         match RegisteredCommands::from_execute_command(params) {
             Err(e) => {
                 return Err(crate::server::api::Error {
@@ -83,27 +84,31 @@ impl SyncRequestHandler for ExecuteCommand {
                 });
             }
             Ok(RegisteredCommands::OpenBamlPanel(args)) => {
-                session
+                let tx = session
                     .playground_tx
                     .send(PreLangServerToWasmMessage::FrontendMessage(
                         FrontendMessage::select_function {
                             // TODO: this can't be correct... but it looks like it is
                             root_path: args.project_id,
                             function_name: args.function_name,
                         },
-                    ))
-                    .unwrap();
+                    ));
+                if let Err(e) = tx {
+                    tracing::warn!("Error forwarding OpenBamlPanel to playground: {}", e);
+                }
             }
             Ok(RegisteredCommands::RunTest(args)) => {
-                session
+                let tx = session
                     .playground_tx
                     .send(PreLangServerToWasmMessage::FrontendMessage(
                         FrontendMessage::run_test {
                             function_name: args.function_name,
                             test_name: args.test_case_name,
                         },
-                    ))
-                    .unwrap();
+                    ));
+                if let Err(e) = tx {
+                    tracing::warn!("Error forwarding RunTest to playground: {}", e);
+                }
             }
         }
 

engine/tools/Cargo.toml

@@ -8,8 +8,8 @@ name = "language-server-hot-reload"
 path = "src/bin/language-server-hot-reload.rs"
 
 [dependencies]
-notify-debouncer-full = "0.3"
-tokio.workspace = true
 anyhow.workspace = true
+notify-debouncer-full = "0.3"
+tokio = { workspace = true, features = ["full"] }
 tracing.workspace = true
 tracing-subscriber.workspace = true

engine/tools/src/bin/language-server-hot-reload.rs

@@ -21,7 +21,6 @@ use tokio::{
 use tracing::{info, warn};
 use tracing_subscriber::{EnvFilter, FmtSubscriber};
 
-const BINARY_PATH: &str = "/Users/sam/baml4/engine/target/debug/baml-cli";
 const MAX_STDIN_BUFFER_SIZE: usize = 1000;
 
 #[derive(Clone, Debug)]
@@ -41,7 +40,7 @@ impl HotReloader {
     fn new() -> Self {
         let (shutdown_tx, _) = watch::channel(false);
         Self {
-            binary_path: BINARY_PATH.to_string(),
+            binary_path: format!("{}/../target/debug/baml-cli", env!("CARGO_MANIFEST_DIR")),
             current_process: None,
             shutdown_tx,
             stdin_buffer: Arc::new(Mutex::new(VecDeque::new())),

integ-tests/baml_src/test-files/functions/input/named-args/single/named-video.baml

@@ -149,6 +149,31 @@ tasks {
     runIde {
         args = listOf("${layout.projectDirectory}/../integ-tests/baml_src")
     }
+    
+    // Configure trusted paths to avoid security prompts in development
+    prepareSandbox {
+        val integTestsPath = layout.projectDirectory.dir("../integ-tests").asFile.absolutePath
+        val bamlSrcPath = layout.projectDirectory.dir("../integ-tests/baml_src").asFile.absolutePath
+        
+        doLast {
+            val trustedPathsFile = sandboxConfigDirectory.file("options/trusted-paths.xml").get().asFile
+            trustedPathsFile.parentFile.mkdirs()
+            trustedPathsFile.writeText(
+                """
+                <application>
+                  <component name="Trusted.Paths">
+                    <option name="TRUSTED_PROJECT_PATHS">
+                      <map>
+                        <entry key="$integTestsPath" value="true" />
+                        <entry key="$bamlSrcPath" value="true" />
+                      </map>
+                    </option>
+                  </component>
+                </application>
+                """.trimIndent()
+            )
+        }
+    }
 }
 
 intellijPlatformTesting {

jetbrains/build.gradle.kts

@@ -8,6 +8,17 @@ import java.nio.file.Path
 
 class BamlLanguageServer(private val project: Project) : OSProcessStreamConnectionProvider() {
 
+    private fun findBamlWorkspaceRoot(startPath: Path): Path? {
+        var current = startPath
+        while (current.parent != null) {
+            if (Files.exists(current.resolve("engine/Cargo.toml"))) {
+                return current
+            }
+            current = current.parent
+        }
+        return null
+    }
+
     init {
         val commandLine = if (BamlIdeConfig.isDebugMode) {
             // Kill any orphaned baml-cli processes before starting
@@ -17,8 +28,13 @@ class BamlLanguageServer(private val project: Project) : OSProcessStreamConnecti
 
             // baml-hot-reload is implemented by recording and replaying stdin, but this may be buggy
             // if that happens, comment this out and just use `baml-cli` directly
-            GeneralCommandLine("/Users/sam/baml4/engine/target/debug/baml-hot-reload", "lsp")
+            val projectBasePath = project.basePath ?: throw RuntimeException("Project base path not found")
+            val workspaceRoot = findBamlWorkspaceRoot(Path.of(projectBasePath)) ?: throw RuntimeException("BAML workspace root not found")
+            val hotReloadPath = workspaceRoot.resolve("engine/target/debug/language-server-hot-reload")
+            GeneralCommandLine(hotReloadPath.toString(), "lsp")
                 .withEnvironment("RUST_BACKTRACE", "full")
+                .withEnvironment("BAML_INTERNAL_LOG", "debug")
+                .withEnvironment("RUST_LOG", "debug")
                 .withEnvironment("VSCODE_DEBUG_MODE", "true")
             // Commented debug option:
             // GeneralCommandLine("/Users/sam/baml4/engine/target/debug/baml-cli", "lsp")

jetbrains/src/main/kotlin/com/boundaryml/jetbrains_ext/BamlLanguageServer.kt

@@ -5,21 +5,21 @@ import com.redhat.devtools.lsp4ij.commands.LSPCommandAction
 import org.eclipse.lsp4j.ExecuteCommandParams
 import com.redhat.devtools.lsp4ij.commands.LSPCommand
 import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.diagnostic.logger
 
 class BamlPlaygroundAction : LSPCommandAction() {
 
     override fun commandPerformed(command: LSPCommand, e: AnActionEvent) {
         val project = e.project ?: return
         val toolWindow = ToolWindowManager.getInstance(project)
-            .getToolWindow("BAML Playground")
+            .getToolWindow("BAML Playground (beta)")
 
         val args: List<Any> = command.arguments
 
-        toolWindow?.show {
-            val ls = getLanguageServer(e)?.server ?: return@show
-            ls.workspaceService.executeCommand(
-                ExecuteCommandParams(command.command, command.arguments)
-            )
-        }
+        toolWindow?.show()
+        val ls = getLanguageServer(e)?.server
+        ls?.workspaceService?.executeCommand(
+            ExecuteCommandParams(command.command, command.arguments)
+        )
     }
 }

jetbrains/src/main/kotlin/com/boundaryml/jetbrains_ext/BamlPlaygroundAction.kt

@@ -30,6 +30,7 @@
     </actions>
 
     <extensions defaultExtensionNs="com.intellij">
+        <!-- When changing toolWindow.id, also make sure to change BamlPlaygroundAction.kt -->
         <toolWindow
                 id="BAML Playground (beta)"
                 factoryClass="com.boundaryml.jetbrains_ext.BamlToolWindowFactory"