tronic 0.5.0

use std::collections::HashMap;

use derivative::Derivative;
use serde::Deserialize;
use serde::Serialize;
use time::OffsetDateTime;

use super::account::AccountType;
use crate::domain::{address::TronAddress, permission::Permission, trx::Trx};

use super::HexMessage;
use super::Message;

pub mod methods;

#[derive(Default, Debug, Clone, PartialEq)]
#[repr(i32)]
pub enum ContractType {
    AccountCreateContract(AccountCreateContract) = 0,
    TransferContract(TransferContract) = 1,
    TransferAssetContract(TransferAssetContract) = 2,
    VoteAssetContract = 3,
    VoteWitnessContract(VoteWitnessContract) = 4,
    WitnessCreateContract(WitnessCreateContract) = 5,
    AssetIssueContract(AssetIssueContract) = 6,
    WitnessUpdateContract(WitnessUpdateContract) = 8,
    ParticipateAssetIssueContract(ParticipateAssetIssueContract) = 9,
    AccountUpdateContract(AccountUpdateContract) = 10,
    FreezeBalanceContract(FreezeBalanceContract) = 11,
    UnfreezeBalanceContract(UnfreezeBalanceContract) = 12,
    WithdrawBalanceContract(WithdrawBalanceContract) = 13,
    UnfreezeAssetContract(UnfreezeAssetContract) = 14,
    UpdateAssetContract(UpdateAssetContract) = 15,
    ProposalCreateContract(ProposalCreateContract) = 16,
    ProposalApproveContract(ProposalApproveContract) = 17,
    ProposalDeleteContract(ProposalDeleteContract) = 18,
    SetAccountIdContract(SetAccountIdContract) = 19,
    #[default]
    CustomContract = 20,
    CreateSmartContract(CreateSmartContract) = 30,
    TriggerSmartContract(TriggerSmartContract) = 31,
    GetContract = 32,
    UpdateSettingContract(UpdateSettingContract) = 33,
    ExchangeCreateContract(ExchangeCreateContract) = 41,
    ExchangeInjectContract(ExchangeInjectContract) = 42,
    ExchangeWithdrawContract(ExchangeWithdrawContract) = 43,
    ExchangeTransactionContract(ExchangeTransactionContract) = 44,
    UpdateEnergyLimitContract(UpdateEnergyLimitContract) = 45,
    AccountPermissionUpdateContract(AccountPermissionUpdateContract) = 46,
    ClearAbiContract(ClearAbiContract) = 48,
    UpdateBrokerageContract(UpdateBrokerageContract) = 49,
    ShieldedTransferContract(ShieldedTransferContract) = 51,
    MarketSellAssetContract(MarketSellAssetContract) = 52,
    MarketCancelOrderContract(MarketCancelOrderContract) = 53,
    FreezeBalanceV2Contract(FreezeBalanceV2Contract) = 54,
    UnfreezeBalanceV2Contract(UnfreezeBalanceV2Contract) = 55,
    WithdrawExpireUnfreezeContract(WithdrawExpireUnfreezeContract) = 56,
    DelegateResourceContract(DelegateResourceContract) = 57,
    UnDelegateResourceContract(UnDelegateResourceContract) = 58,
    CancelAllUnfreezeV2Contract(CancelAllUnfreezeV2Contract) = 59,
}

impl ContractType {
    #[must_use]
    pub const fn id(&self) -> i32 {
        // SAFETY: This is safe because `Foo` is repr(i32).
        #[allow(unsafe_code)]
        unsafe {
            *std::ptr::from_ref::<Self>(self).cast::<i32>()
        }
    }
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct Contract {
    pub contract_type: ContractType,
    pub provider: Vec<u8>,
    pub contract_name: Message,
    pub permission_id: i32,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct TriggerSmartContract {
    pub owner_address: TronAddress,
    pub contract_address: TronAddress,
    pub call_value: Trx,
    pub data: HexMessage,
    pub call_token_value: Trx,
    pub token_id: i64,
}

#[derive(Default, Debug, Clone, Copy, PartialEq)]
pub struct FrozenSupply {
    pub frozen_amount: Trx,
    pub frozen_days: i64,
}

#[derive(Derivative, Debug, Clone, PartialEq)]
#[derivative(Default)]
pub struct AssetIssueContract {
    pub id: String,
    pub owner_address: TronAddress,
    pub name: Message,
    pub abbr: Message,
    pub total_supply: i64,
    pub frozen_supply: Vec<FrozenSupply>,
    pub trx_num: Trx,
    pub precision: i32,
    pub num: i32,
    #[derivative(Default(value = "OffsetDateTime::UNIX_EPOCH"))]
    pub start_time: OffsetDateTime,
    #[derivative(Default(value = "OffsetDateTime::UNIX_EPOCH"))]
    pub end_time: OffsetDateTime,
    /// useless
    pub order: i64,
    pub vote_score: i32,
    pub description: Message,
    pub url: Message,
    pub free_asset_net_limit: i64,
    pub public_free_asset_net_limit: i64,
    pub public_free_asset_net_usage: i64,
    #[derivative(Default(value = "OffsetDateTime::UNIX_EPOCH"))]
    pub public_latest_free_net_time: OffsetDateTime,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct TransferAssetContract {
    /// this field is token name before the proposal ALLOW_SAME_TOKEN_NAME is active, otherwise it is token id and token is should be in string format.
    pub asset_name: Message,
    pub owner_address: TronAddress,
    pub to_address: TronAddress,
    pub amount: Trx,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct UnfreezeAssetContract {
    pub owner_address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct UpdateAssetContract {
    pub owner_address: TronAddress,
    pub description: Message,
    pub url: Message,
    pub new_limit: i64,        // Not TRX
    pub new_public_limit: i64, // Not TRX
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ParticipateAssetIssueContract {
    pub owner_address: TronAddress,
    pub to_address: TronAddress,
    /// this field is token name before the proposal ALLOW_SAME_TOKEN_NAME is active, otherwise it is token id and token is should be in string format.
    pub asset_name: Message,
    /// the amount of drops
    pub amount: Trx,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct AccountCreateContract {
    pub owner_address: TronAddress,
    pub account_address: TronAddress,
    pub account_type: AccountType,
}

// Update account name. Account name is not unique now.
#[derive(Default, Debug, Clone, PartialEq)]
pub struct AccountUpdateContract {
    pub account_name: Message,
    pub owner_address: TronAddress,
}

// Set account id if the account has no id. Account id is unique and case insensitive.
#[derive(Default, Debug, Clone, PartialEq)]
pub struct SetAccountIdContract {
    pub account_id: Message,
    pub owner_address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct AccountPermissionUpdateContract {
    pub owner_address: TronAddress,
    /// Empty is invalidate
    pub owner: Option<Permission>,
    /// Can be empty
    pub witness: Option<Permission>,
    /// Empty is invalidate
    pub actives: Vec<Permission>,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct WitnessCreateContract {
    pub owner_address: TronAddress,
    pub url: Message,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct WitnessUpdateContract {
    pub owner_address: TronAddress,
    pub update_url: Message,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct Vote {
    pub vote_address: TronAddress,
    pub vote_count: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct VoteWitnessContract {
    pub owner_address: TronAddress,
    pub votes: Vec<Vote>,
    pub support: bool,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct FreezeBalanceContract {
    pub owner_address: TronAddress,
    pub frozen_balance: Trx,
    pub frozen_duration: time::Duration,
    pub resource: ResourceCode,
    pub receiver_address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct UnfreezeBalanceContract {
    pub owner_address: TronAddress,
    pub resource: ResourceCode,
    pub receiver_address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct WithdrawBalanceContract {
    pub owner_address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct TransferContract {
    pub owner_address: TronAddress,
    pub to_address: TronAddress,
    pub amount: Trx,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct TransactionBalanceTrace {
    pub transaction_identifier: HexMessage,
    pub operation: Vec<Operation>,
    pub r#type: String,
    pub status: String,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct Operation {
    pub operation_identifier: i64,
    pub address: TronAddress,
    pub amount: Trx,
}

#[derive(Derivative, Debug, Clone, PartialEq)]
#[derivative(Default)]
pub struct BlockBalanceTrace {
    pub block_identifier: BlockIdentifier,
    #[derivative(Default(value = "OffsetDateTime::UNIX_EPOCH"))]
    pub timestamp: OffsetDateTime,
    ///   BlockIdentifier parent_block_identifier = 4;
    pub transaction_balance_trace: Vec<TransactionBalanceTrace>,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct BlockIdentifier {
    pub hash: HexMessage,
    pub number: i64,
}

#[derive(Default, Debug, Clone, Copy, PartialEq)]
pub struct AccountTrace {
    pub balance: Trx,
    pub placeholder: Trx,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct AccountIdentifier {
    pub address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct AccountBalanceRequest {
    pub account_identifier: AccountIdentifier,
    pub block_identifier: BlockIdentifier,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct AccountBalanceResponse {
    pub balance: Trx,
    pub block_identifier: BlockIdentifier,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct FreezeBalanceV2Contract {
    pub owner_address: TronAddress,
    pub frozen_balance: Trx,
    pub resource: ResourceCode,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct UnfreezeBalanceV2Contract {
    pub owner_address: TronAddress,
    pub unfreeze_balance: Trx,
    pub resource: ResourceCode,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct WithdrawExpireUnfreezeContract {
    pub owner_address: TronAddress,
}

#[derive(Default, Debug, Clone, Copy, PartialEq)]
pub enum ResourceCode {
    #[default]
    Bandwidth = 0,
    Energy = 1,
    #[doc = "Deprecated in FreezeV2. Use `tron_power` field instead."]
    TronPower = 2,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct DelegateResourceContract {
    pub owner_address: TronAddress,
    pub resource: ResourceCode,
    pub balance: Trx,
    pub receiver_address: TronAddress,
    pub lock_period: Option<time::Duration>,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct UnDelegateResourceContract {
    pub owner_address: TronAddress,
    pub resource: ResourceCode,
    pub balance: Trx,
    pub receiver_address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct CancelAllUnfreezeV2Contract {
    pub owner_address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ProposalApproveContract {
    pub owner_address: TronAddress,
    pub proposal_id: i64,
    /// add or remove approval
    pub is_add_approval: bool,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ProposalCreateContract {
    pub owner_address: TronAddress,
    pub parameters: HashMap<i64, i64>,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ProposalDeleteContract {
    pub owner_address: TronAddress,
    pub proposal_id: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct BuyStorageBytesContract {
    pub owner_address: TronAddress,
    /// storage bytes for buy
    pub bytes: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct BuyStorageContract {
    pub owner_address: TronAddress,
    /// trx quantity for buy storage (sun)
    pub quant: Trx,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct SellStorageContract {
    pub owner_address: TronAddress,
    pub storage_bytes: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct UpdateBrokerageContract {
    pub owner_address: TronAddress,
    /// 1 mean 1%
    pub brokerage: i32,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ExchangeCreateContract {
    pub owner_address: TronAddress,
    pub first_token_id: Message,
    pub first_token_balance: i64,
    pub second_token_id: Message,
    pub second_token_balance: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ExchangeInjectContract {
    pub owner_address: TronAddress,
    pub exchange_id: i64,
    pub token_id: Message,
    pub quant: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ExchangeWithdrawContract {
    pub owner_address: TronAddress,
    pub exchange_id: i64,
    pub token_id: Message,
    pub quant: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ExchangeTransactionContract {
    pub owner_address: TronAddress,
    pub exchange_id: i64,
    pub token_id: Message,
    pub quant: i64,
    pub expected: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct MarketSellAssetContract {
    pub owner_address: TronAddress,
    pub sell_token_id: Message,
    pub sell_token_quantity: i64,
    pub buy_token_id: Message,
    /// min to receive
    pub buy_token_quantity: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct MarketCancelOrderContract {
    pub owner_address: TronAddress,
    pub order_id: HexMessage,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct SmartContract {
    pub origin_address: TronAddress,
    pub contract_address: TronAddress,
    pub abi: Abi,
    pub bytecode: Vec<u8>,
    pub call_value: Trx,
    pub consume_user_resource_percent: i64,
    pub name: String,
    pub origin_energy_limit: i64,
    pub code_hash: HexMessage,
    pub trx_hash: HexMessage,
    pub version: i32,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct Abi {
    pub entrys: Vec<Entry>,
}

#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct Entry {
    #[serde(default)]
    pub anonymous: bool,
    #[serde(default)]
    pub constant: bool,
    #[serde(default)]
    pub name: String,
    #[serde(default)]
    pub inputs: Vec<Param>,
    #[serde(default)]
    pub outputs: Vec<Param>,
    #[serde(rename = "type")]
    pub entry_type: EntryType,
    #[serde(default)]
    pub payable: bool,
    #[serde(default)]
    pub state_mutability: StateMutabilityType,
}

#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Param {
    #[serde(default)]
    pub indexed: bool,
    #[serde(default)]
    pub name: String,
    /// SolidityType type = 3;
    #[serde(rename = "type")]
    pub param_type: String,
}

#[derive(
    Default,
    Clone,
    Copy,
    Debug,
    PartialEq,
    Eq,
    Hash,
    PartialOrd,
    Ord,
    Serialize,
    Deserialize,
)]
#[serde(rename_all = "lowercase")]
pub enum EntryType {
    #[default]
    UnknownEntryType = 0,
    Constructor = 1,
    Function = 2,
    Event = 3,
    Fallback = 4,
    Receive = 5,
    Error = 6,
}

#[derive(
    Default,
    Clone,
    Copy,
    Debug,
    PartialEq,
    Eq,
    Hash,
    PartialOrd,
    Ord,
    Serialize,
    Deserialize,
)]
#[serde(rename_all = "lowercase")]
pub enum StateMutabilityType {
    #[default]
    UnknownMutabilityType = 0,
    Pure = 1,
    View = 2,
    Nonpayable = 3,
    Payable = 4,
}

#[derive(Default, Debug, Clone, Copy, PartialEq)]
pub struct ContractState {
    pub energy_usage: i64,
    pub energy_factor: i64,
    pub update_cycle: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct CreateSmartContract {
    pub owner_address: TronAddress,
    pub new_contract: SmartContract,
    pub call_token_value: Trx,
    pub token_id: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ClearAbiContract {
    pub owner_address: TronAddress,
    pub contract_address: TronAddress,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct UpdateSettingContract {
    pub owner_address: TronAddress,
    pub contract_address: TronAddress,
    pub consume_user_resource_percent: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct UpdateEnergyLimitContract {
    pub owner_address: TronAddress,
    pub contract_address: TronAddress,
    pub origin_energy_limit: i64,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct SpendDescription {
    pub value_commitment: Vec<u8>,
    /// merkle root
    pub anchor: Vec<u8>,
    /// used for check double spend
    pub nullifier: Vec<u8>,
    /// used for check spend authority signature
    pub rk: Vec<u8>,
    pub zkproof: Vec<u8>,
    pub spend_authority_signature: Vec<u8>,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ReceiveDescription {
    pub value_commitment: Vec<u8>,
    pub note_commitment: Vec<u8>,
    /// for Encryption
    pub epk: Vec<u8>,
    /// Encryption for incoming, decrypt it with ivk
    pub c_enc: Vec<u8>,
    /// Encryption for audit, decrypt it with ovk
    pub c_out: Vec<u8>,
    pub zkproof: Vec<u8>,
}

#[derive(Default, Debug, Clone, PartialEq)]
pub struct ShieldedTransferContract {
    /// transparent address
    pub transparent_from_address: TronAddress,
    pub from_amount: i64,
    pub spend_description: Vec<SpendDescription>,
    pub receive_description: Vec<ReceiveDescription>,
    pub binding_signature: Vec<u8>,
    /// transparent address
    pub transparent_to_address: TronAddress,
    /// the amount to transparent to_address
    pub to_amount: i64,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct ContractArtifact {
    pub contract_name: String,
    pub abi: Vec<Entry>,
    pub bytecode: String,
    pub deployed_bytecode: String,
    pub source_map: Option<String>,
    pub deployed_source_map: Option<String>,
    pub source: Option<String>,
    pub source_path: Option<String>,
    pub schema_version: String,
    pub updated_at: String,
}