Hide CSS language completions inside import source, theme, and prefix fns

Author: thecrypticace

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

@@ -13,7 +13,7 @@ import {
   CompletionItemKind,
   Connection,
 } from 'vscode-languageserver/node'
-import { TextDocument } from 'vscode-languageserver-textdocument'
+import { Position, TextDocument } from 'vscode-languageserver-textdocument'
 import { Utils, URI } from 'vscode-uri'
 import { getLanguageModelCache } from './languageModelCache'
 import { Stylesheet } from 'vscode-css-languageservice'
@@ -137,8 +137,58 @@ export class CssServer {
       })
     }
 
+    function isInImportDirective(doc: TextDocument, pos: Position) {
+      let text = doc.getText({
+        start: { line: pos.line, character: 0 },
+        end: pos,
+      })
+
+      // Scan backwards to see if we're inside an `@import` directive
+      let foundImport = false
+      let foundDirective = false
+
+      for (let i = text.length - 1; i >= 0; i--) {
+        let char = text[i]
+        if (char === '\n') break
+
+        if (char === '(' && !foundDirective) {
+          if (text.startsWith(' source(', i - 7)) {
+            foundDirective = true
+          }
+
+          //
+          else if (text.startsWith(' theme(', i - 6)) {
+            foundDirective = true
+          }
+
+          //
+          else if (text.startsWith(' prefix(', i - 7)) {
+            foundDirective = true
+          }
+        }
+
+        //
+        else if (char === '@' && !foundImport) {
+          if (text.startsWith('@import ', i)) {
+            foundImport = true
+          }
+        }
+      }
+
+      return foundImport && foundDirective
+    }
+
     connection.onCompletion(async ({ textDocument, position }, _token) =>
       withDocumentAndSettings(textDocument.uri, async ({ original, document, settings }) => {
+        // If we're inside source(…), prefix(…), or theme(…), don't show
+        // completions from the CSS language server
+        if (isInImportDirective(original, position)) {
+          return {
+            isIncomplete: false,
+            items: [],
+          }
+        }
+
         let result = await cssLanguageService.doComplete2(
           document,
           position,

packages/tailwindcss-language-server/tests/css/css-server.test.ts

@@ -689,3 +689,49 @@ defineTest({
     expect(await doc.diagnostics()).toEqual([])
   },
 })
+
+defineTest({
+  name: 'completions are hidden inside @import source(…)/theme(…)/prefix(…) functions',
+  prepare: async ({ root }) => ({
+    client: await createClient({
+      server: 'css',
+      root,
+    }),
+  }),
+  handle: async ({ client }) => {
+    let doc = await client.open({
+      lang: 'tailwindcss',
+      name: 'file-1.css',
+      text: css`
+        @import './file.css' source(none);
+        @import './file.css' theme(inline);
+        @import './file.css' prefix(tw);
+        @import './file.css' source(none) theme(inline) prefix(tw);
+      `,
+    })
+
+    // @import './file.css' source(none)
+    //                             ^
+    // @import './file.css' theme(inline);
+    //                            ^
+    // @import './file.css' prefix(tw);
+    //                             ^
+    let completionsA = await doc.completions({ line: 0, character: 29 })
+    let completionsB = await doc.completions({ line: 1, character: 28 })
+    let completionsC = await doc.completions({ line: 2, character: 29 })
+
+    expect(completionsA).toEqual({ isIncomplete: false, items: [] })
+    expect(completionsB).toEqual({ isIncomplete: false, items: [] })
+    expect(completionsC).toEqual({ isIncomplete: false, items: [] })
+
+    // @import './file.css' source(none) theme(inline) prefix(tw);
+    //                             ^           ^              ^
+    let completionsD = await doc.completions({ line: 3, character: 29 })
+    let completionsE = await doc.completions({ line: 3, character: 41 })
+    let completionsF = await doc.completions({ line: 3, character: 56 })
+
+    expect(completionsD).toEqual({ isIncomplete: false, items: [] })
+    expect(completionsE).toEqual({ isIncomplete: false, items: [] })
+    expect(completionsF).toEqual({ isIncomplete: false, items: [] })
+  },
+})