Essentials/Essentials/src/main/java/com/earth2me/essentials/perm/PermissionsHandler.java

245 lines
9.2 KiB
Java

package com.earth2me.essentials.perm;
import com.earth2me.essentials.Essentials;
import com.earth2me.essentials.User;
import com.earth2me.essentials.perm.impl.AbstractVaultHandler;
import com.earth2me.essentials.perm.impl.ConfigPermissionsHandler;
import com.earth2me.essentials.perm.impl.GenericVaultHandler;
import com.earth2me.essentials.perm.impl.LuckPermsHandler;
import com.earth2me.essentials.perm.impl.ModernVaultHandler;
import com.earth2me.essentials.perm.impl.SuperpermsHandler;
import com.earth2me.essentials.utils.TriState;
import com.google.common.collect.ImmutableSet;
import org.bukkit.OfflinePlayer;
import org.bukkit.entity.Player;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.logging.Level;
public class PermissionsHandler implements IPermissionsHandler {
private final transient String defaultGroup = "default";
private final transient Essentials ess;
private transient IPermissionsHandler handler = null;
private transient boolean useSuperperms;
private Class<?> lastHandler = null;
public PermissionsHandler(final Essentials plugin, final boolean useSuperperms) {
this.ess = plugin;
this.useSuperperms = useSuperperms;
}
@Override
public String getGroup(final OfflinePlayer base) {
final long start = System.nanoTime();
String group = handler.getGroup(base);
if (group == null) {
group = defaultGroup;
}
checkPermLag(start, String.format("Getting group for %s", base.getName()));
return group;
}
@Override
public List<String> getGroups(final OfflinePlayer base) {
final long start = System.nanoTime();
final List<String> groups = new ArrayList<>();
groups.add(defaultGroup);
groups.addAll(handler.getGroups(base));
checkPermLag(start, String.format("Getting groups for %s", base.getName()));
return Collections.unmodifiableList(groups);
}
@Override
public List<String> getGroups() {
final long start = System.nanoTime();
List<String> groups = handler.getGroups();
if (groups == null || groups.isEmpty()) {
groups = Collections.singletonList(defaultGroup);
}
checkPermLag(start, "Getting all groups");
return Collections.unmodifiableList(groups);
}
@Override
public boolean addToGroup(OfflinePlayer base, String group) {
final long start = System.nanoTime();
final boolean result = handler.addToGroup(base, group);
checkPermLag(start, String.format("Adding group to %s", base.getName()));
return result;
}
@Override
public boolean removeFromGroup(OfflinePlayer base, String group) {
final long start = System.nanoTime();
final boolean result = handler.removeFromGroup(base, group);
checkPermLag(start, String.format("Removing group from %s", base.getName()));
return result;
}
@Override
public boolean canBuild(final Player base, final String group) {
return handler.canBuild(base, group);
}
@Override
public boolean inGroup(final Player base, final String group) {
final long start = System.nanoTime();
final boolean result = handler.inGroup(base, group);
checkPermLag(start, String.format("Checking if %s is in group %s", base.getName(), group));
return result;
}
@Override
public boolean hasPermission(final Player base, final String node) {
return handler.hasPermission(base, node);
}
@Override
public boolean isPermissionSet(final Player base, final String node) {
return handler.isPermissionSet(base, node);
}
@Override
public TriState isPermissionSetExact(Player base, String node) {
return handler.isPermissionSetExact(base, node);
}
@Override
public String getPrefix(final Player base) {
final long start = System.nanoTime();
String prefix = handler.getPrefix(base);
if (prefix == null) {
prefix = "";
}
checkPermLag(start, String.format("Getting prefix for %s", base.getName()));
return prefix;
}
@Override
public String getSuffix(final Player base) {
final long start = System.nanoTime();
String suffix = handler.getSuffix(base);
if (suffix == null) {
suffix = "";
}
checkPermLag(start, String.format("Getting suffix for %s", base.getName()));
return suffix;
}
@Override
public void registerContext(final String context, final Function<User, Iterable<String>> calculator, final Supplier<Iterable<String>> suggestions) {
handler.registerContext(context, calculator, suggestions);
}
@Override
public void unregisterContexts() {
handler.unregisterContexts();
}
@Override
public String getBackendName() {
return handler.getBackendName();
}
@Override
public boolean tryProvider(Essentials ess) {
return true;
}
public void checkPermissions() {
// load and assign a handler
final List<Class<? extends SuperpermsHandler>> providerClazz = Arrays.asList(
LuckPermsHandler.class,
ModernVaultHandler.class,
GenericVaultHandler.class,
SuperpermsHandler.class
);
for (final Class<? extends IPermissionsHandler> providerClass : providerClazz) {
try {
final IPermissionsHandler provider = providerClass.newInstance();
if (provider.tryProvider(ess)) {
if (provider.getClass().isInstance(this.handler)) {
return;
}
if (this.handler != null) {
unregisterContexts();
}
this.handler = provider;
initContexts();
break;
}
} catch (final Throwable ignored) {
}
}
if (handler == null) {
if (useSuperperms) {
handler = new SuperpermsHandler();
} else {
handler = new ConfigPermissionsHandler(ess);
}
}
// don't spam logs
final Class<?> handlerClass = handler.getClass();
if (lastHandler != null && lastHandler == handlerClass) {
return;
}
lastHandler = handlerClass;
// output handler info
if (handler instanceof AbstractVaultHandler) {
String enabledPermsPlugin = ((AbstractVaultHandler) handler).getEnabledPermsPlugin();
if (enabledPermsPlugin == null) enabledPermsPlugin = "generic";
ess.getLogger().info("Using Vault based permissions (" + enabledPermsPlugin + ")");
} else if (handler.getClass() == SuperpermsHandler.class) {
if (handler.tryProvider(ess)) {
ess.getLogger().warning("Detected supported permissions plugin " +
((SuperpermsHandler) handler).getEnabledPermsPlugin() + " without Vault installed.");
ess.getLogger().warning("Features such as chat prefixes/suffixes and group-related functionality will not " +
"work until you install Vault.");
}
ess.getLogger().info("Using superperms-based permissions.");
} else if (handler.getClass() == ConfigPermissionsHandler.class) {
ess.getLogger().info("Using config file enhanced permissions.");
ess.getLogger().info("Permissions listed in as player-commands will be given to all users.");
}
}
public void setUseSuperperms(final boolean useSuperperms) {
this.useSuperperms = useSuperperms;
}
public String getName() {
return handler.getClass().getSimpleName().replace("Provider", "");
}
private void checkPermLag(final long start, final String summary) {
final long elapsed = System.nanoTime() - start;
if (elapsed > ess.getSettings().getPermissionsLagWarning()) {
ess.getLogger().log(Level.WARNING, String.format("Permissions lag notice with (%s). Response took %fms. Summary: %s", getName(), elapsed / 1000000.0, summary));
}
}
private void initContexts() {
registerContext("essentials:afk", user -> Collections.singleton(String.valueOf(user.isAfk())), () -> ImmutableSet.of("true", "false"));
registerContext("essentials:muted", user -> Collections.singleton(String.valueOf(user.isMuted())), () -> ImmutableSet.of("true", "false"));
registerContext("essentials:vanished", user -> Collections.singleton(String.valueOf(user.isHidden())), () -> ImmutableSet.of("true", "false"));
registerContext("essentials:jailed", user -> Collections.singleton(String.valueOf(user.isJailed())), () -> ImmutableSet.of("true", "false"));
registerContext("essentials:jail", user -> Optional.ofNullable(user.getJail()).map(Arrays::asList).orElse(Collections.emptyList()), () -> {
try {
return ess.getJails().getList();
} catch (final Exception e) {
return Collections.emptyList();
}
});
}
}