chore: use state

Author: yuyinws

global.d.ts

@@ -1,9 +1,10 @@
 import { defineEventHandler, getQuery } from 'h3'
+import { peersState } from '../utils/state'
 
 export default defineEventHandler(async (event) => {
   const { m, t } = getQuery(event) as { m: string, t: string }
 
-  const peers = globalThis.UNPLUGIN_TURBO_CONSOLE_PEERS_SET
+  const peers = peersState()
 
   if (peers) {
     for (const peer of peers) {

src/core/server/send.ts

@@ -1,14 +1,14 @@
 /* eslint-disable no-console */
 import { defineWebSocketHandler } from 'h3'
 import { PLUGIN_NAME } from '../constants'
-
-if (globalThis.UNPLUGIN_TURBO_CONSOLE_PEERS_SET === undefined) {
-  globalThis.UNPLUGIN_TURBO_CONSOLE_PEERS_SET = new Set<any>()
-}
+import { peersState } from '../utils/state'
 
 export default defineWebSocketHandler({
   open(peer) {
-    globalThis.UNPLUGIN_TURBO_CONSOLE_PEERS_SET!.add(peer)
+    const peers = peersState()
+    peers.add(peer)
+
+    peersState(peers)
   },
   message(_peer, message) {
     try {

src/core/server/ws.ts

@@ -1,3 +1,4 @@
+import type { Peer } from 'crossws'
 import type { ExpressionMeta, ExpressionsMap } from '../../types'
 import { signal } from 'alien-signals'
 import { v4 as uuidv4 } from 'uuid'
@@ -25,3 +26,6 @@ export function addExpression(relativePath: string, expressionMeta: ExpressionMe
     expressionsMapState(newMap)
   }
 }
+
+export const serverState = signal(false)
+export const peersState = signal(new Set<Peer>())

src/core/utils/state.ts

@@ -10,7 +10,7 @@ import { resolveOptions } from './core/options'
 import { createServer } from './core/server/index'
 import { transform } from './core/transform/index'
 import { loadPkg, printInfo } from './core/utils'
-import { expressionsMapState } from './core/utils/state'
+import { expressionsMapState, serverState } from './core/utils/state'
 import { initVirtualModulesGenerator, themeDetectVirtualModule } from './core/utils/virtualModules'
 
 export const unpluginFactory: UnpluginFactory<Options | undefined> = (rawOptions = {}) => {
@@ -24,8 +24,8 @@ export const unpluginFactory: UnpluginFactory<Options | undefined> = (rawOptions
 
   async function startTurboConsoleServer() {
     // Avoid start server multiple times
-    if (!globalThis.UNPLUGIN_TURBO_CONSOLE_LAUNCH_SERVER) {
-      globalThis.UNPLUGIN_TURBO_CONSOLE_LAUNCH_SERVER = true
+    if (!serverState()) {
+      serverState(true)
       await detectPort()
       printInfo(options)
       createServer(options)
@@ -84,8 +84,8 @@ export const unpluginFactory: UnpluginFactory<Options | undefined> = (rawOptions
     vite: {
       async configureServer(server) {
         // Avoid start server multiple times
-        if (!globalThis.UNPLUGIN_TURBO_CONSOLE_LAUNCH_SERVER) {
-          globalThis.UNPLUGIN_TURBO_CONSOLE_LAUNCH_SERVER = true
+        if (!serverState()) {
+          serverState(true)
           await detectPort()
           createServer(options)