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 crate::DbImpl;
use crate::QueryError;
use crate::QueryIds;
use crate::QueryMut;
use crate::QueryResult;
use crate::StorageData;

/// Query to insert or update aliases of existing nodes.
/// All `ids` must exist. None of the `aliases` can be empty.
/// If there is an existing alias for any of the elements it
/// will be overwritten with a new one.
///
/// NOTE: Setting `ids` to a search query will result in an error.
///
/// The result will contain number of aliases inserted/updated but no elements.
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
#[derive(Debug, PartialEq)]
pub struct InsertAliasesQuery {
    /// Ids to be aliased
    pub ids: QueryIds,

    /// Aliases to be inserted
    pub aliases: Vec<String>,
}

impl QueryMut for InsertAliasesQuery {
    fn process<Store: StorageData>(
        &self,
        db: &mut DbImpl<Store>,
    ) -> Result<QueryResult, QueryError> {
        let mut result = QueryResult::default();

        match &self.ids {
            QueryIds::Ids(ids) => {
                if ids.len() != self.aliases.len() {
                    return Err(QueryError::from(
                        "Ids and aliases must have the same length",
                    ));
                }

                for (id, alias) in ids.iter().zip(&self.aliases) {
                    if alias.is_empty() {
                        return Err(QueryError::from("Empty alias is not allowed"));
                    }

                    let db_id = db.db_id(id)?;
                    db.insert_alias(db_id, alias)?;
                    result.result += 1;
                }
            }
            QueryIds::Search(_) => {
                return Err(QueryError::from(
                    "Insert aliases query does not support search queries",
                ));
            }
        }

        Ok(result)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::query::query_id::QueryId;
    use crate::query::search_query::SearchQuery;
    use crate::test_utilities::test_file::TestFile;
    use crate::Db;
    use crate::DbId;
    use crate::SearchQueryAlgorithm;

    #[test]
    fn invalid_query() {
        let test_file = TestFile::new();
        let mut db = Db::new(test_file.file_name()).unwrap();
        let query = InsertAliasesQuery {
            ids: QueryIds::Search(SearchQuery {
                algorithm: SearchQueryAlgorithm::BreadthFirst,
                origin: QueryId::Id(DbId(0)),
                destination: QueryId::Id(DbId(0)),
                limit: 0,
                offset: 0,
                order_by: vec![],
                conditions: vec![],
            }),
            aliases: vec![],
        };
        assert_eq!(
            query.process(&mut db).unwrap_err(),
            QueryError::from("Insert aliases query does not support search queries")
        );
    }
}