feat: show file coverage counts

extension/package.json

@@ -75,7 +75,7 @@
     "@types/node-fetch": "2.6.9",
     "@types/semver": "7.3.4",
     "@types/sinon": "9.0.11",
-    "@types/vscode": "1.75.0",
+    "@types/vscode": "1.88.0",
     "@vscode/debugadapter-testsupport": "1.58.0",
     "@vscode/test-electron": "2.3.8",
     "@vscode/vsce": "2.23.0",

extension/src/goCover.ts

@@ -184,12 +184,28 @@ function disposeDecorators() {
 	}
 }
 
+export function GetCoverageCounts(): CoverageCountsMap {
+	return coverageCounts;
+}
+
 interface CoverageData {
 	uncoveredOptions: vscode.DecorationOptions[];
 	coveredOptions: vscode.DecorationOptions[];
 }
 
+interface CoverageCounts {
+	Covered: number;
+	Total: number;
+}
+
+interface CoverageCountsMap {
+	[key: string]: CoverageCounts;
+}
+
+// type CoverageCounts = Map<string, {Covered: number, Total: number}>;
+
 let coverageData: { [key: string]: CoverageData } = {}; // actual file path to the coverage data.
+let coverageCounts: CoverageCountsMap = {}; // actual file path to the coverage counts.
 let isCoverageApplied = false;
 
 function emptyCoverageData(): CoverageData {
@@ -219,6 +235,7 @@ export function applyCodeCoverageToAllEditors(coverProfilePath: string, dir?: st
 		try {
 			const showCounts = getGoConfig().get('coverShowCounts') as boolean;
 			const coveragePath = new Map<string, CoverageData>(); // <filename> from the cover profile to the coverage data.
+			const covCounts = new Map<string, CoverageCounts>(); // <filename> from the cover profile to the coverage counts.
 
 			// Clear existing coverage files
 			clearCoverage();
@@ -261,6 +278,8 @@ export function applyCodeCoverageToAllEditors(coverProfilePath: string, dir?: st
 
 				// and fill in coveragePath
 				const coverage = coveragePath.get(parse[1]) || emptyCoverageData();
+				// fill in file stats
+				const fileCounts = covCounts.get(filename) || { Covered: 0, Total: 0 };
 				// When line directive is used this information is artificial and
 				// the source code file can be non-existent or wrong (go.dev/issues/41222).
 				// There is no perfect way to guess whether the line/col in coverage profile
@@ -280,6 +299,7 @@ export function applyCodeCoverageToAllEditors(coverProfilePath: string, dir?: st
 					endCol - 1
 				);
 
+				const statements = parseInt(parse[6], 10);
 				const counts = parseInt(parse[7], 10);
 				// If is Covered (CoverCount > 0)
 				if (counts > 0) {
@@ -288,11 +308,19 @@ export function applyCodeCoverageToAllEditors(coverProfilePath: string, dir?: st
 					coverage.uncoveredOptions.push(...elaborate(range, counts, showCounts));
 				}
 
+
+				if (counts > 0) {
+					fileCounts.Covered += statements;
+				}
+				fileCounts.Total += statements;
+
+				covCounts.set(filename, fileCounts);
 				coveragePath.set(filename, coverage);
 			});
 
 			getImportPathToFolder([...seenPaths], dir).then((pathsToDirs) => {
 				createCoverageData(pathsToDirs, coveragePath);
+				createCoverageCounts(pathsToDirs, covCounts);
 				setDecorators();
 				vscode.window.visibleTextEditors.forEach(applyCodeCoverage);
 				resolve();
@@ -349,6 +377,26 @@ function createCoverageData(pathsToDirs: Map<string, string>, coveragePath: Map<
 	});
 }
 
+function createCoverageCounts(pathsToDirs: Map<string, string>, counts: Map<string, CoverageCounts>) {
+	counts.forEach((cc, ip) => {
+		if (path.isAbsolute(ip)) {
+			coverageCounts[ip] = cc;
+			return;
+		}
+
+		const lastSlash = ip.lastIndexOf('/');
+		if (lastSlash === -1) {
+			coverageCounts[ip] = cc;
+			return;
+		}
+
+		const maybePkgPath = ip.slice(0, lastSlash);
+		const fileDir = pathsToDirs.get(maybePkgPath) || path.resolve(maybePkgPath);
+		const file = fileDir + path.sep + ip.slice(lastSlash + 1);
+		coverageCounts[file] = cc;
+	});
+}
+
 /**
  * Set the object that holds the coverage data for given file path.
  * @param filePath

extension/src/goTest/run.ts

@@ -29,6 +29,7 @@ import { debugTestAtCursor } from '../goTest';
 import { GoExtensionContext } from '../context';
 import path = require('path');
 import { escapeRegExp } from '../subTestUtils';
+import { GetCoverageCounts } from '../goCover';
 
 let debugSessionID = 0;
 
@@ -408,6 +409,10 @@ export class GoTestRunner {
 			if (token?.isCancellationRequested) {
 				break;
 			}
+
+			if (applyCodeCoverage) {
+				this.collectCoverage(run);
+			}
 		}
 
 		run.end();
@@ -742,6 +747,19 @@ export class GoTestRunner {
 		// TODO(firelizzard18): Add more sophisticated check for build failures?
 		return output.some((x) => rePkg.test(x));
 	}
+
+	collectCoverage(run: TestRun) {
+		const coverageData = GetCoverageCounts();
+		for (const filename in coverageData) {
+			const data = coverageData[filename];
+			run.addCoverage(
+				new vscode.FileCoverage(
+					Uri.file(filename),
+					new vscode.TestCoverageCount(data.Covered, data.Total),
+				),
+			);
+		}
+	}
 }
 
 // escapeSubTestName escapes regexp-like metacharacters. Unlike