@copilotz/chat-ui 0.3.3 → 0.3.4

package/dist/index.cjs

@@ -142,6 +142,8 @@ var defaultChatConfig = {
     inputHelpText: "Press Enter to send, Shift+Enter to add a new line.",
     thinking: "Thinking...",
     defaultThreadName: "Main Thread",
+    loadOlderMessages: "Load older messages",
+    loadingOlderMessages: "Loading older messages...",
     showMoreMessage: "Show more",
     showLessMessage: "Show less"
   },
@@ -5063,7 +5065,10 @@ var ChatUI = ({
   sidebar: _sidebar,
   isGenerating = false,
   isMessagesLoading = false,
+  isLoadingOlderMessages = false,
+  hasMoreMessagesBefore = false,
   callbacks = {},
+  onLoadOlderMessages,
   user,
   assistant,
   suggestions = [],
@@ -5130,6 +5135,7 @@ var ChatUI = ({
     }
   }, [initialInput]);
   const scrollAreaRef = (0, import_react8.useRef)(null);
+  const prependSnapshotRef = (0, import_react8.useRef)(null);
   const stateRef = (0, import_react8.useRef)(state);
   const inputValueRef = (0, import_react8.useRef)(inputValue);
   const attachmentsRef = (0, import_react8.useRef)(attachments);
@@ -5187,6 +5193,10 @@ var ChatUI = ({
       prevMessageCountRef.current = 0;
       return;
     }
+    if (prependSnapshotRef.current) {
+      prevMessageCountRef.current = messages.length;
+      return;
+    }
     const wasEmpty = prevMessageCountRef.current === 0;
     prevMessageCountRef.current = messages.length;
     if (wasEmpty) {
@@ -5211,6 +5221,46 @@ var ChatUI = ({
   (0, import_react8.useEffect)(() => {
     virtualizer.measure();
   }, [expandedMessageIds, virtualizer]);
+  (0, import_react8.useEffect)(() => {
+    prependSnapshotRef.current = null;
+  }, [currentThreadId]);
+  (0, import_react8.useEffect)(() => {
+    const snapshot = prependSnapshotRef.current;
+    if (!snapshot) return;
+    if (messages.length <= snapshot.messageCount) {
+      if (!isLoadingOlderMessages) {
+        prependSnapshotRef.current = null;
+      }
+      return;
+    }
+    if ((messages[0]?.id ?? null) === snapshot.firstMessageId) {
+      if (!isLoadingOlderMessages) {
+        prependSnapshotRef.current = null;
+      }
+      return;
+    }
+    requestAnimationFrame(() => {
+      virtualizer.measure();
+      requestAnimationFrame(() => {
+        const viewport = scrollAreaRef.current;
+        if (!viewport) return;
+        const heightDelta = viewport.scrollHeight - snapshot.scrollHeight;
+        viewport.scrollTop = snapshot.scrollTop + heightDelta;
+        prependSnapshotRef.current = null;
+      });
+    });
+  }, [messages, isLoadingOlderMessages, virtualizer]);
+  const requestOlderMessages = (0, import_react8.useCallback)(() => {
+    if (!onLoadOlderMessages || !hasMoreMessagesBefore || isLoadingOlderMessages) return;
+    const viewport = scrollAreaRef.current;
+    prependSnapshotRef.current = viewport ? {
+      scrollHeight: viewport.scrollHeight,
+      scrollTop: viewport.scrollTop,
+      firstMessageId: messages[0]?.id ?? null,
+      messageCount: messages.length
+    } : null;
+    onLoadOlderMessages();
+  }, [hasMoreMessagesBefore, isLoadingOlderMessages, messages, onLoadOlderMessages]);
   (0, import_react8.useEffect)(() => {
     const validMessageIds = new Set(messages.map((message) => message.id));
     setExpandedMessageIds((prev) => {
@@ -5229,11 +5279,15 @@ var ChatUI = ({
   const handleScroll = (0, import_react8.useCallback)((e) => {
     const { scrollTop, scrollHeight, clientHeight } = e.currentTarget;
     const isAtBottom = scrollHeight - scrollTop - clientHeight < 50;
+    const isNearTop = scrollTop < 120;
+    if (isNearTop && hasMoreMessagesBefore && !isLoadingOlderMessages) {
+      requestOlderMessages();
+    }
     setState((prev) => {
       if (prev.isAtBottom === isAtBottom) return prev;
       return { ...prev, isAtBottom };
     });
-  }, []);
+  }, [hasMoreMessagesBefore, isLoadingOlderMessages, requestOlderMessages]);
   const handleSendMessage = (0, import_react8.useCallback)((content, messageAttachments = []) => {
     if (!content.trim() && messageAttachments.length === 0) return;
     callbacks.onSendMessage?.(content, messageAttachments, createStateCallback());
@@ -5489,48 +5543,59 @@ var ChatUI = ({
               viewportClassName: "p-4 overscroll-contain",
               onScrollCapture: handleScroll,
               style: { contain: "strict" },
-              children: /* @__PURE__ */ (0, import_jsx_runtime26.jsx)("div", { className: "max-w-4xl mx-auto pb-4", children: isMessagesLoading ? renderMessageLoadingSkeleton() : messages.length === 0 ? renderSuggestions() : /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(
-                "div",
-                {
-                  style: {
-                    height: `${virtualizer.getTotalSize()}px`,
-                    width: "100%",
-                    position: "relative"
-                  },
-                  children: virtualizer.getVirtualItems().map((virtualRow) => {
-                    const message = messages[virtualRow.index];
-                    const prevMessage = virtualRow.index > 0 ? messages[virtualRow.index - 1] : null;
-                    const isGrouped = prevMessage !== null && prevMessage.role === message.role && getMessageSpeakerKey(prevMessage) === getMessageSpeakerKey(message);
-                    return /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(
-                      "div",
-                      {
-                        "data-index": virtualRow.index,
-                        ref: virtualizer.measureElement,
-                        style: {
-                          position: "absolute",
-                          top: 0,
-                          left: 0,
-                          width: "100%",
-                          transform: `translateY(${virtualRow.start}px)`
+              children: /* @__PURE__ */ (0, import_jsx_runtime26.jsxs)("div", { className: "max-w-4xl mx-auto pb-4", children: [
+                messages.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime26.jsx)("div", { className: "flex justify-center py-2", children: isLoadingOlderMessages ? /* @__PURE__ */ (0, import_jsx_runtime26.jsx)("span", { className: "text-xs text-muted-foreground", children: config.labels.loadingOlderMessages }) : hasMoreMessagesBefore ? /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(
+                  "button",
+                  {
+                    type: "button",
+                    onClick: requestOlderMessages,
+                    className: "text-xs text-muted-foreground transition-colors hover:text-foreground",
+                    children: config.labels.loadOlderMessages
+                  }
+                ) : null }),
+                isMessagesLoading ? renderMessageLoadingSkeleton() : messages.length === 0 ? renderSuggestions() : /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(
+                  "div",
+                  {
+                    style: {
+                      height: `${virtualizer.getTotalSize()}px`,
+                      width: "100%",
+                      position: "relative"
+                    },
+                    children: virtualizer.getVirtualItems().map((virtualRow) => {
+                      const message = messages[virtualRow.index];
+                      const prevMessage = virtualRow.index > 0 ? messages[virtualRow.index - 1] : null;
+                      const isGrouped = prevMessage !== null && prevMessage.role === message.role && getMessageSpeakerKey(prevMessage) === getMessageSpeakerKey(message);
+                      return /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(
+                        "div",
+                        {
+                          "data-index": virtualRow.index,
+                          ref: virtualizer.measureElement,
+                          style: {
+                            position: "absolute",
+                            top: 0,
+                            left: 0,
+                            width: "100%",
+                            transform: `translateY(${virtualRow.start}px)`
+                          },
+                          children: /* @__PURE__ */ (0, import_jsx_runtime26.jsxs)("div", { className: virtualRow.index === 0 ? "" : isGrouped ? "pt-2" : "pt-4", children: [
+                            /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(
+                              Message,
+                              {
+                                message,
+                                ...messageProps,
+                                isGrouped,
+                                isExpanded: Boolean(expandedMessageIds[message.id])
+                              }
+                            ),
+                            message.role === "assistant" && renderInlineSuggestions(message.id)
+                          ] })
                         },
-                        children: /* @__PURE__ */ (0, import_jsx_runtime26.jsxs)("div", { className: virtualRow.index === 0 ? "" : isGrouped ? "pt-2" : "pt-4", children: [
-                          /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(
-                            Message,
-                            {
-                              message,
-                              ...messageProps,
-                              isGrouped,
-                              isExpanded: Boolean(expandedMessageIds[message.id])
-                            }
-                          ),
-                          message.role === "assistant" && renderInlineSuggestions(message.id)
-                        ] })
-                      },
-                      message.id
-                    );
-                  })
-                }
-              ) })
+                        message.id
+                      );
+                    })
+                  }
+                )
+              ] })
             }
           ),
           /* @__PURE__ */ (0, import_jsx_runtime26.jsxs)("div", { className: "bg-background pb-[env(safe-area-inset-bottom)]", children: [