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

use crate::structure::{GValue, Property, GID};
use crate::{GremlinError, GremlinResult};

use crate::conversion::{BorrowFromGValue, FromGValue};

#[derive(Debug, PartialEq, Clone)]
pub enum GProperty {
    VertexProperty(VertexProperty),
    Property(Property),
}

impl GProperty {
    pub fn value(&self) -> &GValue {
        match self {
            GProperty::Property(p) => p.value(),
            GProperty::VertexProperty(p) => p.value(),
        }
    }

    pub fn take<T>(self) -> GremlinResult<T>
    where
        T: FromGValue,
    {
        match self {
            GProperty::Property(p) => p.take(),
            GProperty::VertexProperty(p) => p.take(),
        }
    }

    pub fn get<'a, T>(&'a self) -> GremlinResult<&'a T>
    where
        T: BorrowFromGValue,
    {
        match self {
            GProperty::Property(p) => p.get(),
            GProperty::VertexProperty(p) => p.get(),
        }
    }

    pub fn label(&self) -> &String {
        match self {
            GProperty::Property(p) => p.label(),
            GProperty::VertexProperty(p) => p.label(),
        }
    }
}

impl FromGValue for GProperty {
    fn from_gvalue(v: GValue) -> GremlinResult<Self> {
        match v {
            GValue::VertexProperty(p) => Ok(GProperty::VertexProperty(p)),
            GValue::Property(p) => Ok(GProperty::Property(p)),
            _ => Err(GremlinError::Cast(String::from(
                "Value not allowed for a property",
            ))),
        }
    }
}
#[derive(Debug, PartialEq, Clone)]
pub struct VertexProperty {
    label: String,
    id: GID,
    value: Box<GValue>,
}

impl VertexProperty {
    pub fn new<G, T, GT>(id: G, label: T, value: GT) -> VertexProperty
    where
        G: Into<GID>,
        T: Into<String>,
        GT: Into<GValue>,
    {
        VertexProperty {
            id: id.into(),
            label: label.into(),
            value: Box::new(value.into()),
        }
    }

    pub fn id(&self) -> &GID {
        &self.id
    }

    pub fn value(&self) -> &GValue {
        &self.value
    }

    pub fn take<T>(self) -> GremlinResult<T>
    where
        T: FromGValue,
    {
        T::from_gvalue(*self.value)
    }

    pub fn get<'a, T>(&'a self) -> GremlinResult<&'a T>
    where
        T: BorrowFromGValue,
    {
        T::from_gvalue(&self.value)
    }
    pub fn label(&self) -> &String {
        &self.label
    }
}