fanxb/open-renamer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 16 Wiki Activity

feat：优化季识别

Browse Source
This commit is contained in:
fanxb 2023-04-09 21:50:37 +08:00
parent 11f28f12fa
commit 9a55460f11
3 changed files with 105 additions and 88 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
openRenamerBackend/entity/bo/rules/AutoRule.ts
View File

@ -1,6 +1,7 @@
import RuleInterface from "./RuleInterface";
import FileObj from "../../vo/FileObj";
import path from 'path';
import {getSeason} from "../../../util/MediaUtil";
let pattern = new RegExp(/s(eason)?(\d+)/);
@ -35,14 +36,12 @@ export default class InsertRule implements RuleInterface {
deal(file: FileObj): void {
//识别到的内容
let getStr = null;
let patternRes = path.basename(file.path).replace(/[ ]+/, "").toLocaleLowerCase().match(pattern);
let season = getSeason(path.basename(file.path));
if (this.type === 'season') {
if (patternRes && patternRes[2]) {
getStr = patternRes[2];
}
getStr = season;
} else if (this.type === 'name') {
let originName = null;
if (patternRes && patternRes[2]) {
if (season && season.length > 0) {
//说明是剧集,取父文件夹的父文件夹名称
originName = path.basename(path.resolve(file.path, '..'));
} else {

163
openRenamerBackend/entity/bo/rules/InsertRule.ts
View File

@ -1,98 +1,95 @@
import RuleInterface from "./RuleInterface";
import FileObj from "../../vo/FileObj";
import path from 'path';
import {getSeason} from "../../../util/MediaUtil";
let pattern = new RegExp(/s(eason)?(\d+)/);
export default class InsertRule implements RuleInterface {
/**
* 插入内容
*/
insertContent: string;
/**
* 操作类别，front：前缀，backend：后缀，at:位置，replace:替换当前文件名
*/
type: string;
/**
* 当type为at,时的位置,从1开始
*/
atInput: number;
/**
* 当type为at,时的方向，true:从右到左，false:从左到右
*/
atIsRightToleft: boolean;
/**
* 忽略拓展名，true:忽略，false：不忽略
*/
ignorePostfix: boolean;
/**
自动识别季号
*/
autoSeason: boolean;
/**
后缀过滤是否开启
*/
endFilter: boolean;
/**
有效后缀
*/
validEnd: Array<String>;
/**
* 插入内容
*/
insertContent: string;
/**
* 操作类别，front：前缀，backend：后缀，at:位置，replace:替换当前文件名
*/
type: string;
/**
* 当type为at,时的位置,从1开始
*/
atInput: number;
/**
* 当type为at,时的方向，true:从右到左，false:从左到右
*/
atIsRightToleft: boolean;
/**
* 忽略拓展名，true:忽略，false：不忽略
*/
ignorePostfix: boolean;
/**
自动识别季号
*/
autoSeason: boolean;
/**
后缀过滤是否开启
*/
endFilter: boolean;
/**
有效后缀
*/
validEnd: Array<String>;
constructor(data: any) {
this.insertContent = data.insertContent;
this.type = data.type;
this.atInput = data.atInput;
this.atIsRightToleft = data.atIsRightToleft;
this.ignorePostfix = data.ignorePostfix;
this.autoSeason = data.autoSeason;
this.endFilter = data.endFilter;
this.validEnd = data.validEnd;
}
constructor(data: any) {
this.insertContent = data.insertContent;
this.type = data.type;
this.atInput = data.atInput;
this.atIsRightToleft = data.atIsRightToleft;
this.ignorePostfix = data.ignorePostfix;
this.autoSeason = data.autoSeason;
this.endFilter = data.endFilter;
this.validEnd = data.validEnd;
}
deal(file: FileObj): void {
if (this.endFilter && file.expandName.length > 0 && this.validEnd.indexOf(file.expandName.substring(1)) == -1) {
//拓展名不符，跳过
return;
}
let str = this.ignorePostfix ? file.realName : file.name;
let season = '';
deal(file: FileObj): void {
if (this.endFilter && file.expandName.length > 0 && this.validEnd.indexOf(file.expandName.substring(1)) == -1) {
//拓展名不符，跳过
return;
}
let str = this.ignorePostfix ? file.realName : file.name;
let season = '';
if (this.autoSeason) {
let patternRes = path.basename(file.path).replace(/[ ]+/, "").toLocaleLowerCase().match(pattern);
if (patternRes && patternRes[2]) {
season = patternRes[2];
}
}
switch (this.type) {
case "front":
str = this.insertContent + season + str;
break;
case "backend":
str = str + this.insertContent + season;
break;
case "at":
let index = this.atIsRightToleft ? str.length - this.atInput + 1 : this.atInput - 1;
str = str.substring(0, index) + this.insertContent + season + str.substring(index);
break;
case "replace":
str = this.insertContent + season;
break;
}
if (this.autoSeason) {
season = getSeason(path.basename(file.path));
}
switch (this.type) {
case "front":
str = this.insertContent + season + str;
break;
case "backend":
str = str + this.insertContent + season;
break;
case "at":
let index = this.atIsRightToleft ? str.length - this.atInput + 1 : this.atInput - 1;
str = str.substring(0, index) + this.insertContent + season + str.substring(index);
break;
case "replace":
str = this.insertContent + season;
break;
}
if (this.ignorePostfix) {
file.realName = str;
} else {
file.expandName = path.extname(str);
if (file.expandName.length > 0) {
file.realName = str.substring(0, str.lastIndexOf("."));
} else {
file.realName = str;
}
}
if (this.ignorePostfix) {
file.realName = str;
} else {
file.expandName = path.extname(str);
if (file.expandName.length > 0) {
file.realName = str.substring(0, str.lastIndexOf("."));
} else {
file.realName = str;
}
}
file.name = file.realName + file.expandName;
}
file.name = file.realName + file.expandName;
}
}

21
openRenamerBackend/util/MediaUtil.ts
View File

@ -1,3 +1,4 @@
const path = require("path")
const videoSet = new Set(["flv", 'avi', 'wmv', 'dat', 'vob', 'mpg', 'mpeg', 'mp4', '3gp', '3g2', 'mkv', 'rm', 'rmvb', 'mov', 'qt', 'ogg', 'ogv', 'oga', 'mod']);
/**
@ -33,4 +34,24 @@ export function isNfo(str: string) {
return false;
}
return "nfo" == str;
}
let pattern1 = new RegExp(/s(eason)?\.?(\d+)/);
let pattern2 = new RegExp(/(\d+)/);
/**
* 识别季号
* @param str
*/
export function getSeason(name: string) {
name = name.replace(/[ ]+/, "").toLocaleLowerCase();
let patternRes = name.match(pattern1);
if (patternRes && patternRes[2]) {
return patternRes[2];
}
patternRes = name.match(pattern2);
if (patternRes && patternRes[1]) {
return patternRes[1];
}
return "";
}