ghost-portal/packages/react-components/src/Balance.tsx
2020-01-06 16:16:50 +01:00

76 lines
1.9 KiB
TypeScript

// Copyright 2017-2020 @polkadot/react-components authors & contributors
// This software may be modified and distributed under the terms
// of the Apache-2.0 license. See the LICENSE file for details.
import { AccountId, AccountIndex, Address } from '@polkadot/types/interfaces';
import { BareProps } from './types';
import BN from 'bn.js';
import React from 'react';
import { Balance, FormatBalance } from '@polkadot/react-query';
import { classes } from './util';
export interface RenderProps extends BareProps {
className?: string;
label?: React.ReactNode;
value?: BN | BN[];
}
export interface Props extends BareProps {
balance?: BN | BN[];
label?: React.ReactNode;
params?: AccountId | AccountIndex | Address | string | Uint8Array | null;
withLabel?: boolean;
}
export function renderProvided ({ className, label, value }: RenderProps): React.ReactNode {
let others: undefined | React.ReactNode;
if (Array.isArray(value)) {
const totals = value.filter((_, index): boolean => index !== 0);
const total = totals.reduce((total, value): BN => total.add(value), new BN(0)).gtn(0);
if (total) {
others = totals.map((balance, index): React.ReactNode =>
<FormatBalance key={index} value={balance} />
);
}
}
return (
<FormatBalance
className={classes('ui--Balance', className)}
label={label}
value={Array.isArray(value) ? value[0] : value}
>
{others && (
<span>&nbsp;(+{others})</span>
)}
</FormatBalance>
);
}
export default function BalanceDisplay (props: Props): React.ReactElement<Props> | null {
const { balance, className, label, params, style } = props;
if (!params) {
return null;
}
return balance
? (
<>
{renderProvided({ className, label, value: balance })}
</>
)
: (
<Balance
className={classes('ui--Balance', className)}
label={label}
params={params}
style={style}
/>
);
}