ink_ir/ir/item_impl/
iter.rs

1// Copyright (C) Use Ink (UK) Ltd.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use super::{
16    CallableWithSelector,
17    ImplItem,
18    ItemImpl,
19};
20use crate::ir;
21
22/// Iterator yielding all ink! constructor within a source ink!
23/// [`ir::ItemImpl`](`crate::ir::ItemImpl`).
24pub struct IterConstructors<'a> {
25    item_impl: &'a ir::ItemImpl,
26    impl_items: core::slice::Iter<'a, ImplItem>,
27}
28
29impl<'a> IterConstructors<'a> {
30    /// Creates a new ink! messages iterator.
31    pub(super) fn new(item_impl: &'a ItemImpl) -> Self {
32        Self {
33            item_impl,
34            impl_items: item_impl.items.iter(),
35        }
36    }
37}
38
39impl<'a> Iterator for IterConstructors<'a> {
40    type Item = CallableWithSelector<'a, ir::Constructor>;
41
42    fn next(&mut self) -> Option<Self::Item> {
43        'repeat: loop {
44            match self.impl_items.next() {
45                None => return None,
46                Some(impl_item) => {
47                    if let Some(constructor) = impl_item.filter_map_constructor() {
48                        return Some(CallableWithSelector::new(
49                            self.item_impl,
50                            constructor,
51                        ))
52                    }
53                    continue 'repeat
54                }
55            }
56        }
57    }
58}
59
60/// Iterator yielding all ink! messages within a source ink!
61/// [`ir::ItemImpl`](`crate::ir::ItemImpl`).
62pub struct IterMessages<'a> {
63    item_impl: &'a ir::ItemImpl,
64    impl_items: core::slice::Iter<'a, ImplItem>,
65}
66
67impl<'a> IterMessages<'a> {
68    /// Creates a new ink! messages iterator.
69    pub(super) fn new(item_impl: &'a ItemImpl) -> Self {
70        Self {
71            item_impl,
72            impl_items: item_impl.items.iter(),
73        }
74    }
75}
76
77impl<'a> Iterator for IterMessages<'a> {
78    type Item = CallableWithSelector<'a, ir::Message>;
79
80    fn next(&mut self) -> Option<Self::Item> {
81        'repeat: loop {
82            match self.impl_items.next() {
83                None => return None,
84                Some(impl_item) => {
85                    if let Some(message) = impl_item.filter_map_message() {
86                        return Some(CallableWithSelector::new(self.item_impl, message))
87                    }
88                    continue 'repeat
89                }
90            }
91        }
92    }
93}
ink_ir/ir/item_impl/iter.rs

ink_ir/ir/item_impl/
iter.rs
