Add support for reference to ref links in md

For #146277

extensions/markdown-language-features/src/languageFeatures/documentLinkProvider.ts

@@ -15,6 +15,7 @@ const localize = nls.loadMessageBundle();
 
 export interface ExternalLinkTarget {
 	readonly kind: 'external';
+
 	readonly uri: vscode.Uri;
 }
 
@@ -30,12 +31,18 @@ export interface InternalLinkTarget {
 export interface ReferenceLinkTarget {
 	readonly kind: 'reference';
 
+	readonly fromResource: vscode.Uri;
+
+	readonly ref: string;
 	readonly position: vscode.Position;
 }
 
 export interface DefinitionLinkTarget {
 	readonly kind: 'definition';
 
+	readonly fromResource: vscode.Uri;
+
+	readonly ref: string;
 	readonly target: ExternalLinkTarget | InternalLinkTarget;
 }
 
@@ -271,23 +278,26 @@ export class MdLinkProvider implements vscode.DocumentLinkProvider {
 					sourceRange: new vscode.Range(linkStart, linkEnd),
 					target: {
 						kind: 'reference',
+						fromResource: document.uri,
+						ref: reference,
 						position: link.linkRange.start
 					}
 				};
-
 			}
 		}
 	}
 
 	public *getDefinitionLinks(document: SkinnyTextDocument): Iterable<LinkData> {
 		const definitions = this.getDefinitions(document);
-		for (const definition of definitions.values()) {
+		for (const [ref, definition] of definitions) {
 			try {
 				const target = parseLink(document, definition.link);
 				if (target) {
 					yield {
 						sourceRange: definition.linkRange,
 						target: {
+							fromResource: document.uri,
+							ref,
 							kind: 'definition',
 							target
 						}

extensions/markdown-language-features/src/languageFeatures/references.ts

@@ -68,16 +68,29 @@ export class MdReferencesProvider extends Disposable implements vscode.Reference
 	}
 
 	private async getReferencesToLink(document: SkinnyTextDocument, position: vscode.Position, context: vscode.ReferenceContext): Promise<vscode.Location[] | undefined> {
-		const links = (await Promise.all(await this._linkCache.getAll())).flat();
+		const links = (await this._linkCache.getAll()).flat();
 
-		const docLinks = await this.linkProvider.getInlineLinks(document);
+		const docLinks = await this.linkProvider.getAllLinks(document);
 		const sourceLink = docLinks.find(link => link.sourceRange.contains(position));
 
+		if (sourceLink?.target.kind === 'reference') {
+			const references: vscode.Location[] = [];
+
+			for (const link of links) {
+				if (link.target.kind === 'reference' || link.target.kind === 'definition') {
+					if (link.target.ref === sourceLink.target.ref && link.target.fromResource.fsPath === document.uri.fsPath) {
+						references.push(new vscode.Location(document.uri, link.sourceRange));
+					}
+				}
+			}
+
+			return references;
+		}
+
 		if (sourceLink?.target.kind !== 'internal') {
 			return undefined;
 		}
 
-
 		let targetDoc = await this.workspaceContents.getMarkdownDocument(sourceLink.target.path);
 		if (!targetDoc) {
 			// We don't think the file exists. If it doesn't already have an extension, try tacking on a `.md` and using that instead