Merge pull request #4418 from leo-labs/options-codelens-filteredsymbo…

…lnames Add option to to exclude custom symbols from codelens
dotnet · May 3, 2021 · 76182f2 · 76182f2
2 parents e0b8959 + 256f50d
commit 76182f2
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -663,6 +663,14 @@
           "default": true,
           "description": "Specifies whether the references CodeLens should be shown."
         },
+        "csharp.referencesCodeLens.filteredSymbols": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "default": [],
+          "description": "Array of custom symbol names for which CodeLens should be disabled."
+        },
         "csharp.testsCodeLens.enabled": {
           "type": "boolean",
           "default": true,

diff --git a/src/features/codeLensProvider.ts b/src/features/codeLensProvider.ts
@@ -194,7 +194,7 @@ function createCodeLenses(elements: Structure.CodeElement[], fileName: string, o
 function createCodeLensesForElement(element: Structure.CodeElement, fileName: string, options: Options): vscode.CodeLens[] {
     let results: vscode.CodeLens[] = [];
 
-    if (options.showReferencesCodeLens && isValidElementForReferencesCodeLens(element)) {
+    if (options.showReferencesCodeLens && isValidElementForReferencesCodeLens(element, options)) {
         let range = element.Ranges[SymbolRangeNames.Name];
         if (range) {
             results.push(new ReferencesCodeLens(range, fileName));
@@ -246,7 +246,7 @@ const filteredSymbolNames: { [name: string]: boolean } = {
     'GetEnumerator': true,
 };
 
-function isValidElementForReferencesCodeLens(element: Structure.CodeElement): boolean {
+function isValidElementForReferencesCodeLens(element: Structure.CodeElement, options: Options): boolean {
     if (element.Kind === SymbolKinds.Namespace) {
         return false;
     }
@@ -255,6 +255,10 @@ function isValidElementForReferencesCodeLens(element: Structure.CodeElement): bo
         return false;
     }
 
+    if(options.filteredSymbolsCodeLens.includes(element.Name)) {
+        return false;
+    }
+
     return true;
 }
 

diff --git a/src/omnisharp/options.ts b/src/omnisharp/options.ts
@@ -19,6 +19,7 @@ export class Options {
         public organizeImportsOnFormat: boolean,
         public showReferencesCodeLens: boolean,
         public showTestsCodeLens: boolean,
+        public filteredSymbolsCodeLens: string[],
         public disableCodeActions: boolean,
         public disableMSBuildDiagnosticWarning: boolean,
         public showOmnisharpLogOnError: boolean,
@@ -80,6 +81,7 @@ export class Options {
 
         const showReferencesCodeLens = csharpConfig.get<boolean>('referencesCodeLens.enabled', true);
         const showTestsCodeLens = csharpConfig.get<boolean>('testsCodeLens.enabled', true);
+        const filteredSymbolsCodeLens = csharpConfig.get<string[]>('referencesCodeLens.filteredSymbols', []);
 
         const useSemanticHighlighting = csharpConfig.get<boolean>('semanticHighlighting.enabled', false);
 
@@ -115,6 +117,7 @@ export class Options {
             organizeImportsOnFormat,
             showReferencesCodeLens,
             showTestsCodeLens,
+            filteredSymbolsCodeLens,
             disableCodeActions,
             disableMSBuildDiagnosticWarning,
             showOmnisharpLogOnError,

diff --git a/test/unitTests/Fakes/FakeOptions.ts b/test/unitTests/Fakes/FakeOptions.ts
@@ -6,5 +6,5 @@
 import { Options } from "../../../src/omnisharp/options";
 
 export function getEmptyOptions(): Options {
-    return new Options("", "", false, "", false, 0, 0, false, false, false, false, false, false, false, false, 0, 0, false, false, false, false, false, false, false, undefined, "", "");
+    return new Options("", "", false, "", false, 0, 0, false, false, false, false, false, [], false, false, false, 0, 0, false, false, false, false, false, false, false, undefined, "", "");
 }