From e4c4a6f8d278ae88d50cb030e04f8f2dd582ccd2 Mon Sep 17 00:00:00 2001
From: Kuingsmile <ma_shiqing@163.com>
Date: Tue, 30 Apr 2024 14:31:33 +0800
Subject: [PATCH] :sparkles: Feature(custom): optimize upload api

---
 src/main/server/routerManager.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/main/server/routerManager.ts b/src/main/server/routerManager.ts
index 0b0c44a..bf85388 100644
--- a/src/main/server/routerManager.ts
+++ b/src/main/server/routerManager.ts
@@ -46,6 +46,7 @@ router.post('/upload', async ({
   urlparams?: URLSearchParams
 }): Promise<void> => {
   try {
+    const picbed = urlparams?.get('picbed')
     const passedKey = urlparams?.get('key')
     const serverKey = picgo.getConfig<string>(configPaths.settings.serverKey) || ''
     if (serverKey && passedKey !== serverKey) {
@@ -58,17 +59,16 @@ router.post('/upload', async ({
       })
       return
     }
-    const picbed = urlparams?.get('picbed')
     let currentPicBedType = ''
     let currentPicBedConfig = {} as IStringKeyMap
     let currentPicBedConfigId = ''
     let needRestore = false
     if (picbed) {
-      const configName = urlparams?.get('configName') || 'Default'
       const currentPicBed = picgo.getConfig<IStringKeyMap>('picBed') || {} as IStringKeyMap
-      currentPicBedType = currentPicBed?.current
-      currentPicBedConfig = currentPicBed?.[currentPicBedType]
-      currentPicBedConfigId = currentPicBedConfig?._id
+      currentPicBedType = currentPicBed.current || ''
+      currentPicBedConfig = currentPicBed[currentPicBedType] || {} as IStringKeyMap
+      currentPicBedConfigId = currentPicBedConfig._id
+      const configName = urlparams?.get('configName') || currentPicBed[picbed]?._configName
       if (picbed === currentPicBedType && configName === currentPicBedConfig._configName) {
         // do nothing
       } else {