NullValkyrie

UserDataManager.java (1710B)

---

 package me.night.nullvalkyrie.database;
 
 import me.night.nullvalkyrie.ui.player.ScoreboardListener;
 import org.bson.Document;
 import org.bson.conversions.Bson;
 
 import java.util.HashMap;
 
 public class UserDataManager {
     public void createUserBank(String uuid) {
         Document document = new Document();
         document.put("UUID", uuid);
         document.put("Bank", 0);
         new DatabaseManager().getUsersDB().insertOne(document);
     }
 
     public void updateUserBank(String uuid, Integer coins, ScoreboardListener... listener) {
         Document document = new DatabaseManager().getUsersDB().find(new Document("UUID", uuid)).first();
         if (document != null) {
             Integer coinsBefore = document.getInteger("Bank");
             Bson updated = new Document("Bank", coins + coinsBefore);
             Bson update = new Document("$set", updated);
             new DatabaseManager().getUsersDB().updateOne(document, update);
             listener[0].sideBarManager.addBank(uuid, coins);
         } else {
             Document doc = new Document();
             doc.put("UUID", uuid);
             doc.put("Bank", coins);
             new DatabaseManager().getUsersDB().insertOne(doc);
             listener[0].sideBarManager.addBank(uuid, coins);
         }
 
 
     }
 
     public HashMap<String, Object> getUser(String uuid) {
         Document document = new DatabaseManager().getUsersDB().find(new Document("UUID", uuid)).first();
         if (document != null) {
             HashMap<String, Object> map = new HashMap<>();
             for (String key : document.keySet()) map.put(key, document.get(key));
             map.remove("_id");
             return map;
         }
         return null;
     }
 }