iota-sdk 0.2.0

The IOTA SDK provides developers with a seamless experience to develop on IOTA by providing account abstractions and clients to interact with node APIs.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

// Copyright 2022 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0

#[cfg(not(target_family = "wasm"))]
use std::collections::HashSet;

#[cfg(not(target_family = "wasm"))]
use futures::FutureExt;

#[cfg(not(target_family = "wasm"))]
use crate::types::api::plugins::indexer::OutputIdsResponse;
use crate::{
    client::node_api::indexer::query_parameters::QueryParameter, types::block::output::OutputId,
    wallet::account::handle::AccountHandle,
};

impl AccountHandle {
    /// Returns output ids of basic outputs that have only the address unlock condition
    pub(crate) async fn get_basic_output_ids_with_address_unlock_condition_only(
        &self,
        bech32_address: String,
    ) -> crate::client::Result<Vec<OutputId>> {
        // Only request basic outputs with `AddressUnlockCondition` only
        Ok(self
            .client
            .basic_output_ids(vec![
                QueryParameter::Address(bech32_address),
                QueryParameter::HasExpiration(false),
                QueryParameter::HasTimelock(false),
                QueryParameter::HasStorageDepositReturn(false),
            ])
            .await?
            .items)
    }

    /// Returns output ids of basic outputs that have the address in the `AddressUnlockCondition`,
    /// `ExpirationUnlockCondition` or `StorageDepositReturnUnlockCondition`
    pub(crate) async fn get_basic_output_ids_with_any_unlock_condition(
        &self,
        bech32_address: &str,
    ) -> crate::wallet::Result<Vec<OutputId>> {
        // aliases and foundries
        #[cfg(target_family = "wasm")]
        {
            let mut output_ids = vec![];
            output_ids.extend(
                self.client()
                    .basic_output_ids(vec![QueryParameter::Address(bech32_address.to_string())])
                    .await?
                    .items,
            );
            output_ids.extend(
                self.client()
                    .basic_output_ids(vec![QueryParameter::StorageDepositReturnAddress(
                        bech32_address.to_string(),
                    )])
                    .await?
                    .items,
            );
            output_ids.extend(
                self.client()
                    .basic_output_ids(vec![QueryParameter::ExpirationReturnAddress(
                        bech32_address.to_string(),
                    )])
                    .await?
                    .items,
            );

            Ok(output_ids)
        }

        #[cfg(not(target_family = "wasm"))]
        {
            let client = self.client();
            let tasks = vec![
                // Get basic outputs
                async move {
                    let bech32_address = bech32_address.to_string();
                    let client = client.clone();
                    tokio::spawn(async move {
                        client
                            .basic_output_ids(vec![QueryParameter::Address(bech32_address)])
                            .await
                            .map_err(From::from)
                    })
                    .await
                }
                .boxed(),
                // Get outputs where the address is in the storage deposit return unlock condition
                async move {
                    let bech32_address = bech32_address.to_string();
                    let client = client.clone();
                    tokio::spawn(async move {
                        client
                            .basic_output_ids(vec![QueryParameter::StorageDepositReturnAddress(bech32_address)])
                            .await
                            .map_err(From::from)
                    })
                    .await
                }
                .boxed(),
                // Get outputs where the address is in an expired expiration unlock condition
                async move {
                    let bech32_address = bech32_address.to_string();
                    let client = client.clone();
                    tokio::spawn(async move {
                        client
                            .basic_output_ids(vec![QueryParameter::ExpirationReturnAddress(bech32_address)])
                            .await
                            .map_err(From::from)
                    })
                    .await
                }
                .boxed(),
            ];

            // Get all results
            let mut output_ids = HashSet::new();
            let results: Vec<crate::wallet::Result<OutputIdsResponse>> = futures::future::try_join_all(tasks).await?;

            for res in results {
                let found_output_ids = res?;
                output_ids.extend(found_output_ids.items);
            }

            Ok(output_ids.into_iter().collect())
        }
    }
}