From 4fbf3fbba82abe1fc462d43dd70d66ab3b504587 Mon Sep 17 00:00:00 2001
From: lecepin <383810086@qq.com>
Date: Thu, 7 Dec 2023 20:27:40 +0800
Subject: [PATCH] chore:

---
 electron/ipc.js         | 13 ++++++-------
 electron/proxyServer.js | 41 +++++++++++++++++++++++++++++------------
 src/App.jsx             |  7 ++++++-
 src/fsm.js              |  2 +-
 4 files changed, 42 insertions(+), 21 deletions(-)

diff --git a/electron/ipc.js b/electron/ipc.js
index 26519c8..50cfdaa 100644
--- a/electron/ipc.js
+++ b/electron/ipc.js
@@ -36,17 +36,16 @@ export default function initIPC() {
   });
 
   ipcMain.handle('invoke_下载视频', async (event, { url, decodeKey, savePath, description }) => {
-    let re = /(?:^|\s)(?!#\S+)\s*([^#\s]*)/;
-    let m = description.match(re);
-    let fileName = m && m.length > 1 ? m[1].replaceAll(/[,，]/g, "_") : description;
+    let fileName = description?.replaceAll?.(/\\|\/|:|\*|\?|"|<|>|\|/g, '') || Date.now();
+
     console.log('description:', description);
-    console.log("fileName:", fileName);
-    console.log("url:", url)
-    console.log("decodeKey:", decodeKey);
+    console.log('fileName:', fileName);
+    console.log('url:', url);
+    console.log('decodeKey:', decodeKey);
+
     return downloadFile(
       url,
       decodeKey,
-      // `${savePath}/${Date.now()}.mp4`,
       `${savePath}/${fileName}.mp4`,
       throttle(value => win?.webContents?.send?.('e_进度变化', value), 1000),
     ).catch(err => {
diff --git a/electron/proxyServer.js b/electron/proxyServer.js
index ff604c6..0a74827 100644
--- a/electron/proxyServer.js
+++ b/electron/proxyServer.js
@@ -1,9 +1,7 @@
 import fs from 'fs';
-import os from 'os';
 import hoxy from 'hoxy';
 import getPort from 'get-port';
 import log from 'electron-log';
-import md5 from 'md5';
 import { app } from 'electron';
 import CONFIG from './const';
 import { setProxy, closeProxy } from './setProxy';
@@ -25,18 +23,24 @@ const injection_html = `
 const injection_script = `
 setTimeout(() => {
   if (window.wvds !== undefined) return;
-  ${WVDS_DEBUG ? `
+  ${
+    WVDS_DEBUG
+      ? `
   document.body.style.border = "2px solid #0000FF";
   let ele_app = document.getElementById("app");
   let ele_btn1 = document.createElement("a");
   let ele_btn2 = document.createElement("a");
   let ele_debug = document.createElement("textarea");
-  ` : ""}
+  `
+      : ''
+  }
   function debug_wvds(msg) {
-    ${WVDS_DEBUG ? `ele_debug.value += "\\n" + msg;` : ""}
+    ${WVDS_DEBUG ? `ele_debug.value += "\\n" + msg;` : ''}
   }
 
-  ${WVDS_DEBUG ? `
+  ${
+    WVDS_DEBUG
+      ? `
   ele_btn1.style = "position:absolute;top:3px;left:20px;width:80px;height:30px;cursor:pointer;";
   ele_btn1.text = "Source";
   ele_btn1.onclick = () => {
@@ -59,7 +63,9 @@ setTimeout(() => {
   ele_debug.style = "position:absolute;top:600px;left:20px;width:600px;height:300px;border:2px solid #FF00FF;";
   ele_debug.value = "Debug:\\n";
   ele_app.appendChild(ele_debug);
-  ` : ""}
+  `
+      : ''
+  }
   let receiver_url = "https://aaaa.com";
 
   function send_response_if_is_video(response) {
@@ -125,8 +131,8 @@ export async function startServer({ win, setProxyErrorCallback = f => f }) {
       .createServer({
         certAuthority: {
           key: fs.readFileSync(CONFIG.CERT_PRIVATE_PATH),
-          cert: fs.readFileSync(CONFIG.CERT_PUBLIC_PATH)
-        }
+          cert: fs.readFileSync(CONFIG.CERT_PUBLIC_PATH),
+        },
       })
       .listen(port, () => {
         setProxy('127.0.0.1', port)
@@ -162,7 +168,7 @@ export async function startServer({ win, setProxyErrorCallback = f => f }) {
       },
       async (req, res) => {
         if (req.url.includes('/web/pages/feed')) {
-          res.string = res.string.replace('</body>', injection_html + "\n</body>");
+          res.string = res.string.replace('</body>', injection_html + '\n</body>');
           res.statusCode = 200;
           console.log('inject[channels.weixin.qq.com]:', req.url, res.string.length);
         }
@@ -173,10 +179,9 @@ export async function startServer({ win, setProxyErrorCallback = f => f }) {
       {
         phase: 'response',
         hostname: 'res.wx.qq.com',
-        as: 'string'
+        as: 'string',
       },
       async (req, res) => {
-        // console.log('response[res.wx.qq.com]:', req.url);
         if (req.url.includes('wvds.inject.js')) {
           console.log('inject[res.wx.qq.com]:', req.url, res.string.length);
           res.string = injection_script;
@@ -185,6 +190,18 @@ export async function startServer({ win, setProxyErrorCallback = f => f }) {
       },
     );
 
+    proxy.intercept(
+      {
+        phase: 'response',
+        hostname: 'res.wx.qq.com',
+        as: 'string',
+      },
+      async (req, res) => {
+        if (req.url.includes('polyfills.publish')) {
+          res.string = res.string + '\n' + injection_script;
+        }
+      },
+    );
   });
 }
 
diff --git a/src/App.jsx b/src/App.jsx
index c42babc..e322338 100644
--- a/src/App.jsx
+++ b/src/App.jsx
@@ -80,7 +80,12 @@ function App() {
                         icon={<DownloadOutlined />}
                         type="primary"
                         onClick={() => {
-                          send({ type: 'e_下载', url: hdUrl || url, decodeKey: decodeKey, description: description });
+                          send({
+                            type: 'e_下载',
+                            url: hdUrl || url,
+                            decodeKey: decodeKey,
+                            description: description,
+                          });
                         }}
                         size="small"
                       >
diff --git a/src/fsm.js b/src/fsm.js
index f345a5d..bad52ec 100644
--- a/src/fsm.js
+++ b/src/fsm.js
@@ -190,7 +190,7 @@ export default createMachine(
           .catch(() => send('e_取消'));
       },
       invoke_下载视频:
-        ({ currentUrl, savePath, decodeKey, description }) => 
+        ({ currentUrl, savePath, decodeKey, description }) =>
         send => {
           ipcRenderer
             .invoke('invoke_下载视频', {