From 72bf034bf0eebffab05b00c782715dd264861ab8 Mon Sep 17 00:00:00 2001
From: Jing Hua <tohjinghua123@gmail.com>
Date: Sun, 5 Mar 2023 01:41:44 +0800
Subject: [PATCH] feat: mobile add chat button

---
 src/assets/icons/PlusIcon.tsx          |  4 +--
 src/components/Menu/NewChat.tsx        | 24 ++--------------
 src/components/MobileBar/MobileBar.tsx |  9 ++++--
 src/hooks/useAddChat.ts                | 40 ++++++++++++++++++++++++++
 4 files changed, 50 insertions(+), 27 deletions(-)
 create mode 100644 src/hooks/useAddChat.ts

diff --git a/src/assets/icons/PlusIcon.tsx b/src/assets/icons/PlusIcon.tsx
index 2b2c838..e439519 100644
--- a/src/assets/icons/PlusIcon.tsx
+++ b/src/assets/icons/PlusIcon.tsx
@@ -1,6 +1,6 @@
 import React from 'react';
 
-const PlusIcon = () => {
+const PlusIcon = ({ className }: { className?: string }) => {
   return (
     <svg
       stroke='currentColor'
@@ -9,7 +9,7 @@ const PlusIcon = () => {
       viewBox='0 0 24 24'
       strokeLinecap='round'
       strokeLinejoin='round'
-      className='h-4 w-4'
+      className={className ? className : 'h-4 w-4'}
       height='1em'
       width='1em'
       xmlns='http://www.w3.org/2000/svg'
diff --git a/src/components/Menu/NewChat.tsx b/src/components/Menu/NewChat.tsx
index 8e99c21..28e4a79 100644
--- a/src/components/Menu/NewChat.tsx
+++ b/src/components/Menu/NewChat.tsx
@@ -3,8 +3,7 @@ import useStore from '@store/store';
 
 import PlusIcon from '@icon/PlusIcon';
 
-import { ChatInterface } from '@type/chat';
-import { defaultSystemMessage } from '@constants/chat';
+import useAddChat from '@hooks/useAddChat';
 
 const NewChat = () => {
   const [chats, setChats, setCurrentChatIndex, setMessages] = useStore(
@@ -16,26 +15,7 @@ const NewChat = () => {
     ]
   );
 
-  const addChat = () => {
-    if (chats) {
-      const updatedChats: ChatInterface[] = JSON.parse(JSON.stringify(chats));
-      let titleIndex = 1;
-      let title = `New Chat ${titleIndex}`;
-
-      while (chats.some((chat) => chat.title === title)) {
-        titleIndex += 1;
-        title = `New Chat ${titleIndex}`;
-      }
-
-      updatedChats.unshift({
-        title,
-        messages: [{ role: 'system', content: defaultSystemMessage }],
-      });
-      setChats(updatedChats);
-      setMessages(updatedChats[0].messages);
-      setCurrentChatIndex(0);
-    }
-  };
+  const addChat = useAddChat();
 
   return (
     <a
diff --git a/src/components/MobileBar/MobileBar.tsx b/src/components/MobileBar/MobileBar.tsx
index 1ef7790..da84f9c 100644
--- a/src/components/MobileBar/MobileBar.tsx
+++ b/src/components/MobileBar/MobileBar.tsx
@@ -1,14 +1,15 @@
 import React from 'react';
 
 import useStore from '@store/store';
-
-import NewChat from '../Menu/NewChat';
+import PlusIcon from '@icon/PlusIcon';
+import useAddChat from '@hooks/useAddChat';
 
 const MobileBar = () => {
   const [chats, currentChatIndex] = useStore((state) => [
     state.chats,
     state.currentChatIndex,
   ]);
+  const addChat = useAddChat();
 
   return (
     <div className='sticky top-0 left-0 w-full z-50 flex items-center border-b border-white/20 bg-gray-800 pl-1 pt-1 text-gray-200 sm:pl-3 md:hidden'>
@@ -46,7 +47,9 @@ const MobileBar = () => {
           ? chats[currentChatIndex]?.title
           : 'New Chat'}
       </h1>
-      <NewChat />
+      <button type='button' className='px-3 text-gray-400' onClick={addChat}>
+        <PlusIcon className='h-6 w-6' />
+      </button>
     </div>
   );
 };
diff --git a/src/hooks/useAddChat.ts b/src/hooks/useAddChat.ts
new file mode 100644
index 0000000..f020971
--- /dev/null
+++ b/src/hooks/useAddChat.ts
@@ -0,0 +1,40 @@
+import React from 'react';
+import useStore from '@store/store';
+import { defaultSystemMessage } from '@constants/chat';
+import { ChatInterface } from '@type/chat';
+
+const useAddChat = () => {
+  const [chats, setChats, setCurrentChatIndex, setMessages] = useStore(
+    (state) => [
+      state.chats,
+      state.setChats,
+      state.setCurrentChatIndex,
+      state.setMessages,
+    ]
+  );
+
+  const addChat = () => {
+    if (chats) {
+      const updatedChats: ChatInterface[] = JSON.parse(JSON.stringify(chats));
+      let titleIndex = 1;
+      let title = `New Chat ${titleIndex}`;
+
+      while (chats.some((chat) => chat.title === title)) {
+        titleIndex += 1;
+        title = `New Chat ${titleIndex}`;
+      }
+
+      updatedChats.unshift({
+        title,
+        messages: [{ role: 'system', content: defaultSystemMessage }],
+      });
+      setChats(updatedChats);
+      setMessages(updatedChats[0].messages);
+      setCurrentChatIndex(0);
+    }
+  };
+
+  return addChat;
+};
+
+export default useAddChat;