Merge pull request #88 from harrysun2006/master

fixed few issues
2025-04-05 04:22:50 +08:00 · 2023-12-07 14:42:24 +08:00 · 2023-12-07 14:42:24 +08:00 · 4c3dd69540
commit 4c3dd69540
parent 685acd4233 b79af80f18
6 changed files with 150 additions and 69 deletions
--- a/.gitignore
+++ b/.gitignore
@ -27,4 +27,6 @@ yarn-error.log*
 yarn.lock
 package-lock.json

-.parcel-cache
+.parcel-cache
+.history
+test.mjs
--- a/electron/ipc.js
+++ b/electron/ipc.js
@ -35,12 +35,19 @@ export default function initIPC() {
    return result?.[0];
  });

-  ipcMain.handle('invoke_下载视频', async (event, { url, decodeKey, savePath }) => {
-    console.log(url, decodeKey);
+  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;
+    console.log('description:', description);
+    console.log("fileName:", fileName);
+    console.log("url:", url)
+    console.log("decodeKey:", decodeKey);
    return downloadFile(
      url,
      decodeKey,
-      `${savePath}/${Date.now()}.mp4`,
+      // `${savePath}/${Date.now()}.mp4`,
+      `${savePath}/${fileName}.mp4`,
      throttle(value => win?.webContents?.send?.('e_进度变化', value), 1000),
    ).catch(err => {
      console;
--- a/electron/proxyServer.js
+++ b/electron/proxyServer.js
@ -1,7 +1,9 @@
 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';
@ -11,62 +13,110 @@ if (process.platform === 'win32') {
  process.env.OPENSSL_CONF = CONFIG.OPEN_SSL_CNF_PATH;
 }

-const injection_script = `
-(function () {
-    if (window.wvds !== undefined) {
-        return
-    }
-    let receiver_url = "https://aaaa.com"
-    function send_response_if_is_video(response) {
-        if (response == undefined) {
-            return;
-        }
-        if (response["err_msg"] != "H5ExtTransfer:ok") {
-            return;
-        }
-        let value = JSON.parse(response["jsapi_resp"]["resp_json"]);
-        if (value["object"] == undefined || value["object"]["object_desc"] == undefined  || value["object"]["object_desc"]["media"].length == 0) {
-            return
-        }
-        let media = value["object"]["object_desc"]["media"][0]
-        let video_data = {
-            "decode_key": media["decode_key"],
-            "url": media["url"]+media["url_token"],
-            "size": media["file_size"],
-            "description":  value["object"]["object_desc"]["description"].trim(),
-            "uploader": value["object"]["nickname"]
-        }
-        fetch(receiver_url, {
-            method: 'POST',
-            mode: 'no-cors',
-            body: JSON.stringify(video_data),
-        }).then((resp) => {
-            console.log(\`video data for \${video_data["description"]} sent\`);
-        });
-    }
-    function wrapper(name,origin) {
-        console.log(\`injecting \${name}\`);
-        return function() {
-            let cmdName = arguments[0];
-            if (arguments.length == 3) {
-                let original_callback = arguments[2];
-                arguments[2] = async function () {
-                    if (arguments.length == 1) {
-                        send_response_if_is_video(arguments[0]);
-                    }
-                    return await original_callback.apply(this, arguments);
-                }
-            }
-            let result = origin.apply(this,arguments);
-            return result;
-        }
-    }
-    console.log(\`------- Invoke WechatVideoDownloader Service ---------\`);
-    window.WeixinJSBridge.invoke = wrapper("WeixinJSBridge.invoke",window.WeixinJSBridge.invoke);
-    window.wvds = true;
-})()
+const WVDS_DEBUG = process.env.WVDS_DEBUG !== undefined;
+
+const injection_html = `
+<script type="text/javascript" src="//res.wx.qq.com/t/wx_fed/finder/web/web-finder/res/js/wvds.inject.js"></script>
 `;

+// setTimeout to allow working in macOS
+// in windows: H5ExtTransfer:ok
+// in macOS: finderH5ExtTransfer:ok
+const injection_script = `
+setTimeout(() => {
+  if (window.wvds !== undefined) return;
+  ${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_btn1.style = "position:absolute;top:3px;left:20px;width:80px;height:30px;cursor:pointer;";
+  ele_btn1.text = "Source";
+  ele_btn1.onclick = () => {
+    var source = "<html>";
+    source += document.getElementsByTagName('html')[0].innerHTML;
+    source += "</html>";
+    debug_wvds(source);
+  };
+  ele_app.appendChild(ele_btn1);
+
+  ele_btn2.style = "position:absolute;top:3px;left:120px;width:80px;height:30px;cursor:pointer;";
+  ele_btn2.text = "Test";
+  ele_btn2.onclick = () => {
+    debug_wvds("Hello WeChatVideo Downloader!");
+  };
+  ele_app.appendChild(ele_btn2);
+
+  ele_debug.setAttribute("rows", "60");
+  ele_debug.setAttribute("cols", "60");
+  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) {
+    if (response == undefined) return;
+    // debug_wvds(JSON.stringify(response));
+    debug_wvds("send 1: " + response["err_msg"]);
+    if (!response["err_msg"].includes("H5ExtTransfer:ok")) return;
+    let value = JSON.parse(response["jsapi_resp"]["resp_json"]);
+    // debug_wvds("send 2: " + JSON.stringify(value));
+    if (value["object"] == undefined || value["object"]["object_desc"] == undefined  || value["object"]["object_desc"]["media"].length == 0) {
+      return;
+    }
+    let media = value["object"]["object_desc"]["media"][0];
+    // debug_wvds("send 3: " + JSON.stringify(media));
+    let description = value["object"]["object_desc"]["description"].trim();
+    debug_wvds("send x decode key: " + media["decode_key"] + " for " + description);
+    let video_data = {
+      "decode_key": media["decode_key"],
+      "url": media["url"]+media["url_token"],
+      "size": media["file_size"],
+      "description":  description,
+      "uploader": value["object"]["nickname"]
+    };
+    fetch(receiver_url, {
+      method: "POST",
+      mode: "no-cors",
+      body: JSON.stringify(video_data),
+    }).then((resp) => {
+      debug_wvds(\`video data for \${video_data["description"]} sent!\`);
+    });
+  }
+
+  function wrapper(name,origin) {
+    return function() {
+      let cmdName = arguments[0];
+      if (arguments.length == 3) {
+        let original_callback = arguments[2];
+        arguments[2] = async function () {
+          if (arguments.length == 1) {
+            debug_wvds("wrapper 3: " + JSON.stringify(arguments[0]));
+            send_response_if_is_video(arguments[0]);
+          }
+          return await original_callback.apply(this, arguments);
+        }
+        debug_wvds("wrapper 1: " + cmdName + ", " + typeof(arguments[1]) + ", " + typeof(arguments[2]));
+      } else {
+        debug_wvds("wrapper 2: " + cmdName + ", " + arguments.length + ", " + arguments[1] + ", " + typeof(arguments[2]));
+      }
+      let result = origin.apply(this,arguments);
+      return result;
+    }
+  }
+  window.WeixinJSBridge.invoke = wrapper("WeixinJSBridge.invoke", window.WeixinJSBridge.invoke);
+  window.wvds = true;
+  debug_wvds("Invoke WechatVideoDownloader Service!");
+}, 1000);`;
+
 export async function startServer({ win, setProxyErrorCallback = f => f }) {
  const port = await getPort();

@ -75,8 +125,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)
@ -97,6 +147,7 @@ export async function startServer({ win, setProxyErrorCallback = f => f }) {
        as: 'json',
      },
      (req, res) => {
+        console.log('request(aaaa.com):', req.json);
        res.string = 'ok';
        res.statusCode = 200;
        win?.webContents?.send?.('VIDEO_CAPTURE', req.json);
@ -106,15 +157,34 @@ export async function startServer({ win, setProxyErrorCallback = f => f }) {
    proxy.intercept(
      {
        phase: 'response',
-        hostname: 'res.wx.qq.com',
+        hostname: 'channels.weixin.qq.com',
        as: 'string',
      },
      async (req, res) => {
-        if (req.url.includes('polyfills.publish')) {
-          res.string = res.string + '\n' + injection_script;
+        if (req.url.includes('/web/pages/feed')) {
+          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);
        }
      },
    );
+
+    proxy.intercept(
+      {
+        phase: 'response',
+        hostname: 'res.wx.qq.com',
+        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;
+          res.statusCode = 200;
+        }
+      },
+    );
+
  });
 }

--- a/package.json
+++ b/package.json
@ -1,6 +1,6 @@
 {
  "name": "wechat-video-downloader",
-  "version": "2.0.0",
+  "version": "2.0.1",
  "description": "",
  "main": "./build-electron/index.js",
  "homepage": "./",
--- a/src/App.jsx
+++ b/src/App.jsx
@ -61,7 +61,7 @@ function App() {
                dataIndex: 'action',
                key: 'action',
                width: '210px',
-                render: (_, { url, decodeKey, hdUrl, fixUrl, fullFileName }) => (
+                render: (_, { url, decodeKey, hdUrl, fixUrl, description, fullFileName }) => (
                  <div>
                    {fullFileName ? (
                      <Button
@ -80,7 +80,7 @@ function App() {
                        icon={<DownloadOutlined />}
                        type="primary"
                        onClick={() => {
-                          send({ type: 'e_下载', url: hdUrl || url, decodeKey: decodeKey });
+                          send({ type: 'e_下载', url: hdUrl || url, decodeKey: decodeKey, description: description });
                        }}
                        size="small"
                      >
--- a/src/fsm.js
+++ b/src/fsm.js
@ -190,13 +190,14 @@ export default createMachine(
          .catch(() => send('e_取消'));
      },
      invoke_下载视频:
-        ({ currentUrl, savePath, decodeKey }) =>
+        ({ currentUrl, savePath, decodeKey, description }) => 
        send => {
          ipcRenderer
            .invoke('invoke_下载视频', {
              url: currentUrl,
-              decodeKey: decodeKey,
+              decodeKey,
              savePath,
+              description,
            })
            .then(({ fullFileName }) => {
              send({ type: 'e_下载完成', fullFileName, currentUrl });
@ -239,10 +240,11 @@ export default createMachine(
          captureList: [],
        };
      }),
-      action_设置当前地址: actions.assign((_, { url, decodeKey }) => {
+      action_设置当前地址: actions.assign((_, { url, decodeKey, description }) => {
        return {
          currentUrl: url,
          decodeKey: decodeKey,
+          description: description,
        };
      }),
      action_存储下载位置: actions.assign((_, { data }) => {