vw_small

commit 36b5350f9b3949acf9a6d3bb0041bb9c08cced4f
parent 367e1ce289cea6a3251b7350a6707c700bd8a544
Author: BlackDex <black.dex@gmail.com>
Date:   Wed, 11 Jan 2023 21:45:11 +0100

Add avatar color support

The new web-vault v2023.1.0 supports a custom color for the avatar.
https://github.com/bitwarden/server/pull/2330

This PR adds this feature.

Diffstat:
A
migrations/mysql/2023-01-11-205851_add_avatar_color/down.sql
 | 
0

A
migrations/mysql/2023-01-11-205851_add_avatar_color/up.sql
 | 
2
++
A
migrations/postgresql/2023-01-11-205851_add_avatar_color/down.sql
 | 
0

A
migrations/postgresql/2023-01-11-205851_add_avatar_color/up.sql
 | 
2
++
A
migrations/sqlite/2023-01-11-205851_add_avatar_color/down.sql
 | 
0

A
migrations/sqlite/2023-01-11-205851_add_avatar_color/up.sql
 | 
2
++
M
src/api/core/accounts.rs
 | 
27
+++++++++++++++++++++++++++
M
src/db/models/user.rs
 | 
5
+++++
M
src/db/schemas/mysql/schema.rs
 | 
1
+
M
src/db/schemas/postgresql/schema.rs
 | 
1
+
M
src/db/schemas/sqlite/schema.rs
 | 
1
+
M
src/main.rs
 | 
2
+-

12 files changed, 42 insertions(+), 1 deletion(-)
diff --git a/migrations/mysql/2023-01-11-205851_add_avatar_color/down.sql b/migrations/mysql/2023-01-11-205851_add_avatar_color/down.sql
diff --git a/migrations/mysql/2023-01-11-205851_add_avatar_color/up.sql b/migrations/mysql/2023-01-11-205851_add_avatar_color/up.sql
@@ -0,0 +1,2 @@
+ALTER TABLE users
+ADD COLUMN avatar_color VARCHAR(7);
diff --git a/migrations/postgresql/2023-01-11-205851_add_avatar_color/down.sql b/migrations/postgresql/2023-01-11-205851_add_avatar_color/down.sql
diff --git a/migrations/postgresql/2023-01-11-205851_add_avatar_color/up.sql b/migrations/postgresql/2023-01-11-205851_add_avatar_color/up.sql
@@ -0,0 +1,2 @@
+ALTER TABLE users
+ADD COLUMN avatar_color TEXT;
diff --git a/migrations/sqlite/2023-01-11-205851_add_avatar_color/down.sql b/migrations/sqlite/2023-01-11-205851_add_avatar_color/down.sql
diff --git a/migrations/sqlite/2023-01-11-205851_add_avatar_color/up.sql b/migrations/sqlite/2023-01-11-205851_add_avatar_color/up.sql
@@ -0,0 +1,2 @@
+ALTER TABLE users
+ADD COLUMN avatar_color TEXT;
diff --git a/src/api/core/accounts.rs b/src/api/core/accounts.rs
@@ -39,6 +39,7 @@ pub fn routes() -> Vec<rocket::Route> {
         api_key,
         rotate_api_key,
         get_known_device,
+        put_avatar,
     ]
 }
 
@@ -228,6 +229,32 @@ async fn post_profile(data: JsonUpcase<ProfileData>, headers: Headers, mut conn:
     Ok(Json(user.to_json(&mut conn).await))
 }
 
+#[derive(Deserialize)]
+#[allow(non_snake_case)]
+struct AvatarData {
+    AvatarColor: Option<String>,
+}
+
+#[put("/accounts/avatar", data = "<data>")]
+async fn put_avatar(data: JsonUpcase<AvatarData>, headers: Headers, mut conn: DbConn) -> JsonResult {
+    let data: AvatarData = data.into_inner().data;
+
+    // It looks like it only supports the 6 hex color format.
+    // If you try to add the short value it will not show that color.
+    // Check and force 7 chars, including the #.
+    if let Some(color) = &data.AvatarColor {
+        if color.len() != 7 {
+            err!("The field AvatarColor must be a HTML/Hex color code with a length of 7 characters")
+        }
+    }
+
+    let mut user = headers.user;
+    user.avatar_color = data.AvatarColor;
+
+    user.save(&mut conn).await?;
+    Ok(Json(user.to_json(&mut conn).await))
+}
+
 #[get("/users/<uuid>/public-key")]
 async fn get_public_keys(uuid: String, _headers: Headers, mut conn: DbConn) -> JsonResult {
     let user = match User::find_by_uuid(&uuid, &mut conn).await {
diff --git a/src/db/models/user.rs b/src/db/models/user.rs
@@ -46,6 +46,8 @@ db_object! {
         pub client_kdf_iter: i32,
 
         pub api_key: Option<String>,
+
+        pub avatar_color: Option<String>,
     }
 
     #[derive(Identifiable, Queryable, Insertable)]
@@ -113,6 +115,8 @@ impl User {
             client_kdf_iter: Self::CLIENT_KDF_ITER_DEFAULT,
 
             api_key: None,
+
+            avatar_color: None,
         }
     }
 
@@ -226,6 +230,7 @@ impl User {
             "Providers": [],
             "ProviderOrganizations": [],
             "ForcePasswordReset": false,
+            "AvatarColor": self.avatar_color,
             "Object": "profile",
         })
     }
diff --git a/src/db/schemas/mysql/schema.rs b/src/db/schemas/mysql/schema.rs
@@ -200,6 +200,7 @@ table! {
         client_kdf_type -> Integer,
         client_kdf_iter -> Integer,
         api_key -> Nullable<Text>,
+        avatar_color -> Nullable<Text>,
     }
 }
 
diff --git a/src/db/schemas/postgresql/schema.rs b/src/db/schemas/postgresql/schema.rs
@@ -200,6 +200,7 @@ table! {
         client_kdf_type -> Integer,
         client_kdf_iter -> Integer,
         api_key -> Nullable<Text>,
+        avatar_color -> Nullable<Text>,
     }
 }
 
diff --git a/src/db/schemas/sqlite/schema.rs b/src/db/schemas/sqlite/schema.rs
@@ -200,6 +200,7 @@ table! {
         client_kdf_type -> Integer,
         client_kdf_iter -> Integer,
         api_key -> Nullable<Text>,
+        avatar_color -> Nullable<Text>,
     }
 }
 
diff --git a/src/main.rs b/src/main.rs
@@ -34,7 +34,7 @@
 // The more key/value pairs there are the more recursion occurs.
 // We want to keep this as low as possible, but not higher then 128.
 // If you go above 128 it will cause rust-analyzer to fail,
-#![recursion_limit = "94"]
+#![recursion_limit = "97"]
 
 // When enabled use MiMalloc as malloc instead of the default malloc
 #[cfg(feature = "enable_mimalloc")]