fix: ensure StreamContext is a singltone

brillout · Dec 7, 2023 · b2172d7 · b2172d7
1 parent 6272686
commit b2172d7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/server/useStream.ts b/src/server/useStream.ts
@@ -3,12 +3,16 @@ export { StreamProvider }
 export type { StreamUtils }
 
 import React, { useContext } from 'react'
-import { assert } from './utils'
+import { assert, getGlobalObject } from './utils'
 
 type StreamUtils = {
   injectToStream: (htmlChunk: string) => void
 }
-const StreamContext = React.createContext<StreamUtils | null>(null)
+
+const globalObject = getGlobalObject('useStream.ts', {
+  StreamContext: React.createContext<StreamUtils | null>(null)
+})
+const { StreamContext } = globalObject
 const StreamProvider = StreamContext.Provider
 
 function useStream(): StreamUtils | null {

diff --git a/src/server/utils.ts b/src/server/utils.ts
@@ -2,3 +2,4 @@ export * from '../utils/assert'
 export * from '../utils/debug'
 export * from '../utils/isPromise'
 export * from '../utils/isBrowser'
+export * from '../utils/getGlobalObject'