import type {ClientPerspective} from '@sanity/client'
import {createNode, createNodeMachine} from '@sanity/comlink'
import {setPerspectiveCookie} from 'next-sanity/live/server-actions'
import {
  createCompatibilityActors,
  type LoaderControllerMsg,
  type LoaderNodeMsg,
} from '@sanity/presentation-comlink'
import {useRouter} from 'next/navigation'
import {startTransition, useEffect} from 'react'
import {useEffectEvent} from 'use-effect-event'
import {
  setComlink,
  setComlinkClientConfig,
  setPerspective,
  perspective,
} from '../../live/hooks/context'
import {sanitizePerspective} from '../../live/utils'

export default function PresentationComlink(props: {
  projectId: string
  dataset: string
  draftModeEnabled: boolean
}): React.JSX.Element | null {
  const {projectId, dataset, draftModeEnabled} = props
  const router = useRouter()

  useEffect(() => {
    setComlinkClientConfig(projectId, dataset)
  }, [dataset, projectId])

  const handlePerspectiveChange = useEffectEvent(
    (_perspective: ClientPerspective, signal: AbortSignal) => {
      const nextPerspective = sanitizePerspective(_perspective, 'drafts')
      if (draftModeEnabled && perspective.toString() !== nextPerspective.toString()) {
        setPerspective(nextPerspective)
        startTransition(() =>
          setPerspectiveCookie(nextPerspective)
            .then(() => {
              if (signal.aborted) return
              router.refresh()
            })
            .catch((reason) =>
              console.error('Failed to set the preview perspective cookie', reason),
            ),
        )
      }
    },
  )

  useEffect(() => {
    const comlink = createNode<LoaderNodeMsg, LoaderControllerMsg>(
      {name: 'loaders', connectTo: 'presentation'},
      createNodeMachine<LoaderNodeMsg, LoaderControllerMsg>().provide({
        actors: createCompatibilityActors<LoaderNodeMsg>(),
      }),
    )

    let controller: AbortController | undefined
    comlink.on('loader/perspective', (data) => {
      controller?.abort()
      controller = new AbortController()
      handlePerspectiveChange(data.perspective, controller.signal)
    })

    const stop = comlink.start()
    setComlink(comlink)
    return () => {
      stop()
    }
  }, [])

  return null
}