fix tiktok + download blob

Author: hoang.tran12

scripts/content-scripts/scripts/ufs_global_webpage_context.js

@@ -289,32 +289,37 @@ const UsefulScriptGlobalPageContext = {
         alert("Error: " + error);
       }
     },
-    async downloadBlobUrl(url, fileName, progressCallback) {
-      try {
-        const response = await fetch(url);
-        if (!response.ok) {
-          throw new Error(`Error: ${response.status} - ${response.statusText}`);
-        }
-        const contentLength = response.headers.get("content-length");
-        const total = parseInt(contentLength, 10);
-        let loaded = 0;
-        const reader = response.body.getReader();
-        const chunks = [];
+    async downloadBlobUrlWithProgress(url, fileName, progressCallback) {
+      const response = await fetch(url);
+      if (!response.ok) {
+        throw new Error(`Error: ${response.status} - ${response.statusText}`);
+      }
+      const contentLength = response.headers.get("content-length");
+      const total = parseInt(contentLength, 10);
+      let loaded = 0;
+      const reader = response.body.getReader();
+      const chunks = [];
 
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          loaded += value.byteLength;
-          progressCallback?.(loaded, total);
-          chunks.push(value);
-        }
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        loaded += value.byteLength;
+        progressCallback?.(loaded, total);
+        chunks.push(value);
+      }
 
-        const blob = new Blob(chunks, {
-          type: response.headers.get("content-type"),
-        });
-        UsefulScriptGlobalPageContext.Utils.downloadBlob(blob, fileName);
-      } catch (error) {
-        alert("Error: " + error);
+      const blob = new Blob(chunks, {
+        type: response.headers.get("content-type"),
+      });
+      UsefulScriptGlobalPageContext.Utils.downloadBlob(blob, fileName);
+    },
+    async downloadBlobUrl(url, title) {
+      try {
+        let res = await fetch(url);
+        let blob = await res.blob();
+        UsefulScriptGlobalPageContext.Utils.downloadBlob(blob, title);
+      } catch (e) {
+        alert("Error: " + e);
       }
     },
     downloadBlob(blob, filename) {

scripts/tiktok_downloadWatchingVideo.js

@@ -13,36 +13,47 @@ export default {
 
   onClickExtension: async function () {
     const { closeLoading, setLoadingText } = showLoading("Đang lấy video id..");
-    try {
-      const videoIds = await shared.getListVideoIdInWebsite();
-      if (!videoIds.length) throw Error("Không tìm thấy video nào");
-      else {
-        setLoadingText(`Đang tìm link tải video ${videoIds[0]}...`);
-        const link =
-          await UsefulScriptGlobalPageContext.Tiktok.downloadTiktokVideoFromId(
+
+    const getLinkFuncs = [
+      async () => {
+        setLoadingText("Đang tìm videoid...");
+        const videoIds = await shared.getListVideoIdInWebsite();
+        if (videoIds.length) {
+          setLoadingText(`Đang tìm link tải video ${videoIds[0]}...`);
+          return await UsefulScriptGlobalPageContext.Tiktok.downloadTiktokVideoFromId(
             videoIds[0]
           );
-        if (link) {
-          // await UsefulScriptGlobalPageContext.Utils.downloadBlobUrl(
-          //   link,
-          //   "video.mp4",
-          //   (loaded, total) => {
-          //     let loadedMB = ~~(loaded / 1024 / 1024);
-          //     let totalMB = ~~(total / 1024 / 1024);
-          //     let percent = ((loaded / total) * 100) | 0;
-          //     setLoadingText(
-          //       `Đang tải video... (${loadedMB}/${totalMB}MB - ${percent}%)`
-          //     );
-          //   }
-          // );
-          window.open(link);
-        } else alert("Không tìm được video đang xem");
+        }
+      },
+
+      async () => {
+        setLoadingText("Đang tìm video url từ DOM...");
+        return await runScriptInCurrentTab(
+          async () =>
+            await UsefulScriptGlobalPageContext.DOM.getWatchingVideoSrc()
+        );
+      },
+    ];
+
+    let link;
+    for (let func of getLinkFuncs) {
+      try {
+        link = await func();
+        if (link) break;
+      } catch (e) {
+        alert("lol");
       }
-    } catch (e) {
-      alert("ERROR: " + e);
-    } finally {
-      closeLoading();
     }
+
+    if (!link) alert("Không tìm được link video");
+    else {
+      UsefulScriptGlobalPageContext.Utils.downloadBlobUrl(
+        link,
+        "tiktok_video.mp4"
+      );
+    }
+
+    closeLoading();
   },
 };