Fix stack trace parsing and enable feature by default

Co-authored-by: sheremet-va <[email protected]>

Author: Copilot

README.md

@@ -11,7 +11,7 @@
 - **Run**, **debug**, and **watch** Vitest tests in Visual Studio Code.
 - **Coverage** support (requires VS Code >= 1.88)
 - An `@open` tag can be used when filtering tests, to only show the tests open in the editor.
-- **Inline console.log display**: Optionally show `console.log` output inline in the editor (enable via `vitest.showConsoleLogInline` setting)
+- **Inline console.log display**: Show `console.log` output inline in the editor next to the line that produced it (enabled by default, requires `printConsoleTrace: true` in Vitest config)
 
 ## Requirements
 
@@ -100,7 +100,7 @@ These options are resolved relative to the [workspace file](https://code.visuals
 - `vitest.logLevel`: How verbose should the logger be in the "Output" channel. Default: `info`
 - `vitest.applyDiagnostic`: Show a squiggly line where the error was thrown. This also enables the error count in the File Tab. Default: `true`
 - `vitest.experimentalStaticAstCollect`: uses AST parses to collect tests instead of running files and collecting them at runtime. Default: `true`
-- `vitest.showConsoleLogInline`: Show `console.log` output inline in the editor next to the line that produced it. This can be useful for debugging. Default: `false`
+- `vitest.showConsoleLogInline`: Show `console.log` output inline in the editor next to the line that produced it. Note: This requires `printConsoleTrace: true` in your Vitest config to include stack trace information. Default: `true`
 
 ### Commands
 

package.json

@@ -255,7 +255,7 @@
         "vitest.showConsoleLogInline": {
           "description": "Show console.log output inline in the editor next to the line that produced it.",
           "type": "boolean",
-          "default": false,
+          "default": true,
           "scope": "resource"
         }
       }

packages/extension/src/config.ts

@@ -72,7 +72,7 @@ export function getConfig(workspaceFolder?: WorkspaceFolder) {
   const debugOutFiles = get<string[]>('debugOutFiles', [])
   const applyDiagnostic = get<boolean>('applyDiagnostic', true)
   const ignoreWorkspace = get<boolean>('ignoreWorkspace', false) ?? false
-  const showConsoleLogInline = get<boolean>('showConsoleLogInline', false)!
+  const showConsoleLogInline = get<boolean>('showConsoleLogInline', true)!
 
   return {
     env: get<null | Record<string, string>>('nodeEnv', null),

packages/extension/src/inlineConsoleLog.ts

@@ -110,21 +110,41 @@ export class InlineConsoleLogManager extends vscode.Disposable {
       return null
     }
 
-    // Origin format is typically: file:line:column
-    // Example: /path/to/file.ts:10:5
-    const match = origin.match(/^(.+):(\d+):(\d+)$/)
-    if (!match) {
-      return null
-    }
-
-    const [, file, lineStr] = match
-    const line = Number.parseInt(lineStr, 10) - 1 // Convert to 0-based line number
-
-    if (Number.isNaN(line)) {
-      return null
+    // Origin is a stack trace string. We need to extract the file path and line number.
+    // Stack trace formats vary but typically look like:
+    //   at functionName (file:///path/to/file.ts:10:5)
+    //   at /path/to/file.ts:10:5
+    //   at Object.<anonymous> (/path/to/file.ts:10:5)
+    // We look for the first line that contains a file path with line:column
+
+    const lines = origin.split('\n')
+    for (const line of lines) {
+      // Match various stack trace formats
+      // Handles: (file:///path/to/file.ts:10:5) or (/path/to/file.ts:10:5) or just /path/to/file.ts:10:5
+      const match = line.match(/(?:file:\/\/)?([^():\s]+\.(?:ts|js|jsx|tsx|mjs|cjs|cts|mts)):(\d+):(\d+)/)
+      if (match) {
+        const [, file, lineStr] = match
+        const lineNum = Number.parseInt(lineStr, 10) - 1 // Convert to 0-based line number
+
+        if (!Number.isNaN(lineNum) && lineNum >= 0) {
+          // Clean up file:// protocol if present and decode URI components
+          let cleanPath = file
+          if (cleanPath.startsWith('file://')) {
+            cleanPath = cleanPath.substring(7)
+          }
+          try {
+            cleanPath = decodeURIComponent(cleanPath)
+          }
+          catch {
+            // If decoding fails, use the original path
+          }
+
+          return { file: cleanPath, line: lineNum }
+        }
+      }
     }
 
-    return { file, line }
+    return null
   }
 
   private updateDecorations(editor: vscode.TextEditor): void {

test/unit/inlineConsoleLog.test.ts

@@ -3,11 +3,11 @@ import { expect } from 'chai'
 import { InlineConsoleLogManager } from '../../packages/extension/src/inlineConsoleLog'
 
 describe('InlineConsoleLogManager', () => {
-  it('correctly parses origin with file path', () => {
+  it('correctly parses stack trace with file path', () => {
     const manager = new InlineConsoleLogManager()
     const log: UserConsoleLog = {
       content: 'test message',
-      origin: '/path/to/file.ts:10:5',
+      origin: '    at Object.<anonymous> (/path/to/file.ts:10:5)',
       type: 'stdout',
       taskId: 'test-id',
       time: Date.now(),
@@ -25,6 +25,46 @@ describe('InlineConsoleLogManager', () => {
     manager.dispose()
   })
 
+  it('correctly parses stack trace with file:// protocol', () => {
+    const manager = new InlineConsoleLogManager()
+    const log: UserConsoleLog = {
+      content: 'test message',
+      origin: '    at functionName (file:///path/to/file.ts:10:5)',
+      type: 'stdout',
+      taskId: 'test-id',
+      time: Date.now(),
+      size: 12,
+    }
+
+    // @ts-expect-error accessing private method for testing
+    const result = manager.parseOrigin(log.origin)
+
+    expect(result).to.deep.equal({
+      file: '/path/to/file.ts',
+      line: 9, // 0-based
+    })
+
+    manager.dispose()
+  })
+
+  it('correctly parses multi-line stack trace', () => {
+    const manager = new InlineConsoleLogManager()
+    const stackTrace = `    at consoleLog (file:///path/to/helper.ts:5:3)
+    at test (file:///path/to/file.ts:10:5)
+    at Object.<anonymous> (file:///path/to/other.ts:20:3)`
+
+    // @ts-expect-error accessing private method for testing
+    const result = manager.parseOrigin(stackTrace)
+
+    // Should parse the first valid line
+    expect(result).to.deep.equal({
+      file: '/path/to/helper.ts',
+      line: 4, // 0-based
+    })
+
+    manager.dispose()
+  })
+
   it('returns null for invalid origin', () => {
     const manager = new InlineConsoleLogManager()
 
@@ -33,7 +73,7 @@ describe('InlineConsoleLogManager', () => {
     // @ts-expect-error accessing private method for testing
     expect(manager.parseOrigin('invalid')).to.be.null
     // @ts-expect-error accessing private method for testing
-    expect(manager.parseOrigin('/path/to/file.ts')).to.be.null
+    expect(manager.parseOrigin('no file path here')).to.be.null
 
     manager.dispose()
   })