cynic_parser/values/
objects.rs

1use std::fmt;
2
3use crate::{AstLookup, Span};
4
5use super::{
6    Cursor, ValueStoreId,
7    const_objects::ConstObject,
8    ids::{FieldId, StringId, ValueId},
9    iter::{Iter, ValueStoreReader},
10    value::Value,
11};
12
13pub struct FieldRecord {
14    pub name: StringId,
15    pub name_span: Span,
16    pub value: ValueId,
17}
18
19#[derive(Clone, Copy)]
20pub struct Object<'a>(pub(super) super::Cursor<'a, ValueId>);
21
22impl<'a> Object<'a> {
23    pub fn is_empty(&self) -> bool {
24        self.len() == 0
25    }
26
27    pub fn len(&self) -> usize {
28        let store = self.0.store;
29        store.lookup(self.0.id).kind.as_object().unwrap().len()
30    }
31
32    pub fn span(&self) -> Span {
33        let store = &self.0.store;
34        store.lookup(self.0.id).span
35    }
36
37    pub fn fields(&self) -> Iter<'a, ObjectField<'a>> {
38        let store = self.0.store;
39        Iter::new(store.lookup(self.0.id).kind.as_object().unwrap(), store)
40    }
41
42    pub fn get(&self, name: &str) -> Option<Value<'a>> {
43        Some(self.fields().find(|field| field.name() == name)?.value())
44    }
45}
46
47impl PartialEq for Object<'_> {
48    fn eq(&self, other: &Self) -> bool {
49        self.len() == other.len()
50            && self.fields().all(|field| {
51                let needle = field.name();
52                let Some(b_field) = other
53                    .fields()
54                    .find(|other_field| other_field.name() == needle)
55                else {
56                    return false;
57                };
58
59                field.value() == b_field.value()
60            })
61    }
62}
63
64impl Eq for Object<'_> {}
65
66impl fmt::Debug for Object<'_> {
67    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
68        f.debug_map()
69            .entries(self.fields().map(|field| (field.name(), field.value())))
70            .finish()
71    }
72}
73
74impl<'a> From<ConstObject<'a>> for Object<'a> {
75    fn from(value: ConstObject<'a>) -> Self {
76        let Cursor { id, store } = value.0;
77
78        let id = id.into();
79
80        Object(Cursor { id, store })
81    }
82}
83
84impl<'a> IntoIterator for Object<'a> {
85    type Item = ObjectField<'a>;
86
87    type IntoIter = Iter<'a, ObjectField<'a>>;
88
89    fn into_iter(self) -> Self::IntoIter {
90        self.fields()
91    }
92}
93
94// TODO: Make sure this ObjectField name has no obvious clashes
95// and rename if it does
96#[derive(Clone, Copy)]
97pub struct ObjectField<'a>(super::Cursor<'a, FieldId>);
98
99impl<'a> ObjectField<'a> {
100    pub fn name(&self) -> &'a str {
101        let store = self.0.store;
102        store.lookup(store.lookup(self.0.id).name)
103    }
104
105    pub fn name_span(&self) -> Span {
106        let store = self.0.store;
107        store.lookup(self.0.id).name_span
108    }
109
110    pub fn value(&self) -> Value<'a> {
111        let store = self.0.store;
112        store.read(store.lookup(self.0.id).value)
113    }
114}
115
116impl<'a> ValueStoreReader<'a> for ObjectField<'a> {
117    type Id = FieldId;
118}
119
120impl ValueStoreId for FieldId {
121    type Reader<'a> = ObjectField<'a>;
122
123    fn read(self, store: &super::ValueStore) -> Self::Reader<'_> {
124        ObjectField(super::Cursor { id: self, store })
125    }
126}
cynic_parser/values/objects.rs

cynic_parser/values/
objects.rs
