chat-store-effect.js 875 B

12345678910111213141516171819202122232425262728293031323334353637
  1. import { useState, useEffect } from 'react'
  2. import { ChatStore } from './chat-store'
  3. let chatStore
  4. export function resetChatStore() {
  5. chatStore = undefined
  6. }
  7. export function useChatStore() {
  8. if (!chatStore) {
  9. chatStore = new ChatStore()
  10. }
  11. function getStateFromStore() {
  12. return {
  13. userId: window.user.id,
  14. atEnd: chatStore.atEnd,
  15. loading: chatStore.loading,
  16. messages: chatStore.messages,
  17. loadMoreMessages: () => chatStore.loadMoreMessages(),
  18. sendMessage: message => chatStore.sendMessage(message)
  19. }
  20. }
  21. const [storeState, setStoreState] = useState(getStateFromStore())
  22. useEffect(() => {
  23. function handleStoreUpdated() {
  24. setStoreState(getStateFromStore())
  25. }
  26. chatStore.on('updated', handleStoreUpdated)
  27. return () => chatStore.off('updated', handleStoreUpdated)
  28. }, [])
  29. return storeState
  30. }