93 lines
1.9 KiB
TypeScript
93 lines
1.9 KiB
TypeScript
import RuleInterface from "./RuleInterface";
|
||
import FileObj from "../../vo/FileObj";
|
||
import path from 'path';
|
||
|
||
export default class DeleteRule implements RuleInterface {
|
||
/**
|
||
* 类别：deletePart:部分删除，deleteAll:全部删除
|
||
*/
|
||
type: string;
|
||
/**
|
||
* 部分删除时的开始信息
|
||
*/
|
||
start: DeleteRuleItem;
|
||
/**
|
||
* 部分删除时的结束信息
|
||
|
||
*/
|
||
end: DeleteRuleItem;
|
||
/**
|
||
* 忽略拓展名，true:忽略，false：不忽略
|
||
*/
|
||
ignorePostfix: boolean;
|
||
|
||
constructor(data: any) {
|
||
this.type = data.type;
|
||
this.start = new DeleteRuleItem(data.start);
|
||
this.end = new DeleteRuleItem(data.end);
|
||
this.ignorePostfix = data.ignorePostfix;
|
||
}
|
||
|
||
|
||
|
||
deal(file: FileObj): void {
|
||
if (this.type === 'deleteAll') {
|
||
file.realName = "";
|
||
if (!this.ignorePostfix) {
|
||
file.expandName = "";
|
||
}
|
||
} else {
|
||
let str = file.realName + (this.ignorePostfix ? "" : file.expandName);
|
||
let startIndex = this.start.calIndex(str);
|
||
let endIndex = this.end.calIndex(str);
|
||
if (startIndex < 0 || endIndex < 0) {
|
||
return;
|
||
}
|
||
str = str.substring(0, startIndex) + str.substring(endIndex + 1);
|
||
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;
|
||
}
|
||
|
||
}
|
||
|
||
class DeleteRuleItem {
|
||
/**
|
||
* location:位置，text:文本，end:直到末尾
|
||
*/
|
||
type: string;
|
||
/**
|
||
* 对应的值
|
||
*/
|
||
value: string;
|
||
|
||
constructor(data: any) {
|
||
this.type = data.type;
|
||
this.value = data.value;
|
||
}
|
||
|
||
/**
|
||
* 计算位置
|
||
*/
|
||
calIndex(str: string): number {
|
||
if (this.type === 'location') {
|
||
return parseInt(this.value) - 1;
|
||
} else if (this.type === 'text') {
|
||
return str.indexOf(this.value);
|
||
} else if (this.type === 'end') {
|
||
return str.length - 1;
|
||
}
|
||
return -1;
|
||
}
|
||
}
|