sanitizeName

Author: HoangTran0410

scripts/content-scripts/ufs_global.js

@@ -32,6 +32,7 @@ export const UfsGlobal = {
     getWatchingVideoSrc,
   },
   Utils: {
+    sanitizeName,
     debounce,
     makeUrlValid,
     getNumberFormatter,
@@ -517,16 +518,48 @@ function getWatchingVideoSrc() {
     }
   }
 }
+// #endregion
+
+// #region Utils
+
+// https://github.com/parshap/node-sanitize-filename/blob/master/index.js
+// https://github.com/Dinoosauro/tiktok-to-ytdlp/blob/main/script.js
+function sanitizeName(name, modifyIfPosible = true) {
+  if (typeof name !== "string") {
+    throw new Error("Input must be string");
+  }
+  const replacement = "";
+  const illegalRe = /[\/\?<>\\:\*\|"]/g;
+  const controlRe = /[\x00-\x1f\x80-\x9f]/g;
+  const reservedRe = /^\.+$/;
+  const windowsReservedRe = /^(con|prn|aux|nul|com[0-9]|lpt[0-9])(\..*)?$/i;
+  const windowsTrailingRe = /[\. ]+$/;
+  if (modifyIfPosible) {
+    name = name
+      .replaceAll("<", "‹")
+      .replaceAll(">", "›")
+      .replaceAll(":", "∶")
+      .replaceAll('"', "″")
+      .replaceAll("/", "∕")
+      .replaceAll("\\", "∖")
+      .replaceAll("|", "¦")
+      .replaceAll("?", "¿");
+  }
+  const sanitized = name
+    .replace(illegalRe, replacement)
+    .replace(controlRe, replacement)
+    .replace(reservedRe, replacement)
+    .replace(windowsReservedRe, replacement)
+    .replace(windowsTrailingRe, replacement);
+  return sanitized; // TODO truncates to length of 255
+}
 function debounce(func, delay) {
   let timeout;
   return (...args) => {
     clearTimeout(timeout);
     timeout = setTimeout(() => func.apply(this, args), delay);
   };
 }
-// #endregion
-
-// #region Utils
 
 const numberFormatCached = {};
 /**

scripts/tiktok_batchDownload.js

@@ -224,7 +224,7 @@ export default {
         return checkbox;
       }
 
-      const sleep = UfsGlobal.Utils.sleep;
+      const { sleep, sanitizeName } = UfsGlobal.Utils;
 
       async function getLinkVideos(videoUrls) {
         if (!videoUrls.length) return;
@@ -263,14 +263,14 @@ export default {
               //   conflictAction: "overwrite",
               //   filename:
               //     "tiktok/" +
-              //     sanitizeFileName(
+              //     sanitizeName(
               //       CACHED.videoById.get(id)?.name || id
               //     ) +
               //     ".mp4",
               // });
               links.push({
                 url: link,
-                name: sanitizeFileName(cached?.name || id) + ".mp4",
+                name: sanitizeName(cached?.name || id) + ".mp4",
               });
             } else {
               progressDiv.innerText = `[LỖI] Không thể tải video ${url}.`;
@@ -337,19 +337,3 @@ export default {
     },
   },
 };
-
-function sanitizeFileName(fileName) {
-  // Định nghĩa các ký tự hợp lệ (chữ cái, số, dấu gạch dưới, dấu gạch ngang và dấu chấm)
-  const validChars = /^[a-zA-Z0-9_\-.]+$/;
-
-  // Lược bỏ các ký tự không hợp lệ
-  let sanitizedFileName = "";
-  for (let char of fileName) {
-    if (validChars.test(char)) {
-      sanitizedFileName += char;
-    }
-  }
-
-  // Trả về tên tệp đã lược bỏ các ký tự không hợp lệ
-  return sanitizedFileName;
-}