fix: handle init documents

Author: hyoban

packages/tailwindcss-language-server/src/tw.ts

@@ -629,7 +629,10 @@ export class TW {
         this.getProject(document)
           ?.provideAnnotations(document)
           .then((annotations) => {
-            this.connection.sendRequest('@/tailwindCSS/annotations', annotations)
+            this.connection.sendRequest('@/tailwindCSS/annotations', {
+              uri: document.uri,
+              annotations,
+            })
           })
       }),
     )
@@ -656,11 +659,25 @@ export class TW {
         this.getProject(document)
           ?.provideAnnotations(document)
           .then((annotations) => {
-            this.connection.sendRequest('@/tailwindCSS/annotations', annotations)
+            this.connection.sendRequest('@/tailwindCSS/annotations', {
+              uri: document.uri,
+              annotations,
+            })
           })
       }),
     )
 
+    this.documentService.getAllDocuments().forEach((document) => {
+      this.getProject(document)
+        ?.provideAnnotations(document)
+        .then((annotations) => {
+          this.connection.sendRequest('@/tailwindCSS/annotations', {
+            uri: document.uri,
+            annotations,
+          })
+        })
+    })
+
     if (this.initializeParams.capabilities.workspace.workspaceFolders) {
       this.disposables.push(
         this.connection.workspace.onDidChangeWorkspaceFolders(async (evt) => {

packages/vscode-tailwindcss/src/extension.ts

@@ -543,8 +543,10 @@ export async function activate(context: ExtensionContext) {
     client.onNotification('@/tailwindCSS/projectInitialized', updateActiveTextEditorContext)
     client.onNotification('@/tailwindCSS/projectReset', updateActiveTextEditorContext)
     client.onNotification('@/tailwindCSS/projectsDestroyed', resetActiveTextEditorContext)
-    client.onRequest('@/tailwindCSS/annotations', (ranges) => {
-      Window.activeTextEditor.setDecorations(underlineDecorationType, ranges)
+    client.onRequest('@/tailwindCSS/annotations', ({ uri, annotations }) => {
+      Window.visibleTextEditors
+        .find((editor) => editor.document.uri.toString() === uri)
+        ?.setDecorations(underlineDecorationType, annotations)
     })
     client.onRequest('@/tailwindCSS/getDocumentSymbols', showSymbols)