package emu.grasscutter.database; import java.util.List; import com.mongodb.WriteResult; import dev.morphia.query.FindOptions; import dev.morphia.query.Query; import dev.morphia.query.internal.MorphiaCursor; import emu.grasscutter.GenshinConstants; import emu.grasscutter.Grasscutter; import emu.grasscutter.game.Account; import emu.grasscutter.game.GenshinPlayer; import emu.grasscutter.game.avatar.GenshinAvatar; import emu.grasscutter.game.friends.Friendship; import emu.grasscutter.game.inventory.GenshinItem; public class DatabaseHelper { protected static FindOptions FIND_ONE = new FindOptions().limit(1); public static Account createAccount(String username) { return createAccountWithId(username, 0); } public static Account createAccountWithId(String username, int reservedId) { // Unique names only Account exists = DatabaseHelper.getAccountByName(username); if (exists != null) { return null; } // Make sure there are no id collisions if (reservedId > 0) { // Cannot make account with the same uid as the server console if (reservedId == GenshinConstants.SERVER_CONSOLE_UID) { return null; } exists = DatabaseHelper.getAccountByPlayerId(reservedId); if (exists != null) { return null; } } // Account Account account = new Account(); account.setUsername(username); account.setId(Integer.toString(DatabaseManager.getNextId(account))); if (reservedId > 0) { account.setPlayerId(reservedId); } DatabaseHelper.saveAccount(account); return account; } @Deprecated public static Account createAccountWithPassword(String username, String password) { // Unique names only Account exists = DatabaseHelper.getAccountByName(username); if (exists != null) { return null; } // Account Account account = new Account(); account.setId(Integer.toString(DatabaseManager.getNextId(account))); account.setUsername(username); account.setPassword(password); DatabaseHelper.saveAccount(account); return account; } public static void saveAccount(Account account) { DatabaseManager.getDatastore().save(account); } public static Account getAccountByName(String username) { MorphiaCursor cursor = DatabaseManager.getDatastore().createQuery(Account.class).field("username").equalIgnoreCase(username).find(FIND_ONE); if (!cursor.hasNext()) return null; return cursor.next(); } public static Account getAccountByToken(String token) { if (token == null) return null; MorphiaCursor cursor = DatabaseManager.getDatastore().createQuery(Account.class).field("token").equal(token).find(FIND_ONE); if (!cursor.hasNext()) return null; return cursor.next(); } public static Account getAccountById(String uid) { MorphiaCursor cursor = DatabaseManager.getDatastore().createQuery(Account.class).field("_id").equal(uid).find(FIND_ONE); if (!cursor.hasNext()) return null; return cursor.next(); } private static Account getAccountByPlayerId(int playerId) { MorphiaCursor cursor = DatabaseManager.getDatastore().createQuery(Account.class).field("playerId").equal(playerId).find(FIND_ONE); if (!cursor.hasNext()) return null; return cursor.next(); } public static boolean deleteAccount(String username) { Query q = DatabaseManager.getDatastore().createQuery(Account.class).field("username").equalIgnoreCase(username); return DatabaseManager.getDatastore().findAndDelete(q) != null; } public static GenshinPlayer getPlayerById(int id) { Query query = DatabaseManager.getDatastore().createQuery(GenshinPlayer.class).field("_id").equal(id); MorphiaCursor cursor = query.find(FIND_ONE); if (!cursor.hasNext()) return null; return cursor.next(); } public static boolean checkPlayerExists(int id) { MorphiaCursor query = DatabaseManager.getDatastore().createQuery(GenshinPlayer.class).field("_id").equal(id).find(FIND_ONE); return query.hasNext(); } public static synchronized GenshinPlayer createPlayer(GenshinPlayer character, int reservedId) { // Check if reserved id int id = 0; if (reservedId > 0 && !checkPlayerExists(reservedId)) { id = reservedId; character.setId(id); } else { do { id = DatabaseManager.getNextId(character); } while (checkPlayerExists(id)); character.setId(id); } // Save to database DatabaseManager.getDatastore().save(character); return character; } public static synchronized int getNextPlayerId(int reservedId) { // Check if reserved id int id = 0; if (reservedId > 0 && !checkPlayerExists(reservedId)) { id = reservedId; } else { do { id = DatabaseManager.getNextId(GenshinPlayer.class); } while (checkPlayerExists(id)); } return id; } public static void savePlayer(GenshinPlayer character) { DatabaseManager.getDatastore().save(character); } public static void saveAvatar(GenshinAvatar avatar) { DatabaseManager.getDatastore().save(avatar); } public static List getAvatars(GenshinPlayer player) { Query query = DatabaseManager.getDatastore().createQuery(GenshinAvatar.class).filter("ownerId", player.getId()); return query.find().toList(); } public static void saveItem(GenshinItem item) { DatabaseManager.getDatastore().save(item); } public static boolean deleteItem(GenshinItem item) { WriteResult result = DatabaseManager.getDatastore().delete(item); return result.wasAcknowledged(); } public static List getInventoryItems(GenshinPlayer player) { Query query = DatabaseManager.getDatastore().createQuery(GenshinItem.class).filter("ownerId", player.getId()); return query.find().toList(); } public static List getFriends(GenshinPlayer player) { Query query = DatabaseManager.getDatastore().createQuery(Friendship.class).filter("ownerId", player.getId()); return query.find().toList(); } public static List getReverseFriends(GenshinPlayer player) { Query query = DatabaseManager.getDatastore().createQuery(Friendship.class).filter("friendId", player.getId()); return query.find().toList(); } public static void saveFriendship(Friendship friendship) { DatabaseManager.getDatastore().save(friendship); } public static void deleteFriendship(Friendship friendship) { DatabaseManager.getDatastore().delete(friendship); } public static Friendship getReverseFriendship(Friendship friendship) { Query query = DatabaseManager.getDatastore().createQuery(Friendship.class); query.and( query.criteria("ownerId").equal(friendship.getFriendId()), query.criteria("friendId").equal(friendship.getOwnerId()) ); MorphiaCursor reverseFriendship = query.find(FIND_ONE); if (!reverseFriendship.hasNext()) return null; return reverseFriendship.next(); } }