All files / apps/host/src/hooks useWebSocket.ts

100% Statements 5/5
50% Branches 1/2
100% Functions 2/2
100% Lines 5/5

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24                          1x 11x     4x 4x 4x        
import { getAuth, getIdToken } from '@react-native-firebase/auth';
 
import { useWebSocketListener as useSharedWebSocket } from '@repo/hooks';
 
import { FLASH_WS_URL } from '@/constants/apis';
 
export interface WSMessage {
  type: string;
  channel?: string;
  payload?: any;
  data?: unknown;
}
 
export const useWebSocketListener = () => {
  return useSharedWebSocket({
    url: FLASH_WS_URL,
    getToken: async () => {
      const auth = getAuth();
      const user = auth.currentUser;
      return user ? getIdToken(user) : null;
    },
  });
};