upstream-tldraw/apps/docs/components/common/autorefresh.tsx

'use client'

import { useRouter } from 'next/navigation'
import { ReactNode, useEffect } from 'react'

let AutoRefresh = ({ children }: { children: ReactNode }) => {
	return children
}

if (process.env.NODE_ENV === 'development') {
	AutoRefresh = function AutoRefresh({ children }) {
		const router = useRouter()

		useEffect(() => {
			const ws = new WebSocket('ws://localhost:3201')
			ws.onmessage = (event) => {
				if (event.data === 'refresh') {
					router.refresh()
				}
			}
			return () => {
				ws.close()
			}
		}, [router])

		return children
	}
}

export default AutoRefresh