// 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 { Props } from '../types';

import BN from 'bn.js';
import React, { useMemo } from 'react';
import { GenericVote } from '@polkadot/types';
import { Dropdown } from '@polkadot/react-components';

import { useTranslation } from '../translate';
import Bare from './Bare';

function doChange (onChange?: (value: any) => void): (_: number) => void {
  return function (value: number): void {
    onChange && onChange({
      isValid: true,
      value
    });
  };
}

function Vote ({ className, defaultValue: { value }, isDisabled, isError, onChange, style, withLabel }: Props): React.ReactElement<Props> {
  const { t } = useTranslation();
  const optAye = useMemo(() => {
    return [
      { text: t('Nay'), value: 0 },
      { text: t('Aye'), value: -1 }
    ];
  }, [t]);
  const optConv = useMemo(() => {
    return [
      { text: t('None'), value: 0 },
      { text: t('Locked1x'), value: 1 },
      { text: t('Locked2x'), value: 2 },
      { text: t('Locked3x'), value: 3 },
      { text: t('Locked4x'), value: 4 },
      { text: t('Locked5x'), value: 5 },
      { text: t('Locked6x'), value: 6 }
    ];
  }, [t]);
  const defaultValue = value instanceof BN
    ? value.toNumber()
    : value instanceof GenericVote
      ? (value.isAye ? -1 : 0)
      : value as number;
  const defaultConv = value instanceof GenericVote
    ? value.conviction.index
    : 0;

  return (
    <Bare
      className={className}
      style={style}
    >
      <Dropdown
        className='full'
        defaultValue={defaultValue}
        isDisabled={isDisabled}
        isError={isError}
        label={t('aye: bool')}
        onChange={doChange(onChange)}
        options={optAye}
        withLabel={withLabel}
      />
      {isDisabled && (
        <Dropdown
          className='full'
          defaultValue={defaultConv}
          isDisabled={isDisabled}
          isError={isError}
          label={t('conviction: Conviction')}
          options={optConv}
          withLabel={withLabel}
        />
      )}
    </Bare>
  );
}

export default React.memo(Vote);