systemprompt_users/
extension.rs

1//! `Extension` registration for the users domain: schema and required flag.
2//!
3//! Copyright (c) systemprompt.io — Business Source License 1.1.
4//! See <https://systemprompt.io> for licensing details.
5
6use systemprompt_extension::prelude::*;
7
8#[derive(Debug, Clone, Copy, Default)]
9pub struct UsersExtension;
10
11impl Extension for UsersExtension {
12    fn metadata(&self) -> ExtensionMetadata {
13        ExtensionMetadata {
14            id: "users",
15            name: "Users",
16            version: env!("CARGO_PKG_VERSION"),
17        }
18    }
19
20    fn is_required(&self) -> bool {
21        true
22    }
23
24    fn schemas(&self) -> Vec<SchemaDefinition> {
25        vec![
26            SchemaDefinition::new("users", include_str!("../schema/users.sql"))
27                .with_required_columns(vec![
28                    "id".into(),
29                    "name".into(),
30                    "email".into(),
31                    "created_at".into(),
32                ]),
33            SchemaDefinition::new("user_sessions", include_str!("../schema/user_sessions.sql"))
34                .with_required_columns(vec!["session_id".into(), "started_at".into()]),
35            SchemaDefinition::new("banned_ips", include_str!("../schema/banned_ips.sql"))
36                .with_required_columns(vec![
37                    "ip_address".into(),
38                    "reason".into(),
39                    "banned_at".into(),
40                ]),
41            SchemaDefinition::new("user_api_keys", include_str!("../schema/user_api_keys.sql"))
42                .with_required_columns(vec![
43                    "id".into(),
44                    "user_id".into(),
45                    "key_prefix".into(),
46                    "key_hash".into(),
47                    "model_allowlist".into(),
48                    "request_window_seconds".into(),
49                ]),
50            SchemaDefinition::new(
51                "user_api_key_scopes",
52                include_str!("../schema/user_api_key_scopes.sql"),
53            )
54            .with_required_columns(vec![
55                "key_id".into(),
56                "dimension".into(),
57                "value".into(),
58            ]),
59            SchemaDefinition::new(
60                "user_device_certs",
61                include_str!("../schema/user_device_certs.sql"),
62            )
63            .with_required_columns(vec![
64                "id".into(),
65                "user_id".into(),
66                "fingerprint".into(),
67                "label".into(),
68            ]),
69            SchemaDefinition::new(
70                "user_rate_limit_buckets",
71                include_str!("../schema/user_rate_limit_buckets.sql"),
72            )
73            .with_required_columns(vec![
74                "user_id".into(),
75                "scope".into(),
76                "window_start".into(),
77                "hits".into(),
78            ]),
79            SchemaDefinition::new(
80                "federated_identities",
81                include_str!("../schema/federated_identities.sql"),
82            )
83            .with_required_columns(vec![
84                "issuer".into(),
85                "external_sub".into(),
86                "user_id".into(),
87            ]),
88        ]
89    }
90
91    fn dependencies(&self) -> Vec<&'static str> {
92        vec!["events"]
93    }
94
95    fn migrations(&self) -> Vec<Migration> {
96        extension_migrations!()
97    }
98}
99
100register_extension!(UsersExtension);
systemprompt_users/extension.rs

systemprompt_users/
extension.rs
