From b48d964c2c053573440b287279db93433fd5d7d7 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Tue, 21 Mar 2023 21:12:27 +0800
Subject: [PATCH] FileManager add layout switch button;store searchText

---
 .../components/FileManager/FileManager.tsx    | 44 ++++++++++++++++---
 .../app/src/components/shared/Modal.tsx       |  2 +-
 lama_cleaner/app/src/store/Atoms.tsx          | 11 +++++
 3 files changed, 51 insertions(+), 6 deletions(-)

diff --git a/lama_cleaner/app/src/components/FileManager/FileManager.tsx b/lama_cleaner/app/src/components/FileManager/FileManager.tsx
index 2a848e9..6c73789 100644
--- a/lama_cleaner/app/src/components/FileManager/FileManager.tsx
+++ b/lama_cleaner/app/src/components/FileManager/FileManager.tsx
@@ -11,7 +11,11 @@ import * as Tabs from '@radix-ui/react-tabs'
 import { useRecoilState, useSetRecoilState } from 'recoil'
 import PhotoAlbum from 'react-photo-album'
 import { BarsArrowDownIcon, BarsArrowUpIcon } from '@heroicons/react/24/outline'
-import { MagnifyingGlassIcon } from '@radix-ui/react-icons'
+import {
+  MagnifyingGlassIcon,
+  ViewHorizontalIcon,
+  ViewGridIcon,
+} from '@radix-ui/react-icons'
 import { useDebounce } from 'react-use'
 import { Id, Index, IndexSearchResult } from 'flexsearch'
 import * as ScrollArea from '@radix-ui/react-scroll-area'
@@ -19,6 +23,7 @@ import Modal from '../shared/Modal'
 import Flex from '../shared/Layout'
 import {
   fileManagerLayout,
+  fileManagerSearchText,
   fileManagerSortBy,
   fileManagerSortOrder,
   SortBy,
@@ -74,10 +79,11 @@ export default function FileManager(props: Props) {
   const [sortBy, setSortBy] = useRecoilState<SortBy>(fileManagerSortBy)
   const [sortOrder, setSortOrder] = useRecoilState(fileManagerSortOrder)
   const [layout, setLayout] = useRecoilState(fileManagerLayout)
-
+  const [debouncedSearchText, setDebouncedSearchText] = useRecoilState(
+    fileManagerSearchText
+  )
   const ref = useRef(null)
-  const [searchText, setSearchText] = useState('')
-  const [debouncedSearchText, setDebouncedSearchText] = useState('')
+  const [searchText, setSearchText] = useState(debouncedSearchText)
   const [tab, setTab] = useState(IMAGE_TAB)
   const [photos, setPhotos] = useState<Photo[]>([])
 
@@ -163,11 +169,39 @@ export default function FileManager(props: Props) {
     onPhotoClick(tab, photos[index].name)
   }
 
+  const renderTitle = () => {
+    return (
+      <Flex
+        style={{ justifyContent: 'flex-start', alignItems: 'center', gap: 12 }}
+      >
+        <div>{`Images (${photos.length})`}</div>
+        <Flex>
+          <Button
+            icon={<ViewHorizontalIcon />}
+            toolTip="Rows layout"
+            onClick={() => {
+              setLayout('rows')
+            }}
+            className={layout !== 'rows' ? 'sort-btn-inactive' : ''}
+          />
+          <Button
+            icon={<ViewGridIcon />}
+            toolTip="Grid layout"
+            onClick={() => {
+              setLayout('masonry')
+            }}
+            className={layout !== 'masonry' ? 'sort-btn-inactive' : ''}
+          />
+        </Flex>
+      </Flex>
+    )
+  }
+
   return (
     <Modal
       onClose={onClose}
       // TODO：layout switch 放到标题中
-      title={`Images (${photos.length})`}
+      title={renderTitle()}
       className="file-manager-modal"
       show={show}
     >
diff --git a/lama_cleaner/app/src/components/shared/Modal.tsx b/lama_cleaner/app/src/components/shared/Modal.tsx
index 0489797..4ab10e0 100644
--- a/lama_cleaner/app/src/components/shared/Modal.tsx
+++ b/lama_cleaner/app/src/components/shared/Modal.tsx
@@ -9,7 +9,7 @@ export interface ModalProps {
   show: boolean
   children?: ReactNode
   onClose?: () => void
-  title: string
+  title: string | ReactNode
   showCloseIcon?: boolean
   className?: string
 }
diff --git a/lama_cleaner/app/src/store/Atoms.tsx b/lama_cleaner/app/src/store/Atoms.tsx
index b1695f9..fc555e3 100644
--- a/lama_cleaner/app/src/store/Atoms.tsx
+++ b/lama_cleaner/app/src/store/Atoms.tsx
@@ -701,6 +701,7 @@ interface FileManagerState {
   sortBy: SortBy
   sortOrder: SortOrder
   layout: 'rows' | 'masonry'
+  searchText: string
 }
 
 const FILE_MANAGER_STATE_KEY = 'fileManagerState'
@@ -711,6 +712,7 @@ export const fileManagerState = atom<FileManagerState>({
     sortBy: SortBy.CTIME,
     sortOrder: SortOrder.DESCENDING,
     layout: 'masonry',
+    searchText: '',
   },
   effects: [localStorageEffect(FILE_MANAGER_STATE_KEY)],
 })
@@ -741,3 +743,12 @@ export const fileManagerLayout = selector({
     set(fileManagerState, { ...val, layout: newValue })
   },
 })
+
+export const fileManagerSearchText = selector({
+  key: 'fileManagerSearchText',
+  get: ({ get }) => get(fileManagerState).searchText,
+  set: ({ get, set }, newValue: any) => {
+    const val = get(fileManagerState)
+    set(fileManagerState, { ...val, searchText: newValue })
+  },
+})