🐛 Fix(custom): fix compatiblity with typora

Kuingsmile · Apr 5, 2024 · 762913c · 762913c
1 parent e6e9472
commit 762913c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/main/server/index.ts b/src/main/server/index.ts
@@ -92,6 +92,17 @@ class Server {
         }
       })
     } else {
+      const remoteAddress = request.socket.remoteAddress || 'unknown'
+      logger.info('[PicList Server] get a POST request from IP:', remoteAddress)
+      let urlSP = query ? new URLSearchParams(query) : undefined
+      if (remoteAddress === '::1' || remoteAddress === '127.0.0.1') {
+        const serverKey = picgo.getConfig<string>('settings.serverKey') || ''
+        if (urlSP) {
+          urlSP.set('key', serverKey)
+        } else {
+          urlSP = new URLSearchParams('key=' + serverKey)
+        }
+      }
       if (request.headers['content-type'] && request.headers['content-type'].startsWith('multipart/form-data')) {
         // @ts-ignore
         uploadMulter.any()(request, response, (err: any) => {
@@ -113,7 +124,7 @@ class Server {
             handler({
               list,
               response,
-              urlparams: query ? new URLSearchParams(query) : undefined
+              urlparams: urlSP
             })
           }
         })
@@ -141,7 +152,7 @@ class Server {
           handler!({
             ...postObj,
             response,
-            urlparams: query ? new URLSearchParams(query) : undefined
+            urlparams: urlSP
           })
         })
       }

diff --git a/src/main/server/routerManager.ts b/src/main/server/routerManager.ts
@@ -46,7 +46,7 @@ router.post('/upload', async ({
 }): Promise<void> => {
   try {
     const passedKey = urlparams?.get('key')
-    const serverKey = picgo.getConfig('settings.serverKey') || ''
+    const serverKey = picgo.getConfig<string>('settings.serverKey') || ''
     if (serverKey && passedKey !== serverKey) {
       handleResponse({
         response,