Plexus/Plex
1
0
Fork 0
mirror of https://github.com/plexusorg/Plex.git synced 2025-07-01 23:56:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

add get all admins method to get all admin objects

Browse Source
This commit is contained in:
Taah
2022-03-31 22:47:15 -07:00
parent f0a05a8256
commit 8ebb1c91a0
8 changed files with 381 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
src/main/java/dev/plex/admin/AdminList.java
View File

@ -2,12 +2,15 @@ package dev.plex.admin;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.reflect.TypeToken;
import com.google.gson.Gson;
import dev.morphia.Datastore;
import dev.morphia.query.Query;
import dev.plex.PlexBase;
import dev.plex.player.PlexPlayer;
import dev.plex.rank.enums.Rank;
import dev.plex.storage.StorageType;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
@ -23,8 +26,7 @@ import java.util.stream.Collectors;
* @see Admin
*/
public class AdminList extends PlexBase
{
public class AdminList extends PlexBase {
/**
* Key/Value storage, where the key is the unique ID of the admin
*/
@ -35,8 +37,7 @@ public class AdminList extends PlexBase
*
* @param admin The admin object
*/
public void addToCache(Admin admin)
{
public void addToCache(Admin admin) {
admins.put(admin.getUuid(), admin);
}
@ -46,8 +47,7 @@ public class AdminList extends PlexBase
* @param uuid The unique ID of the admin
* @see UUID
*/
public void removeFromCache(UUID uuid)
{
public void removeFromCache(UUID uuid) {
admins.remove(uuid);
}
@ -56,35 +56,76 @@ public class AdminList extends PlexBase
*
* @return An array list of the names of every admin
*/
public List<String> getAllAdmins()
{
public List<String> getAllAdmins() {
List<String> admins = Lists.newArrayList();
if (plugin.getStorageType() == StorageType.MONGODB)
{
if (plugin.getStorageType() == StorageType.MONGODB) {
Datastore store = plugin.getMongoConnection().getDatastore();
Query<PlexPlayer> query = store.find(PlexPlayer.class);
admins.addAll(query.stream().filter(plexPlayer -> plexPlayer.getRankFromString().isAtLeast(Rank.ADMIN)).map(PlexPlayer::getName).collect(Collectors.toList()));
}
else
{
try (Connection con = plugin.getSqlConnection().getCon())
{
} else {
try (Connection con = plugin.getSqlConnection().getCon()) {
PreparedStatement statement = con.prepareStatement("SELECT * FROM `players` WHERE rank IN(?, ?, ?)");
statement.setString(1, Rank.ADMIN.name().toLowerCase());
statement.setString(2, Rank.SENIOR_ADMIN.name().toLowerCase());
statement.setString(3, Rank.EXECUTIVE.name().toLowerCase());
ResultSet set = statement.executeQuery();
while (set.next())
{
while (set.next()) {
admins.add(set.getString("name"));
}
}
catch (SQLException throwables)
{
} catch (SQLException throwables) {
throwables.printStackTrace();
}
}
return admins;
}
/**
* Gathers every admin (cached and databsed)
*
* @return An array list of the names of every admin
*/
public List<PlexPlayer> getAllAdminPlayers() {
List<PlexPlayer> plexPlayers = Lists.newArrayList();
if (plugin.getStorageType() == StorageType.MONGODB) {
Datastore store = plugin.getMongoConnection().getDatastore();
Query<PlexPlayer> query = store.find(PlexPlayer.class);
return query.stream().toList().stream().filter(player -> plugin.getRankManager().isAdmin(player)).collect(Collectors.toList());
} else {
try (Connection con = plugin.getSqlConnection().getCon()) {
PreparedStatement statement = con.prepareStatement("SELECT * FROM `players` WHERE rank IN(?, ?, ?)");
statement.setString(1, Rank.ADMIN.name().toLowerCase());
statement.setString(2, Rank.SENIOR_ADMIN.name().toLowerCase());
statement.setString(3, Rank.EXECUTIVE.name().toLowerCase());
ResultSet set = statement.executeQuery();
while (set.next()) {
String uuid = set.getString("uuid");
String name = set.getString("name");
String loginMSG = set.getString("login_msg");
String prefix = set.getString("prefix");
String rankName = set.getString("rank").toUpperCase();
long coins = set.getLong("coins");
boolean vanished = set.getBoolean("vanished");
boolean commandspy = set.getBoolean("commandspy");
List<String> ips = new Gson().fromJson(set.getString("ips"), new TypeToken<List<String>>() {
}.getType());
PlexPlayer plexPlayer = new PlexPlayer(UUID.fromString(uuid));
plexPlayer.setName(name);
plexPlayer.setLoginMessage(loginMSG);
plexPlayer.setPrefix(prefix);
plexPlayer.setRank(rankName);
plexPlayer.setIps(ips);
plexPlayer.setCoins(coins);
plexPlayer.setVanished(vanished);
plexPlayer.setCommandSpy(commandspy);
plexPlayers.add(plexPlayer);
}
} catch (SQLException throwables) {
throwables.printStackTrace();
}
}
return plexPlayers;
}
}