From f1ca8a5b7a483c5a30b6252526b57d984c9604a8 Mon Sep 17 00:00:00 2001
From: pompurin404 <pompurin404@mihomo.party>
Date: Thu, 12 Sep 2024 18:57:24 +0800
Subject: [PATCH] fix override update logic

---
 .../src/components/override/edit-info-modal.tsx     |  2 ++
 .../src/components/override/override-item.tsx       | 13 +++++++++----
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/renderer/src/components/override/edit-info-modal.tsx b/src/renderer/src/components/override/edit-info-modal.tsx
index ba9a5e4..f573ed4 100644
--- a/src/renderer/src/components/override/edit-info-modal.tsx
+++ b/src/renderer/src/components/override/edit-info-modal.tsx
@@ -10,6 +10,7 @@ import {
 } from '@nextui-org/react'
 import React, { useState } from 'react'
 import SettingItem from '../base/base-setting-item'
+import { restartCore } from '@renderer/utils/ipc'
 interface Props {
   item: IOverrideItem
   updateOverrideItem: (item: IOverrideItem) => Promise<void>
@@ -21,6 +22,7 @@ const EditInfoModal: React.FC<Props> = (props) => {
 
   const onSave = async (): Promise<void> => {
     await updateOverrideItem(values)
+    await restartCore()
     onClose()
   }
 
diff --git a/src/renderer/src/components/override/override-item.tsx b/src/renderer/src/components/override/override-item.tsx
index 7531d9a..8d4622b 100644
--- a/src/renderer/src/components/override/override-item.tsx
+++ b/src/renderer/src/components/override/override-item.tsx
@@ -16,7 +16,7 @@ import EditInfoModal from './edit-info-modal'
 import { useSortable } from '@dnd-kit/sortable'
 import { CSS } from '@dnd-kit/utilities'
 import ExecLogModal from './exec-log-modal'
-import { openFile } from '@renderer/utils/ipc'
+import { openFile, restartCore } from '@renderer/utils/ipc'
 
 interface Props {
   info: IOverrideItem
@@ -186,11 +186,16 @@ const OverrideItem: React.FC<Props> = (props) => {
                   variant="light"
                   color="default"
                   disabled={updating}
-                  onPress={() => {
+                  onPress={async () => {
                     setUpdating(true)
-                    addOverrideItem(info).finally(() => {
+                    try {
+                      await addOverrideItem(info)
+                      await restartCore()
+                    } catch (e) {
+                      alert(e)
+                    } finally {
                       setUpdating(false)
-                    })
+                    }
                   }}
                 >
                   <IoMdRefresh