cloak sec and hats

2025-03-30 12:18:54 +02:00
parent dfe7b0e1f7
commit caae6ea0fc
6 changed files with 173 additions and 58 deletions
@@ -1,24 +1,32 @@
 use mongodb::bson::doc;

-use crate::response::{Response, Result};
+use crate::response::{Error, Response, Result};

-use super::Session;
+use super::{player, Session};

 pub fn set_cloak(session: &Session, cloak: String) -> Result {
-    session
-        .database
-        .players
-        .update_one(
-            doc! {
-                "uuid": session.local_player.id.clone(),
-            },
-            doc! {
-                "$set": {
-                    "cloak": cloak,
-                },
-            },
-        )
-        .run()?;
-
-    Ok(Response::Success)
+    match player::player(session, session.local_player.id.clone())? {
+        Response::Player(crate::response::PlayerResponse::Player { cloaks, .. }) => {
+            if cloaks.contains(&cloak) || cloak.is_empty() {
+                session
+                    .database
+                    .players
+                    .update_one(
+                        doc! {
+                            "uuid": session.local_player.id.clone(),
+                        },
+                        doc! {
+                            "$set": {
+                                "cloak": cloak,
+                            },
+                        },
+                    )
+                    .run()?;
+                Ok(Response::Success)
+            } else {
+                Err(Error::SomethingWentWrong)
+            }
+        }
+        _ => Err(Error::SomethingWentWrong),
+    }
 }