tronic 0.5.0

use bitvec::{bitvec, view::BitView};

use crate::{define_fixed_string, domain::address::TronAddress, error::Error};

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

impl Ops {
    /// Encodes a list of contract types into a 32-byte operations string
    pub fn encode_ops(ops: &[Ops]) -> Vec<u8> {
        if ops.is_empty() {
            return Vec::new();
        }

        let mut bits = bitvec![u8, bitvec::order::Lsb0; 0; 256]; // 32 bytes * 8 bits

        for op in ops {
            let num: usize = (*op) as usize;
            if (0..256).contains(&num) {
                bits.set(num, true);
            }
        }

        bits.into_vec()
    }

    /// Decodes an operations string into a list of contract types
    pub fn decode_ops(operations: &[u8]) -> Vec<Ops> {
        let mut result = Vec::new();

        let bits = operations.view_bits::<bitvec::order::Lsb0>();

        for (pos, bit) in bits.iter().enumerate() {
            if *bit {
                let contract = Ops::from_repr(pos).unwrap();
                result.push(contract);
            }
        }

        result
    }
}

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

#[derive(Debug, Default, Clone, PartialEq)]
pub enum PermissionType {
    #[default]
    Owner = 0,
    Witness = 1,
    Active = 2,
}

define_fixed_string!(PermissionName, 32, "TRON permission name (max 32 bytes)");

#[derive(
    Debug, Default, Clone, PartialEq, getset::Getters, getset::CopyGetters,
)]
pub struct Permission {
    pub(crate) permission_type: PermissionType,
    /// Owner id=0, Witness id=1, Active id start by 2
    #[getset(get_copy = "pub")]
    pub(crate) id: i32,
    #[getset(get_copy = "pub")]
    pub(crate) permission_name: PermissionName,
    #[getset(get_copy = "pub")]
    pub(crate) threshold: i64,
    #[getset(get_copy = "pub")]
    pub(crate) parent_id: i32,
    #[getset(get = "pub")]
    pub(crate) operations: Vec<Ops>,
    #[getset(get = "pub")]
    pub(crate) keys: Vec<Key>,
}

#[derive(Debug, Default, Clone, PartialEq, bon::Builder)]
#[builder(start_fn = with_name)]
pub struct PermissionParams {
    pub permission_name: PermissionName,
    pub threshold: i64,
    pub parent_id: i32,
    pub operations: Vec<Ops>,
    pub keys: Vec<Key>,
}

impl From<Permission> for PermissionParams {
    fn from(value: Permission) -> Self {
        PermissionParams {
            permission_name: value.permission_name,
            threshold: value.threshold,
            parent_id: value.parent_id,
            operations: value.operations.clone(),
            keys: value.keys.clone(),
        }
    }
}

#[bon::bon]
impl Permission {
    #[builder]
    pub(crate) fn owner(params: PermissionParams) -> Self {
        Permission {
            permission_type: PermissionType::Owner,
            id: 0,
            permission_name: params.permission_name,
            threshold: params.threshold,
            parent_id: params.parent_id,
            operations: params.operations,
            keys: params.keys,
        }
    }
    #[builder]
    pub(crate) fn witness(params: PermissionParams) -> Self {
        Permission {
            permission_type: PermissionType::Witness,
            id: 1,
            permission_name: params.permission_name,
            threshold: params.threshold,
            parent_id: params.parent_id,
            operations: params.operations,
            keys: params.keys,
        }
    }
    #[builder]
    pub(crate) fn actives(params: Vec<PermissionParams>) -> Vec<Permission> {
        params
            .into_iter()
            .enumerate()
            .map(|(idx, p)| Permission {
                permission_type: PermissionType::Active,
                id: idx as i32 + 2,
                permission_name: p.permission_name,
                threshold: p.threshold,
                parent_id: p.parent_id,
                operations: p.operations,
                keys: p.keys,
            })
            .collect()
    }
    /// Returns the minimum number of signatures needed (greedy selection)
    pub(crate) fn required_signatures(&self) -> Option<i64> {
        if self.threshold <= 0 || self.keys.is_empty() {
            return Some(0);
        }

        let mut weights: Vec<i64> =
            self.keys.iter().map(|k| k.weight).collect();
        weights.sort_by(|a, b| b.cmp(a)); // Descending

        let mut remaining = self.threshold;
        let mut count = 0;

        for &w in &weights {
            remaining -= w;
            count += 1;
            if remaining <= 0 {
                return Some(count);
            }
        }
        // If we get here, the threshold isn't met even with all signatures
        None
    }

    /// Checks if ANY combination of keys meets the threshold (combinatorial)
    pub(crate) fn can_meet_threshold(&self) -> crate::Result<()> {
        let total_weight: i64 = self.keys.iter().map(|k| k.weight).sum();
        if total_weight >= self.threshold {
            Ok(())
        } else {
            Err(Error::InvalidInput(
                "insufficient key weight for threshold".into(),
            ))
        }
    }

    pub fn contains(&self, address: TronAddress) -> bool {
        self.keys.iter().any(|k| k.address.eq(&address))
    }

    pub fn enough_sign_weight(&self, addresses: Vec<TronAddress>) -> bool {
        if self.threshold <= 0 {
            return true;
        }

        // Calculate total weight of the provided addresses
        let total_weight: i64 = self
            .keys
            .iter()
            .filter(|key| addresses.contains(&key.address))
            .map(|key| key.weight)
            .sum();

        // Check if the combined weight meets or exceeds threshold
        total_weight >= self.threshold
    }
}