overleaf/services/web/frontend/js/shared/hooks/use-detach-state.js

import { useEffect, useState, useCallback } from 'react'
import { useDetachContext } from '../context/detach-context'
import getMeta from '../../utils/meta'

const debugPdfDetach = getMeta('ol-debugPdfDetach')

export default function useDetachState(
  key,
  defaultValue,
  senderRole,
  targetRole
) {
  const [value, setValue] = useState(defaultValue)

  const {
    role,
    broadcastEvent,
    addEventHandler,
    deleteEventHandler,
  } = useDetachContext()

  const eventName = `state-${key}`

  useEffect(() => {
    if (role === senderRole) {
      broadcastEvent(eventName, { value })
    }
  }, [role, senderRole, eventName, value, broadcastEvent])

  const handleStateEvent = useCallback(
    message => {
      if (message.event !== eventName) {
        return
      }
      if (role !== targetRole) {
        return
      }
      if (debugPdfDetach) {
        console.log(`Set ${message.data.value} for ${eventName}`)
      }
      setValue(message.data.value)
    },
    [role, targetRole, eventName, setValue]
  )

  useEffect(() => {
    addEventHandler(handleStateEvent)
    return () => deleteEventHandler(handleStateEvent)
  }, [addEventHandler, deleteEventHandler, handleStateEvent])

  return [value, setValue]
}