|
| 1 | +import { useEffect } from 'react'; |
| 2 | +import uniqBy from 'lodash/uniqBy'; |
| 3 | + |
| 4 | +import { useChatContext } from '../../../../contexts/chatContext/ChatContext'; |
| 5 | +import { getChannel } from '../../utils'; |
| 6 | + |
| 7 | +import type { Channel, Event } from 'stream-chat'; |
| 8 | + |
| 9 | +import type { |
| 10 | + DefaultAttachmentType, |
| 11 | + DefaultChannelType, |
| 12 | + DefaultCommandType, |
| 13 | + DefaultEventType, |
| 14 | + DefaultMessageType, |
| 15 | + DefaultReactionType, |
| 16 | + DefaultUserType, |
| 17 | + UnknownType, |
| 18 | +} from '../../../../types/types'; |
| 19 | + |
| 20 | +type Parameters< |
| 21 | + At extends UnknownType = DefaultAttachmentType, |
| 22 | + Ch extends UnknownType = DefaultChannelType, |
| 23 | + Co extends string = DefaultCommandType, |
| 24 | + Ev extends UnknownType = DefaultEventType, |
| 25 | + Me extends UnknownType = DefaultMessageType, |
| 26 | + Re extends UnknownType = DefaultReactionType, |
| 27 | + Us extends UnknownType = DefaultUserType |
| 28 | +> = { |
| 29 | + setChannels: React.Dispatch< |
| 30 | + React.SetStateAction<Channel<At, Ch, Co, Ev, Me, Re, Us>[]> |
| 31 | + >; |
| 32 | + onChannelVisible?: ( |
| 33 | + setChannels: React.Dispatch< |
| 34 | + React.SetStateAction<Channel<At, Ch, Co, Ev, Me, Re, Us>[]> |
| 35 | + >, |
| 36 | + event: Event<At, Ch, Co, Ev, Me, Re, Us>, |
| 37 | + ) => void; |
| 38 | +}; |
| 39 | + |
| 40 | +export const useChannelVisible = < |
| 41 | + At extends UnknownType = DefaultAttachmentType, |
| 42 | + Ch extends UnknownType = DefaultChannelType, |
| 43 | + Co extends string = DefaultCommandType, |
| 44 | + Ev extends UnknownType = DefaultEventType, |
| 45 | + Me extends UnknownType = DefaultMessageType, |
| 46 | + Re extends UnknownType = DefaultReactionType, |
| 47 | + Us extends UnknownType = DefaultUserType |
| 48 | +>({ |
| 49 | + onChannelVisible, |
| 50 | + setChannels, |
| 51 | +}: Parameters<At, Ch, Co, Ev, Me, Re, Us>) => { |
| 52 | + const { client } = useChatContext<At, Ch, Co, Ev, Me, Re, Us>(); |
| 53 | + |
| 54 | + useEffect(() => { |
| 55 | + const handleEvent = async (event: Event<At, Ch, Co, Ev, Me, Re, Us>) => { |
| 56 | + if (typeof onChannelVisible === 'function') { |
| 57 | + onChannelVisible(setChannels, event); |
| 58 | + } else { |
| 59 | + if (event.channel_id && event.channel_type) { |
| 60 | + const channel = await getChannel<At, Ch, Co, Ev, Me, Re, Us>({ |
| 61 | + client, |
| 62 | + id: event.channel_id, |
| 63 | + type: event.channel_type, |
| 64 | + }); |
| 65 | + setChannels((channels) => uniqBy([channel, ...channels], 'cid')); |
| 66 | + } |
| 67 | + } |
| 68 | + }; |
| 69 | + |
| 70 | + client.on('channel.visible', handleEvent); |
| 71 | + return () => client.off('channel.visible', handleEvent); |
| 72 | + }, []); |
| 73 | +}; |
0 commit comments