mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-16 18:09:22 +08:00
4b2a94e383
* chore: code optimization * Update components/theme/util/calc/CSSCalculator.ts Co-authored-by: afc163 <afc163@gmail.com> Signed-off-by: lijianan <574980606@qq.com> --------- Signed-off-by: lijianan <574980606@qq.com> Co-authored-by: afc163 <afc163@gmail.com>
89 lines
2.5 KiB
TypeScript
89 lines
2.5 KiB
TypeScript
import AbstractCalculator from './calculator';
|
|
|
|
const CALC_UNIT = 'CALC_UNIT';
|
|
|
|
const regexp = new RegExp(CALC_UNIT, 'g');
|
|
|
|
function unit(value: string | number) {
|
|
if (typeof value === 'number') {
|
|
return `${value}${CALC_UNIT}`;
|
|
}
|
|
return value;
|
|
}
|
|
|
|
export default class CSSCalculator extends AbstractCalculator {
|
|
result: string = '';
|
|
|
|
lowPriority?: boolean;
|
|
|
|
constructor(num: number | string | AbstractCalculator) {
|
|
super();
|
|
if (num instanceof CSSCalculator) {
|
|
this.result = `(${num.result})`;
|
|
} else if (typeof num === 'number') {
|
|
this.result = unit(num);
|
|
} else if (typeof num === 'string') {
|
|
this.result = num;
|
|
}
|
|
}
|
|
|
|
add(num: number | string | AbstractCalculator): this {
|
|
if (num instanceof CSSCalculator) {
|
|
this.result = `${this.result} + ${num.getResult()}`;
|
|
} else if (typeof num === 'number' || typeof num === 'string') {
|
|
this.result = `${this.result} + ${unit(num)}`;
|
|
}
|
|
this.lowPriority = true;
|
|
return this;
|
|
}
|
|
|
|
sub(num: number | string | AbstractCalculator): this {
|
|
if (num instanceof CSSCalculator) {
|
|
this.result = `${this.result} - ${num.getResult()}`;
|
|
} else if (typeof num === 'number' || typeof num === 'string') {
|
|
this.result = `${this.result} - ${unit(num)}`;
|
|
}
|
|
this.lowPriority = true;
|
|
return this;
|
|
}
|
|
|
|
mul(num: number | string | AbstractCalculator): this {
|
|
if (this.lowPriority) {
|
|
this.result = `(${this.result})`;
|
|
}
|
|
if (num instanceof CSSCalculator) {
|
|
this.result = `${this.result} * ${num.getResult(true)}`;
|
|
} else if (typeof num === 'number' || typeof num === 'string') {
|
|
this.result = `${this.result} * ${num}`;
|
|
}
|
|
this.lowPriority = false;
|
|
return this;
|
|
}
|
|
|
|
div(num: number | string | AbstractCalculator): this {
|
|
if (this.lowPriority) {
|
|
this.result = `(${this.result})`;
|
|
}
|
|
if (num instanceof CSSCalculator) {
|
|
this.result = `${this.result} / ${num.getResult(true)}`;
|
|
} else if (typeof num === 'number' || typeof num === 'string') {
|
|
this.result = `${this.result} / ${num}`;
|
|
}
|
|
this.lowPriority = false;
|
|
return this;
|
|
}
|
|
|
|
getResult(force?: boolean): string {
|
|
return this.lowPriority || force ? `(${this.result})` : this.result;
|
|
}
|
|
|
|
equal(options?: { unit?: boolean }): string {
|
|
const { unit: cssUnit = true } = options || {};
|
|
this.result = this.result.replace(regexp, cssUnit ? 'px' : '');
|
|
if (typeof this.lowPriority !== 'undefined') {
|
|
return `calc(${this.result})`;
|
|
}
|
|
return this.result;
|
|
}
|
|
}
|