alibi-cli 0.3.0

Schema generator for Alibi authentication
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

use alibi_schema_registry::{self as registry, EntityRole, ExtraEntitySchema, FieldDef};
use std::collections::HashSet;

pub(crate) fn list_plugins() -> Vec<&'static str> {
    registry::plugin_schemas().iter().map(|p| p.name).collect()
}

/// Core and selected plugin fields of every generated entity.
pub(crate) struct Selection {
    pub(crate) user: Vec<FieldDef>,
    pub(crate) session: Vec<FieldDef>,
    pub(crate) extra: Vec<&'static ExtraEntitySchema>,
}

pub(crate) fn select(plugins: &[String]) -> Selection {
    let mut selection = Selection {
        user: Vec::new(),
        session: Vec::new(),
        extra: Vec::new(),
    };
    let mut selected = HashSet::new();
    for plugin_name in plugins {
        if !selected.insert(plugin_name.as_str()) {
            continue;
        }
        if let Some(schema) = registry::plugin_schemas()
            .iter()
            .find(|p| p.name == plugin_name.as_str())
        {
            selection.user.extend_from_slice(schema.user_fields);
            selection.session.extend_from_slice(schema.session_fields);
            selection.extra.extend(schema.extra_entities.iter());
        }
    }
    selection
}

pub(crate) const CORE_ENTITIES: [(&str, &str, EntityRole); 4] = [
    ("user", "users", EntityRole::User),
    ("session", "sessions", EntityRole::Session),
    ("account", "accounts", EntityRole::Account),
    ("verification", "verifications", EntityRole::Verification),
];

pub(crate) fn role_name(role: EntityRole) -> &'static str {
    match role {
        EntityRole::User => "user",
        EntityRole::Session => "session",
        EntityRole::Account => "account",
        EntityRole::Verification => "verification",
    }
}

/// Fields of a core entity: its core fields, then the selected plugin fields.
pub(crate) fn entity_fields(selection: &Selection, role: EntityRole) -> Vec<&FieldDef> {
    let plugin: &[FieldDef] = match role {
        EntityRole::User => &selection.user,
        EntityRole::Session => &selection.session,
        EntityRole::Account | EntityRole::Verification => &[],
    };
    registry::core_fields(role).iter().chain(plugin).collect()
}