cynic_parser/values/
objects.rs

1use std::fmt;
2
3use crate::{AstLookup, Span};
4
5use super::{
6    const_objects::ConstObject,
7    ids::{FieldId, StringId, ValueId},
8    iter::{Iter, ValueStoreReader},
9    value::Value,
10    Cursor, ValueStoreId,
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 fmt::Debug for Object<'_> {
65    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
66        f.debug_map()
67            .entries(self.fields().map(|field| (field.name(), field.value())))
68            .finish()
69    }
70}
71
72impl<'a> From<ConstObject<'a>> for Object<'a> {
73    fn from(value: ConstObject<'a>) -> Self {
74        let Cursor { id, store } = value.0;
75
76        let id = id.into();
77
78        Object(Cursor { id, store })
79    }
80}
81
82impl<'a> IntoIterator for Object<'a> {
83    type Item = ObjectField<'a>;
84
85    type IntoIter = Iter<'a, ObjectField<'a>>;
86
87    fn into_iter(self) -> Self::IntoIter {
88        self.fields()
89    }
90}
91
92// TODO: Make sure this ObjectField name has no obvious clashes
93// and rename if it does
94#[derive(Clone, Copy)]
95pub struct ObjectField<'a>(super::Cursor<'a, FieldId>);
96
97impl<'a> ObjectField<'a> {
98    pub fn name(&self) -> &'a str {
99        let store = self.0.store;
100        store.lookup(store.lookup(self.0.id).name)
101    }
102
103    pub fn name_span(&self) -> Span {
104        let store = self.0.store;
105        store.lookup(self.0.id).name_span
106    }
107
108    pub fn value(&self) -> Value<'a> {
109        let store = self.0.store;
110        store.read(store.lookup(self.0.id).value)
111    }
112}
113
114impl<'a> ValueStoreReader<'a> for ObjectField<'a> {
115    type Id = FieldId;
116}
117
118impl ValueStoreId for FieldId {
119    type Reader<'a> = ObjectField<'a>;
120
121    fn read(self, store: &super::ValueStore) -> Self::Reader<'_> {
122        ObjectField(super::Cursor { id: self, store })
123    }
124}
cynic_parser/values/objects.rs

cynic_parser/values/
objects.rs
