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

use std::collections::HashMap;

use sea_orm::JsonValue;
use serde::{Deserialize, Serialize};

use super::{EdgeJsonBatch, NodeJsonBatch};

/// Structure of a mutate request, deserialized as struct from json
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum MutateJson {
    /// Insert new data; Use option "upsert" to allow insert-or-update
    Insert(MutateInsertJson),
    /// Update selected data
    Update(MutateUpdateJson),
    /// Delete selected data
    Delete(MutateDeleteJson),
    /// Calculate and store the connectitivties of relations with the supplied *unformatted* names
    CalConn(Vec<String>),
}

/// Structure of a mutate insert request
#[derive(Debug, Clone, Serialize, Deserialize)]
#[allow(non_camel_case_types)]
pub enum MutateInsertJson {
    /// Insert nodes
    node(NodeJsonBatch),
    /// Insert edges
    edge(EdgeJsonBatch),
}

/// Structure of a mutate update request
#[derive(Debug, Clone, Serialize, Deserialize)]
#[allow(non_camel_case_types)]
pub enum MutateUpdateJson {
    /// Update nodes
    node {
        /// Selector to select nodes for updating
        selector: MutateNodeSelectorJson,
        /// Specify how to update the selected nodes
        content: HashMap<String, JsonValue>,
    },
    /// Update edges
    edge {
        /// Selector to select edges for updating
        selector: MutateEdgeSelectorJson,
        /// Specify how to update the selected edges
        content: MutateEdgeContentJson,
    },
}

/// Structure of a mutate delete request
#[derive(Debug, Clone, Serialize, Deserialize)]
#[allow(non_camel_case_types)]
pub enum MutateDeleteJson {
    /// Delete nodes
    node(MutateNodeSelectorJson),
    /// Delete edges
    edge(MutateEdgeSelectorJson),
}

/// Structure of a node selector of a mutate update/delete request
#[derive(Debug, Clone, Serialize, Deserialize)]
#[allow(non_camel_case_types)]
pub struct MutateNodeSelectorJson {
    /// Name of entity this node belongs to
    pub of: String,
    /// Name of this node
    pub name: Option<String>,
    /// Additional attributes of node
    #[serde(default)]
    pub attributes: HashMap<String, JsonValue>,
}

/// Structure of an edge selector of a mutate update/delete request
#[derive(Debug, Clone, Serialize, Deserialize)]
#[allow(non_camel_case_types)]
pub struct MutateEdgeSelectorJson {
    /// Name of relation this edge belongs to
    pub of: String,
    /// Specify what edges to look for
    #[serde(flatten)]
    pub edge_content: MutateEdgeContentJson,
}

/// Structure of the update content of a mutate update request
#[derive(Debug, Clone, Serialize, Deserialize)]
#[allow(non_camel_case_types)]
pub struct MutateEdgeContentJson {
    /// Name of related node (from side, if any)
    pub from_node: Option<String>,
    /// Name of related node (to side, if any)
    pub to_node: Option<String>,
}