mark/vibn-agent-runner
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(preview): re-trigger dev server ensure check when user clicks refresh button

Browse Source
This commit is contained in:
mawkone
2026-06-11 12:07:39 -07:00
parent 2036df6c2b
commit 7337e2c5b0
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
vibn-frontend/app/[workspace]/project/[projectId]/(home)/preview/page.tsx
View File

@@ -153,6 +153,18 @@ export default function PreviewTab() {
const deviceMode = usePreviewToolbarStore((s) => s.deviceMode); const deviceMode = usePreviewToolbarStore((s) => s.deviceMode);
const refreshKey = usePreviewToolbarStore((s) => s.refreshKey); const refreshKey = usePreviewToolbarStore((s) => s.refreshKey);
// When the user clicks the manual refresh button in the toolbar, we don't
// just want to reload the iframe — we also want to trigger the same ghost/zombie
// check as the initial mount, in case the server died while they were looking at it.
const prevRefreshKeyRef = useRef(refreshKey);
useEffect(() => {
if (refreshKey === prevRefreshKeyRef.current) return;
prevRefreshKeyRef.current = refreshKey;
// Reset the ensure called flag so the ensure effect (below) fires again.
ensureCalledRef.current = false;
}, [refreshKey]);
useLayoutEffect(() => { useLayoutEffect(() => {
setIframeSrc(primaryRunning?.url ?? null); setIframeSrc(primaryRunning?.url ?? null);
}, [primaryRunning?.url]); }, [primaryRunning?.url]);