memtable-macros 0.2.0

Macro library that provides ability to derive typed tables and generate tables compile-time.
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

use super::{TableColumn, TableMode};
use syn::{parse_quote, Generics, Ident, ItemImpl, Path, Type};

pub struct Args<'a> {
    pub root: &'a Path,
    pub mode: TableMode,
    pub table_name: &'a Ident,
    pub generics: &'a Generics,
    pub table_data_name: &'a Ident,
    pub columns: &'a [&'a TableColumn],
}

pub fn make(args: Args) -> ItemImpl {
    let Args {
        root,
        mode,
        table_name,
        generics,
        table_data_name,
        columns,
    } = args;

    let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
    let cols = columns.len();

    let row_t: Type = match mode {
        TableMode::Dynamic => {
            parse_quote!(#root::list::DynamicList<Self::Data>)
        }
        TableMode::Fixed { .. } | TableMode::FixedColumn => {
            parse_quote!(#root::list::FixedList<Self::Data, #cols>)
        }
    };
    let col_t: Type = match mode {
        TableMode::Dynamic | TableMode::FixedColumn => {
            parse_quote!(#root::list::DynamicList<Self::Data>)
        }
        TableMode::Fixed { rows } => parse_quote!(#root::list::FixedList<Self::Data, #rows>),
    };

    parse_quote! {
        impl #impl_generics #root::Table for #table_name #ty_generics #where_clause {
            type Data = #table_data_name #ty_generics;
            type Row = #row_t;
            type Column = #col_t;

            fn max_row_capacity(&self) -> #root::Capacity {
                #root::Table::max_row_capacity(&self.0)
            }

            fn max_column_capacity(&self) -> #root::Capacity {
                #root::Table::max_column_capacity(&self.0)
            }

            fn row_cnt(&self) -> ::core::primitive::usize {
                #root::Table::row_cnt(&self.0)
            }

            fn col_cnt(&self) -> ::core::primitive::usize {
                #root::Table::col_cnt(&self.0)
            }

            fn cell(
                &self,
                row: ::core::primitive::usize,
                col: ::core::primitive::usize,
            ) -> ::core::option::Option<&Self::Data> {
                #root::Table::cell(&self.0, row, col)
            }

            fn mut_cell(
                &mut self,
                row: ::core::primitive::usize,
                col: ::core::primitive::usize,
            ) -> ::core::option::Option<&mut Self::Data> {
                #root::Table::mut_cell(&mut self.0, row, col)
            }

            fn insert_cell(
                &mut self,
                row: ::core::primitive::usize,
                col: ::core::primitive::usize,
                value: Self::Data,
            ) -> ::core::option::Option<Self::Data> {
                #root::Table::insert_cell(&mut self.0, row, col, value)
            }

            fn remove_cell(
                &mut self,
                row: ::core::primitive::usize,
                col: ::core::primitive::usize,
            ) -> ::core::option::Option<Self::Data> {
                #root::Table::remove_cell(&mut self.0, row, col)
            }

            fn set_preferred_row_cnt(&mut self, capacity: ::core::primitive::usize) {
                #root::Table::set_preferred_row_cnt(&mut self.0, capacity);
            }

            fn set_preferred_col_cnt(&mut self, capacity: ::core::primitive::usize) {
                #root::Table::set_preferred_col_cnt(&mut self.0, capacity);
            }
        }
    }
}