feat:增加优化

openRenamerBackend/entity/vo/FileObj.ts

@@ -1,55 +1,71 @@
 import * as pathUtil from "path";
+import {isVideo, isSub, isNfo} from "../../util/MediaUtil"
+
 export default class FileObj {
-	/**
+    /**
-	 * 文件名
+     * 文件名
-	 */
+     */
-	name: string;
+    name: string;
-	/**
+    /**
-	原始名字
+     原始名字
-	 */
+     */
-	originName: string;
+    originName: string;
-	/**
+    /**
-	 * 拓展名
+     * 拓展名
-	 */
+     */
-	expandName: string;
+    expandName: string;
-	/** 
+    /**
-	 * 去掉拓展名后的名字
+     * 去掉拓展名后的名字
-	 */
+     */
-	realName: string;
+    realName: string;
-	/**
+    /**
-	 * 所属路径
+     * 所属路径
-	 */
+     */
-	path: string;
+    path: string;
-	/**
+    /**
-	 * 是否文件夹
+     * 是否文件夹
-	 */
+     */
-	isFolder: boolean;
+    isFolder: boolean;
-	/**
+    /**
-	 * 重命名错误原因
+     * 文件大小
-	 */
+     */
-	errorMessage: string;
+    size: number;
-	/**
+    /**
-	 * 创建时间ms
+     * 重命名错误原因
-	 */
+     */
-	createdTime: number;
+    errorMessage: string;
-	/**
+    /**
-	 * 更新时间ms
+     * 创建时间ms
-	 */
+     */
-	updatedTime: number;
+    createdTime: number;
+    /**
+     * 更新时间ms
+     */
+    updatedTime: number;
+    /**
+     * 是否广告文件
+     */
+    isAdFile: boolean;
 
 
-	constructor(name: string, path, isFolder, createdTime, updatedTime) {
+    constructor(name: string, path, isFolder, size: number, createdTime, updatedTime) {
-		this.name = name;
+        this.name = name;
-		this.originName = name;
+        this.originName = name;
-		this.expandName = pathUtil.extname(name);
+        this.expandName = pathUtil.extname(name);
-		if (this.expandName.length > 0) {
+        if (this.expandName.length > 0) {
-			this.realName = name.substring(0, name.lastIndexOf("."));
+            this.realName = name.substring(0, name.lastIndexOf("."));
-		} else {
+            let end = this.expandName.toLowerCase().replace(".", "");
-			this.realName = name;
+            if (isVideo(end)) {
-		}
+                this.isAdFile = size < 5 * 1024 * 1024;
-		this.path = path;
+            } else this.isAdFile = !(isSub(end) || isNfo(end));
-		this.isFolder = isFolder;
+        } else {
-		this.createdTime = createdTime;
+            this.realName = name;
-		this.updatedTime = updatedTime;
+        }
-	}
+        this.path = path;
+        this.isFolder = isFolder;
+        this.size = size;
+        this.createdTime = createdTime;
+        this.updatedTime = updatedTime;
+
+    }
 }

openRenamerBackend/service/FileService.ts

@@ -42,12 +42,12 @@ class FileService {
                 if (fileList[index].startsWith('.')) {
                     if (showHidden) {
                         (stat.isDirectory() ? folderList : files).push(
-                            new FileObj(fileList[index], pathStr, stat.isDirectory(), stat.birthtime.getTime(), stat.mtime.getTime()),
+                            new FileObj(fileList[index], pathStr, stat.isDirectory(), stat.size, stat.birthtime.getTime(), stat.mtime.getTime()),
                         );
                     }
                 } else {
                     (stat.isDirectory() ? folderList : files).push(
-                        new FileObj(fileList[index], pathStr, stat.isDirectory(), stat.birthtime.getTime(), stat.mtime.getTime()),
+                        new FileObj(fileList[index], pathStr, stat.isDirectory(), stat.size, stat.birthtime.getTime(), stat.mtime.getTime()),
                     );
                 }
             } catch (e) {
@@ -82,7 +82,7 @@ class FileService {
                 let filePath = path.join(file.path, file.name);
                 let temp = (await fs.readdir(filePath)).map(item => {
                     let stat = fs.statSync(path.join(filePath, item));
-                    return new FileObj(item, filePath, stat.isDirectory(), stat.birthtime.getTime(), stat.mtime.getTime());
+                    return new FileObj(item, filePath, stat.isDirectory(), stat.size, stat.birthtime.getTime(), stat.mtime.getTime());
                 });
                 await FileService.readDirRecursion(res, temp, depth + 1);
             }

openRenamerBackend/util/MediaUtil.ts

@@ -1,23 +1,36 @@
 const videoSet = new Set(["flv", 'avi', 'wmv', 'dat', 'vob', 'mpg', 'mpeg', 'mp4', '3gp', '3g2', 'mkv', 'rm', 'rmvb', 'mov', 'qt', 'ogg', 'ogv', 'oga', 'mod']);
+
 /**
  * 判断文件后缀是否为视频类型
  * @param str 文件后缀
  */
 export function isVideo(str: string) {
-	if (!str) {
+    if (!str) {
-		return false;
+        return false;
-	}
+    }
-	return videoSet.has(str.toLowerCase());
+    return videoSet.has(str.toLowerCase());
 }
 
 const subSet = new Set(['sub', 'sst', 'son', 'srt', 'ssa', 'ass', 'smi', 'psb', 'pjs', 'stl', 'tts', 'vsf', 'zeg']);
+
 /**
  * 判断文件是否为字幕文件
  * @param str 文件后缀
  */
 export function isSub(str: string) {
-	if (!str) {
+    if (!str) {
-		return false;
+        return false;
-	}
+    }
-	return subSet.has(str.toLowerCase());
+    return subSet.has(str.toLowerCase());
+}
+
+/**
+ * 判断文件是否为字幕文件
+ * @param str 文件后缀
+ */
+export function isNfo(str: string) {
+    if (!str) {
+        return false;
+    }
+    return "nfo" == str;
 }

openRenamerFront/src/views/home/Home.vue

@@ -24,7 +24,7 @@
       <div style="margin-top: 5px">
         <el-button type="primary" size="small" @click="selectAllFiles">反选</el-button>
         <el-tooltip effect="dark" content="一键选中所有的非视频、字幕文件和小于5MB的视频文件" placement="bottom">
-          <el-button type="success" size="small" @click="">一键选择</el-button>
+          <el-button type="success" size="small" @click="choseAdFile">一键选择</el-button>
         </el-tooltip>
         <el-tooltip effect="dark" content="移除（非删除）需要重命名的文件" placement="bottom">
           <el-button type="warning" size="small" @click="removeCheckedFiles">移除</el-button>
@@ -199,7 +199,6 @@ export default {
       this.newName = list[0].name;
       this.currentEditFile = list[0];
       this.showNameEditDialog = true;
-      await this.showResult();
     },
     async doEditFile() {
       if (!this.newName) {
@@ -282,6 +281,9 @@ export default {
     async refreshSavePathList() {
       this.savePathList = await HttpUtil.get("/file/path");
     },
+    async choseAdFile() {
+      this.fileList.forEach(item => item.checked = item.isAdFile);
+    }
   },
 };