feat(core): storage creator

antfu · antfu · commit 6b35809e5e4c · 2025-12-16T14:28:14.000+09:00
diff --git a/packages/core/src/node/context.ts b/packages/core/src/node/context.ts
@@ -2,6 +2,7 @@ import type { DevToolsNodeContext } from '@vitejs/devtools-kit'
 import type { ResolvedConfig, ViteDevServer } from 'vite'
 import Debug from 'debug'
 import { debounce } from 'perfect-debounce'
+import { searchForWorkspaceRoot } from 'vite'
 import { ContextUtils } from './context-utils'
 import { DevToolsDockHost } from './host-docks'
 import { RpcFunctionsHost } from './host-functions'
@@ -19,6 +20,7 @@ export async function createDevToolsContext(
 
   const context: DevToolsNodeContext = {
     cwd,
+    workspaceRoot: searchForWorkspaceRoot(cwd) ?? cwd,
     viteConfig,
     viteServer,
     mode: viteConfig.command === 'serve' ? 'dev' : 'build',
diff --git a/packages/core/src/node/storage.ts b/packages/core/src/node/storage.ts
@@ -0,0 +1,28 @@
+import fs from 'node:fs'
+import { createSharedState } from '@vitejs/devtools-kit/utils/shared-state'
+
+export interface CreateStorageOptions<T extends object> {
+  filepath: string
+  initialValue: T
+}
+
+export async function createStorage<T extends object>(options: CreateStorageOptions<T>) {
+  let initialState: T
+  if (fs.existsSync(options.filepath)) {
+    initialState = JSON.parse(await fs.readFileSync(options.filepath, 'utf-8')) as T
+  }
+  else {
+    initialState = options.initialValue
+  }
+
+  const state = createSharedState<T>({
+    initialState,
+    enablePatches: false,
+  })
+
+  state.on('updated', (newState) => {
+    fs.writeFileSync(options.filepath, `${JSON.stringify(newState, null, 2)}\n`)
+  })
+
+  return state
+}
diff --git a/packages/kit/src/types/vite-plugin.ts b/packages/kit/src/types/vite-plugin.ts
@@ -19,6 +19,10 @@ export interface DevToolsPluginOptions {
 }
 
 export interface DevToolsNodeContext {
+  /**
+   * Workspace root directory of Vite DevTools
+   */
+  readonly workspaceRoot: string
   /**
    * Current working directory of Vite DevTools
    */
