From 37f7042f0f2b716789a925c11f0e35b6d4d6f846 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Wed, 8 Oct 2025 21:42:36 +0800
Subject: [PATCH] refactor: update styling and layout in Message component and
 NotesSidebar

- Adjusted class names in Message component for better layout management.
- Modified margin in DropHintNode of NotesSidebar for improved spacing.
- Enhanced BackupService to remove 'notes_tree' from indexedDB during data restoration.
---
 src/renderer/src/pages/home/Messages/Message.tsx           | 2 +-
 .../src/pages/home/Tabs/components/AssistantItem.tsx       | 1 +
 src/renderer/src/pages/notes/NotesSidebar.tsx              | 2 +-
 src/renderer/src/services/BackupService.ts                 | 7 +++++++
 4 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/pages/home/Messages/Message.tsx b/src/renderer/src/pages/home/Messages/Message.tsx
index d4a7ceef71..926aa66414 100644
--- a/src/renderer/src/pages/home/Messages/Message.tsx
+++ b/src/renderer/src/pages/home/Messages/Message.tsx
@@ -231,7 +231,7 @@ const MessageItem: FC<Props> = ({
                 <HorizontalScrollContainer
                   classNames={{
                     content: cn(
-                      'items-center',
+                      'flex-1 items-center justify-between',
                       isLastMessage && messageStyle === 'plain' ? 'flex-row-reverse' : 'flex-row'
                     )
                   }}>
diff --git a/src/renderer/src/pages/home/Tabs/components/AssistantItem.tsx b/src/renderer/src/pages/home/Tabs/components/AssistantItem.tsx
index d20c69b052..c7d6c7a870 100644
--- a/src/renderer/src/pages/home/Tabs/components/AssistantItem.tsx
+++ b/src/renderer/src/pages/home/Tabs/components/AssistantItem.tsx
@@ -386,6 +386,7 @@ const Container = styled.div`
   border-radius: var(--list-item-border-radius);
   border: 0.5px solid transparent;
   width: calc(var(--assistants-width) - 20px);
+  cursor: pointer;
 
   &:hover {
     background-color: var(--color-list-item-hover);
diff --git a/src/renderer/src/pages/notes/NotesSidebar.tsx b/src/renderer/src/pages/notes/NotesSidebar.tsx
index a53fc5b5f7..e7f3db9494 100644
--- a/src/renderer/src/pages/notes/NotesSidebar.tsx
+++ b/src/renderer/src/pages/notes/NotesSidebar.tsx
@@ -1129,7 +1129,7 @@ const DragOverIndicator = styled.div`
 `
 
 const DropHintNode = styled.div`
-  margin: 8px;
+  margin: 6px 0;
   margin-bottom: 20px;
 
   ${TreeNodeContainer} {
diff --git a/src/renderer/src/services/BackupService.ts b/src/renderer/src/services/BackupService.ts
index 985247e43f..c168bd489b 100644
--- a/src/renderer/src/services/BackupService.ts
+++ b/src/renderer/src/services/BackupService.ts
@@ -89,6 +89,7 @@ export async function restore() {
       }
 
       await handleData(data)
+
       notificationService.send({
         id: uuid(),
         type: 'success',
@@ -850,6 +851,12 @@ export async function handleData(data: Record<string, any>) {
 
   if (data.version >= 2) {
     localStorage.setItem('persist:cherry-studio', data.localStorage['persist:cherry-studio'])
+
+    // remove notes_tree from indexedDB
+    if (data.indexedDB['notes_tree']) {
+      delete data.indexedDB['notes_tree']
+    }
+
     await restoreDatabase(data.indexedDB)
 
     if (data.version === 3) {