From 233e26520c2de0a13d787c1d9d9e243d240658a4 Mon Sep 17 00:00:00 2001
From: Philippe Loctaux
Date: Tue, 25 Apr 2023 23:37:24 +0200
Subject: [PATCH] totp: new crate, sql migration + queries, enable totp page,
save secret in database
---
Cargo.lock | 151 +++-
.../20230423210500_users_totp.down.sql | 4 +
.../20230423210500_users_totp.up.sql | 4 +
crates/database/queries/users/get_all.sql | 4 +-
.../queries/users/get_initial_admin.sql | 4 +-
.../queries/users/get_one_by_email.sql | 4 +-
.../database/queries/users/get_one_by_id.sql | 4 +-
.../queries/users/get_one_by_username.sql | 4 +-
.../users/get_one_from_authorization_code.sql | 4 +-
.../get_one_from_password_reset_token.sql | 4 +-
.../users/get_one_from_refresh_token.sql | 4 +-
.../queries/users/set_totp_backup.sql | 5 +
.../queries/users/set_totp_secret.sql | 5 +
crates/database/sqlx-data.json | 822 ++++++++++--------
crates/database/src/tables/users.rs | 30 +
crates/ezidam/Cargo.toml | 3 +-
crates/ezidam/src/error/conversion.rs | 20 +
crates/ezidam/src/page.rs | 5 +
crates/ezidam/src/routes/settings.rs | 12 +
crates/ezidam/src/routes/settings/security.rs | 182 ++++
.../pages/settings/security.html.tera | 66 ++
.../templates/pages/settings/totp.html.tera | 70 ++
crates/totp/Cargo.toml | 8 +
crates/totp/src/lib.rs | 34 +
crates/users/src/database.rs | 22 +
crates/users/src/lib.rs | 5 +
26 files changed, 1116 insertions(+), 364 deletions(-)
create mode 100644 crates/database/migrations/20230423210500_users_totp.down.sql
create mode 100644 crates/database/migrations/20230423210500_users_totp.up.sql
create mode 100644 crates/database/queries/users/set_totp_backup.sql
create mode 100644 crates/database/queries/users/set_totp_secret.sql
create mode 100644 crates/ezidam/templates/pages/settings/totp.html.tera
create mode 100644 crates/totp/Cargo.toml
create mode 100644 crates/totp/src/lib.rs
diff --git a/Cargo.lock b/Cargo.lock
index 2abe08f..43756a0 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -12,6 +12,12 @@ dependencies = [
"regex",
]
+[[package]]
+name = "adler"
+version = "1.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f26201604c87b1e01bd3d98f8d5d9a8fcbb815e8cedb41ffccbeb4bf593a35fe"
+
[[package]]
name = "ahash"
version = "0.7.6"
@@ -145,6 +151,12 @@ version = "0.1.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "349a06037c7bf932dd7e7d1f653678b2038b9ad46a74102f1fc7bd7872678cce"
+[[package]]
+name = "base32"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "23ce669cd6c8588f79e15cf450314f9638f967fc5770ff1c7c1deb0925ea7cfa"
+
[[package]]
name = "base64"
version = "0.13.1"
@@ -215,6 +227,12 @@ version = "3.12.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0d261e256854913907f67ed06efbc3338dfe6179796deefc1ff763fc1aee5535"
+[[package]]
+name = "bytemuck"
+version = "1.13.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "17febce684fd15d89027105661fec94afb475cb995fbc59d2865198446ba2eea"
+
[[package]]
name = "byteorder"
version = "1.4.3"
@@ -309,12 +327,24 @@ dependencies = [
"unicode-width",
]
+[[package]]
+name = "color_quant"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3d7b894f5411737b7867f4827955924d7c254fc9f4d91a6aad6b097804b1018b"
+
[[package]]
name = "const-oid"
version = "0.9.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "520fbf3c07483f94e3e3ca9d0cfd913d7718ef2483d2cfd91c0d9e91474ab913"
+[[package]]
+name = "constant_time_eq"
+version = "0.2.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "13418e745008f7349ec7e449155f419a61b92b58a99cc3616942b926825ec76b"
+
[[package]]
name = "cookie"
version = "0.17.0"
@@ -356,6 +386,15 @@ version = "2.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9cace84e55f07e7301bae1c519df89cdad8cc3cd868413d3fdbdeca9ff3db484"
+[[package]]
+name = "crc32fast"
+version = "1.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b540bd8bc810d3885c6ea91e2018302f68baba2129ab3e88f32389ee9370880d"
+dependencies = [
+ "cfg-if",
+]
+
[[package]]
name = "crossbeam-channel"
version = "0.5.7"
@@ -749,6 +788,7 @@ dependencies = [
"rocket_db_pools",
"rocket_dyn_templates",
"settings",
+ "totp",
"url",
"users",
]
@@ -798,6 +838,16 @@ dependencies = [
"windows-sys 0.45.0",
]
+[[package]]
+name = "flate2"
+version = "1.0.25"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a8a2db397cb1c8772f31494cb8917e48cd1e64f0fa7efac59fbd741a0a8ce841"
+dependencies = [
+ "crc32fast",
+ "miniz_oxide",
+]
+
[[package]]
name = "flume"
version = "0.10.14"
@@ -1268,6 +1318,20 @@ dependencies = [
"winapi-util",
]
+[[package]]
+name = "image"
+version = "0.24.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "527909aa81e20ac3a44803521443a765550f09b5130c2c2fa1ea59c2f8f50a3a"
+dependencies = [
+ "bytemuck",
+ "byteorder",
+ "color_quant",
+ "num-rational",
+ "num-traits",
+ "png",
+]
+
[[package]]
name = "indexmap"
version = "1.9.2"
@@ -1566,6 +1630,15 @@ version = "0.2.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "68354c5c6bd36d73ff3feceb05efa59b6acb7626617f4962be322a825e61f79a"
+[[package]]
+name = "miniz_oxide"
+version = "0.6.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b275950c28b37e794e8c55d88aeb5e139d0ce23fdbbeda68f8d7174abdf9e8fa"
+dependencies = [
+ "adler",
+]
+
[[package]]
name = "mio"
version = "0.8.6"
@@ -1786,6 +1859,17 @@ dependencies = [
"num-traits",
]
+[[package]]
+name = "num-rational"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0638a1c9d0a3c0914158145bc76cff373a75a627e6ecbfb71cbe6f453a5a19b0"
+dependencies = [
+ "autocfg",
+ "num-integer",
+ "num-traits",
+]
+
[[package]]
name = "num-traits"
version = "0.2.15"
@@ -2204,6 +2288,18 @@ version = "0.3.26"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6ac9a59f73473f1b8d852421e59e64809f025994837ef743615c6d0c5b305160"
+[[package]]
+name = "png"
+version = "0.17.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5d708eaf860a19b19ce538740d2b4bdeeb8337fa53f7738455e706623ad5c638"
+dependencies = [
+ "bitflags 1.3.2",
+ "crc32fast",
+ "flate2",
+ "miniz_oxide",
+]
+
[[package]]
name = "ppv-lite86"
version = "0.2.17"
@@ -2232,6 +2328,12 @@ dependencies = [
"yansi",
]
+[[package]]
+name = "qrcodegen"
+version = "1.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4339fc7a1021c9c1621d87f5e3505f2805c8c105420ba2f2a4df86814590c142"
+
[[package]]
name = "quote"
version = "1.0.26"
@@ -2771,6 +2873,17 @@ dependencies = [
"url",
]
+[[package]]
+name = "sha1"
+version = "0.10.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f04293dc80c3993519f2d7f6f511707ee7094fe0c6d3406feb330cdb3540eba3"
+dependencies = [
+ "cfg-if",
+ "cpufeatures",
+ "digest",
+]
+
[[package]]
name = "sha2"
version = "0.10.6"
@@ -3273,6 +3386,34 @@ dependencies = [
"serde",
]
+[[package]]
+name = "totp"
+version = "0.0.0"
+dependencies = [
+ "thiserror",
+ "totp-rs",
+]
+
+[[package]]
+name = "totp-rs"
+version = "5.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "332e333b188e843cb4cc477b2911160a533bcfc6e9e488d7bef25011f9e2ba1b"
+dependencies = [
+ "base32",
+ "base64 0.21.0",
+ "constant_time_eq",
+ "hmac",
+ "image",
+ "qrcodegen",
+ "rand",
+ "sha1",
+ "sha2",
+ "url",
+ "urlencoding",
+ "zeroize",
+]
+
[[package]]
name = "tower-service"
version = "0.3.2"
@@ -3510,6 +3651,12 @@ dependencies = [
"serde",
]
+[[package]]
+name = "urlencoding"
+version = "2.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e8db7427f936968176eaa7cdf81b7f98b980b18495ec28f1b5791ac3bfe3eea9"
+
[[package]]
name = "users"
version = "0.0.0"
@@ -3810,9 +3957,9 @@ checksum = "09041cd90cf85f7f8b2df60c646f853b7f535ce68f85244eb6731cf89fa498ec"
[[package]]
name = "zeroize"
-version = "1.5.7"
+version = "1.6.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "c394b5bd0c6f669e7275d9c20aa90ae064cb22e75a1cad54e1b34088034b149f"
+checksum = "2a0956f1ba7c7909bfb66c2e9e4124ab6f6482560f6628b5aaeba39207c9aad9"
dependencies = [
"zeroize_derive",
]
diff --git a/crates/database/migrations/20230423210500_users_totp.down.sql b/crates/database/migrations/20230423210500_users_totp.down.sql
new file mode 100644
index 0000000..73bb79a
--- /dev/null
+++ b/crates/database/migrations/20230423210500_users_totp.down.sql
@@ -0,0 +1,4 @@
+alter table users
+ drop column totp_secret;
+alter table users
+ drop column totp_backup;
diff --git a/crates/database/migrations/20230423210500_users_totp.up.sql b/crates/database/migrations/20230423210500_users_totp.up.sql
new file mode 100644
index 0000000..0333e14
--- /dev/null
+++ b/crates/database/migrations/20230423210500_users_totp.up.sql
@@ -0,0 +1,4 @@
+alter table users
+ add column totp_secret BLOB;
+alter table users
+ add column totp_backup TEXT;
diff --git a/crates/database/queries/users/get_all.sql b/crates/database/queries/users/get_all.sql
index f2479ff..15a7548 100644
--- a/crates/database/queries/users/get_all.sql
+++ b/crates/database/queries/users/get_all.sql
@@ -9,7 +9,9 @@ select id,
password_recover,
paper_key,
is_archived as "is_archived: bool",
- timezone
+ timezone,
+ totp_secret,
+ totp_backup
from users
order by created_at desc
diff --git a/crates/database/queries/users/get_initial_admin.sql b/crates/database/queries/users/get_initial_admin.sql
index f9effa2..daa7d23 100644
--- a/crates/database/queries/users/get_initial_admin.sql
+++ b/crates/database/queries/users/get_initial_admin.sql
@@ -9,7 +9,9 @@ select u.id,
u.password_recover,
u.paper_key,
u.is_archived as "is_archived: bool",
- u.timezone
+ u.timezone,
+ u.totp_secret,
+ u.totp_backup
from users u
inner join settings s on u.id = s.first_admin
diff --git a/crates/database/queries/users/get_one_by_email.sql b/crates/database/queries/users/get_one_by_email.sql
index 3b20840..ad14533 100644
--- a/crates/database/queries/users/get_one_by_email.sql
+++ b/crates/database/queries/users/get_one_by_email.sql
@@ -9,7 +9,9 @@ select id,
password_recover,
paper_key,
is_archived as "is_archived: bool",
- timezone
+ timezone,
+ totp_secret,
+ totp_backup
from users
where email is (?)
diff --git a/crates/database/queries/users/get_one_by_id.sql b/crates/database/queries/users/get_one_by_id.sql
index a0dff1d..73c61f7 100644
--- a/crates/database/queries/users/get_one_by_id.sql
+++ b/crates/database/queries/users/get_one_by_id.sql
@@ -9,7 +9,9 @@ select id,
password_recover,
paper_key,
is_archived as "is_archived: bool",
- timezone
+ timezone,
+ totp_secret,
+ totp_backup
from users
where id is (?)
diff --git a/crates/database/queries/users/get_one_by_username.sql b/crates/database/queries/users/get_one_by_username.sql
index bd03c86..155d389 100644
--- a/crates/database/queries/users/get_one_by_username.sql
+++ b/crates/database/queries/users/get_one_by_username.sql
@@ -9,7 +9,9 @@ select id,
password_recover,
paper_key,
is_archived as "is_archived: bool",
- timezone
+ timezone,
+ totp_secret,
+ totp_backup
from users
where username is (?)
diff --git a/crates/database/queries/users/get_one_from_authorization_code.sql b/crates/database/queries/users/get_one_from_authorization_code.sql
index 3f13a1f..775f66f 100644
--- a/crates/database/queries/users/get_one_from_authorization_code.sql
+++ b/crates/database/queries/users/get_one_from_authorization_code.sql
@@ -9,7 +9,9 @@ select u.id,
u.password_recover,
u.paper_key,
u.is_archived as "is_archived: bool",
- u.timezone
+ u.timezone,
+ u.totp_secret,
+ u.totp_backup
from users u
inner join authorization_codes ac on u.id = ac.user
diff --git a/crates/database/queries/users/get_one_from_password_reset_token.sql b/crates/database/queries/users/get_one_from_password_reset_token.sql
index e2ff3c7..fbbafdb 100644
--- a/crates/database/queries/users/get_one_from_password_reset_token.sql
+++ b/crates/database/queries/users/get_one_from_password_reset_token.sql
@@ -9,7 +9,9 @@ select id,
password_recover,
paper_key,
is_archived as "is_archived: bool",
- timezone
+ timezone,
+ totp_secret,
+ totp_backup
from users
where password_recover is (?)
diff --git a/crates/database/queries/users/get_one_from_refresh_token.sql b/crates/database/queries/users/get_one_from_refresh_token.sql
index bedf5fc..1468cbb 100644
--- a/crates/database/queries/users/get_one_from_refresh_token.sql
+++ b/crates/database/queries/users/get_one_from_refresh_token.sql
@@ -9,7 +9,9 @@ select u.id,
u.password_recover,
u.paper_key,
u.is_archived as "is_archived: bool",
- u.timezone
+ u.timezone,
+ u.totp_secret,
+ u.totp_backup
from users u
inner join refresh_tokens rt on u.id = rt.user
diff --git a/crates/database/queries/users/set_totp_backup.sql b/crates/database/queries/users/set_totp_backup.sql
new file mode 100644
index 0000000..32545f8
--- /dev/null
+++ b/crates/database/queries/users/set_totp_backup.sql
@@ -0,0 +1,5 @@
+update users
+
+set totp_backup = ?
+
+where id is ?
\ No newline at end of file
diff --git a/crates/database/queries/users/set_totp_secret.sql b/crates/database/queries/users/set_totp_secret.sql
new file mode 100644
index 0000000..92ab2d6
--- /dev/null
+++ b/crates/database/queries/users/set_totp_secret.sql
@@ -0,0 +1,5 @@
+update users
+
+set totp_secret = ?
+
+where id is ?
\ No newline at end of file
diff --git a/crates/database/sqlx-data.json b/crates/database/sqlx-data.json
index 69b1c18..f9e28a9 100644
--- a/crates/database/sqlx-data.json
+++ b/crates/database/sqlx-data.json
@@ -120,7 +120,7 @@
},
"query": "update users\n\nset password_recover = ?\n\nwhere id is ?"
},
- "35de1a35e6cf6c683a1b2ca3605791aea9cbb852ac1d3df151cc21c341046361": {
+ "37000307a757b4912ca300c346b5bd8ee5d50331d94abeee076ae4db46215a56": {
"describe": {
"columns": [
{
@@ -182,6 +182,16 @@
"name": "timezone",
"ordinal": 11,
"type_info": "Text"
+ },
+ {
+ "name": "totp_secret",
+ "ordinal": 12,
+ "type_info": "Blob"
+ },
+ {
+ "name": "totp_backup",
+ "ordinal": 13,
+ "type_info": "Text"
}
],
"nullable": [
@@ -196,13 +206,15 @@
true,
true,
false,
- false
+ false,
+ true,
+ true
],
"parameters": {
"Right": 1
}
},
- "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone\nfrom users\n\nwhere id is (?)\n"
+ "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone,\n totp_secret,\n totp_backup\nfrom users\n\nwhere email is (?)\n"
},
"37681902a5f5d87492812a525a6488e75d20c1c436a3ba2c5aa3f54da62fe861": {
"describe": {
@@ -294,7 +306,7 @@
},
"query": "update users\n\nset username = ?\n\nwhere id is ?"
},
- "56a88e7e68cfa94a055008510e3bc4389d7a7f64b43479d5fc8e4495ade0f84a": {
+ "5567f75b47f739e85da3469bec0202be315c42dfc820d3c37bba498b1b783811": {
"describe": {
"columns": [
{
@@ -356,6 +368,16 @@
"name": "timezone",
"ordinal": 11,
"type_info": "Text"
+ },
+ {
+ "name": "totp_secret",
+ "ordinal": 12,
+ "type_info": "Blob"
+ },
+ {
+ "name": "totp_backup",
+ "ordinal": 13,
+ "type_info": "Text"
}
],
"nullable": [
@@ -370,13 +392,15 @@
true,
true,
false,
- false
+ false,
+ true,
+ true
],
"parameters": {
"Right": 1
}
},
- "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone\nfrom users\n\nwhere password_recover is (?)\n"
+ "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone,\n totp_secret,\n totp_backup\nfrom users\n\nwhere username is (?)\n"
},
"56a9c0dff010858189a95087d014c7d0ce930da5d841b9d788a9c0e84b580bc6": {
"describe": {
@@ -540,7 +564,7 @@
},
"query": "update users\n\nset paper_key = ?\n\nwhere id is ?"
},
- "6a4a17c69175a677961779db048bff43d01e8773e7c4a6b25b9ee9b3c6fbacd5": {
+ "71c74369dc5d374d8ec5aa347b5f599728b74e545df3e986e3e7e66882f73ba0": {
"describe": {
"columns": [
{
@@ -602,89 +626,15 @@
"name": "timezone",
"ordinal": 11,
"type_info": "Text"
- }
- ],
- "nullable": [
- false,
- false,
- false,
- false,
- false,
- true,
- true,
- true,
- true,
- true,
- false,
- false
- ],
- "parameters": {
- "Right": 0
- }
- },
- "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone\nfrom users\n\norder by created_at desc\n"
- },
- "73f0d480c8dbef497a458070a32e65f0140f9a6e098ea082870a9c904629a97b": {
- "describe": {
- "columns": [
- {
- "name": "id",
- "ordinal": 0,
- "type_info": "Text"
},
{
- "name": "created_at: DateTime",
- "ordinal": 1,
- "type_info": "Text"
+ "name": "totp_secret",
+ "ordinal": 12,
+ "type_info": "Blob"
},
{
- "name": "updated_at: DateTime",
- "ordinal": 2,
- "type_info": "Text"
- },
- {
- "name": "is_admin: bool",
- "ordinal": 3,
- "type_info": "Int64"
- },
- {
- "name": "username",
- "ordinal": 4,
- "type_info": "Text"
- },
- {
- "name": "name",
- "ordinal": 5,
- "type_info": "Text"
- },
- {
- "name": "email",
- "ordinal": 6,
- "type_info": "Text"
- },
- {
- "name": "password",
- "ordinal": 7,
- "type_info": "Text"
- },
- {
- "name": "password_recover",
- "ordinal": 8,
- "type_info": "Text"
- },
- {
- "name": "paper_key",
- "ordinal": 9,
- "type_info": "Text"
- },
- {
- "name": "is_archived: bool",
- "ordinal": 10,
- "type_info": "Int64"
- },
- {
- "name": "timezone",
- "ordinal": 11,
+ "name": "totp_backup",
+ "ordinal": 13,
"type_info": "Text"
}
],
@@ -700,13 +650,15 @@
true,
true,
false,
- false
+ false,
+ true,
+ true
],
"parameters": {
"Right": 1
}
},
- "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone\nfrom users\n\nwhere username is (?)\n"
+ "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone,\n totp_secret,\n totp_backup\nfrom users\n\nwhere id is (?)\n"
},
"7b7f2430b2a719b3d5ce504c0a9302731b3ff82da99ba7771c2728d88aee642a": {
"describe": {
@@ -768,17 +720,7 @@
},
"query": "update users\n\nset password = ?\n\nwhere id is ?"
},
- "9f1885c4786f73335b4d614f562bb7cad49c91bfe7f084d8c25c6c571673ab90": {
- "describe": {
- "columns": [],
- "nullable": [],
- "parameters": {
- "Right": 0
- }
- },
- "query": "update refresh_tokens\n\nset revoked_at = CURRENT_TIMESTAMP\n\nwhere revoked_at is null"
- },
- "a2afd616a7d9742e5d416f308c30838ee11520d38a5812432af549a8a25b1393": {
+ "979c7ba41dac5ea5593fc169ea95b68d2311c5ae05c1e18db43424c8ed2f2e4e": {
"describe": {
"columns": [
{
@@ -840,6 +782,16 @@
"name": "timezone",
"ordinal": 11,
"type_info": "Text"
+ },
+ {
+ "name": "totp_secret",
+ "ordinal": 12,
+ "type_info": "Blob"
+ },
+ {
+ "name": "totp_backup",
+ "ordinal": 13,
+ "type_info": "Text"
}
],
"nullable": [
@@ -854,13 +806,121 @@
true,
true,
false,
- false
+ false,
+ true,
+ true
],
"parameters": {
"Right": 0
}
},
- "query": "select u.id,\n u.created_at as \"created_at: DateTime\",\n u.updated_at as \"updated_at: DateTime\",\n u.is_admin as \"is_admin: bool\",\n u.username,\n u.name,\n u.email,\n u.password,\n u.password_recover,\n u.paper_key,\n u.is_archived as \"is_archived: bool\",\n u.timezone\nfrom users u\n\n inner join settings s on u.id = s.first_admin\n\nwhere u.is_admin is 1\n and u.is_archived is 0\n and u.id is s.first_admin\n\nlimit 1"
+ "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone,\n totp_secret,\n totp_backup\nfrom users\n\norder by created_at desc\n"
+ },
+ "9dc379b0f3a3c944a33ef01dc40489b29b2a61b9f73f6ca4b5df7e9fb9bccf90": {
+ "describe": {
+ "columns": [
+ {
+ "name": "id",
+ "ordinal": 0,
+ "type_info": "Text"
+ },
+ {
+ "name": "created_at: DateTime",
+ "ordinal": 1,
+ "type_info": "Text"
+ },
+ {
+ "name": "updated_at: DateTime",
+ "ordinal": 2,
+ "type_info": "Text"
+ },
+ {
+ "name": "is_admin: bool",
+ "ordinal": 3,
+ "type_info": "Int64"
+ },
+ {
+ "name": "username",
+ "ordinal": 4,
+ "type_info": "Text"
+ },
+ {
+ "name": "name",
+ "ordinal": 5,
+ "type_info": "Text"
+ },
+ {
+ "name": "email",
+ "ordinal": 6,
+ "type_info": "Text"
+ },
+ {
+ "name": "password",
+ "ordinal": 7,
+ "type_info": "Text"
+ },
+ {
+ "name": "password_recover",
+ "ordinal": 8,
+ "type_info": "Text"
+ },
+ {
+ "name": "paper_key",
+ "ordinal": 9,
+ "type_info": "Text"
+ },
+ {
+ "name": "is_archived: bool",
+ "ordinal": 10,
+ "type_info": "Int64"
+ },
+ {
+ "name": "timezone",
+ "ordinal": 11,
+ "type_info": "Text"
+ },
+ {
+ "name": "totp_secret",
+ "ordinal": 12,
+ "type_info": "Blob"
+ },
+ {
+ "name": "totp_backup",
+ "ordinal": 13,
+ "type_info": "Text"
+ }
+ ],
+ "nullable": [
+ false,
+ false,
+ false,
+ false,
+ false,
+ true,
+ true,
+ true,
+ true,
+ true,
+ false,
+ false,
+ true,
+ true
+ ],
+ "parameters": {
+ "Right": 0
+ }
+ },
+ "query": "select u.id,\n u.created_at as \"created_at: DateTime\",\n u.updated_at as \"updated_at: DateTime\",\n u.is_admin as \"is_admin: bool\",\n u.username,\n u.name,\n u.email,\n u.password,\n u.password_recover,\n u.paper_key,\n u.is_archived as \"is_archived: bool\",\n u.timezone,\n u.totp_secret,\n u.totp_backup\nfrom users u\n\n inner join settings s on u.id = s.first_admin\n\nwhere u.is_admin is 1\n and u.is_archived is 0\n and u.id is s.first_admin\n\nlimit 1"
+ },
+ "9f1885c4786f73335b4d614f562bb7cad49c91bfe7f084d8c25c6c571673ab90": {
+ "describe": {
+ "columns": [],
+ "nullable": [],
+ "parameters": {
+ "Right": 0
+ }
+ },
+ "query": "update refresh_tokens\n\nset revoked_at = CURRENT_TIMESTAMP\n\nwhere revoked_at is null"
},
"a55b17a3a70e6445517f19536220f0dafc78a0e8b69221dee4715f84841839da": {
"describe": {
@@ -882,6 +942,16 @@
},
"query": "update settings\n\nset first_admin = ?\n\nwhere id is 0\n"
},
+ "afc08ba1fd47656c7d987e3817941004f988e683d28df2f533ff6272bc71b356": {
+ "describe": {
+ "columns": [],
+ "nullable": [],
+ "parameters": {
+ "Right": 2
+ }
+ },
+ "query": "update users\n\nset totp_secret = ?\n\nwhere id is ?"
+ },
"b3936ae237f02cc048ca084372f7975921c79aef8fcd6d15a4d63c10f2377d24": {
"describe": {
"columns": [],
@@ -892,6 +962,16 @@
},
"query": "insert into refresh_tokens (token, ip_address, user, app, expires_at)\nvalues (?, ?, ?, ?, datetime(?, 'unixepoch'))\n"
},
+ "baaae9952348da17e2f4d099c6903c50bc90daed80ee2df1fd0dc1d9d2c67b26": {
+ "describe": {
+ "columns": [],
+ "nullable": [],
+ "parameters": {
+ "Right": 2
+ }
+ },
+ "query": "update users\n\nset totp_backup = ?\n\nwhere id is ?"
+ },
"c00e5fce25caebdeeb24db20880e6c2210f583cddb0d478075f78124258712dd": {
"describe": {
"columns": [],
@@ -922,6 +1002,198 @@
},
"query": "update refresh_tokens\n\nset revoked_at = CURRENT_TIMESTAMP\n\nwhere token is ?"
},
+ "ca7d100a9440fb7854a27f9aafd91ce94d1df9fa1ccd65b549be92d16741f9d2": {
+ "describe": {
+ "columns": [
+ {
+ "name": "id",
+ "ordinal": 0,
+ "type_info": "Text"
+ },
+ {
+ "name": "created_at: DateTime",
+ "ordinal": 1,
+ "type_info": "Text"
+ },
+ {
+ "name": "updated_at: DateTime",
+ "ordinal": 2,
+ "type_info": "Text"
+ },
+ {
+ "name": "is_admin: bool",
+ "ordinal": 3,
+ "type_info": "Int64"
+ },
+ {
+ "name": "username",
+ "ordinal": 4,
+ "type_info": "Text"
+ },
+ {
+ "name": "name",
+ "ordinal": 5,
+ "type_info": "Text"
+ },
+ {
+ "name": "email",
+ "ordinal": 6,
+ "type_info": "Text"
+ },
+ {
+ "name": "password",
+ "ordinal": 7,
+ "type_info": "Text"
+ },
+ {
+ "name": "password_recover",
+ "ordinal": 8,
+ "type_info": "Text"
+ },
+ {
+ "name": "paper_key",
+ "ordinal": 9,
+ "type_info": "Text"
+ },
+ {
+ "name": "is_archived: bool",
+ "ordinal": 10,
+ "type_info": "Int64"
+ },
+ {
+ "name": "timezone",
+ "ordinal": 11,
+ "type_info": "Text"
+ },
+ {
+ "name": "totp_secret",
+ "ordinal": 12,
+ "type_info": "Blob"
+ },
+ {
+ "name": "totp_backup",
+ "ordinal": 13,
+ "type_info": "Text"
+ }
+ ],
+ "nullable": [
+ false,
+ false,
+ false,
+ false,
+ false,
+ true,
+ true,
+ true,
+ true,
+ true,
+ false,
+ false,
+ true,
+ true
+ ],
+ "parameters": {
+ "Right": 1
+ }
+ },
+ "query": "select u.id,\n u.created_at as \"created_at: DateTime\",\n u.updated_at as \"updated_at: DateTime\",\n u.is_admin as \"is_admin: bool\",\n u.username,\n u.name,\n u.email,\n u.password,\n u.password_recover,\n u.paper_key,\n u.is_archived as \"is_archived: bool\",\n u.timezone,\n u.totp_secret,\n u.totp_backup\nfrom users u\n\n inner join refresh_tokens rt on u.id = rt.user\n\nwhere rt.token is ?\n"
+ },
+ "cf12bd3112b2047f263118649ba89800509d94f3a858be99fb7ce8c8bc05d8b7": {
+ "describe": {
+ "columns": [
+ {
+ "name": "id",
+ "ordinal": 0,
+ "type_info": "Text"
+ },
+ {
+ "name": "created_at: DateTime",
+ "ordinal": 1,
+ "type_info": "Text"
+ },
+ {
+ "name": "updated_at: DateTime",
+ "ordinal": 2,
+ "type_info": "Text"
+ },
+ {
+ "name": "is_admin: bool",
+ "ordinal": 3,
+ "type_info": "Int64"
+ },
+ {
+ "name": "username",
+ "ordinal": 4,
+ "type_info": "Text"
+ },
+ {
+ "name": "name",
+ "ordinal": 5,
+ "type_info": "Text"
+ },
+ {
+ "name": "email",
+ "ordinal": 6,
+ "type_info": "Text"
+ },
+ {
+ "name": "password",
+ "ordinal": 7,
+ "type_info": "Text"
+ },
+ {
+ "name": "password_recover",
+ "ordinal": 8,
+ "type_info": "Text"
+ },
+ {
+ "name": "paper_key",
+ "ordinal": 9,
+ "type_info": "Text"
+ },
+ {
+ "name": "is_archived: bool",
+ "ordinal": 10,
+ "type_info": "Int64"
+ },
+ {
+ "name": "timezone",
+ "ordinal": 11,
+ "type_info": "Text"
+ },
+ {
+ "name": "totp_secret",
+ "ordinal": 12,
+ "type_info": "Blob"
+ },
+ {
+ "name": "totp_backup",
+ "ordinal": 13,
+ "type_info": "Text"
+ }
+ ],
+ "nullable": [
+ false,
+ false,
+ false,
+ false,
+ false,
+ true,
+ true,
+ true,
+ true,
+ true,
+ false,
+ false,
+ true,
+ true
+ ],
+ "parameters": {
+ "Right": 1
+ }
+ },
+ "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone,\n totp_secret,\n totp_backup\nfrom users\n\nwhere password_recover is (?)\n"
+ },
"cf624c4e122477228e3bab09f7cd0dedf4776f73e7a86f19e06772a0adf83406": {
"describe": {
"columns": [
@@ -1012,6 +1284,102 @@
},
"query": "select id,\n created_at as \"created_at: DateTime\",\n revoked_at as \"revoked_at: DateTime\",\n private_der,\n public_der\n\nfrom keys\nwhere revoked_at is null\norder by created_at desc\n"
},
+ "d6d3a401f5563ce339099ccbdedec1ac248865810cafafd5d78d6cc3f74d5237": {
+ "describe": {
+ "columns": [
+ {
+ "name": "id",
+ "ordinal": 0,
+ "type_info": "Text"
+ },
+ {
+ "name": "created_at: DateTime",
+ "ordinal": 1,
+ "type_info": "Text"
+ },
+ {
+ "name": "updated_at: DateTime",
+ "ordinal": 2,
+ "type_info": "Text"
+ },
+ {
+ "name": "is_admin: bool",
+ "ordinal": 3,
+ "type_info": "Int64"
+ },
+ {
+ "name": "username",
+ "ordinal": 4,
+ "type_info": "Text"
+ },
+ {
+ "name": "name",
+ "ordinal": 5,
+ "type_info": "Text"
+ },
+ {
+ "name": "email",
+ "ordinal": 6,
+ "type_info": "Text"
+ },
+ {
+ "name": "password",
+ "ordinal": 7,
+ "type_info": "Text"
+ },
+ {
+ "name": "password_recover",
+ "ordinal": 8,
+ "type_info": "Text"
+ },
+ {
+ "name": "paper_key",
+ "ordinal": 9,
+ "type_info": "Text"
+ },
+ {
+ "name": "is_archived: bool",
+ "ordinal": 10,
+ "type_info": "Int64"
+ },
+ {
+ "name": "timezone",
+ "ordinal": 11,
+ "type_info": "Text"
+ },
+ {
+ "name": "totp_secret",
+ "ordinal": 12,
+ "type_info": "Blob"
+ },
+ {
+ "name": "totp_backup",
+ "ordinal": 13,
+ "type_info": "Text"
+ }
+ ],
+ "nullable": [
+ false,
+ false,
+ false,
+ false,
+ false,
+ true,
+ true,
+ true,
+ true,
+ true,
+ false,
+ false,
+ true,
+ true
+ ],
+ "parameters": {
+ "Right": 1
+ }
+ },
+ "query": "select u.id,\n u.created_at as \"created_at: DateTime\",\n u.updated_at as \"updated_at: DateTime\",\n u.is_admin as \"is_admin: bool\",\n u.username,\n u.name,\n u.email,\n u.password,\n u.password_recover,\n u.paper_key,\n u.is_archived as \"is_archived: bool\",\n u.timezone,\n u.totp_secret,\n u.totp_backup\nfrom users u\n\n inner join authorization_codes ac on u.id = ac.user\n\nwhere ac.code is ?"
+ },
"d8517e5faa5292da25d21a85fcb2f676dce0f67275466566d5bbedf0f4b7f4f5": {
"describe": {
"columns": [
@@ -1132,90 +1500,6 @@
},
"query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n label,\n redirect_uri,\n secret,\n is_confidential as \"is_confidential: bool\",\n is_archived as \"is_archived: bool\"\nfrom apps\n\nwhere id is (?)\n"
},
- "e6d72e5ef795c6daa506d27560d6daeb36ffede94228636e2b0fa47a0855c1b2": {
- "describe": {
- "columns": [
- {
- "name": "id",
- "ordinal": 0,
- "type_info": "Text"
- },
- {
- "name": "created_at: DateTime",
- "ordinal": 1,
- "type_info": "Text"
- },
- {
- "name": "updated_at: DateTime",
- "ordinal": 2,
- "type_info": "Text"
- },
- {
- "name": "is_admin: bool",
- "ordinal": 3,
- "type_info": "Int64"
- },
- {
- "name": "username",
- "ordinal": 4,
- "type_info": "Text"
- },
- {
- "name": "name",
- "ordinal": 5,
- "type_info": "Text"
- },
- {
- "name": "email",
- "ordinal": 6,
- "type_info": "Text"
- },
- {
- "name": "password",
- "ordinal": 7,
- "type_info": "Text"
- },
- {
- "name": "password_recover",
- "ordinal": 8,
- "type_info": "Text"
- },
- {
- "name": "paper_key",
- "ordinal": 9,
- "type_info": "Text"
- },
- {
- "name": "is_archived: bool",
- "ordinal": 10,
- "type_info": "Int64"
- },
- {
- "name": "timezone",
- "ordinal": 11,
- "type_info": "Text"
- }
- ],
- "nullable": [
- false,
- false,
- false,
- false,
- false,
- true,
- true,
- true,
- true,
- true,
- false,
- false
- ],
- "parameters": {
- "Right": 1
- }
- },
- "query": "select u.id,\n u.created_at as \"created_at: DateTime\",\n u.updated_at as \"updated_at: DateTime\",\n u.is_admin as \"is_admin: bool\",\n u.username,\n u.name,\n u.email,\n u.password,\n u.password_recover,\n u.paper_key,\n u.is_archived as \"is_archived: bool\",\n u.timezone\nfrom users u\n\n inner join authorization_codes ac on u.id = ac.user\n\nwhere ac.code is ?"
- },
"eaf0744f65a1de803fa8cc21b67bad4bdf22760d431265cf97b911e6456b2fd8": {
"describe": {
"columns": [
@@ -1356,174 +1640,6 @@
},
"query": "insert into apps (id, label, redirect_uri, secret, is_confidential)\nvalues (?, ?, ?, ?, ?)\n"
},
- "f472d850bc00dc01a95c837f22879535d8234aa3085a0087edc79231387b4f3d": {
- "describe": {
- "columns": [
- {
- "name": "id",
- "ordinal": 0,
- "type_info": "Text"
- },
- {
- "name": "created_at: DateTime",
- "ordinal": 1,
- "type_info": "Text"
- },
- {
- "name": "updated_at: DateTime",
- "ordinal": 2,
- "type_info": "Text"
- },
- {
- "name": "is_admin: bool",
- "ordinal": 3,
- "type_info": "Int64"
- },
- {
- "name": "username",
- "ordinal": 4,
- "type_info": "Text"
- },
- {
- "name": "name",
- "ordinal": 5,
- "type_info": "Text"
- },
- {
- "name": "email",
- "ordinal": 6,
- "type_info": "Text"
- },
- {
- "name": "password",
- "ordinal": 7,
- "type_info": "Text"
- },
- {
- "name": "password_recover",
- "ordinal": 8,
- "type_info": "Text"
- },
- {
- "name": "paper_key",
- "ordinal": 9,
- "type_info": "Text"
- },
- {
- "name": "is_archived: bool",
- "ordinal": 10,
- "type_info": "Int64"
- },
- {
- "name": "timezone",
- "ordinal": 11,
- "type_info": "Text"
- }
- ],
- "nullable": [
- false,
- false,
- false,
- false,
- false,
- true,
- true,
- true,
- true,
- true,
- false,
- false
- ],
- "parameters": {
- "Right": 1
- }
- },
- "query": "select id,\n created_at as \"created_at: DateTime\",\n updated_at as \"updated_at: DateTime\",\n is_admin as \"is_admin: bool\",\n username,\n name,\n email,\n password,\n password_recover,\n paper_key,\n is_archived as \"is_archived: bool\",\n timezone\nfrom users\n\nwhere email is (?)\n"
- },
- "f50157586b672a848294aca06fbb1b5a2d3c91ff83f7e81b42f32339bed0f35f": {
- "describe": {
- "columns": [
- {
- "name": "id",
- "ordinal": 0,
- "type_info": "Text"
- },
- {
- "name": "created_at: DateTime",
- "ordinal": 1,
- "type_info": "Text"
- },
- {
- "name": "updated_at: DateTime",
- "ordinal": 2,
- "type_info": "Text"
- },
- {
- "name": "is_admin: bool",
- "ordinal": 3,
- "type_info": "Int64"
- },
- {
- "name": "username",
- "ordinal": 4,
- "type_info": "Text"
- },
- {
- "name": "name",
- "ordinal": 5,
- "type_info": "Text"
- },
- {
- "name": "email",
- "ordinal": 6,
- "type_info": "Text"
- },
- {
- "name": "password",
- "ordinal": 7,
- "type_info": "Text"
- },
- {
- "name": "password_recover",
- "ordinal": 8,
- "type_info": "Text"
- },
- {
- "name": "paper_key",
- "ordinal": 9,
- "type_info": "Text"
- },
- {
- "name": "is_archived: bool",
- "ordinal": 10,
- "type_info": "Int64"
- },
- {
- "name": "timezone",
- "ordinal": 11,
- "type_info": "Text"
- }
- ],
- "nullable": [
- false,
- false,
- false,
- false,
- false,
- true,
- true,
- true,
- true,
- true,
- false,
- false
- ],
- "parameters": {
- "Right": 1
- }
- },
- "query": "select u.id,\n u.created_at as \"created_at: DateTime\",\n u.updated_at as \"updated_at: DateTime\",\n u.is_admin as \"is_admin: bool\",\n u.username,\n u.name,\n u.email,\n u.password,\n u.password_recover,\n u.paper_key,\n u.is_archived as \"is_archived: bool\",\n u.timezone\nfrom users u\n\n inner join refresh_tokens rt on u.id = rt.user\n\nwhere rt.token is ?\n"
- },
"f705411720bd037562f7e3622832262ac4c0a8fc0921fbd934d2b98146d3f413": {
"describe": {
"columns": [],
diff --git a/crates/database/src/tables/users.rs b/crates/database/src/tables/users.rs
index 46d348b..7f84f5f 100644
--- a/crates/database/src/tables/users.rs
+++ b/crates/database/src/tables/users.rs
@@ -17,6 +17,8 @@ pub struct Users {
pub paper_key: Option,
pub is_archived: bool,
pub timezone: String,
+ pub totp_secret: Option>,
+ pub totp_backup: Option,
}
impl Users {
@@ -213,4 +215,32 @@ impl Users {
Ok((query.rows_affected() == 1).then_some(()))
}
+
+ pub async fn set_totp_secret(
+ conn: impl SqliteExecutor<'_>,
+ id: &str,
+ secret: Option<&[u8]>,
+ ) -> Result