cloak sec and hats

src/methods/player.rs

@@ -19,37 +19,18 @@ pub fn player(session: &Session, uuid: String) -> Result {
     match cursor.next() {
         Some(Ok(player)) => {
             // Player exists, deserialize and return their data
-            Ok(Response::Player {
+            Ok(Response::Player(crate::response::PlayerResponse::Player {
                 cloak: player.cloak,
                 uuid,
                 cloaks: player.cloaks,
                 hats: player.hats,
                 hat: player.hat,
-            })
+            }))
         }
 
-        // No document found or deserialization error
-        _ => {
-            let player = Player {
-                uuid: uuid.clone(),
-                cloaks: Vec::new(),
-                cloak: "".to_string(),
-                hats: Vec::new(),
-                hat: String::new(),
-            };
-
-            // Insert the new player
-            session.database.players.insert_one(player.clone()).run()?;
-
-            // Return the newly created player's data
-            Ok(Response::Player {
-                cloak: player.cloak,
-                uuid,
-                cloaks: player.cloaks,
-                hats: player.hats,
-                hat: player.hat,
-            })
-        }
+        _ => Ok(crate::response::Response::Player(
+            crate::response::PlayerResponse::NonSaturnPlayer,
+        )),
     }
 }
 
@@ -66,11 +47,59 @@ pub fn create(session: &Session) -> Result {
 
     session.database.players.insert_one(player.clone()).run()?;
 
-    Ok(Response::Player {
+    Ok(Response::Player(crate::response::PlayerResponse::Player {
         cloak: player.cloak,
         uuid,
         cloaks: player.cloaks,
         hats: player.hats,
         hat: player.hat,
-    })
+    }))
+}
+
+pub fn login(session: &Session) -> Result {
+    let uuid = session.local_player.id.clone();
+
+    // Try to find the player with the given UUID
+    let mut cursor = session
+        .database
+        .players
+        .find(doc! { "uuid": uuid.clone() })
+        .run()?;
+
+    // Check if there's a document in the cursor
+    match cursor.next() {
+        Some(Ok(player)) => {
+            // Player exists, deserialize and return their data
+            Ok(Response::Player(crate::response::PlayerResponse::Player {
+                cloak: player.cloak,
+                uuid,
+                cloaks: player.cloaks,
+                hats: player.hats,
+                hat: player.hat,
+            }))
+        }
+
+        // No document found or deserialization error
+        _ => {
+            let player = Player {
+                uuid: uuid.clone(),
+                cloaks: Vec::new(),
+                cloak: "".to_string(),
+                hats: Vec::new(),
+                hat: String::new(),
+            };
+
+            // Insert the new player
+            session.database.players.insert_one(player.clone()).run()?;
+
+            // Return the newly created player's data
+            Ok(Response::Player(crate::response::PlayerResponse::Player {
+                cloak: player.cloak,
+                uuid,
+                cloaks: player.cloaks,
+                hats: player.hats,
+                hat: player.hat,
+            }))
+        }
+    }
 }