import { MILLISECONDS_PER_SECOND } from '../../../../common/time';
import { formatDecimal } from '../../../../common/decimal';
import {
  TbankMoney,
  TbankOperationItem,
} from '../../../../integrations/tbank/tbank.api';
import { decimal } from '../../../../integrations/tbank/decimal';
import {
  BrokerageOperationDto,
  OperationCategory,
} from '../presentation/brokerage-operation.dto';

const NANOSECONDS_PER_MILLISECOND = 1_000_000;

const types: Record<string, OperationCategory> = {
  OPERATION_TYPE_BUY: 'trade',
  OPERATION_TYPE_BUY_CARD: 'trade',
  OPERATION_TYPE_SELL: 'trade',
  OPERATION_TYPE_SELL_CARD: 'trade',
  OPERATION_TYPE_BUY_MARGIN: 'trade',
  OPERATION_TYPE_SELL_MARGIN: 'trade',
  OPERATION_TYPE_COUPON: 'income',
  OPERATION_TYPE_DIVIDEND: 'income',
  OPERATION_TYPE_DIVIDEND_TRANSFER: 'income',
  OPERATION_TYPE_DIV_EXT: 'income',
  OPERATION_TYPE_TRANS_IIS_BS: 'transfer',
  OPERATION_TYPE_TRANS_BS_BS: 'transfer',
  OPERATION_TYPE_BOND_TAX: 'tax',
  OPERATION_TYPE_TAX: 'tax',
  OPERATION_TYPE_DIVIDEND_TAX: 'tax',
  OPERATION_TYPE_SERVICE_FEE: 'fee',
  OPERATION_TYPE_MARGIN_FEE: 'fee',
  OPERATION_TYPE_BROKER_FEE: 'fee',
  OPERATION_TYPE_SUCCESS_FEE: 'fee',
  OPERATION_TYPE_INPUT: 'transfer',
  OPERATION_TYPE_OUTPUT: 'transfer',
  OPERATION_TYPE_INPUT_SECURITIES: 'transfer',
  OPERATION_TYPE_OUTPUT_SECURITIES: 'transfer',
};

export function operationCategory(type: string | undefined): OperationCategory {
  return types[type ?? ''] ?? 'other';
}

export function mapBrokerageOperation(
  operation: TbankOperationItem,
  occurredAt = operationTimestamp(operation.date),
): BrokerageOperationDto {
  return {
    id: operation.id ?? operation.cursor ?? '',
    occurredAt: occurredAt.toISOString(),
    type: operation.type ?? 'OPERATION_TYPE_UNSPECIFIED',
    category: operationCategory(operation.type),
    description: operation.description?.trim() || operation.name?.trim() || '',
    figi: text(operation.figi),
    instrumentUid: text(operation.instrumentUid),
    ticker: text(operation.ticker),
    instrumentType: text(operation.instrumentType),
    logoName: null,
    payment: money(operation.payment),
    quantity:
      (operationCategory(operation.type) === 'trade'
        ? operation.quantityDone
        : operation.quantity) === undefined
        ? null
        : String(
            operationCategory(operation.type) === 'trade'
              ? operation.quantityDone
              : operation.quantity,
          ),
    price: money(operation.price),
    accruedInterest: money(operation.accruedInt),
    commission: money(operation.commission),
  };
}

function text(value: string | undefined): string | null {
  return value?.trim() || null;
}

function money(value: TbankMoney | undefined) {
  const currency = value?.currency?.trim().toUpperCase();
  return currency && value
    ? { currency, value: formatDecimal(decimal(value)) }
    : null;
}

export function operationTimestamp(
  value: { seconds?: string | number; nanos?: number } | undefined,
) {
  return new Date(
    Number(value?.seconds ?? 0) * MILLISECONDS_PER_SECOND +
      Math.floor((value?.nanos ?? 0) / NANOSECONDS_PER_MILLISECOND),
  );
}