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
use std::{fmt, ops::Deref};
use rb_sys::VALUE;
use crate::{
class,
error::Error,
exception,
object::Object,
try_convert::TryConvert,
value::{private, NonZeroValue, ReprValue, Value},
};
#[derive(Clone, Copy)]
#[repr(transparent)]
pub struct Enumerator(NonZeroValue);
impl Enumerator {
#[inline]
pub fn from_value(val: Value) -> Option<Self> {
unsafe {
val.is_kind_of(class::enumerator())
.then(|| Self(NonZeroValue::new_unchecked(val)))
}
}
#[inline]
pub(crate) unsafe fn from_rb_value_unchecked(val: VALUE) -> Self {
Self(NonZeroValue::new_unchecked(Value::new(val)))
}
}
impl Iterator for Enumerator {
type Item = Result<Value, Error>;
fn next(&mut self) -> Option<Self::Item> {
match self.funcall("next", ()) {
Ok(v) => Some(Ok(v)),
Err(e) if e.is_kind_of(exception::stop_iteration()) => None,
Err(e) => Some(Err(e)),
}
}
}
impl Deref for Enumerator {
type Target = Value;
fn deref(&self) -> &Self::Target {
self.0.get_ref()
}
}
impl fmt::Display for Enumerator {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", unsafe { self.to_s_infallible() })
}
}
impl fmt::Debug for Enumerator {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.deref().inspect())
}
}
impl From<Enumerator> for Value {
fn from(val: Enumerator) -> Self {
*val
}
}
impl Object for Enumerator {}
unsafe impl private::ReprValue for Enumerator {
fn to_value(self) -> Value {
*self
}
unsafe fn from_value_unchecked(val: Value) -> Self {
Self(NonZeroValue::new_unchecked(val))
}
}
impl ReprValue for Enumerator {}
impl TryConvert for Enumerator {
fn try_convert(val: Value) -> Result<Self, Error> {
Self::from_value(val).ok_or_else(|| {
Error::new(
exception::type_error(),
format!("no implicit conversion of {} into Enumerator", unsafe {
val.classname()
},),
)
})
}
}