import { BigNumber } from "bignumber.js";
import type { Transaction, TransactionRaw } from "../types";
import { formatTransactionStatus } from "@ledgerhq/coin-framework/formatters";
import {
  fromTransactionCommonRaw,
  fromTransactionStatusRawCommon as fromTransactionStatusRaw,
  toTransactionCommonRaw,
  toTransactionStatusRawCommon as toTransactionStatusRaw,
} from "@ledgerhq/coin-framework/serialization";
import type { Account } from "@ledgerhq/types-live";
import { getAccountCurrency } from "@ledgerhq/coin-framework/account";
import { formatCurrencyUnit } from "@ledgerhq/coin-framework/currencies";

export const fromTransactionRaw = (tr: TransactionRaw): Transaction => {
  const common = fromTransactionCommonRaw(tr);
  const { networkInfo } = tr;
  return {
    ...common,
    networkInfo: networkInfo && {
      family: "tron",
      freeNetUsed: new BigNumber(networkInfo.freeNetUsed),
      freeNetLimit: new BigNumber(networkInfo.freeNetLimit),
      netUsed: new BigNumber(networkInfo.netUsed),
      netLimit: new BigNumber(networkInfo.netLimit),
      energyUsed: new BigNumber(networkInfo.energyUsed),
      energyLimit: new BigNumber(networkInfo.energyLimit),
    },
    family: tr.family,
    mode: tr.mode,
    resource: tr.resource || null,
    duration: tr.duration || 3,
    votes: tr.votes,
  };
};

export const toTransactionRaw = (t: Transaction): TransactionRaw => {
  const common = toTransactionCommonRaw(t);
  const { networkInfo } = t;
  return {
    ...common,
    networkInfo: networkInfo && {
      family: "tron",
      freeNetUsed: networkInfo.freeNetUsed.toString(),
      freeNetLimit: networkInfo.freeNetLimit.toString(),
      netUsed: networkInfo.netUsed.toString(),
      netLimit: networkInfo.netLimit.toString(),
      energyUsed: networkInfo.energyUsed.toString(),
      energyLimit: networkInfo.energyLimit.toString(),
    },
    family: t.family,
    mode: t.mode,
    resource: t.resource || null,
    duration: t.duration || 3,
    votes: t.votes,
  };
};

export const formatTransaction = (t: Transaction, mainAccount: Account): string => {
  const account =
    (t.subAccountId && (mainAccount.subAccounts || []).find(a => a.id === t.subAccountId)) ||
    mainAccount;
  return `
${t.mode.toUpperCase()}${t.resource ? " " + t.resource : ""} ${
    t.useAllAmount
      ? "MAX"
      : t.amount.isZero()
        ? ""
        : " " +
          formatCurrencyUnit(getAccountCurrency(account).units[0], t.amount, {
            showCode: true,
            disableRounding: true,
          })
  }${!t.votes ? "" : " " + t.votes.map(v => v.voteCount + "->" + v.address).join(" ")}
TO ${t.recipient}`;
};

export default {
  formatTransaction,
  fromTransactionRaw,
  toTransactionRaw,
  fromTransactionStatusRaw,
  toTransactionStatusRaw,
  formatTransactionStatus,
};