import { TronService } from '@tatumio/api-client'
import { Blockchain } from '@tatumio/shared-core'
import { SDKArguments } from '@tatumio/shared-abstract-sdk'
import { abstractBlockchainSdk } from '@tatumio/shared-blockchain-abstract'
import { tronWeb } from './services/tron.web'
import { tronTx } from './services/tron.tx'
import { tronWallet } from './services/tron.wallet'
import { tronRecord } from './services/tron.record'

const blockchain = Blockchain.TRON

export const TatumTronSDK = (args: SDKArguments) => {
  const web = tronWeb()
  const txService = tronTx({ tronWeb: web })
  const { nft, ...abstractSdk } = abstractBlockchainSdk({ ...args, blockchain })

  const {
    deployNFTSmartContract,
    mintNFT,
    transferNFT,
    mintMultipleNFTs,
    burnNFT,
    updateNFTRoyalty,
    getNFTTransaction,
    getNFTAccountBalance,
    getNFTMetadataURI,
    getNFTRoyalty,
  } = nft

  return {
    ...abstractSdk,
    transaction: txService.native,
    trc10: txService.trc10,
    trc20: txService.trc20,
    nft: {
      ...txService.trc721,
      deployNFTSmartContract,
      mintNFT,
      transferNFT,
      mintMultipleNFTs,
      burnNFT,
      updateNFTRoyalty,
      getNFTTransaction,
      getNFTAccountBalance,
      getNFTMetadataURI,
      getNFTRoyalty,
    },
    marketplace: txService.marketplace,
    smartContract: txService.smartContract,
    custodial: txService.custodial,
    wallet: tronWallet({ tronWeb: web }),
    tronWeb: web,
    record: tronRecord(),
    blockchain: {
      broadcast: TronService.tronBroadcast,
      getCurrentBlock: TronService.tronGetCurrentBlock,
      getBlock: TronService.tronGetBlock,
      getTrc10Detail: TronService.tronTrc10Detail,
      getAccount: TronService.tronGetAccount,
      getTransaction: TronService.tronGetTransaction,
      sendTransaction: TronService.tronTransfer,
    },
  }
}