ic-canister-kit 1.3.3

pub use ic_cdk_bitcoin_canister::{
    Address as BitcoinAddress, BlockHash, BlockHeader, GetBalanceRequest, GetBlockHeadersRequest,
    GetBlockHeadersResponse, GetCurrentFeePercentilesRequest, GetUtxosRequest, GetUtxosResponse, Height,
    MillisatoshiPerByte, Network as BitcoinNetwork, Satoshi, SendTransactionRequest, Utxo, UtxosFilter,
    UtxosFilterInRequest,
};

use crate::types::CanisterCallError;

/// 查询余额
/// <https://docs.internetcomputer.org/references/management-canister/#bitcoin_get_balance>
pub async fn bitcoin_get_balance(
    network: BitcoinNetwork,
    address: BitcoinAddress,
    min_confirmations: Option<u32>,
) -> super::types::CanisterCallResult<Satoshi> {
    ic_cdk_bitcoin_canister::bitcoin_get_balance(&GetBalanceRequest {
        network: network.into(),
        address,
        min_confirmations,
    })
    .await
    .map_err(|err| CanisterCallError {
        canister_id: crate::identity::CanisterId::management_canister(),
        method: "ic#bitcoin_get_balance".to_string(),
        message: err.to_string(),
    })
}

/// 查询指定高度范围内的区块头
/// <https://docs.internetcomputer.org/references/management-canister/#bitcoin_get_block_headers>
pub async fn bitcoin_get_block_headers(
    network: BitcoinNetwork,
    start_height: Height,
    end_height: Option<Height>,
) -> super::types::CanisterCallResult<GetBlockHeadersResponse> {
    ic_cdk_bitcoin_canister::bitcoin_get_block_headers(&GetBlockHeadersRequest {
        start_height,
        end_height,
        network: network.into(),
    })
    .await
    .map_err(|err| CanisterCallError {
        canister_id: crate::identity::CanisterId::management_canister(),
        method: "ic#bitcoin_get_block_headers".to_string(),
        message: err.to_string(),
    })
}

/// 查询网络费用
/// <https://docs.internetcomputer.org/references/management-canister/#bitcoin_get_current_fee_percentiles>
pub async fn bitcoin_get_current_fee_percentiles(
    network: BitcoinNetwork,
) -> super::types::CanisterCallResult<Vec<MillisatoshiPerByte>> {
    ic_cdk_bitcoin_canister::bitcoin_get_current_fee_percentiles(&GetCurrentFeePercentilesRequest {
        network: network.into(),
    })
    .await
    .map_err(|err| CanisterCallError {
        canister_id: crate::identity::CanisterId::management_canister(),
        method: "ic#bitcoin_get_current_fee_percentiles".to_string(),
        message: err.to_string(),
    })
}

/// 查询 UTXO
/// <https://docs.internetcomputer.org/references/management-canister/#bitcoin_get_utxos>
pub async fn bitcoin_get_utxos(
    network: BitcoinNetwork,
    address: BitcoinAddress,
    filter: Option<UtxosFilter>,
) -> super::types::CanisterCallResult<GetUtxosResponse> {
    ic_cdk_bitcoin_canister::bitcoin_get_utxos(&GetUtxosRequest {
        address,
        network: network.into(),
        filter: filter.map(|f| match f {
            UtxosFilter::MinConfirmations(x) => UtxosFilterInRequest::MinConfirmations(x),
            UtxosFilter::Page(p) => UtxosFilterInRequest::Page(p),
        }),
    })
    .await
    .map_err(|err| CanisterCallError {
        canister_id: crate::identity::CanisterId::management_canister(),
        method: "ic#bitcoin_get_utxos".to_string(),
        message: err.to_string(),
    })
}

/// 发送交易
/// <https://docs.internetcomputer.org/references/management-canister/#bitcoin_send_transaction>
pub async fn bitcoin_send_transaction(
    network: BitcoinNetwork,
    transaction: Vec<u8>,
) -> super::types::CanisterCallResult<()> {
    ic_cdk_bitcoin_canister::bitcoin_send_transaction(&SendTransactionRequest {
        transaction,
        network: network.into(),
    })
    .await
    .map_err(|err| CanisterCallError {
        canister_id: crate::identity::CanisterId::management_canister(),
        method: "ic#bitcoin_send_transaction".to_string(),
        message: err.to_string(),
    })
}