fix(one): improve import / code env isolation to avoid server code ge…

…tting into client

packages/one/src/utils/serverContext.tsx

@@ -113,5 +113,7 @@ export function getServerData(key: keyof One.ClientData) {
   if (process.env.VITE_ENVIRONMENT === 'ssr') {
     throw new Error(`Cannot getServerData on the server`)
   }
-  return getServerContext()?.postRenderData?.[key]
+  if (process.env.VITE_ENVIRONMENT !== 'ssr') {
+    return getServerContext()?.postRenderData?.[key]
+  }
 }

packages/one/src/vite/server.ts

@@ -1,26 +1,39 @@
 import { AsyncLocalStorage } from 'node:async_hooks'
 import { ensureAsyncLocalID } from '../utils/one__ensureAsyncLocalID'
 
-const newContext = new AsyncLocalStorage()
-const existing = globalThis['__vxrnrequestAsyncLocalStore'] as typeof newContext
-export const requestAsyncLocalStore = existing ?? newContext
-
-const newCache = new WeakMap<any, Headers>()
-export const asyncHeadersCache =
-  (globalThis['__vxrnasyncHeadersCache'] as typeof newCache) ?? newCache
-
 // NOTE: we have a big issue right now, we have "one" in optimizeDeps
 // because otherwise you get an error when Root imports Text/View from react-native
 // even though react-native is in optimizeDeps, it seems thats not working
 // but what happens is that somehow one is imported directly by node right away
 // and then later its imported through optimizeDeps, a seaprate instance, creating a separate requestAsyncLocalStore
-globalThis['__vxrnrequestAsyncLocalStore'] ||= requestAsyncLocalStore
+type ALSInstance = AsyncLocalStorage<unknown>
+
+const key = '__vxrnrequestAsyncLocalStore'
+const read = () => globalThis[key] as ALSInstance | undefined
+
+const ASYNC_LOCAL_STORE = {
+  get current() {
+    if (read()) return read()
+    const _ = new AsyncLocalStorage()
+    globalThis[key] = _
+    return _
+  },
+}
+
+export const requestAsyncLocalStore =
+  process.env.VITE_ENVIRONMENT === 'ssr' ? ASYNC_LOCAL_STORE.current : null
+
+const newCache = new WeakMap<any, Headers>()
+
+export const asyncHeadersCache =
+  (globalThis['__vxrnasyncHeadersCache'] as typeof newCache) ?? newCache
+
 globalThis['__vxrnasyncHeadersCache'] ||= asyncHeadersCache
 
 export async function runWithAsyncLocalContext<A>(cb: (id: Object) => Promise<A>): Promise<A> {
   const id = { _id: Math.random() }
   let out: A = null as any
-  await requestAsyncLocalStore.run(id, async () => {
+  await ASYNC_LOCAL_STORE.current!.run(id, async () => {
     out = await cb(id)
   })
   return out

packages/one/types/vite/server.d.ts

@@ -1,7 +1,9 @@
 import { AsyncLocalStorage } from 'node:async_hooks';
-export declare const requestAsyncLocalStore: AsyncLocalStorage<unknown>;
+type ALSInstance = AsyncLocalStorage<unknown>;
+export declare const requestAsyncLocalStore: ALSInstance | null | undefined;
 export declare const asyncHeadersCache: WeakMap<any, Headers>;
 export declare function runWithAsyncLocalContext<A>(cb: (id: Object) => Promise<A>): Promise<A>;
 export declare function setResponseHeaders(cb: (headers: Headers) => void): Promise<void>;
 export declare function mergeHeaders(onto: Headers, from: Headers): void;
+export {};
 //# sourceMappingURL=server.d.ts.map