feat: add method to get locations with provider name

add TextDocumentLocationProviderRegistry.getLocationsWithProviderName method

Author: beyang

src/client/features/hover.ts

@@ -42,7 +42,10 @@ export class TextDocumentHoverFeature extends Feature<TextDocumentRegistrationOp
         }
         this.register(this.messages, {
             id: uuidv4(),
-            registerOptions: { documentSelector },
+            registerOptions: {
+                documentSelector,
+                providerName: this.client.name,
+            },
         })
     }
 

src/client/features/location.ts

@@ -58,7 +58,10 @@ export abstract class TextDocumentLocationFeature<
         }
         this.register(this.messages, {
             id: uuidv4(),
-            registerOptions: { documentSelector },
+            registerOptions: {
+                documentSelector,
+                providerName: this.client.name,
+            },
         })
     }
 

src/client/features/textDocument.ts

@@ -124,7 +124,10 @@ export class TextDocumentDidOpenFeature extends TextDocumentNotificationFeature<
         if (documentSelector && textDocumentSyncOptions && textDocumentSyncOptions.openClose) {
             this.register(this.messages, {
                 id: uuidv4(),
-                registerOptions: { documentSelector },
+                registerOptions: {
+                    documentSelector,
+                    providerName: this.client.name,
+                },
             })
         }
     }
@@ -171,7 +174,10 @@ export class TextDocumentDidCloseFeature extends TextDocumentNotificationFeature
         if (documentSelector && textDocumentSyncOptions && textDocumentSyncOptions.openClose) {
             this.register(this.messages, {
                 id: uuidv4(),
-                registerOptions: { documentSelector },
+                registerOptions: {
+                    documentSelector,
+                    providerName: this.client.name,
+                },
             })
         }
     }

src/environment/providers/location.ts

@@ -3,7 +3,7 @@ import { map, switchMap } from 'rxjs/operators'
 import { Location } from 'vscode-languageserver-types'
 import { ReferenceParams, TextDocumentPositionParams, TextDocumentRegistrationOptions } from '../../protocol'
 import { FeatureProviderRegistry } from './registry'
-import { flattenAndCompact } from './util'
+import { flattenAndCompact, flattenAndCompactNonNull } from './util'
 
 /**
  * Function signature for retrieving related locations given a location (e.g., definition, implementation, and type
@@ -22,6 +22,23 @@ export class TextDocumentLocationProviderRegistry<
     public getLocation(params: P): Observable<L | L[] | null> {
         return getLocation<P, L>(this.providers, params)
     }
+
+    public getLocationsWithProviderName(params: P): Observable<{ providerName: string; location: L }[] | null> {
+        return getLocationsWithProviderName<P, L>(this.providersWithName, params)
+    }
+
+    public readonly providersWithName: Observable<
+        { name: string; provider: ProvideTextDocumentLocationSignature<P, L> }[]
+    > = this.entries.pipe(
+        map(providers =>
+            providers.map(({ provider, registrationOptions }) => {
+                return {
+                    name: registrationOptions.providerName,
+                    provider,
+                }
+            })
+        )
+    )
 }
 
 /**
@@ -62,6 +79,35 @@ export function getLocations<
     )
 }
 
+export function getLocationsWithProviderName<
+    P extends TextDocumentPositionParams = TextDocumentPositionParams,
+    L extends Location = Location
+>(
+    providersWithName: Observable<{ name: string; provider: ProvideTextDocumentLocationSignature<P, L> }[]>,
+    params: P
+): Observable<{ providerName: string; location: L }[]> {
+    return providersWithName.pipe(
+        switchMap(async providersWithName => {
+            const resultsByProvider = await Promise.all(
+                providersWithName.map(async ({ provider, name }) => {
+                    const providerRes = await provider(params).toPromise()
+                    return flattenAndCompactNonNull([providerRes]).map(loc => ({
+                        providerName: name,
+                        location: loc,
+                    }))
+                })
+            )
+            const flattenedResults: { providerName: string; location: L }[] = []
+            for (const providerResults of resultsByProvider) {
+                for (const res of providerResults) {
+                    flattenedResults.push(res)
+                }
+            }
+            return flattenedResults
+        })
+    )
+}
+
 /**
  * Provides reference results from all extensions.
  *

src/environment/providers/registry.ts

@@ -9,7 +9,7 @@ interface Entry<O, P> {
 
 /** Base class for provider registries for features. */
 export abstract class FeatureProviderRegistry<O, P> {
-    private entries = new BehaviorSubject<Entry<O, P>[]>([])
+    protected entries = new BehaviorSubject<Entry<O, P>[]>([])
 
     public constructor(initialEntries?: Entry<O, P>[]) {
         if (initialEntries) {

src/environment/providers/util.ts

@@ -8,3 +8,8 @@ export function flattenAndCompact<T>(value: (T | T[] | null)[] | null): T[] | nu
     const merged = flatten(compact(value))
     return merged.length === 0 ? null : merged
 }
+
+/** Flattens and compacts the argument. If it is null or if the result is empty, it returns null. */
+export function flattenAndCompactNonNull<T>(value: (T | T[] | null)[] | null): T[] {
+    return value ? flatten(compact(value)) : []
+}

src/protocol/textDocument.ts

@@ -358,6 +358,11 @@ export interface TextDocumentRegistrationOptions {
      * the document selector provided on the client side will be used.
      */
     documentSelector: DocumentSelector | null
+
+    /**
+     * User-visible name of the of the provider being registered.
+     */
+    providerName: string
 }
 
 /**