From 49bd75b8563db181517eb97679ca8349177eb102 Mon Sep 17 00:00:00 2001
From: Jim <james@colderwood.com>
Date: Thu, 23 Jun 2022 06:58:34 +0000
Subject: [PATCH] Lightning bolt optional

---
 conduit-example.toml         |  3 +++
 src/client_server/account.rs | 22 +++++++++-------------
 src/config.rs                |  6 ++++++
 src/database/admin.rs        |  8 +++++++-
 src/database/globals.rs      |  4 ++++
 5 files changed, 29 insertions(+), 14 deletions(-)

diff --git a/conduit-example.toml b/conduit-example.toml
index 362f7e7e..5eed0708 100644
--- a/conduit-example.toml
+++ b/conduit-example.toml
@@ -40,6 +40,9 @@ allow_registration = true
 
 allow_federation = true
 
+# Enable the display name lightning bolt on registration.
+enable_lightning_bolt = true
+
 trusted_servers = ["matrix.org"]
 
 #max_concurrent_requests = 100 # How many requests Conduit sends to other servers at the same time
diff --git a/src/client_server/account.rs b/src/client_server/account.rs
index dc0782d1..1484bf66 100644
--- a/src/client_server/account.rs
+++ b/src/client_server/account.rs
@@ -1,10 +1,7 @@
-use std::sync::Arc;
-
 use super::{DEVICE_ID_LENGTH, SESSION_ID_LENGTH, TOKEN_LENGTH};
 use crate::{
     database::{admin::make_user_admin, DatabaseGuard},
-    pdu::PduBuilder,
-    utils, Database, Error, Result, Ruma,
+    utils, Error, Result, Ruma,
 };
 use ruma::{
     api::client::{
@@ -15,16 +12,9 @@ use ruma::{
         error::ErrorKind,
         uiaa::{AuthFlow, AuthType, UiaaInfo},
     },
-    events::{
-        room::{
-            member::{MembershipState, RoomMemberEventContent},
-            message::RoomMessageEventContent,
-        },
-        GlobalAccountDataEventType, RoomEventType,
-    },
+    events::{room::message::RoomMessageEventContent, GlobalAccountDataEventType},
     push, UserId,
 };
-use serde_json::value::to_raw_value;
 use tracing::{info, warn};
 
 use register::RegistrationKind;
@@ -181,7 +171,13 @@ pub async fn register_route(
     db.users.create(&user_id, password)?;
 
     // Default to pretty displayname
-    let displayname = format!("{} ⚡️", user_id.localpart());
+    let mut displayname = user_id.localpart().to_owned();
+
+    // If enabled append lightning bolt to display name (default true)
+    if db.globals.enable_lightning_bolt() {
+        displayname.push_str(" ⚡️");
+    }
+
     db.users
         .set_displayname(&user_id, Some(displayname.clone()))?;
 
diff --git a/src/config.rs b/src/config.rs
index 29af8839..7d81d0f8 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -26,6 +26,8 @@ pub struct Config {
     pub database_path: String,
     #[serde(default = "default_db_cache_capacity_mb")]
     pub db_cache_capacity_mb: f64,
+    #[serde(default = "true_fn")]
+    pub enable_lightning_bolt: bool,
     #[serde(default = "default_conduit_cache_capacity_modifier")]
     pub conduit_cache_capacity_modifier: f64,
     #[serde(default = "default_rocksdb_max_open_files")]
@@ -135,6 +137,10 @@ impl fmt::Display for Config {
                 &self.max_concurrent_requests.to_string(),
             ),
             ("Allow registration", &self.allow_registration.to_string()),
+            (
+                "Enabled lightning bolt",
+                &self.enable_lightning_bolt.to_string(),
+            ),
             ("Allow encryption", &self.allow_encryption.to_string()),
             ("Allow federation", &self.allow_federation.to_string()),
             ("Allow room creation", &self.allow_room_creation.to_string()),
diff --git a/src/database/admin.rs b/src/database/admin.rs
index 6f418ea8..edc76914 100644
--- a/src/database/admin.rs
+++ b/src/database/admin.rs
@@ -598,7 +598,13 @@ async fn process_admin_command(
             db.users.create(&user_id, Some(password.as_str()))?;
 
             // Default to pretty displayname
-            let displayname = format!("{} ⚡️", user_id.localpart());
+            let mut displayname = user_id.localpart().to_owned();
+
+            // If enabled append lightning bolt to display name (default true)
+            if db.globals.enable_lightning_bolt() {
+                displayname.push_str(" ⚡️");
+            }
+
             db.users
                 .set_displayname(&user_id, Some(displayname.clone()))?;
 
diff --git a/src/database/globals.rs b/src/database/globals.rs
index 7e09128e..7d7b7fd6 100644
--- a/src/database/globals.rs
+++ b/src/database/globals.rs
@@ -267,6 +267,10 @@ impl Globals {
         self.config.default_room_version.clone()
     }
 
+    pub fn enable_lightning_bolt(&self) -> bool {
+        self.config.enable_lightning_bolt
+    }
+
     pub fn trusted_servers(&self) -> &[Box<ServerName>] {
         &self.config.trusted_servers
     }