showCovered/showUncovered

package.json

@@ -446,9 +446,10 @@
           "description": "Run 'go test' on save for current package. It is not advised to set this to `true` when you have Auto Save enabled."
         },
         "go.coverOnSave": {
-          "type": "boolean",
+          "type": ["boolean", "string"],
+          "enum": [true, false, "showCoveredOnly", "showUncoveredOnly"],
           "default": false,
-          "description": "Run 'go test -coverprofile' on save"
+          "description": "If not false, runs 'go test -coverprofile' on save and shows test coverage. If set to showCoveredOnly, then only covered code is highlighted. If set to showUncoveredOnly, then only uncovered code is highlighted."
         },
         "go.testTimeout": {
           "type": "string",

src/goCover.ts

@@ -106,8 +106,12 @@ function applyCoverage(remove: boolean = false) {
 }
 
 function highlightCoverage(editor: vscode.TextEditor, file: CoverageFile, remove: boolean) {
-	editor.setDecorations(uncoveredHighLight, remove ? [] : file.uncoveredRange);
-	editor.setDecorations(coveredHighLight, remove ? [] : file.coveredRange);
+	let cfg = vscode.workspace.getConfiguration('go');
+	let coverOnSave = cfg.get('coverOnSave', false);
+	let hideUncovered = remove || !coverOnSave || coverOnSave === 'showCoveredOnly';
+	let hideCovered = remove || !coverOnSave || coverOnSave === 'showUncoveredOnly';
+	editor.setDecorations(uncoveredHighLight, hideUncovered ? [] : file.uncoveredRange);
+	editor.setDecorations(coveredHighLight, hideCovered ? [] : file.coveredRange);
 }
 
 export function getCoverage(coverProfilePath: string, showErrOutput: boolean = false): Promise<any[]> {