systemprompt_users/
extension.rs1use 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 ]),
48 SchemaDefinition::new(
49 "user_device_certs",
50 include_str!("../schema/user_device_certs.sql"),
51 )
52 .with_required_columns(vec![
53 "id".into(),
54 "user_id".into(),
55 "fingerprint".into(),
56 "label".into(),
57 ]),
58 SchemaDefinition::new(
59 "federated_identities",
60 include_str!("../schema/federated_identities.sql"),
61 )
62 .with_required_columns(vec![
63 "issuer".into(),
64 "external_sub".into(),
65 "user_id".into(),
66 ]),
67 ]
68 }
69
70 fn dependencies(&self) -> Vec<&'static str> {
71 vec![]
72 }
73
74 fn migrations(&self) -> Vec<Migration> {
75 extension_migrations!()
76 }
77}
78
79register_extension!(UsersExtension);