require a magic response for dev reload server to reconnect

src/npm-fastui/src/dev.tsx

@@ -3,6 +3,7 @@ import { FC, useContext, useEffect } from 'react'
 import { sleep } from './tools'
 import { ErrorContext } from './hooks/error'
 import { fireLoadEvent } from './events'
+import { ConfigContext } from './hooks/config'
 
 let devConnected = false
 
@@ -20,6 +21,7 @@ export const DevReload: FC<{ enabled?: boolean }> = ({ enabled }) => {
 
 const DevReloadActive = () => {
   const { setError } = useContext(ErrorContext)
+  const { rootUrl } = useContext(ConfigContext)
 
   useEffect(() => {
     let listening = true
@@ -34,7 +36,7 @@ const DevReloadActive = () => {
         if (!listening || failCount >= 5) {
           return count
         }
-        const response = await fetch('/api/__dev__/reload')
+        const response = await fetch(rootUrl + '/__dev__/reload')
         count++
         console.debug(`dev reload connected ${count}...`)
         // if the response is okay, and we previously failed, clear error
@@ -46,9 +48,10 @@ const DevReloadActive = () => {
         if (response.status === 404) {
           console.log('dev reload endpoint not found, disabling dev reload')
           return count
-        } else if (response.ok) {
+        } else if (response.ok && text.startsWith('fastui-dev-reload')) {
           failCount = 0
-          const value = parseInt(text.replace(/\./g, '')) || 0
+          const valueMatch = text.match(/(\d+)$/)
+          const value = valueMatch ? parseInt(valueMatch[1]!) : 0
           if (value !== lastValue) {
             lastValue = value
             // wait long enough for the server to be back online
@@ -57,6 +60,9 @@ const DevReloadActive = () => {
             fireLoadEvent({ reloadValue: value })
             setError(null)
           }
+        } else if (response.ok) {
+          console.log("dev reload endpoint didn't return magic value, disabling dev reload")
+          return count
         } else {
           failCount++
           await sleep(2000)
@@ -72,6 +78,6 @@ const DevReloadActive = () => {
         devConnected = false
       }
     }
-  }, [setError])
+  }, [setError, rootUrl])
   return <></>
 }

src/python-fastui/fastui/dev.py

@@ -43,6 +43,7 @@ def _on_signal(self, *_args: _t.Any):
 
     async def ping(self):
         # print('connected', os.getpid())
+        yield b'fastui-dev-reload\n'
         yield b'.'
         while True:
             try: