Merge pull request jest-community#198 from seanpoulter/issue-178

Rework Debug CodeLens to display on correct documents

Author: orta

CHANGELOG.md

@@ -3,6 +3,7 @@
 
 ## Master
 
+* Improve where the Debug CodeLens is displayed - seanpoulter
 *
 
 ### 2.5.4

src/CodeLens.ts

@@ -2,12 +2,10 @@ import { createSourceMapStore } from 'istanbul-lib-source-maps'
 import { createCoverageMap, CoverageMap } from 'istanbul-lib-coverage'
 
 export class Coverage {
-  private rootPath: string
   private sourceMapStore
   private transformedCoverageMap: CoverageMap
 
-  constructor(rootPath: string) {
-    this.rootPath = rootPath
+  constructor() {
     this.sourceMapStore = createSourceMapStore()
     this.transformedCoverageMap = createCoverageMap()
   }

src/Coverage/Coverage.ts

@@ -0,0 +1,12 @@
+import * as vscode from 'vscode'
+
+export class DebugCodeLens extends vscode.CodeLens {
+  readonly fileName: string
+  readonly testName: string
+
+  constructor(range: vscode.Range, fileName: string, testName: string) {
+    super(range)
+    this.fileName = fileName
+    this.testName = testName
+  }
+}

src/DebugCodeLens/DebugCodeLens.ts

@@ -0,0 +1,71 @@
+import * as vscode from 'vscode'
+import { extensionName } from '../appGlobals'
+import { basename } from 'path'
+import { DebugCodeLens } from './DebugCodeLens'
+import { TestReconciliationState } from '../TestReconciliationState'
+import { TestResultProvider } from '../TestResultProvider'
+
+export class DebugCodeLensProvider implements vscode.CodeLensProvider {
+  private _enabled: boolean
+  onDidChange: vscode.EventEmitter<void>
+  testResultProvider: TestResultProvider
+
+  constructor(testResultProvider: TestResultProvider, enabled: boolean) {
+    this.testResultProvider = testResultProvider
+    this._enabled = enabled
+    this.onDidChange = new vscode.EventEmitter()
+  }
+
+  get enabled() {
+    return this._enabled
+  }
+
+  set enabled(value: boolean) {
+    this._enabled = value
+    this.onDidChange.fire()
+  }
+
+  get onDidChangeCodeLenses(): vscode.Event<void> {
+    return this.onDidChange.event
+  }
+
+  provideCodeLenses(document: vscode.TextDocument, _: vscode.CancellationToken): vscode.CodeLens[] {
+    const result = []
+
+    if (!this._enabled || document.isUntitled) {
+      return result
+    }
+
+    const filePath = document.fileName
+    const testResults = this.testResultProvider.getResults(filePath)
+    const fileName = basename(document.fileName)
+    for (const test of testResults) {
+      if (test.status === TestReconciliationState.KnownSuccess) {
+        continue
+      }
+
+      const start = new vscode.Position(test.start.line, test.start.column)
+      const end = new vscode.Position(test.end.line, test.start.column + 5)
+      const range = new vscode.Range(start, end)
+      result.push(new DebugCodeLens(range, fileName, test.name))
+    }
+
+    return result
+  }
+
+  resolveCodeLens(codeLens: vscode.CodeLens, _: vscode.CancellationToken): vscode.ProviderResult<vscode.CodeLens> {
+    if (codeLens instanceof DebugCodeLens) {
+      codeLens.command = {
+        arguments: [codeLens.fileName, codeLens.testName],
+        command: `${extensionName}.run-test`,
+        title: 'Debug',
+      }
+    }
+
+    return codeLens
+  }
+
+  didChange() {
+    this.onDidChange.fire()
+  }
+}

src/DebugCodeLens/DebugCodeLensProvider.ts

@@ -0,0 +1 @@
+export { DebugCodeLensProvider } from './DebugCodeLensProvider'