packages/instant/src/util/big_number.ts


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

import { BigNumber } from '@0x/utils';
import * as _ from 'lodash';

/**
 *  A BigNumber extension that is more flexible about decimal strings.
 *  Such as allowing:
 *  new BigNumberInput(0.) => 0
 *  new BigNumberInput(1.) => 1
 *  new BigNumberInput(1..) => still throws
 */
export class BigNumberInput extends BigNumber {
    private _hasDecimalPeriod: boolean;
    constructor(bigNumberString: string) {
        const hasDecimalPeriod = _.endsWith(bigNumberString, '.');
        let internalString = bigNumberString;
        if (hasDecimalPeriod) {
            internalString = bigNumberString.slice(0, bigNumberString.length - 1);
        }
        super(internalString);
        this._hasDecimalPeriod = hasDecimalPeriod;
    }
    public toString(): string {
        const internalString = super.toString();
        if (this._hasDecimalPeriod) {
            return `${internalString}.`;
        }
        return internalString;
    }
}