actually emit the version when opening baml file (#2612)

<!-- CURSOR_SUMMARY -->
> [!NOTE]
> Emit generator version on BAML file open and track the last active
BAML file in VS Code to scope version updates; improve reload/logging
and simplify debug/telemetry handling.
> 
> - **Engine (LSP)**:
> - **didOpen**: Reload project and send `baml_src_generator_version`;
add tracing around generator version retrieval.
> - **didSave**: Add info log when no generator version map is
available.
> - **Session.reload**: Enhance logging (per-project and total file
counts) during initial reload.
> - **VS Code Extension**:
> - **Active file tracking**: Introduce `LAST_ACTIVE_BAML_FILE` and
update it on cursor selection; use it to validate
`baml_src_generator_version` root before acting.
> - **Debug/telemetry**: Replace `isDebugOrTestSession` with
`isDebugMode`; always initialize telemetry; adjust disposal condition;
add debug-mode CLI logs.
> 
> <sup>Written by [Cursor
Bugbot](https://cursor.com/dashboard?tab=bugbot) for commit
95dc502. This will update automatically
on new commits. Configure
[here](https://cursor.com/dashboard?tab=bugbot).</sup>
<!-- /CURSOR_SUMMARY -->

Author: aaronvg

engine/language_server/src/server/api/notifications/did_open.rs

@@ -75,7 +75,15 @@ impl SyncNotificationHandler for DidOpenTextDocumentHandler {
             )
             .internal_error()?;
 
-        // tracing::info!("before get_or_create_project");
+        // session.open_text_document(
+        //    q DocumentKey::from_path(&file_path, &file_path).internal_error()?,
+        //     TextDocument::new(params.text_document.text, params.text_document.version),
+        // );
+
+        session.reload(Some(notifier.clone())).internal_error()?;
+
+        // We do this after the project reload since we may be loading this baml project with files (and creating a new runtime) in the .reload(), and we only want to send the generator version if we've had a runtime created. Ideally we don't depend on the runtime being created (since our version of the LSP may not be able to read all baml files), and it only reads the generator config blocks.
+        tracing::info!("before send_generator_version");
         {
             let locked = project.lock();
             let default_flags = vec!["beta".to_string()];
@@ -87,16 +95,10 @@ impl SyncNotificationHandler for DidOpenTextDocumentHandler {
             let client_version = session.baml_settings.get_client_version();
 
             let generator_version = locked.get_common_generator_version();
+            tracing::info!("common generator version {:?}", generator_version);
             send_generator_version(&notifier, &locked, generator_version.as_ref().ok());
         }
 
-        // session.open_text_document(
-        //     DocumentKey::from_path(&file_path, &file_path).internal_error()?,
-        //     TextDocument::new(params.text_document.text, params.text_document.version),
-        // );
-
-        session.reload(Some(notifier.clone())).internal_error()?;
-
         publish_session_lsp_diagnostics(&notifier, session, &url)?;
 
         Ok(())

engine/language_server/src/server/api/notifications/did_save_text_document.rs

@@ -164,6 +164,8 @@ pub(crate) fn send_generator_version(
                     "Failed to send baml_src_generator_version notification to IDE: {e}"
                 );
             });
+    } else {
+        tracing::info!("No version map available");
     }
 }
 

engine/language_server/src/session.rs

@@ -245,12 +245,19 @@ impl Session {
 
         let project_updates: Vec<HashMap<_, _>> = baml_src_projects
             .iter_mut()
-            .map(|(_project_root, project)| {
+            .map(|(project_root, project)| {
                 let files_map = project
                     .lock()
                     .baml_project
                     .load_files()
                     .map_err(|e| anyhow::anyhow!("Failed to load project files: {}", e))?;
+
+                tracing::info!(
+                    "Loaded {} files for project root: {:?}",
+                    files_map.len(),
+                    project_root
+                );
+
                 {
                     let default_flags = vec!["beta".to_string()];
                     project.lock().update_runtime(
@@ -268,7 +275,13 @@ impl Session {
                 Ok(files_map)
             })
             .collect::<anyhow::Result<Vec<_>>>()?;
-        tracing::info!("Initial reload of {} files", project_updates.len());
+
+        let total_files: usize = project_updates.iter().map(|m| m.len()).sum();
+        tracing::info!(
+            "Initial reload complete: {} projects, {} total files",
+            project_updates.len(),
+            total_files
+        );
 
         // Guard no longer used. We can drop now instead of waiting for the end
         // of scope.

typescript/apps/vscode-ext/src/extension.ts

@@ -17,6 +17,7 @@ import type { Express } from 'express'
 import { Socket } from 'net'
 import StatusBarPanel from './panels/StatusBarPanel'
 import { Server } from 'http'
+import { LAST_ACTIVE_BAML_FILE } from './helpers/get-open-file'
 
 const outputChannel = vscode.window.createOutputChannel('baml')
 const diagnosticsCollection =
@@ -292,7 +293,18 @@ export function activate(context: vscode.ExtensionContext) {
   })
 
   // Add cursor movement listener
+  vscode.window.onDidChangeActiveTextEditor
   vscode.window.onDidChangeTextEditorSelection((event) => {
+
+    // properly track the actual last known baml file.
+    if (event.textEditor.document.fileName.endsWith(".baml")) {
+      LAST_ACTIVE_BAML_FILE.uri = event.textEditor.document.uri;
+    }
+
+
+
+    // The code below this line may not be fully accurate. The vscode.window.activeTextEditor may go
+    // undefined in some cases.
     const position = event.selections[0]?.active
 
     const editor = vscode.window.activeTextEditor
@@ -303,6 +315,8 @@ export function activate(context: vscode.ExtensionContext) {
       return
     }
 
+
+
     // TODO: buggy when used with multiple functions, needs a fix.
     WebviewPanelHost.currentPanel?.sendCommandToWebview({
       source: 'ide_message',

typescript/apps/vscode-ext/src/helpers/get-open-file.ts

@@ -13,3 +13,7 @@ export const getCurrentOpenedFile = () => {
     return visibleEditors[0]?.document.uri?.toString() ?? '';
   }
 };
+
+export const LAST_ACTIVE_BAML_FILE: { uri: undefined | vscode.Uri } = {
+  uri: undefined
+};

typescript/apps/vscode-ext/src/plugins/language-server-client/index.ts

@@ -20,14 +20,13 @@ import {
 import { URI } from 'vscode-uri';
 import { z } from 'zod';
 import packageJson from '../../../package.json';
-import { getCurrentOpenedFile } from '../../helpers/get-open-file';
+import { getCurrentOpenedFile, LAST_ACTIVE_BAML_FILE } from '../../helpers/get-open-file';
 import StatusBarPanel from '../../panels/StatusBarPanel';
 import { WebviewPanelHost } from '../../panels/WebviewPanelHost';
 import TelemetryReporter from '../../telemetryReporter';
 import {
   checkForMinimalColorTheme,
   createLanguageServer,
-  isDebugOrTestSession,
 } from '../../util';
 import type { BamlVSCodePlugin } from '../types';
 import {
@@ -544,15 +543,16 @@ export const registerClientEventHandlers = (client: LanguageClient, context: Ext
         }
 
         // Check if this version update is for the currently active baml_src directory
-        const activeEditor =
-          window.activeTextEditor || (window.visibleTextEditors.length > 0 ? window.visibleTextEditors[0] : null);
+        const activeEditor = LAST_ACTIVE_BAML_FILE.uri;
         if (activeEditor) {
           try {
-            const currentFilePath = URI.parse(activeEditor.document.uri.toString()).fsPath;
+
+            const currentFilePath = activeEditor.fsPath;
+
             const rootPathUri = URI.file(payload.root_path).fsPath;
             if (!isPathWithinParent(currentFilePath, rootPathUri)) {
               bamlOutputChannel.appendLine(
-                `baml_src_generator_version ignored: root path does not match active editor ${currentFilePath} ${rootPathUri}`,
+                `baml_src_generator_version ignored: root path does not match active editor ${currentFilePath} root: ${rootPathUri}`,
               );
               return;
             }
@@ -795,7 +795,7 @@ const plugin: BamlVSCodePlugin = {
   name: 'baml-language-server',
   enabled: () => true,
   activate: async (context, _outputChannel) => {
-    const isDebugOrTest = isDebugOrTestSession();
+    const isDebugOrTest = isDebugMode();
     bamlOutputChannel = _outputChannel;
     context.subscriptions.push(bamlOutputChannel);
     bamlOutputChannel.appendLine('Activating BAML Language Server plugin...');
@@ -809,6 +809,8 @@ const plugin: BamlVSCodePlugin = {
 
     let serverAbsolutePath: string | null = null;
     if (isDebugOrTest) {
+      console.log('Using debug cli in debug mode');
+      bamlOutputChannel.append('Using debug cli in debug mode');
       serverAbsolutePath = process.env.VSCODE_DEBUG_BAML_CLI_PATH || null;
     } else {
       try {
@@ -1029,22 +1031,22 @@ const plugin: BamlVSCodePlugin = {
 
     activateClient(context, serverOptions, clientOptions);
 
-    if (!isDebugOrTest) {
-      try {
-        const extensionId = `Boundary.${packageJson.name}`;
-        const extensionVersion: string = packageJson.version;
-        console.log(
-          `Initializing telemetry for ${extensionId} v${extensionVersion}`,
-        );
-        telemetry = new TelemetryReporter(extensionId, extensionVersion);
-        context.subscriptions.push(telemetry);
-        await telemetry.initialize();
-        console.log('Telemetry initialized.');
-      } catch (err) {
-        console.error('Failed to initialize telemetry:', err);
-      }
+
+    try {
+      const extensionId = `Boundary.${packageJson.name}`;
+      const extensionVersion: string = packageJson.version;
+      console.log(
+        `Initializing telemetry for ${extensionId} v${extensionVersion}`,
+      );
+      telemetry = new TelemetryReporter(extensionId, extensionVersion);
+      context.subscriptions.push(telemetry);
+      await telemetry.initialize();
+      console.log('Telemetry initialized.');
+    } catch (err) {
+      console.error('Failed to initialize telemetry:', err);
     }
 
+
     checkForMinimalColorTheme();
     console.log('BAML Language Server plugin activation finished.');
   },
@@ -1061,7 +1063,7 @@ const plugin: BamlVSCodePlugin = {
       console.log('Client not running or already stopped.');
     }
 
-    if (!isDebugOrTestSession() && telemetry) {
+    if (!isDebugMode() && telemetry) {
       console.log('Disposing telemetry.');
       await telemetry.dispose().catch((err) => {
         console.error('Error disposing telemetry:', err);

typescript/apps/vscode-ext/src/util.ts

@@ -17,9 +17,7 @@ import type {
 } from 'vscode-languageclient';
 import { LanguageClient, type ServerOptions } from 'vscode-languageclient/node';
 
-export function isDebugOrTestSession(): boolean {
-  return env.sessionId === 'someValue.sessionId';
-}
+
 
 export function checkForOtherPrismaExtension(): void {
   const files = readdirSync(path.join(homedir(), '.vscode/extensions')).filter(