commit 1f0b7387a7d017cbd32a882d79b23414f7011696 · gmstn.systems/gemstone-app

+25 -11
src/providers/authed/SessionsProvider.tsx
···

       6
       6
        
       import { useHandshakes } from "@/providers/authed/HandshakesProvider";

     

       7
       7
        
       import { useQueries } from "@tanstack/react-query";

     

       8
       8
        
       import type { ReactNode } from "react";

     

       9
       9
       -
       import { createContext, useContext } from "react";

     

       9
       9
       +
       import { createContext, useContext, useEffect, useState } from "react";

     

       10
       10
        
       

     

       11
       11
        
       type SessionsMap = Map<LatticeSessionInfo, WebSocket>;

     

       12
       12
        
       

     
···

       46
       46
        
                       return await endpointQueryFn(handshake);

     

       47
       47
        
                   },

     

       48
       48
        
                   staleTime: DEFAULT_STALE_TIME,

     

       49
       49
       +
                   retries: false,

     

       49
       50
        
               })),

     

       50
       51
        
           });

     

       51
       52
        
       

     
···

       53
       54
        
               handshakesInitialising || endpointQueries.some((q) => q.isLoading);

     

       54
       55
        
           const error = endpointQueries.find((q) => q.error)?.error ?? null;

     

       55
       56
        
       

     

       56
       56
       -
           const sessionsMap = new Map<LatticeSessionInfo, WebSocket>();

     

       57
       57
       +
           const [sessionsMap] = useState<Map<LatticeSessionInfo, WebSocket>>(

     

       58
       58
       +
               new Map<LatticeSessionInfo, WebSocket>(),

     

       59
       59
       +
           );

     

       60
       60
       +
           console.log(sessionsMap);

     

       57
       61
        
       

     

       58
       58
       -
           endpointQueries.forEach((q) => {

     

       59
       59
       -
               const endpoint = q.data;

     

       60
       60
       -
               if (!endpoint) return;

     

       61
       61
       -
               const { sessionInfo, shardUrl } = endpoint;

     

       62
       62
       -
               const websocket = connectToLattice({

     

       63
       63
       -
                   shardUrl,

     

       64
       64
       -
                   sessionToken: sessionInfo.token,

     

       62
       62
       +
           useEffect(() => {

     

       63
       63
       +
               endpointQueries.forEach((q) => {

     

       64
       64
       +
                   const endpoint = q.data;

     

       65
       65
       +
                   if (!endpoint) return;

     

       66
       66
       +
                   const { sessionInfo, shardUrl } = endpoint;

     

       67
       67
       +
                   const existingWs = sessionsMap.get(sessionInfo);

     

       68
       68
       +
                   if (existingWs) return;

     

       69
       69
       +
                   console.log(

     

       70
       70
       +
                       "Initiating lattice connection to",

     

       71
       71
       +
                       shardUrl,

     

       72
       72
       +
                       "with sessionToken",

     

       73
       73
       +
                       sessionInfo.token,

     

       74
       74
       +
                   );

     

       75
       75
       +
                   const websocket = connectToLattice({

     

       76
       76
       +
                       shardUrl,

     

       77
       77
       +
                       sessionToken: sessionInfo.token,

     

       78
       78
       +
                   });

     

       79
       79
       +
                   sessionsMap.set(sessionInfo, websocket);

     

       65
       80
        
               });

     

       66
       66
       -
               sessionsMap.set(sessionInfo, websocket);

     

       67
       67
       -
           });

     

       81
       81
       +
           }, [sessionsMap, endpointQueries]);

     

       68
       82
        
       

     

       69
       83
        
           const value: SessionsContextValue = {

     

       70
       84
        
               sessionsMap,