open-renamer/openRenamerBackend/entity/bo/rules/AutoRule.ts

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+)/);
let eNumPatternArr = [new RegExp(/ep?(\d+)/), new RegExp(/[\(\[（](\d+)[\)\]）]/), new RegExp(/[\.-](\d+)/), new RegExp(/(\d+)/)];
let resolutionPattern = new RegExp(/(\d{3,}[pP])/);
let resolutionArr = ['1k', '1K', '2k', '2K', '4k', '4K', '8k', '8K'];
let charSet = new Set([' ', '[', '.', '(', '（']);
export default class InsertRule implements RuleInterface {

    /**
     * 识别类型，season：季号，name：剧名/电影名识别
     */
    type: string;
    /**
     * 前面追加
     */
    frontAdd: string;
    /**
     * 后面追加
     */
    endAdd: string;
    eNumWidth: number;

    constructor(data: any) {
        this.type = data.type;
        this.frontAdd = data.frontAdd;
        this.endAdd = data.endAdd;
        this.eNumWidth = data.eNumWidth;
    }


    deal(file: FileObj): void {
        //识别到的内容
        let getStr = null;
        let season = getSeason(path.basename(file.path));
        if (this.type === 'season') {
            getStr = season;
        } else if (this.type === 'name') {
            let originName = null;
            if (season && season.length > 0) {
                //说明是剧集,取父文件夹的父文件夹名称
                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;
            }
        } else if (this.type === 'eNum') {
            let lowName = file.originName.toLocaleLowerCase().replace(/ /g, '')
                .replace(/\d+[kp]/g, '')//去除4k,1080p等
                .replace(/[xh]\d+/g, '')//去除x264,h264等 ;
            for (let i in eNumPatternArr) {
                let patternRes = lowName.match(eNumPatternArr[i]);
                if (patternRes && patternRes.length > 1) {
                    getStr = patternRes[1];
                    for (let i = 0; i < this.eNumWidth - getStr.length; i++) {
                        getStr = '0' + getStr;
                    }
                    break;
                }
            }
        } else if (this.type === 'resolution') {
            let res = file.originName.match(resolutionPattern);
            if (res && res.length > 1) {
                getStr = res[1];
            } else {
                for (let i = 0; i < resolutionArr.length; i++) {
                    if (file.originName.indexOf(resolutionArr[i]) > -1) {
                        getStr = resolutionArr[i];
                        break;
                    }
                }
            }
        }
        if (getStr && getStr.length > 0) {
            file.realName = file.realName + this.frontAdd + getStr + this.endAdd;
            file.name = file.realName + file.expandName;
        }
    }
}