feat: migrate document link to language-service

extensions/vscode/src/index.ts

@@ -9,7 +9,6 @@ import { addToIgnore } from './commands/add-to-ignore'
 import { openFileInNpmx } from './commands/open-file-in-npmx'
 import { openInBrowser } from './commands/open-in-browser'
 import { useDecorators } from './providers/decorators'
-import { useDocumentLink } from './providers/document-link'
 import { logger } from './state'
 
 export const { activate, deactivate } = defineExtension((ctx) => {
@@ -22,7 +21,6 @@ export const { activate, deactivate } = defineExtension((ctx) => {
   useWorkspaceContext()
 
   useDecorators()
-  useDocumentLink()
 
   useCommand(ADD_TO_IGNORE_COMMAND, addToIgnore)
 

packages/language-service/src/index.ts

@@ -2,13 +2,15 @@ import type { LanguageServicePlugin } from '@volar/language-service'
 import type { IWorkspaceState } from './types'
 import { create as createNpmxCatalogService } from './plugins/catalog'
 import { create as createNpmxDiagnosticsService } from './plugins/diagnostics'
+import { create as createNpmxDocumentLinkService } from './plugins/document-link'
 import { create as createNpmxHoverService } from './plugins/hover'
 import { create as createNpmxVersionCompletionService } from './plugins/version-completion'
 
 export function createNpmxLanguageServicePlugins(workspace: IWorkspaceState): LanguageServicePlugin[] {
   return [
     createNpmxCatalogService(workspace),
     createNpmxDiagnosticsService(workspace),
+    createNpmxDocumentLinkService(workspace),
     createNpmxHoverService(workspace),
     createNpmxVersionCompletionService(workspace),
   ]

packages/language-service/src/plugins/document-link.ts

@@ -0,0 +1,65 @@
+import type { DocumentLink, LanguageServicePlugin, LanguageServicePluginInstance } from '@volar/language-service'
+import type { IWorkspaceState } from '../types'
+import { npmxPackageUrl } from 'npmx-language-core/links'
+import { isDependencyFile } from 'npmx-language-core/utils'
+import { URI } from 'vscode-uri'
+import { getConfig } from '../config'
+
+export function create(workspaceState: IWorkspaceState): LanguageServicePlugin {
+  return {
+    name: 'npmx-document-link',
+    capabilities: {
+      documentLinkProvider: {},
+    },
+    create(context): LanguageServicePluginInstance {
+      return {
+        async provideDocumentLinks(document): Promise<DocumentLink[] | undefined> {
+          const linkMode = await getConfig(context, 'npmx.packageLinks')
+          if (linkMode === 'off')
+            return
+
+          const uri = URI.parse(document.uri)
+          if (uri.scheme !== 'file' || !isDependencyFile(uri.path))
+            return
+
+          const dependencies = await workspaceState.getResolvedDependencies(document.uri)
+          if (!dependencies)
+            return
+
+          const links: DocumentLink[] = []
+
+          for (const dep of dependencies) {
+            if (dep.resolvedProtocol !== 'npm')
+              continue
+
+            const { resolvedName, resolvedSpec, nameRange } = dep
+
+            let targetVersion: string | undefined
+
+            if (linkMode === 'declared') {
+              targetVersion = resolvedSpec
+            } else if (linkMode === 'resolved') {
+              targetVersion = await dep.resolvedVersion() ?? resolvedSpec
+            }
+
+            const url = targetVersion
+              ? npmxPackageUrl(resolvedName, targetVersion)
+              : npmxPackageUrl(resolvedName)
+
+            const [start, end] = nameRange
+            links.push({
+              range: {
+                start: document.positionAt(start),
+                end: document.positionAt(end),
+              },
+              target: url,
+              tooltip: `Open ${resolvedName}@${targetVersion ?? 'latest'} on npmx`,
+            })
+          }
+
+          return links
+        },
+      }
+    },
+  }
+}