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

use enumflags2::*;

macro_rules! tags {
    ($($name:ident = $pattern:expr,)*) => {
        /// Test-relevant connector tags.
        #[bitflags]
        #[derive(Copy, Clone, Debug, PartialEq)]
        #[repr(u32)]
        pub enum Tags {
            $($name = $pattern,)*
        }

        const ALL_TAG_NAMES: &[(&str, Tags)] = &[
            $(
                (stringify!($name), Tags::$name),
            )*
        ];
    }
}

tags![
    LowerCasesTableNames = 1 << 0,
    Mysql = 1 << 1,
    Mariadb = 1 << 2,
    Postgres = 1 << 3,
    Sqlite = 1 << 4,
    Mysql8 = 1 << 5,
    Mysql56 = 1 << 6,
    Mysql57 = 1 << 7,
    Mssql2017 = 1 << 8,
    Mssql2019 = 1 << 9,
    Postgres12 = 1 << 10,
    Mssql = 1 << 11,
    Vitess = 1 << 12,
    CockroachDb = 1 << 13,
    Postgres14 = 1 << 14,
    Postgres9 = 1 << 15,
    Postgres15 = 1 << 16,
    Postgres11 = 1 << 17,
    Postgres13 = 1 << 18,
    CockroachDb221 = 1 << 19,
    CockroachDb222 = 1 << 20,
    CockroachDb231 = 1 << 21,
    Postgres16 = 1 << 22,
];

pub fn tags_from_comma_separated_list(input: &str) -> BitFlags<Tags> {
    let mut tags = Default::default();

    for s in input.split(',').map(|s| s.trim()) {
        match ALL_TAG_NAMES.iter().find(|(name, _t)| name.eq_ignore_ascii_case(s)) {
            Some((_, tag)) => tags |= *tag,
            None => panic!("unknown tag: {s}"),
        }
    }

    tags
}