62 lines
1.5 KiB
TypeScript
62 lines
1.5 KiB
TypeScript
|
import RuleInterface from "./RuleInterface";
|
|||
|
|
import FileObj from "../../vo/FileObj";
|
|||
|
|
import path from 'path';
|
|||
|
|
|
|||
|
|
|
|||
|
|
let pattern = new RegExp(/s(eason)?(\d+)/);
|
|||
|
|
let charSet = new Set([' ', '[', '.', '(', '（']);
|
|||
|
|
export default class InsertRule implements RuleInterface {
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 识别类型，season：季号，name：剧名/电影名识别
|
|||
|
|
*/
|
|||
|
|
type: string;
|
|||
|
|
/**
|
|||
|
|
* 前面追加
|
|||
|
|
*/
|
|||
|
|
frontAdd: string;
|
|||
|
|
/**
|
|||
|
|
* 后面追加
|
|||
|
|
*/
|
|||
|
|
endAdd: string;
|
|||
|
|
|
|||
|
|
constructor(data: any) {
|
|||
|
|
this.type = data.type;
|
|||
|
|
this.frontAdd = data.frontAdd;
|
|||
|
|
this.endAdd = data.endAdd;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
deal(file: FileObj): void {
|
|||
|
|
//识别到的内容
|
|||
|
|
let getStr = null;
|
|||
|
|
let patternRes = path.basename(file.path).replace(/[ ]+/, "").toLocaleLowerCase().match(pattern);
|
|||
|
|
if (this.type === 'season') {
|
|||
|
|
if (patternRes && patternRes[2]) {
|
|||
|
|
getStr = patternRes[2];
|
|||
|
|
}
|
|||
|
|
} else if (this.type === 'name') {
|
|||
|
|
let originName = null;
|
|||
|
|
if (patternRes && patternRes[2]) {
|
|||
|
|
//说明是剧集,取父文件夹的父文件夹名称
|
|||
|
|
originName = path.basename(path.resolve(file.path, '..'));
|
|||
|
|
} else {
|
|||
|
|
//说明是电影
|
|||
|
|
originName = path.basename(file.path);
|
|||
|
|
}
|
|||
|
|
getStr = '';
|
|||
|
|
for (let i = 0; i < originName.length; i++) {
|
|||
|
|
let char = originName.charAt(i);
|
|||
|
|
if (charSet.has(char)) {
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
getStr += char;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (getStr && getStr.length > 0) {
|
|||
|
|
file.realName = file.realName + this.frontAdd + getStr + this.endAdd;
|
|||
|
|
file.name = file.realName + file.expandName;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|