gatelogue-types 3.1.3+13

Types for loading and reading Gatelogue data
Documentation 
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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

use rusqlite::{Connection, Params, Row};

use crate::error::Result;

pub type ID = u32;

#[macro_export]
#[doc(hidden)]
macro_rules! _get_column {
    ($table_name:literal, $column_name:ident, $ColTy:ty) => {
        pub fn $column_name(self, gd: &$crate::GD) -> $crate::error::Result<$ColTy> {
            #[expect(clippy::allow_attributes)]
            #[allow(unused_imports)]
            use $crate::node::Node;
            gd.0.query_one(
                concat!(
                    "SELECT \"",
                    stringify!($column_name),
                    "\" FROM ",
                    $table_name,
                    " WHERE i = ?"
                ),
                (self.i(),),
                |a| a.get(0),
            )
            .map_err(|e| {
                if e == rusqlite::Error::QueryReturnedNoRows {
                    $crate::error::Error::NoNodeOfType(self.i(), self.ty())
                } else {
                    e.into()
                }
            })
        }
    };
    ($table_name:literal, $fn_name:ident, $column_name:literal, $ColTy:ty) => {
        pub fn $fn_name(self, gd: &$crate::GD) -> $crate::error::Result<$ColTy> {
            #[expect(clippy::allow_attributes)]
            #[allow(unused_imports)]
            use $crate::node::Node;
            gd.0.query_one(
                concat!(
                    "SELECT \"",
                    $column_name,
                    "\" FROM ",
                    $table_name,
                    " WHERE i = ?"
                ),
                (self.i(),),
                |a| a.get(0),
            )
            .map_err(|e| {
                if e == rusqlite::Error::QueryReturnedNoRows {
                    $crate::error::Error::NoNodeOfType(self.i(), self.ty())
                } else {
                    e.into()
                }
            })
        }
    };
}
#[macro_export]
#[doc(hidden)]
macro_rules! _get_set {
    ($table_name:literal, $fn_name:ident, $column_name:literal, $ColTy:ty) => {
        pub fn $fn_name(self, gd: &$crate::GD) -> $crate::error::Result<Vec<$ColTy>> {
            #[expect(clippy::allow_attributes)]
            #[allow(unused_imports)]
            use $crate::node::Node;
            match gd
                .0
                .prepare_cached(concat!(
                    "SELECT DISTINCT \"",
                    $column_name,
                    "\" FROM ",
                    $table_name,
                    " WHERE i = ?"
                ))?
                .query_and_then((self.i(),), |a| a.get(0).map_err(Into::into))
            {
                Ok(a) => Ok(a.collect::<$crate::error::Result<Vec<_>>>()?),
                Err(e) => Err(e.into()),
            }
        }
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! _get_derived_vec {
    ($fn_name:ident, $RetTy:ty, $key:literal) => {
        pub fn $fn_name(self, gd: &$crate::GD) -> $crate::error::Result<Vec<$RetTy>> {
            use $crate::util::ConnectionExt;
            gd.0.query_and_then_get_vec(include_str!($key), (self.0,), |row| Ok(row.get(0)?))
        }
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! _from_sql_for_enum {
    ($Ty:ty) => {
        impl rusqlite::types::FromSql for $Ty {
            fn column_result(
                value: rusqlite::types::ValueRef<'_>,
            ) -> rusqlite::types::FromSqlResult<Self> {
                use std::str::FromStr;
                Self::from_str(value.as_str()?)
                    .map_err(|e| rusqlite::types::FromSqlError::Other(e.into()))
            }
        }
    };
}

pub trait ConnectionExt {
    fn query_and_then_get_vec<T, P, F>(&self, sql: &str, params: P, f: F) -> Result<Vec<T>>
    where
        P: Params,
        F: FnMut(&Row<'_>) -> Result<T>;
}
impl ConnectionExt for Connection {
    fn query_and_then_get_vec<T, P, F>(&self, sql: &str, params: P, f: F) -> Result<Vec<T>>
    where
        P: Params,
        F: FnMut(&Row<'_>) -> Result<T>,
    {
        match self.prepare(sql)?.query_and_then(params, f) {
            Ok(a) => Ok(a.collect::<Result<Vec<_>>>()?),
            Err(e) => Err(e.into()),
        }
    }
}