feat(vscode): If uno.root is not set, auto-detection

packages/vscode/package.json

@@ -111,6 +111,7 @@
     "jiti": "^1.19.3",
     "prettier": "^2.8.8",
     "tsup": "^7.2.0",
-    "unconfig": "^0.3.10"
+    "unconfig": "^0.3.10",
+    "find-up": "^6.3.0"
   }
 }

packages/vscode/src/index.ts

@@ -1,6 +1,7 @@
 import path from 'path'
 import type { ExtensionContext, StatusBarItem } from 'vscode'
 import { StatusBarAlignment, commands, window, workspace } from 'vscode'
+import { findUp } from 'find-up'
 import { version } from '../package.json'
 import { log } from './log'
 import { registerAnnotations } from './annotation'
@@ -11,7 +12,6 @@ import { isFulfilled, isRejected } from './utils'
 
 async function registerRoot(ext: ExtensionContext, status: StatusBarItem, cwd: string) {
   const contextLoader = new ContextLoader(cwd)
-
   await contextLoader.ready
 
   const hasConfig = await contextLoader.loadContextInDirectory(cwd)
@@ -47,7 +47,6 @@ export async function activate(ext: ExtensionContext) {
 
   if (Array.isArray(root) && root.length) {
     const cwds = root.map(dir => path.resolve(projectPath, dir))
-
     const contextLoadersResult = await Promise.allSettled(
       cwds.map(cwd => registerRoot(ext, status, cwd)),
     )
@@ -70,29 +69,41 @@ export async function activate(ext: ExtensionContext) {
     }
     return
   }
-
   // now if the root is an array, then it is an empty array
-  const cwd = (root && !Array.isArray(root))
-    ? path.resolve(projectPath, root)
-    : projectPath
-
-  const contextLoader = await registerRoot(ext, status, cwd)
-  ext.subscriptions.push(
-    commands.registerCommand('unocss.reload', async () => {
-      log.appendLine('🔁 Reloading...')
-      if (contextLoader.contextsMap.get(cwd) === null) {
-        contextLoader.contextsMap.delete(cwd)
-        const hasConfig = await contextLoader.loadContextInDirectory(cwd)
-        if (hasConfig) {
-          registerAutoComplete(cwd, contextLoader, ext)
-          registerAnnotations(cwd, contextLoader, status, ext)
-          registerSelectionStyle(cwd, contextLoader)
-        }
-      }
-      contextLoader.reload()
-      log.appendLine('✅ Reloaded.')
-    }),
-  )
+
+  const cacheMap = new Set()
+
+  const registerUnocss = async () => {
+    const url = window.activeTextEditor?.document.uri.fsPath
+    if (!url)
+      return
+    if (/node_modules/.test(url))
+      return
+    const target = await findUp(['package.json'], { cwd: url })
+    if (!target)
+      return
+    const cwd = target.slice(0, -13)
+    if (cacheMap.has(cwd))
+      return
+    cacheMap.add(cwd)
+    const contextLoader = await registerRoot(ext, status, cwd)
+
+    ext.subscriptions.push(
+      commands.registerCommand('unocss.reload', async () => {
+        log.appendLine('🔁 Reloading...')
+        await contextLoader.reload()
+        log.appendLine('✅ Reloaded.')
+      }),
+    )
+  }
+
+  try {
+    await registerUnocss()
+    ext.subscriptions.push(window.onDidChangeActiveTextEditor(registerUnocss))
+  }
+  catch (e: any) {
+    log.appendLine(String(e.stack ?? e))
+  }
 }
 
-export function deactivate() {}
+export function deactivate() { }