refactor: move revokeAuthToken to internal context, add @vitejs/devtools/internal export

- Attach revokeAuthToken as a method on DevToolsInternalContext instead of
  a standalone exported function
- Create @vitejs/devtools/internal sub-export for getInternalContext
- Remove getInternalContext and revokeAuthToken from main entry point
- Add alias and tsconfig path for @vitejs/devtools/internal
- Update self-inspect to import from @vitejs/devtools/internal

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: antfu

alias.ts

@@ -20,6 +20,7 @@ export const alias = {
   '@vitejs/devtools-kit': r('kit/src/index.ts'),
   '@vitejs/devtools-rolldown': r('rolldown/src/index.ts'),
   '@vitejs/devtools-self-inspect': r('self-inspect/src/index.ts'),
+  '@vitejs/devtools/internal': r('core/src/internal.ts'),
   '@vitejs/devtools/client/inject': r('core/src/client/inject/index.ts'),
   '@vitejs/devtools/client/webcomponents': r('core/src/client/webcomponents/index.ts'),
   '@vitejs/devtools': r('core/src/index.ts'),

packages/core/package.json

@@ -27,6 +27,7 @@
     "./client/webcomponents": "./dist/client/webcomponents.js",
     "./config": "./dist/config.js",
     "./dirs": "./dist/dirs.js",
+    "./internal": "./dist/internal.js",
     "./package.json": "./package.json"
   },
   "types": "./dist/index.d.ts",

packages/core/src/index.ts

@@ -1,6 +1,4 @@
-export { revokeAuthToken } from './node/auth-revoke'
 export { createDevToolsContext } from './node/context'
-export { getInternalContext } from './node/context-internal'
 export type { DevToolsInternalContext, InternalAnonymousAuthStorage } from './node/context-internal'
 export { DevTools } from './node/plugins'
 export { createDevToolsMiddleware } from './node/server'

packages/core/src/internal.ts

@@ -0,0 +1,2 @@
+export { getInternalContext } from './node/context-internal'
+export type { DevToolsInternalContext, InternalAnonymousAuthStorage } from './node/context-internal'

packages/core/src/node/auth-revoke.ts

@@ -1,15 +1,17 @@
 import type { DevToolsNodeContext } from '@vitejs/devtools-kit'
+import type { SharedState } from '@vitejs/devtools-kit/utils/shared-state'
+import type { InternalAnonymousAuthStorage } from './context-internal'
 import type { RpcFunctionsHost } from './host-functions'
-import { getInternalContext } from './context-internal'
 
 /**
  * Revoke an auth token: remove from storage and notify all connected clients
  * using this token that they are no longer trusted.
  */
-export async function revokeAuthToken(context: DevToolsNodeContext, token: string): Promise<void> {
-  const internal = getInternalContext(context)
-  const storage = internal.storage.auth
-
+export async function revokeAuthToken(
+  context: DevToolsNodeContext,
+  storage: SharedState<InternalAnonymousAuthStorage>,
+  token: string,
+): Promise<void> {
   // Remove from persistent storage
   storage.mutate((state) => {
     delete state.trusted[token]

packages/core/src/node/context-internal.ts

@@ -2,6 +2,7 @@ import type { DevToolsNodeContext } from '@vitejs/devtools-kit'
 import type { SharedState } from '@vitejs/devtools-kit/utils/shared-state'
 import { homedir } from 'node:os'
 import { join } from 'pathe'
+import { revokeAuthToken } from './auth-revoke'
 import { createStorage } from './storage'
 
 export interface InternalAnonymousAuthStorage {
@@ -17,21 +18,28 @@ export interface DevToolsInternalContext {
   storage: {
     auth: SharedState<InternalAnonymousAuthStorage>
   }
+  /**
+   * Revoke an auth token: remove from storage and notify all connected clients
+   * using this token that they are no longer trusted.
+   */
+  revokeAuthToken: (token: string) => Promise<void>
 }
 
 export const internalContextMap = new WeakMap<DevToolsNodeContext, DevToolsInternalContext>()
 
 export function getInternalContext(context: DevToolsNodeContext): DevToolsInternalContext {
   if (!internalContextMap.has(context)) {
+    const storage = createStorage<InternalAnonymousAuthStorage>({
+      filepath: join(homedir(), '.vite/devtools/auth.json'),
+      initialValue: {
+        trusted: {},
+      },
+    })
     const internalContext: DevToolsInternalContext = {
       storage: {
-        auth: createStorage<InternalAnonymousAuthStorage>({
-          filepath: join(homedir(), '.vite/devtools/auth.json'),
-          initialValue: {
-            trusted: {},
-          },
-        }),
+        auth: storage,
       },
+      revokeAuthToken: (token: string) => revokeAuthToken(context, storage, token),
     }
     internalContextMap.set(context, internalContext)
   }

packages/core/tsdown.config.ts

@@ -48,6 +48,7 @@ export default defineConfig({
   tsconfig: '../../tsconfig.base.json',
   entry: {
     'index': 'src/index.ts',
+    'internal': 'src/internal.ts',
     'dirs': 'src/dirs.ts',
     'cli': 'src/node/cli.ts',
     'cli-commands': 'src/node/cli-commands.ts',

packages/self-inspect/src/node/rpc/functions/get-auth-tokens.ts

@@ -1,5 +1,5 @@
-import { getInternalContext } from '@vitejs/devtools'
 import { defineRpcFunction } from '@vitejs/devtools-kit'
+import { getInternalContext } from '@vitejs/devtools/internal'
 
 export const getAuthTokens = defineRpcFunction({
   name: 'devtoolskit:self-inspect:get-auth-tokens',

packages/self-inspect/src/node/rpc/functions/revoke-auth-token.ts

@@ -1,13 +1,14 @@
-import { revokeAuthToken } from '@vitejs/devtools'
 import { defineRpcFunction } from '@vitejs/devtools-kit'
+import { getInternalContext } from '@vitejs/devtools/internal'
 
 export const revokeAuthTokenRpc = defineRpcFunction({
   name: 'devtoolskit:self-inspect:revoke-auth-token',
   type: 'action',
   setup: (context) => {
+    const internal = getInternalContext(context)
     return {
       handler: async (authId: string) => {
-        await revokeAuthToken(context, authId)
+        await internal.revokeAuthToken(authId)
       },
     }
   },

test/exports/@vitejs/devtools.yaml

@@ -2,8 +2,6 @@
   createDevToolsContext: function
   createDevToolsMiddleware: function
   DevTools: function
-  getInternalContext: function
-  revokeAuthToken: function
 ./cli-commands:
   build: function
   start: function
@@ -12,3 +10,5 @@
 ./dirs:
   dirClientStandalone: string
   dirDist: string
+./internal:
+  getInternalContext: function