Show @theme in symbol list

Author: thecrypticace

packages/tailwindcss-language-server/src/language/rewriting.test.ts

@@ -65,6 +65,30 @@ test('@utility', () => {
   expect(rewriteCss(input.join('\n'))).toBe(output.join('\n'))
 })
 
+test('@theme', () => {
+  let input = [
+    //
+    '@theme {',
+    '  color: red;',
+    '}',
+    '@theme inline reference static default {',
+    '  color: red;',
+    '}',
+  ]
+
+  let output = [
+    //
+    '.placeholder {', // wrong
+    '  color: red;',
+    '}',
+    '.placeholder                                 {', // wrong
+    '  color: red;',
+    '}',
+  ]
+
+  expect(rewriteCss(input.join('\n'))).toBe(output.join('\n'))
+})
+
 test('@custom-variant', () => {
   let input = [
     //

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

@@ -37,6 +37,7 @@ export function rewriteCss(css: string) {
   css = css.replace(/@variants(\s+[^{]+){/g, replaceWithAtRule())
   css = css.replace(/@responsive(\s*){/g, replaceWithAtRule())
   css = css.replace(/@utility(\s+[^{]+){/g, replaceWithStyleRule())
+  css = css.replace(/@theme(\s+[^{]*){/g, replaceWithStyleRule())
 
   css = css.replace(/@custom-variant(\s+[^;{]+);/g, (match: string) => {
     let spaces = ' '.repeat(match.length - 11)

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

@@ -202,6 +202,46 @@ defineTest({
   },
 })
 
+defineTest({
+  name: '@theme',
+  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 'tailwindcss';
+        @theme {
+          --color-primary: #333;
+        }
+      `,
+    })
+
+    // No errors
+    expect(await doc.diagnostics()).toEqual([])
+
+    // Symbols show up for @custom-variant
+    expect(await doc.symbols()).toMatchObject([
+      {
+        kind: SymbolKind.Class,
+        name: '@theme ',
+        location: {
+          uri: '{workspace:default}/file-1.css',
+          range: {
+            start: { line: 1, character: 0 },
+            end: { line: 3, character: 1 },
+          },
+        },
+      },
+    ])
+  },
+})
+
 defineTest({
   name: '@layer statement',
   prepare: async ({ root }) => ({