@assistant-ui/react-native 0.1.28 → 0.1.30

package/src/primitives/thread/ThreadMessages.test.tsx

@@ -1,6 +1,8 @@
-import { act } from "react";
+import { act, createRef } from "react";
 import { createRoot, type Root } from "react-dom/client";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import type { FlatList } from "react-native";
+import type { ThreadMessage } from "@assistant-ui/core";
 import { ThreadMessages } from "./ThreadMessages";
 
 type Msg = { id: string; role: string };
@@ -189,4 +191,22 @@ describe("ThreadMessages", () => {
     });
     expect(container.querySelector('[data-testid="c-message"]')).toBeNull();
   });
+
+  it("forwards refs to the underlying FlatList", async () => {
+    h.state.thread.messages = [{ id: "1", role: "user" }];
+    const ref = createRef<FlatList<ThreadMessage>>();
+
+    await act(async () => {
+      root.render(
+        <ThreadMessages
+          ref={ref}
+          components={{
+            Message: () => <span data-testid="c-message">message</span>,
+          }}
+        />,
+      );
+    });
+
+    expect(ref.current).not.toBeNull();
+  });
 });

package/src/primitives/thread/ThreadMessages.tsx

@@ -2,6 +2,7 @@ import {
   type ComponentType,
   type FC,
   type ReactNode,
+  forwardRef,
   memo,
   useCallback,
 } from "react";
@@ -154,11 +155,10 @@ const ThreadMessageByChildren = memo(
 );
 ThreadMessageByChildren.displayName = "ThreadPrimitive.MessageByChildren";
 
-export const ThreadMessages = ({
-  components,
-  children,
-  ...flatListProps
-}: ThreadMessagesProps) => {
+export const ThreadMessages = forwardRef<
+  FlatList<ThreadMessage>,
+  ThreadMessagesProps
+>(({ components, children, ...flatListProps }, ref) => {
   const messages = useAuiState((s) => s.thread.messages);
 
   const renderItem = useCallback(
@@ -179,11 +179,12 @@ export const ThreadMessages = ({
 
   return (
     <FlatList
+      ref={ref}
       data={messages as unknown as ThreadMessage[]}
       renderItem={renderItem}
       keyExtractor={keyExtractor}
       {...flatListProps}
     />
   );
-};
+});
 ThreadMessages.displayName = "ThreadPrimitive.Messages";