Add prefix customization command

2025-08-14 18:30:33 -03:00
parent eb6dcb4604
commit 53d61cfbbb
7 changed files with 143 additions and 7 deletions
@@ -253,6 +253,36 @@ impl Server {
        }
    }

+    pub async fn get_or_create_by_id(
+        id: u64,
+        session: &Session,
+        http: &Client,
+    ) -> Result<Self, DatabaseOrDiscordError> {
+        let server = Server::get_by_id(id, session)
+            .await
+            .map_err(DatabaseOrDiscordError::Database)?;
+
+        if let Some(server) = server {
+            Ok(server)
+        } else {
+            let server: Server = http
+                .guild(Id::new(id))
+                .await
+                .map_err(DatabaseOrDiscordError::Discord)?
+                .model()
+                .await
+                .map_err(|_| DatabaseOrDiscordError::DiscordDeserializationFailed)?
+                .into();
+
+            server
+                .insert(session)
+                .await
+                .map_err(DatabaseOrDiscordError::Database)?;
+
+            Ok(server)
+        }
+    }
+
    pub async fn insert(&self, session: &Session) -> Result<(), DatabaseError> {
        session
            .query_unpaged(