From f4da4a30356878870108bce2cdef9d89ffa721ea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?marcin=20miko=C5=82ajczak?= <git@mkljczk.pl>
Date: Thu, 19 Sep 2024 16:14:47 +0200
Subject: [PATCH] pl-fe: move chat search results to @tanstack/virtual
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: marcin mikołajczak <git@mkljczk.pl>
---
 .../components/chat-search/chat-search.tsx    |  4 +--
 .../chats/components/chat-search/results.tsx  | 28 ++++++++++---------
 packages/pl-fe/src/queries/search.ts          | 24 +++++++++-------
 3 files changed, 31 insertions(+), 25 deletions(-)

diff --git a/packages/pl-fe/src/features/chats/components/chat-search/chat-search.tsx b/packages/pl-fe/src/features/chats/components/chat-search/chat-search.tsx
index 8394bc669..fac23a0c3 100644
--- a/packages/pl-fe/src/features/chats/components/chat-search/chat-search.tsx
+++ b/packages/pl-fe/src/features/chats/components/chat-search/chat-search.tsx
@@ -86,7 +86,7 @@ const ChatSearch = (props: IChatSearch) => {
   };
 
   return (
-    <Stack space={4} className='h-full grow'>
+    <Stack space={4} className='h-full relative overflow-auto'>
       <div className='px-4'>
         <Input
           data-testid='search'
@@ -109,7 +109,7 @@ const ChatSearch = (props: IChatSearch) => {
         />
       </div>
 
-      <Stack className='grow'>
+      <Stack className='h-full grow overflow-auto'>
         {renderBody()}
       </Stack>
     </Stack>
diff --git a/packages/pl-fe/src/features/chats/components/chat-search/results.tsx b/packages/pl-fe/src/features/chats/components/chat-search/results.tsx
index 4d8cdac5b..f4b6fc9a8 100644
--- a/packages/pl-fe/src/features/chats/components/chat-search/results.tsx
+++ b/packages/pl-fe/src/features/chats/components/chat-search/results.tsx
@@ -1,7 +1,7 @@
 import clsx from 'clsx';
 import React, { useCallback, useState } from 'react';
-import { Virtuoso } from 'react-virtuoso';
 
+import ScrollableList from 'pl-fe/components/scrollable-list';
 import { Avatar, HStack, Stack, Text } from 'pl-fe/components/ui';
 import VerificationBadge from 'pl-fe/components/verification-badge';
 import useAccountSearch from 'pl-fe/queries/search';
@@ -25,7 +25,7 @@ const Results = ({ accountSearchResult, onSelect }: IResults) => {
     }
   };
 
-  const renderAccount = useCallback((_index: number, account: Account) => (
+  const renderAccount = useCallback((account: Account) => (
     <button
       key={account.id}
       type='button'
@@ -49,17 +49,19 @@ const Results = ({ accountSearchResult, onSelect }: IResults) => {
 
   return (
     <div className='relative grow'>
-      <Virtuoso
-        data={accounts}
-        itemContent={(index, chat) => (
-          <div className='px-2'>
-            {renderAccount(index, chat)}
-          </div>
-        )}
-        endReached={handleLoadMore}
-        atTopStateChange={(atTop) => setNearTop(atTop)}
-        atBottomStateChange={(atBottom) => setNearBottom(atBottom)}
-      />
+      <ScrollableList
+        itemClassName='px-2'
+        loadMoreClassName='mx-4 mb-4'
+        onScroll={(startIndex, endIndex) => {
+          setNearTop(startIndex === 0);
+          setNearBottom(endIndex === accounts?.length);
+        }}
+        isLoading={isFetching}
+        hasMore={hasNextPage}
+        onLoadMore={handleLoadMore}
+      >
+        {(accounts || []).map((chat) => renderAccount(chat))}
+      </ScrollableList>
 
       <>
         <div
diff --git a/packages/pl-fe/src/queries/search.ts b/packages/pl-fe/src/queries/search.ts
index 32de4e5aa..c05aaf6ab 100644
--- a/packages/pl-fe/src/queries/search.ts
+++ b/packages/pl-fe/src/queries/search.ts
@@ -11,18 +11,22 @@ const useAccountSearch = (q: string) => {
   const getAccountSearch = async(q: string, pageParam?: Pick<PaginatedResponse<Account>, 'next'>): Promise<PaginatedResponse<Account>> => {
     if (pageParam?.next) return pageParam.next();
 
-    const response = await client.accounts.searchAccounts(q, {
-      limit: 10,
-      following: true,
-      offset: 0,
-    });
+    const _getAccountSearch = async (offset = 0) => {
+      const response = await client.accounts.searchAccounts(q, {
+        limit: 10,
+        following: true,
+        offset: offset,
+      });
 
-    return {
-      previous: null,
-      next: null,
-      items: response,
-      partial: false,
+      return {
+        previous: null,
+        next: response.length ? () => _getAccountSearch(offset + 10) : null,
+        items: response,
+        partial: false,
+      };
     };
+
+    return _getAccountSearch(0);
   };
 
   const queryInfo = useInfiniteQuery({