mirror of https://github.com/plexusorg/Plex.git
89 lines
2.5 KiB
Java
89 lines
2.5 KiB
Java
package dev.plex.admin;
|
|
|
|
import com.google.common.collect.Lists;
|
|
import com.google.common.collect.Maps;
|
|
import dev.morphia.Datastore;
|
|
import dev.morphia.query.Query;
|
|
import dev.plex.Plex;
|
|
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;
|
|
import java.sql.SQLException;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.UUID;
|
|
import java.util.stream.Collectors;
|
|
|
|
/**
|
|
* Cached storage for Admin objects
|
|
* @see Admin
|
|
*/
|
|
|
|
public class AdminList
|
|
{
|
|
/**
|
|
* Key/Value storage, where the key is the unique ID of the admin
|
|
*/
|
|
private final Map<UUID, Admin> admins = Maps.newHashMap();
|
|
|
|
/**
|
|
* Adds the admin to cache
|
|
* @param admin The admin object
|
|
*/
|
|
public void addToCache(Admin admin)
|
|
{
|
|
admins.put(admin.getUuid(), admin);
|
|
}
|
|
|
|
/**
|
|
* Removes an admin from the cache
|
|
* @param uuid The unique ID of the admin
|
|
* @see UUID
|
|
*/
|
|
public void removeFromCache(UUID uuid)
|
|
{
|
|
admins.remove(uuid);
|
|
}
|
|
|
|
/**
|
|
* Gathers every admin's username (cached and databsed)
|
|
* @return An array list of the names of every admin
|
|
*/
|
|
public List<String> getAllAdmins()
|
|
{
|
|
List<String> admins = Lists.newArrayList();
|
|
if (Plex.get().getStorageType() == StorageType.MONGODB)
|
|
{
|
|
Datastore store = Plex.get().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 = Plex.get().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())
|
|
{
|
|
admins.add(set.getString("name"));
|
|
}
|
|
|
|
}
|
|
catch (SQLException throwables)
|
|
{
|
|
throwables.printStackTrace();
|
|
}
|
|
}
|
|
return admins;
|
|
}
|
|
|
|
}
|