All files / utils Calculator.ts

100% Statements 21/21
100% Branches 18/18
100% Functions 16/16
100% Lines 21/21

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159                                                                                                                                                23x 28x   28x 56x     10x       28x       12x         20x 24x               5x         8x   8x               12x         21x 21x               5x         7x   7x               24x           28x           23x  
import DecimalLight, { Config, Numeric } from 'decimal.js-light'
import { castArray } from 'lodash-uni'
import { OneOrMore } from '../types'
 
export interface CalculatorConfig extends Config {
  /** 小数位数 */
  decimalPlaces?: number
}
 
export type CalculatorPrimitiveValue = Numeric
 
export type CalculatorFunctionValue = (
  calculator: CalculatorInstance<DecimalLight>,
) => OneOrMore<DecimalLight>
 
export type CalculatorValue = CalculatorPrimitiveValue | CalculatorFunctionValue
 
export interface CalculatorInstance<T = number> {
  /**
   * decimal.js 引用。
   */
  decimal: typeof DecimalLight
  /**
   * 根据配置创建一个新的计算器。
   *
   * @param config 配置
   */
  make(config?: CalculatorConfig): CalculatorInstance
  /**
   * 加。
   *
   * - 加数列表为空时返回 0;
   * - 加数列表长度为 1 时返回第 1 个的值。
   *
   * @param values 加数列表
   */
  add(...values: CalculatorValue[]): T
  /**
   * 减。
   *
   * - 减数列表为空时返回 0;
   * - 减数列表长度为 1 时返回第 1 个的值。
   *
   * @param values 减数列表
   */
  sub(...values: CalculatorValue[]): T
  /**
   * 乘。
   *
   * - 乘数列表为空时返回 0;
   * - 乘数列表长度为 1 时返回第 1 个的值。
   *
   * @param values 乘数列表
   */
  mul(...values: CalculatorValue[]): T
  /**
   * 除。
   *
   * - 除数列表为空时返回 0;
   * - 除数列表长度为 1 时返回第 1 个的值。
   *
   * @param values 除数列表
   */
  div(...values: CalculatorValue[]): T
  /**
   * 转换为数字。
   *
   * @param value 值
   */
  toNumber(value: CalculatorPrimitiveValue): T
}
 
const make: CalculatorInstance['make'] = config => {
  const Decimal = DecimalLight.clone(config)
 
  const getValue = (value: CalculatorValue) =>
    typeof value === 'function'
      ? value({
          ...Calculator,
          toNumber: (value: any) => new Decimal(value),
        } as any as CalculatorInstance<DecimalLight>)
      : value
 
  const Calculator: CalculatorInstance = {
    decimal: DecimalLight,
    make: make,
    add(...values) {
      return this.toNumber(
        values.length === 0
          ? 0
          : values.reduce<DecimalLight>(
              (res, value) =>
                castArray(getValue(value)).reduce<DecimalLight>(
                  (res, item) => res.add(item),
                  res,
                ),
              new Decimal(0),
            ),
      )
    },
    sub(...values) {
      return this.toNumber(
        values.length === 0
          ? 0
          : values.reduce<DecimalLight>(
              (res, value, index) =>
                castArray(getValue(value)).reduce<DecimalLight>(
                  (res, item, index2) =>
                    index === 0 && index2 === 0 ? res.add(item) : res.sub(item),
                  res,
                ),
              new Decimal(0),
            ),
      )
    },
    mul(...values) {
      return this.toNumber(
        values.length === 0
          ? 0
          : values.reduce<DecimalLight>(
              (res, value) =>
                castArray(getValue(value)).reduce<DecimalLight>(
                  (res, item) => res.mul(item),
                  res,
                ),
              new Decimal(1),
            ),
      )
    },
    div(...values) {
      return this.toNumber(
        values.length === 0
          ? 0
          : values.reduce<DecimalLight>(
              (res, value, index) =>
                castArray(getValue(value)).reduce<DecimalLight>(
                  (res, item, index2) =>
                    index === 0 && index2 === 0 ? res.add(item) : res.div(item),
                  res,
                ),
              new Decimal(0),
            ),
      )
    },
    toNumber(value) {
      return new Decimal(value)
        .toDecimalPlaces(config?.decimalPlaces)
        .toNumber()
    },
  }
 
  return Calculator
}
 
/**
 * 科学计算器。主要是为了避免 js 的浮点数精度计算问题。
 */
export const Calculator = make()