87 lines
2.5 KiB
TypeScript
87 lines
2.5 KiB
TypeScript
import RuleInterface from "./RuleInterface";
|
||
import FileObj from "../../vo/FileObj";
|
||
import path from 'path';
|
||
|
||
export default class InsertRule implements RuleInterface {
|
||
/**
|
||
* 开始位置
|
||
*/
|
||
start: number;
|
||
/**
|
||
* 记录当前的值是多少
|
||
*/
|
||
currentIndexMap: Map<string, number>;
|
||
/**
|
||
* 增量
|
||
*/
|
||
increment: number;
|
||
/**
|
||
* 是否填充0
|
||
*/
|
||
addZero: boolean;
|
||
/**
|
||
* 填充后长度
|
||
*/
|
||
numLength: number;
|
||
/**
|
||
* 插入位置,front:前缀，backend：后缀，at：位置
|
||
*/
|
||
insertType: string;
|
||
/**
|
||
* 插入的位置
|
||
*/
|
||
insertValue: number;
|
||
/**
|
||
* 忽略拓展名
|
||
*/
|
||
ignorePostfix: boolean;
|
||
/**
|
||
* 拓展名分组
|
||
*/
|
||
postfixGroup: boolean;
|
||
|
||
constructor(data: any) {
|
||
this.start = data.start;
|
||
this.currentIndexMap = new Map<string, number>();
|
||
this.increment = data.increment;
|
||
this.addZero = data.addZero;
|
||
this.numLength = data.numLength;
|
||
this.insertType = data.insertType;
|
||
this.insertValue = data.insertValue;
|
||
this.ignorePostfix = data.ignorePostfix;
|
||
this.postfixGroup = data.postfixGroup;
|
||
}
|
||
|
||
deal(file: FileObj): void {
|
||
let expand = this.postfixGroup ? file.expandName : "";
|
||
let currentIndex = this.currentIndexMap.has(expand) ? this.currentIndexMap.get(expand) : this.start;
|
||
let length = currentIndex.toString().length;
|
||
let numStr = (this.addZero && this.numLength > length ? "0".repeat(this.numLength - length) : "") + currentIndex;
|
||
let str = this.ignorePostfix ? file.realName : file.name;
|
||
switch (this.insertType) {
|
||
case "front":
|
||
str = numStr + str;
|
||
break;
|
||
case "backend":
|
||
str = str + numStr;
|
||
break;
|
||
case "at":
|
||
str = str.substring(0, this.insertValue - 1) + numStr + str.substring(this.insertValue - 1);
|
||
break;
|
||
}
|
||
this.currentIndexMap.set(expand, currentIndex + this.increment);
|
||
|
||
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;
|
||
}
|
||
} |