update ui deep link handler

Author: filipslezaklab

src/pages/client/types.ts

@@ -113,6 +113,7 @@ export enum TauriEventKey {
   DEAD_CONNECTION_DROPPED = 'dead-connection-dropped',
   DEAD_CONNECTION_RECONNECTED = 'dead-connection-reconnected',
   APPLICATION_CONFIG_CHANGED = 'application-config-changed',
+  ADD_INSTANCE = 'add-instance',
   MFA_TRIGGER = 'mfa-trigger',
   VERSION_MISMATCH = 'version-mismatch',
   UUID_MISMATCH = 'uuid-mismatch',

src/shared/components/providers/DeepLinkProvider.tsx

@@ -1,125 +1,30 @@
-import { getCurrent, onOpenUrl } from '@tauri-apps/plugin-deep-link';
+import { listen } from '@tauri-apps/api/event';
 import { error } from '@tauri-apps/plugin-log';
-import { type PropsWithChildren, useCallback, useEffect, useRef } from 'react';
-import z, { string } from 'zod';
+import { type PropsWithChildren, useEffect } from 'react';
+import { type AddInstancePayload, TauriEventKey } from '../../../pages/client/types';
 import useAddInstance from '../../hooks/useAddInstance';
 import { errorDetail } from '../../utils/errorDetail';
 
-enum DeepLink {
-  AddInstance = 'addinstance',
-}
-
 export const linkStorageKey = 'lastSuccessfullyHandledDeepLink';
 
 export const storeLink = (value: string) => {
   sessionStorage.setItem(linkStorageKey, value);
 };
 
-const readStoreLink = (): string | null => {
-  return sessionStorage.getItem(linkStorageKey);
-};
-
-const addInstanceLinkSchema = z.object({
-  token: string().trim().min(1),
-  url: string().trim().min(1).url(),
-});
-
-const AddInstanceLink = z.object({
-  link: z.literal(DeepLink.AddInstance),
-  data: addInstanceLinkSchema,
-});
-
-const validLinkPayload = z.discriminatedUnion('link', [AddInstanceLink]);
-
-type LinkPayload = z.infer<typeof validLinkPayload>;
-
-const linkIntoPayload = (link: URL | null): LinkPayload | null => {
-  if (link == null) return null;
-
-  const searchData = Object.fromEntries(new URLSearchParams(link.search));
-  const linkKey = [link.hostname, link.pathname]
-    .map((l) => l.trim().replaceAll('/', ''))
-    .filter((l) => l !== '')[0] as string;
-  const payload = {
-    link: linkKey,
-    data: searchData,
-  };
-  const result = validLinkPayload.safeParse(payload);
-  if (result.success) {
-    return result.data;
-  } else {
-    error(`Link ${link} was rejected due to schema validation.`);
-  }
-  return null;
-};
-
 export const DeepLinkProvider = ({ children }: PropsWithChildren) => {
-  const mounted = useRef(false);
-
   const { handleAddInstance } = useAddInstance();
 
-  const handleValidLink = useCallback(
-    async (payload: LinkPayload, rawLink?: string) => {
-      const { data, link } = payload;
-      switch (link) {
-        case DeepLink.AddInstance:
-          await handleAddInstance(data, rawLink);
-          break;
-      }
-      if (rawLink) {
-        storeLink(rawLink);
-      }
-    },
-    [handleAddInstance],
-  );
-
-  // biome-ignore lint/correctness/useExhaustiveDependencies: only on mount
   useEffect(() => {
-    if (!mounted.current) {
-      mounted.current = true;
-
-      let unlisten: (() => void) | undefined;
-      (async () => {
-        const start = await getCurrent();
-        if (start != null) {
-          const lastLink = readStoreLink();
-          // if the link is exact as last successfully executed link
-          // this is only necessary bcs in dev mode window is hot reloaded causing the startup link to be handled multiple times over.
-          if (lastLink != null && lastLink === start[0]) {
-            return;
-          }
-          const payload = linkIntoPayload(new URL(start[0]));
-          if (payload != null) {
-            try {
-              handleValidLink(payload, start[0]);
-            } catch (e) {
-              const detail = errorDetail(e);
-              error(`Failed to handle startup deep link "${payload.link}": ${detail}`);
-            }
-          }
-        }
-        unlisten = await onOpenUrl((urls) => {
-          if (urls?.length) {
-            const link = urls[0];
-            const payload = linkIntoPayload(new URL(link));
-            if (payload != null) {
-              try {
-                handleValidLink(payload);
-              } catch (e) {
-                const detail = errorDetail(e);
-                error(
-                  `Failed to handle valid deep link "${payload?.link}" action: ${detail}`,
-                );
-              }
-            }
-          }
-        });
-      })();
-      return () => {
-        unlisten?.();
-      };
-    }
-  }, []);
+    const unlisten = listen<AddInstancePayload>(TauriEventKey.ADD_INSTANCE, (event) => {
+      handleAddInstance(event.payload).catch((e) => {
+        error(`Failed to handle add-instance event: ${errorDetail(e)}`);
+      });
+    });
+
+    return () => {
+      unlisten.then((fn) => fn());
+    };
+  }, [handleAddInstance]);
 
   return <>{children}</>;
 };