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 readonly _isEndingWithDecimal: boolean;
    constructor(numberOrString: string | number) {
        if (_.isString(numberOrString)) {
            const hasDecimalPeriod = _.endsWith(numberOrString, '.');
            let internalString = numberOrString;
            if (hasDecimalPeriod) {
                internalString = numberOrString.slice(0, -1);
            }
            super(internalString);
            this._isEndingWithDecimal = hasDecimalPeriod;
        } else {
            super(numberOrString);
            this._isEndingWithDecimal = false;
        }
    }
    public toDisplayString(): string {
        const internalString = super.toString();
        if (this._isEndingWithDecimal) {
            return `${internalString}.`;
        }
        return internalString;
    }
}