wip

Author: mizdra

packages/ts-plugin/src/index.ts

@@ -2,8 +2,9 @@ import type { CMKConfig } from '@css-modules-kit/core';
 import { createMatchesPattern, createResolver, readConfigFile } from '@css-modules-kit/core';
 import { TsConfigFileNotFoundError } from '@css-modules-kit/core';
 import { createLanguageServicePlugin } from '@volar/typescript/lib/quickstart/createLanguageServicePlugin.js';
-import { createCSSLanguagePlugin } from './language-plugin.js';
+import { CMK_DATA_KEY, createCSSLanguagePlugin, isCSSModuleScript } from './language-plugin.js';
 import { proxyLanguageService } from './language-service/proxy.js';
+import type { DocumentLink } from './type.js';
 
 const plugin = createLanguageServicePlugin((ts, info) => {
   if (info.project.projectKind !== ts.server.ProjectKind.Configured) {
@@ -15,6 +16,7 @@ const plugin = createLanguageServicePlugin((ts, info) => {
     info.project.projectService.logger.info('[@css-modules-kit/ts-plugin] info: Session is not available');
     return { languagePlugins: [] };
   }
+  const session = info.session;
 
   let config: CMKConfig;
   try {
@@ -55,17 +57,6 @@ const plugin = createLanguageServicePlugin((ts, info) => {
   const resolver = createResolver(config.compilerOptions, moduleResolutionCache);
   const matchesPattern = createMatchesPattern(config);
 
-  info.session.addProtocolHandler('_css-modules-kit:rename', (request) => {
-    const { fileName, position } = request.arguments;
-    const result = info.languageService.findRenameLocations(fileName, position, false, false, {});
-    return { response: { result } };
-  });
-  info.session.addProtocolHandler('_css-modules-kit:renameInfo', (request) => {
-    const { fileName, position } = request.arguments;
-    const result = info.languageService.getRenameInfo(fileName, position, {});
-    return { response: { result } };
-  });
-
   return {
     languagePlugins: [createCSSLanguagePlugin(resolver, matchesPattern, config)],
     setup: (language) => {
@@ -77,6 +68,33 @@ const plugin = createLanguageServicePlugin((ts, info) => {
         matchesPattern,
         config,
       );
+      session.addProtocolHandler('_css-modules-kit:rename', (request) => {
+        const { fileName, position } = request.arguments;
+        const result = info.languageService.findRenameLocations(fileName, position, false, false, {});
+        return { response: { result } };
+      });
+      session.addProtocolHandler('_css-modules-kit:renameInfo', (request) => {
+        const { fileName, position } = request.arguments;
+        const result = info.languageService.getRenameInfo(fileName, position, {});
+        return { response: { result } };
+      });
+      session.addProtocolHandler('_css-modules-kit:documentLink', (request) => {
+        const { fileName } = request.arguments;
+        const script = language.scripts.get(fileName);
+        const links: DocumentLink[] = [];
+        if (isCSSModuleScript(script)) {
+          const { tokenImporters } = script.generated.root[CMK_DATA_KEY].cssModule;
+          for (const { from, fromLoc } of tokenImporters) {
+            const resolved = resolver(from, { request: fileName });
+            if (!resolved) continue;
+            links.push({
+              fileName: resolved,
+              textSpan: { start: fromLoc.start.offset, length: fromLoc.end.offset - fromLoc.start.offset },
+            });
+          }
+        }
+        return { response: { result: links } };
+      });
     },
   };
 });

packages/ts-plugin/src/type.ts

@@ -30,6 +30,26 @@ export interface CSSModulesKitRenameInfoResponse extends ts.server.protocol.Resp
   readonly body: CSSModulesKitRenameInfoHandlerResponse['response'];
 }
 
+export interface DocumentLink {
+  fileName: string;
+  textSpan: ts.TextSpan;
+}
+export interface CSSModulesKitDocumentLinkRequest extends ts.server.protocol.Request {
+  command: '_css-modules-kit:documentLink';
+  arguments: {
+    fileName: string;
+  };
+}
+export interface CSSModulesKitDocumentLinkHandlerResponse extends ts.server.HandlerResponse {
+  response: {
+    result: DocumentLink[];
+  };
+}
+export interface CSSModulesKitDocumentLinkResponse extends ts.server.protocol.Response {
+  command: '_css-modules-kit:documentLink';
+  readonly body: CSSModulesKitDocumentLinkHandlerResponse['response'];
+}
+
 declare module 'typescript' {
   // eslint-disable-next-line @typescript-eslint/no-namespace
   namespace server {
@@ -42,6 +62,10 @@ declare module 'typescript' {
         command: '_css-modules-kit:renameInfo',
         handler: (request: CSSModulesKitRenameInfoRequest) => CSSModulesKitRenameInfoHandlerResponse,
       ): void;
+      addProtocolHandler(
+        command: '_css-modules-kit:documentLink',
+        handler: (request: CSSModulesKitDocumentLinkRequest) => CSSModulesKitDocumentLinkHandlerResponse,
+      ): void;
     }
   }
 }

packages/vscode/src/index.ts

@@ -1,6 +1,8 @@
 /* eslint-disable no-console */
 
 import type {
+  CSSModulesKitDocumentLinkRequest,
+  CSSModulesKitDocumentLinkResponse,
   CSSModulesKitRenameInfoRequest,
   CSSModulesKitRenameInfoResponse,
   CSSModulesKitRenameRequest,
@@ -63,9 +65,24 @@ export async function activate(context: vscode.ExtensionContext) {
     vscode.languages.registerDocumentLinkProvider(
       { scheme: 'file', language: 'css' },
       {
-        provideDocumentLinks(document, _token) {
-          // TODO
-          return [];
+        async provideDocumentLinks(document, _token) {
+          const res = await vscode.commands.executeCommand<CSSModulesKitDocumentLinkResponse>(
+            'typescript.tsserverRequest',
+            '_css-modules-kit:documentLink',
+            {
+              fileName: document.fileName,
+            } satisfies CSSModulesKitDocumentLinkRequest['arguments'],
+          );
+          if (!res.success || !res.body.result) return [];
+          return res.body.result.map((link) => {
+            return new vscode.DocumentLink(
+              new vscode.Range(
+                document.positionAt(link.textSpan.start),
+                document.positionAt(link.textSpan.start + link.textSpan.length),
+              ),
+              vscode.Uri.file(link.fileName),
+            );
+          });
         },
       },
     ),