kopia lustrzana https://github.com/Tldraw/Tldraw
31 wiersze
591 B
TypeScript
31 wiersze
591 B
TypeScript
'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
|