flyerhq/react-native-firebase-chat-core

import firestore from '@react-native-firebase/firestore'
import * as React from 'react'

import { ROOMS_COLLECTION_NAME } from '.'
import { Room } from './types'
import { useFirebaseUser } from './useFirebaseUser'
import { processRoomDocument } from './utils'

/** Returns a stream of changes in a room from Firebase */
export const useRoom = (initialRoom: Room) => {
  const [room, setRoom] = React.useState(initialRoom)
  const { firebaseUser } = useFirebaseUser()

  React.useEffect(() => {
    if (!firebaseUser) return

    return firestore()
      .collection(ROOMS_COLLECTION_NAME)
      .doc(initialRoom.id)
      .onSnapshot(async (doc) => {
        const newRoom = await processRoomDocument({ doc, firebaseUser })

        setRoom(newRoom)
      })
  }, [firebaseUser, initialRoom.id])

  return { room }
}