src/components/interface/latent-engine/core/useChildController.ts

'use client'

import { create } from 'zustand'
import { Bellhop } from 'bellhop-iframe'

export const useChildController = create<{
  bellhop: Bellhop

  /**
   * Whether the communication pipeline seems to be working or not
   *
   * Initially we assume that it is, but this can be invalidated
   * if we know for certain it is not, or in case of exception
   */
  canUseBellhop: boolean

  /**
   * 
   */
  hasLoadedBellhop: boolean

  /**
   * Whether Clapper is ready
   * 
   * This will be set upon reception of the event
   */
  clapperIsReady: boolean

  setDomElement: (domElement: HTMLIFrameElement, origin?: string) => void

  setCanUseBellhop: (canUseBellhop: boolean) => void
  setHasLoadedBellhop: (hasLoadedBellhop: boolean) => void
  onMessage: (name: string, callback: Function, priority?: number) => void
  sendMessage: (type: string, data?: any) => void
}>((set, get) => ({
  bellhop: undefined as unknown as Bellhop,
  canUseBellhop: true,
  hasLoadedBellhop: false,
  clapperIsReady: false,
  setDomElement: (domElement: HTMLIFrameElement, origin?: string) => {
    const bellhop = new Bellhop()
    bellhop.connect(domElement, origin)
    set({
      bellhop,
    })
  },
  setCanUseBellhop: (canUseBellhop: boolean) => {
    set({
      canUseBellhop,
    })
  },
  setHasLoadedBellhop: (hasLoadedBellhop: boolean) => {
    set({
      bellhop: hasLoadedBellhop ? new Bellhop() : (undefined as unknown as Bellhop),
      hasLoadedBellhop,
    })
  },
  onMessage: (name: string, callback: Function, priority?: number) => {
    const { bellhop } = get()
    try {
      bellhop.on(name, callback, priority)
    } catch (err) {
      console.log(`failed to subscribe to parent iframe messages:`, err)
    }
  },
  sendMessage: (type: string, data?: any) => {
    const { bellhop } = get()
    try {
      bellhop.send(type, data)
    } catch (err) {
      console.log(`failed to send a message to parent iframe:`, err)
    }
  },
}))