From 60635d07c21ea2417747e97ab662d972dc9c6442 Mon Sep 17 00:00:00 2001
From: Alois <aloisianer@proton.me>
Date: Fri, 10 Apr 2026 22:07:51 +0200
Subject: [PATCH] Deeplink error fix, updated chat ux

---
 apps/tauri/src/deeplinkHandler.tsx       |  6 ++-
 packages/chat/src/components/input.tsx   |  2 +-
 packages/chat/src/components/message.tsx | 15 +++++++-
 packages/chat/src/screen.tsx             | 47 ++++++++++++++++++++++--
 4 files changed, 63 insertions(+), 7 deletions(-)

diff --git a/apps/tauri/src/deeplinkHandler.tsx b/apps/tauri/src/deeplinkHandler.tsx
index e9d0d47..b375771 100644
--- a/apps/tauri/src/deeplinkHandler.tsx
+++ b/apps/tauri/src/deeplinkHandler.tsx
@@ -6,6 +6,7 @@ import {
   type ReactNode,
 } from "react";
 import { getCurrent, onOpenUrl } from "@tauri-apps/plugin-deep-link";
+import { isTauri } from "@tauri-apps/api/core";
 
 type DeeplinkContextValue = {
   deeplinks: readonly string[];
@@ -30,8 +31,11 @@ export default function DeeplinkProvider({
   children: ReactNode;
 }) {
   const [deeplinks, setDeeplinks] = useState<string[]>([]);
+  const isTauriEnv = isTauri();
 
   useEffect(() => {
+    if (!isTauriEnv) return;
+
     let mounted = true;
     let unlisten: (() => void) | undefined;
 
@@ -53,7 +57,7 @@ export default function DeeplinkProvider({
       mounted = false;
       unlisten?.();
     };
-  }, []);
+  }, [isTauriEnv]);
 
   useEffect(() => {
     console.log(deeplinks);
diff --git a/packages/chat/src/components/input.tsx b/packages/chat/src/components/input.tsx
index cb5888a..77dee82 100644
--- a/packages/chat/src/components/input.tsx
+++ b/packages/chat/src/components/input.tsx
@@ -115,7 +115,7 @@ export default function InputComponent({
       <CardHeader className="relative p-0 flex flex-col">
         <div
           aria-hidden="true"
-          className="pointer-events-none absolute left-0 top-0 w-full opacity-0"
+          className="pointer-events-none absolute left-0 top-0 w-full px-2.5 opacity-0"
           style={{ visibility: "hidden" }}
         >
           <Message
diff --git a/packages/chat/src/components/message.tsx b/packages/chat/src/components/message.tsx
index fc06822..8f8bd19 100644
--- a/packages/chat/src/components/message.tsx
+++ b/packages/chat/src/components/message.tsx
@@ -28,7 +28,7 @@ function generateFixedLoadingSize(size: number, height: number) {
  * @param props Parameter props.
  * @returns unknown.
  */
-export default function Message(props: {
+function MessageComponent(props: {
   message: RawMessage & {
     failed?: boolean;
   };
@@ -143,3 +143,16 @@ export default function Message(props: {
     </div>
   );
 }
+
+export default React.memo(MessageComponent, (prev, next) => {
+  return (
+    prev.notEncrypted === next.notEncrypted &&
+    prev.measureRef === next.measureRef &&
+    prev.message.timestamp === next.message.timestamp &&
+    prev.message.content === next.message.content &&
+    prev.message.height === next.message.height &&
+    prev.message.sent_by_self === next.message.sent_by_self &&
+    prev.message.message_state === next.message.message_state &&
+    prev.message.failed === next.message.failed
+  );
+});
diff --git a/packages/chat/src/screen.tsx b/packages/chat/src/screen.tsx
index 0d9c8c5..193edf3 100644
--- a/packages/chat/src/screen.tsx
+++ b/packages/chat/src/screen.tsx
@@ -8,6 +8,11 @@ import Message from "./components/message";
 
 import { PAGE_SIZE } from "./values";
 import { useLayoutEffect } from "react";
+import { useIsMobile } from "@tensamin/ui/utils";
+
+function getDistanceFromBottom(element: HTMLDivElement) {
+  return element.scrollHeight - (element.scrollTop + element.clientHeight);
+}
 
 /**
  * Renders the chat screen with virtualized history and live message updates.
@@ -18,6 +23,8 @@ export default function Screen() {
 
   const inputBoxRef = React.useRef<HTMLDivElement | null>(null);
   const scrollRef = React.useRef<HTMLDivElement | null>(null);
+  const isMobile = useIsMobile();
+  const stickyBottomThreshold = isMobile ? 220 : 140;
 
   const [hasScrolledToBottomInitially, setHasScrolledToBottomInitially] =
     React.useState(false);
@@ -72,11 +79,22 @@ export default function Screen() {
     return [...historicalMessages, ...liveWithoutDuplicates];
   }, [historicalMessages, liveMessagesSnapshot]);
 
+  const messagesRef = React.useRef(messages);
+
+  React.useEffect(() => {
+    messagesRef.current = messages;
+  }, [messages]);
+
+  const getItemKey = React.useCallback((index: number) => {
+    return messagesRef.current[index]?.timestamp ?? index;
+  }, []);
+
   const virtualizer = useVirtualizer({
     count: messages.length,
     getScrollElement: () => scrollRef.current,
+    getItemKey,
     estimateSize: (index) => messages[index].height,
-    overscan: 6,
+    overscan: isMobile ? 10 : 6,
   });
 
   /**
@@ -125,11 +143,16 @@ export default function Screen() {
     }
 
     if (count > lastLiveMessageCount) {
-      scrollRef.current.scrollTop = scrollRef.current.scrollHeight;
+      const shouldStickToBottom =
+        getDistanceFromBottom(scrollRef.current) <= stickyBottomThreshold;
+
+      if (shouldStickToBottom) {
+        scrollRef.current.scrollTop = scrollRef.current.scrollHeight;
+      }
     }
 
     setLastLiveMessageCount(count);
-  }, [lastLiveMessageCount, liveMessagesSnapshot.length]);
+  }, [lastLiveMessageCount, liveMessagesSnapshot.length, stickyBottomThreshold]);
 
   React.useLayoutEffect(() => {
     if (
@@ -163,6 +186,9 @@ export default function Screen() {
     const host = scrollRef.current;
     if (!host || !el || typeof window === "undefined") return;
 
+    const shouldStickToBottom =
+      getDistanceFromBottom(host) <= stickyBottomThreshold;
+
     const viewportHeight = Math.max(
       window.innerHeight,
       window.visualViewport?.height ?? 0,
@@ -170,7 +196,17 @@ export default function Screen() {
     const inputHeight = el.scrollHeight + 56;
 
     host.style.maxHeight = `calc(${viewportHeight}px - ${inputHeight - 62}px - env(safe-area-inset-bottom) - env(safe-area-inset-top))`;
-  }, []);
+
+    if (shouldStickToBottom) {
+      requestAnimationFrame(() => {
+        if (!scrollRef.current) {
+          return;
+        }
+
+        scrollRef.current.scrollTop = scrollRef.current.scrollHeight;
+      });
+    }
+  }, [stickyBottomThreshold]);
 
   useLayoutEffect(() => {
     updateMaxHeight();
@@ -221,6 +257,9 @@ export default function Screen() {
         ref={scrollRef}
         id="chat_container"
         className="flex-1 overflow-y-auto px-2.5"
+        style={{
+          overflowAnchor: "none",
+        }}
         onScroll={handleContainerScroll}
       >
         <div