support multiple project

Author: mizdra

packages/ts-plugin/src/index.ts

@@ -68,33 +68,54 @@ 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 },
-            });
+      try {
+        session.addProtocolHandler('_css-modules-kit:rename', (request) => {
+          const project = info.project.projectService.getDefaultProjectForFile(
+            ts.server.toNormalizedPath(request.arguments.fileName),
+            true,
+          );
+          if (!project) {
+            return { response: { result: [] } };
           }
-        }
-        return { response: { result: links } };
-      });
+          const languageService = project.getLanguageService();
+          const { fileName, position } = request.arguments;
+          const result = languageService.findRenameLocations(fileName, position, false, false, {});
+          return { response: { result } };
+        });
+        session.addProtocolHandler('_css-modules-kit:renameInfo', (request) => {
+          const project = info.project.projectService.getDefaultProjectForFile(
+            ts.server.toNormalizedPath(request.arguments.fileName),
+            true,
+          );
+          if (!project) {
+            return { response: { result: { canRename: false, localizedErrorMessage: '' } as const } };
+          }
+          const languageService = project.getLanguageService();
+          const { fileName, position } = request.arguments;
+          const result = languageService.getRenameInfo(fileName, position, {});
+          return { response: { result } };
+        });
+        session.addProtocolHandler('_css-modules-kit:documentLink', (request) => {
+          // TODO: Get `language` instance related to `request.arguments.fileName`
+          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 } };
+        });
+      } catch {
+        console.log('already registered');
+      }
     },
   };
 });