diff --git a/src/app/go/[...route]/page.tsx b/src/app/go/[...route]/page.tsx index c7a7546..e9e3e4c 100644 --- a/src/app/go/[...route]/page.tsx +++ b/src/app/go/[...route]/page.tsx @@ -24,6 +24,7 @@ declare global { export default function Route({ params }: { params: { route: string[] } }) { const ref = useRef(null); + const contentWindow = ref.current?.contentWindow as ContentWindow; const [open, setOpen] = useState(false); const route = params.route.join("/"); @@ -52,8 +53,7 @@ export default function Route({ params }: { params: { route: string[] } }) { function triggerShortcut() { store.set("shortcuts", [], false); - if (!ref.current || !ref.current.contentWindow) return; - const contentWindow = ref.current.contentWindow as ContentWindow; + const contentWindow = ref.current?.contentWindow as ContentWindow; if (!("__uv$location" in contentWindow)) return; const shortcuts: any[] = store("shortcuts"); @@ -86,9 +86,7 @@ export default function Route({ params }: { params: { route: string[] } }) { } function handleLoad() { - if (!ref.current || !ref.current.contentWindow) return; - const contentWindow = ref.current.contentWindow as ContentWindow; - + const contentWindow = ref.current?.contentWindow as ContentWindow; setTabName(contentWindow.document.title); setTabIcon( ( @@ -104,8 +102,11 @@ export default function Route({ params }: { params: { route: string[] } }) { shortcuts.some((value) => value.url == contentWindow.__uv$location.href) ) { setShortcutted(true); + } else { + setShortcutted(false); } } + return (
@@ -128,7 +129,7 @@ export default function Route({ params }: { params: { route: string[] } }) { - @@ -137,7 +138,16 @@ export default function Route({ params }: { params: { route: string[] } }) { - + + Forward + + + + +