fix(vscode): selection style with media (#1863)

Co-authored-by: cuiluc <cuilulu4043@126.com> Co-authored-by: Anthony Fu <anthonyfu117@hotmail.com>
unocss · Nov 9, 2022 · dbee1fd · dbee1fd
1 parent 9da073a
commit dbee1fd
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/packages/vscode/src/selectionStyle.ts b/packages/vscode/src/selectionStyle.ts
@@ -35,7 +35,7 @@ export async function registerSelectionStyle(cwd: string, contextLoader: Context
         code = `${code} >`
       const ctx = await contextLoader.resolveContext(code, id) || (await contextLoader.resolveClosestContext(code, id))
       const result = await getMatchedPositionsFromCode(ctx.uno, code)
-      if (!result.length)
+      if (result.length <= 1)
         return reset()
 
       const uniqMap = new Map<string, string>()
@@ -47,21 +47,23 @@ export async function registerSelectionStyle(cwd: string, contextLoader: Context
       await Promise.all(Array.from(uniqMap.values())
         .map(async (name) => {
           const tokens = await ctx.uno.parseToken(name, classNamePlaceholder) || []
-          tokens.forEach((token) => {
-            if (token[1] && token[2]) {
-              const key = token[1]
+          tokens.forEach(([, className, cssText, media]) => {
+            if (className && cssText) {
+              let key = className
                 .replace(`.${classNamePlaceholder}`, '&')
                 .replace(regexScopePlaceholder, ' ')
                 .trim()
-              sheetMap.set(key, (sheetMap.get(key) || '') + token[2])
+              if (media)
+                key = `${media}{${key}`
+              sheetMap.set(key, (sheetMap.get(key) || '') + cssText)
             }
           })
         }),
       )
 
       const css = Array.from(sheetMap.keys())
         .sort()
-        .map(key => `${key}{${sheetMap.get(key)}}`)
+        .map(key => `${key}{${sheetMap.get(key)}}${key.includes('{') ? '}' : ''}`)
         .join('\n')
 
       const prettified = prettier.format(css, {