// Copyright 2022 - Nym Technologies SA <contact@nymtech.net>
// SPDX-License-Identifier: Apache-2.0

use super::models::SummedDelegations;
use crate::client::ThreadsafeValidatorClient;
use itertools::Itertools;
use nym_mixnet_contract_common::{Delegation, MixId};
use nym_validator_client::nyxd::contract_traits::PagedMixnetQueryClient;

pub(crate) async fn get_single_mixnode_delegations(
    client: &ThreadsafeValidatorClient,
    mix_id: MixId,
) -> Vec<Delegation> {
    match client
        .0
        .nyxd
        .get_all_single_mixnode_delegations(mix_id)
        .await
    {
        Ok(result) => result,
        Err(e) => {
            error!("Could not get delegations for mix node {}: {:?}", mix_id, e);
            vec![]
        }
    }
}

pub(crate) async fn get_single_mixnode_delegations_summed(
    client: &ThreadsafeValidatorClient,
    mix_id: MixId,
) -> Vec<SummedDelegations> {
    let delegations_by_owner = get_single_mixnode_delegations(client, mix_id)
        .await
        .into_iter()
        .into_group_map_by(|delegation| delegation.owner.clone());

    delegations_by_owner
        .iter()
        .filter_map(|(_, delegations)| SummedDelegations::from(delegations))
        .collect()
}