fix(jotai): Change GuestAtom to AsyncAtom (#12)

pbkit · Aug 18, 2022 · a752f9d · a752f9d
1 parent 337276f
commit a752f9d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/jotai/index.ts b/src/jotai/index.ts
@@ -9,7 +9,7 @@ import { createWrpGuest, WrpGuest } from "../guest.ts";
 
 export type SocketAtom = Atom<Promise<Socket | undefined>>;
 export type ChannelAtom = Atom<WrpChannel | undefined>;
-export type GuestAtom = Atom<WrpGuest | undefined>;
+export type GuestAtom = Atom<Promise<WrpGuest> | undefined>;
 export type ClientImplAtom = Atom<RpcClientImpl | undefined>;
 export interface WrpAtomSet {
   channelAtom: ChannelAtom;
@@ -19,15 +19,15 @@ export interface WrpAtomSet {
 export function createWrpAtomSet(socketAtom: SocketAtom): WrpAtomSet {
   interface ChannelAndGuest {
     channel: WrpChannel;
-    guest: WrpGuest;
+    guest: Promise<WrpGuest>;
   }
   const channelAndGuestAtom = atom<Promise<ChannelAndGuest | undefined>>(
     async (get) => {
       const socket = get(socketAtom);
       if (!socket) return;
       const realChannel = createWrpChannel(socket);
       const listeners: ((message?: WrpMessage) => void)[] = [];
-      const guest = await createWrpGuest({
+      const guest = createWrpGuest({
         channel: {
           ...realChannel,
           async *listen() {