refactor: minor appservice code cleanup

src/api/appservice_server.rs

@@ -17,7 +17,10 @@ pub(crate) async fn send_request<T: OutgoingRequest>(
 where
     T: Debug,
 {
-    if let Some(destination) = registration.url {
+    let Some(destination) = registration.url else {
+        return None;
+    };
+
     let hs_token = registration.hs_token.as_str();
 
     let mut http_request = request
@@ -98,6 +101,7 @@ where
             .body(body)
             .expect("reqwest body is valid http body"),
     );
+
     Some(response.map_err(|_| {
         warn!(
             "Appservice returned invalid response bytes {}\n{}",
@@ -105,7 +109,4 @@ where
         );
         Error::BadServerResponse("Server returned bad response.")
     }))
-    } else {
-        None
-    }
 }

src/api/client_server/alias.rs

@@ -100,13 +100,7 @@ pub(crate) async fn get_alias_helper(
     match services().rooms.alias.resolve_local_alias(&room_alias)? {
         Some(r) => room_id = Some(r),
         None => {
-            for appservice in services()
+            for appservice in services().appservice.all().await {
-                .appservice
-                .registration_info
-                .read()
-                .await
-                .values()
-            {
                 if appservice.aliases.is_match(room_alias.as_str())
                     && if let Some(opt_result) = services()
                         .sending

src/api/ruma_wrapper/axum.rs

@@ -80,19 +80,19 @@ where
 
         let mut json_body = serde_json::from_slice::<CanonicalJsonValue>(&body).ok();
 
-        let appservices = services().appservice.all().unwrap();
+        let appservices = services().appservice.all().await;
         let appservice_registration = appservices
             .iter()
-            .find(|(_id, registration)| Some(registration.as_token.as_str()) == token);
+            .find(|info| Some(info.registration.as_token.as_str()) == token);
 
         let (sender_user, sender_device, sender_servername, from_appservice) =
-            if let Some((_id, registration)) = appservice_registration {
+            if let Some(info) = appservice_registration {
                 match metadata.authentication {
                     AuthScheme::AccessToken => {
                         let user_id = query_params.user_id.map_or_else(
                             || {
                                 UserId::parse_with_server_name(
-                                    registration.sender_localpart.as_str(),
+                                    info.registration.sender_localpart.as_str(),
                                     services().globals.server_name(),
                                 )
                                 .unwrap()

src/database/key_value/appservice.rs

@@ -10,10 +10,6 @@ impl service::appservice::Data for KeyValueDatabase {
             id.as_bytes(),
             serde_yaml::to_string(&yaml).unwrap().as_bytes(),
         )?;
-        self.cached_registrations
-            .write()
-            .unwrap()
-            .insert(id.to_owned(), yaml.to_owned());
 
         Ok(id.to_owned())
     }
@@ -26,33 +22,18 @@ impl service::appservice::Data for KeyValueDatabase {
     fn unregister_appservice(&self, service_name: &str) -> Result<()> {
         self.id_appserviceregistrations
             .remove(service_name.as_bytes())?;
-        self.cached_registrations
-            .write()
-            .unwrap()
-            .remove(service_name);
         Ok(())
     }
 
     fn get_registration(&self, id: &str) -> Result<Option<Registration>> {
-        self.cached_registrations
-            .read()
-            .unwrap()
-            .get(id)
-            .map_or_else(
-                || {
         self.id_appserviceregistrations
             .get(id.as_bytes())?
             .map(|bytes| {
                 serde_yaml::from_slice(&bytes).map_err(|_| {
-                                Error::bad_database(
+                    Error::bad_database("Invalid registration bytes in id_appserviceregistrations.")
-                                    "Invalid registration bytes in id_appserviceregistrations.",
-                                )
                 })
             })
             .transpose()
-                },
-                |r| Ok(Some(r.clone())),
-            )
     }
 
     fn iter_ids<'a>(&'a self) -> Result<Box<dyn Iterator<Item = Result<String>> + 'a>> {

src/database/mod.rs

@@ -10,7 +10,6 @@ use directories::ProjectDirs;
 use lru_cache::LruCache;
 
 use ruma::{
-    api::appservice::Registration,
     events::{
         push_rules::{PushRulesEvent, PushRulesEventContent},
         room::message::RoomMessageEventContent,
@@ -164,7 +163,6 @@ pub struct KeyValueDatabase {
     //pub pusher: pusher::PushData,
     pub(super) senderkey_pusher: Arc<dyn KvTree>,
 
-    pub(super) cached_registrations: Arc<RwLock<HashMap<String, Registration>>>,
     pub(super) pdu_cache: Mutex<LruCache<OwnedEventId, Arc<PduEvent>>>,
     pub(super) shorteventid_cache: Mutex<LruCache<u64, Arc<EventId>>>,
     pub(super) auth_chain_cache: Mutex<LruCache<Vec<u64>, Arc<HashSet<u64>>>>,
@@ -374,7 +372,6 @@ impl KeyValueDatabase {
             global: builder.open_tree("global")?,
             server_signingkeys: builder.open_tree("server_signingkeys")?,
 
-            cached_registrations: Arc::new(RwLock::new(HashMap::new())),
             pdu_cache: Mutex::new(LruCache::new(
                 config
                     .pdu_cache_capacity
@@ -969,22 +966,6 @@ impl KeyValueDatabase {
             );
         }
 
-        // Inserting registrations into cache
-        for appservice in services().appservice.all()? {
-            services()
-                .appservice
-                .registration_info
-                .write()
-                .await
-                .insert(
-                    appservice.0,
-                    appservice
-                        .1
-                        .try_into()
-                        .expect("Should be validated on registration"),
-                );
-        }
-
         // This data is probably outdated
         db.presenceid_presence.clear()?;
 

src/service/appservice/mod.rs

@@ -10,7 +10,8 @@ use tokio::sync::RwLock;
 
 use crate::{services, Result};
 
-/// Compiled regular expressions for a namespace
+/// Compiled regular expressions for a namespace.
+#[derive(Clone, Debug)]
 pub struct NamespaceRegex {
     pub exclusive: Option<RegexSet>,
     pub non_exclusive: Option<RegexSet>,
@@ -72,7 +73,8 @@ impl TryFrom<Vec<Namespace>> for NamespaceRegex {
     type Error = regex::Error;
 }
 
-/// Compiled regular expressions for an appservice
+/// Appservice registration combined with its compiled regular expressions.
+#[derive(Clone, Debug)]
 pub struct RegistrationInfo {
     pub registration: Registration,
     pub users: NamespaceRegex,
@@ -95,11 +97,29 @@ impl TryFrom<Registration> for RegistrationInfo {
 
 pub struct Service {
     pub db: &'static dyn Data,
-    pub registration_info: RwLock<HashMap<String, RegistrationInfo>>,
+    registration_info: RwLock<HashMap<String, RegistrationInfo>>,
 }
 
 impl Service {
-    /// Registers an appservice and returns the ID to the caller
+    pub fn build(db: &'static dyn Data) -> Result<Self> {
+        let mut registration_info = HashMap::new();
+        // Inserting registrations into cache
+        for appservice in db.all()? {
+            registration_info.insert(
+                appservice.0,
+                appservice
+                    .1
+                    .try_into()
+                    .expect("Should be validated on registration"),
+            );
+        }
+
+        Ok(Self {
+            db,
+            registration_info: RwLock::new(registration_info),
+        })
+    }
+    /// Registers an appservice and returns the ID to the caller.
     pub async fn register_appservice(&self, yaml: Registration) -> Result<String> {
         services()
             .appservice
@@ -111,7 +131,7 @@ impl Service {
         self.db.register_appservice(yaml)
     }
 
-    /// Remove an appservice registration
+    /// Removes an appservice registration.
     ///
     /// # Arguments
     ///
@@ -135,7 +155,12 @@ impl Service {
         self.db.iter_ids()
     }
 
-    pub fn all(&self) -> Result<Vec<(String, Registration)>> {
+    pub async fn all(&self) -> Vec<RegistrationInfo> {
-        self.db.all()
+        self.registration_info
+            .read()
+            .await
+            .values()
+            .cloned()
+            .collect()
     }
 }

src/service/mod.rs

@@ -4,7 +4,7 @@ use std::{
 };
 
 use lru_cache::LruCache;
-use tokio::sync::{Mutex, RwLock};
+use tokio::sync::Mutex;
 
 use crate::{Config, Result};
 
@@ -56,10 +56,7 @@ impl Services {
         config: Config,
     ) -> Result<Self> {
         Ok(Self {
-            appservice: appservice::Service {
+            appservice: appservice::Service::build(db)?,
-                db,
-                registration_info: RwLock::new(HashMap::new()),
-            },
             pusher: pusher::Service { db },
             rooms: rooms::Service {
                 alias: rooms::alias::Service { db },

src/service/rooms/timeline/mod.rs

@@ -524,17 +524,11 @@ impl Service {
             }
         }
 
-        for appservice in services()
+        for appservice in services().appservice.all().await {
-            .appservice
-            .registration_info
-            .read()
-            .await
-            .values()
-        {
             if services()
                 .rooms
                 .state_cache
-                .appservice_in_room(&pdu.room_id, appservice)?
+                .appservice_in_room(&pdu.room_id, &appservice)?
             {
                 services()
                     .sending