olehmell/subsocial-sdk

import { hexToBn } from '@polkadot/util';
import BN from 'bn.js';

const ZERO = new BN(0);
const ONE = new BN(1);

type Props = {
  count: number | BN | string,
  singularText: string,
  pluralText?: string,
  hideCount?: boolean
}

export function pluralize ({ count, singularText, pluralText, hideCount }: Props) {
  if (!count) {
    count = ZERO;
  } else if (typeof count === 'string') {

    if (count.startsWith('0x')) {
      count = hexToBn(count);
    } else if (count.startsWith('-0x')) {
      count = hexToBn(count.substring(1), { isNegative: true });
    } else {
      count = new BN(count);
    }

  } else if (typeof count === 'number') {
    count = new BN(count);
  }

  const plural = () => !pluralText
    ? singularText + 's'
    : pluralText;

  const text = count.eq(ONE)
    ? singularText
    : plural();

  const countMsg = hideCount ? '' : `${count.toNumber()} `

  return `${countMsg}${text}`;
}