Rework of the permission API

This commit is contained in:
Felix Cravic 2020-12-10 02:56:56 +01:00
parent 30a7843018
commit 00c144b592
7 changed files with 139 additions and 142 deletions

View File

@ -3,16 +3,15 @@ package net.minestom.server.command;
import net.minestom.server.permission.Permission;
import org.jetbrains.annotations.NotNull;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArraySet;
/**
* Represents the console when sending a command to the server.
*/
public class ConsoleSender implements CommandSender {
private final List<Permission> permissions = new LinkedList<>();
private final Set<Permission> permissions = new CopyOnWriteArraySet<>();
@Override
public void sendMessage(@NotNull String message) {
@ -21,7 +20,7 @@ public class ConsoleSender implements CommandSender {
@NotNull
@Override
public Collection<Permission> getAllPermissions() {
public Set<Permission> getAllPermissions() {
return permissions;
}
}

View File

@ -96,7 +96,7 @@ public abstract class Entity implements Viewable, EventHandler, DataContainer, P
private final int id;
protected final Set<Player> viewers = new CopyOnWriteArraySet<>();
private Data data;
private final List<Permission> permissions = new LinkedList<>();
private final Set<Permission> permissions = new CopyOnWriteArraySet<>();
protected UUID uuid;
private boolean isActive; // False if entity has only been instanced without being added somewhere
@ -372,7 +372,7 @@ public abstract class Entity implements Viewable, EventHandler, DataContainer, P
@NotNull
@Override
public Collection<Permission> getAllPermissions() {
public Set<Permission> getAllPermissions() {
return permissions;
}

View File

@ -1,15 +0,0 @@
package net.minestom.server.permission;
import net.minestom.server.command.CommandSender;
import org.jetbrains.annotations.NotNull;
/**
* Basic {@link Permission} implementation that only requires the permission to be given to the {@link CommandSender} to be considered applied
* (eg. no arguments)
*/
public class BasicPermission implements Permission<Object> {
@Override
public boolean isValidFor(@NotNull PermissionHandler permissionHandler, Object data) {
return true;
}
}

View File

@ -1,47 +1,70 @@
package net.minestom.server.permission;
import net.minestom.server.command.CommandSender;
import net.minestom.server.data.Data;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jglrxavpok.hephaistos.nbt.NBTCompound;
import java.util.Objects;
/**
* Representation of a permission granted to a {@link CommandSender}.
*
* @param <T> the type of data that this permission can handle in {@link #isValidFor(PermissionHandler, Object)}.
* Used if you want to allow passing additional data to check if the permission is valid in a certain situation,
* you can default it to {@link Object} if you do not need it.
*/
@FunctionalInterface
public interface Permission<T> {
public class Permission {
private String permissionName;
private NBTCompound data;
/**
* Does the given {@link CommandSender} have the permission represented by this object?
* <p>
* Called with {@link CommandSender#hasPermission(Permission)}, the {@link CommandSender} requires to both
* have this permission and validate the condition in this method.
* Creates a new permission object with optional data.
*
* @param permissionHandler the permission handler
* @param data the optional data (eg the number of home possible, placing a block at X position)
* @return true if the commandSender possesses this permission
* @param permissionName the name of the permission
* @param data the optional data of the permission
*/
boolean isValidFor(@NotNull PermissionHandler permissionHandler, @Nullable T data);
/**
* Writes any required data for this permission inside the given destination.
*
* @param destination the {@link Data} to write to
*/
default void write(@NotNull Data destination) {
public Permission(@NotNull String permissionName, @Nullable NBTCompound data) {
this.permissionName = permissionName;
this.data = data;
}
/**
* Reads any required data for this permission from the given destination.
* Creates a new permission object without additional data
*
* @param source the {@link Data} to read from
* @return this for chaining
* @param permissionName the name of the permission
*/
default Permission read(@Nullable Data source) {
return this;
public Permission(@NotNull String permissionName) {
this(permissionName, null);
}
/**
* Gets the name of the permission.
*
* @return the permission name
*/
@NotNull
public String getPermissionName() {
return permissionName;
}
/**
* Gets the data associated to this permission.
*
* @return the nbt data of this permission, can be null if not any
*/
@Nullable
public NBTCompound getNBTData() {
return data;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Permission that = (Permission) o;
return permissionName.equals(that.permissionName) && Objects.equals(data, that.data);
}
@Override
public int hashCode() {
return Objects.hash(permissionName, data);
}
}

View File

@ -3,7 +3,7 @@ package net.minestom.server.permission;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.util.Collection;
import java.util.Set;
/**
* Represents an object which can have permissions.
@ -11,16 +11,16 @@ import java.util.Collection;
public interface PermissionHandler {
/**
* Returns all permissions associated to this command sender.
* Returns all permissions associated to this handler.
* The returned collection should be modified only by subclasses.
*
* @return the permissions of this command sender.
* @return the permissions of this handler.
*/
@NotNull
Collection<Permission> getAllPermissions();
Set<Permission> getAllPermissions();
/**
* Adds a {@link Permission} to this commandSender
* Adds a {@link Permission} to this handler.
*
* @param permission the permission to add
*/
@ -29,7 +29,7 @@ public interface PermissionHandler {
}
/**
* Removes a {@link Permission} from this commandSender
* Removes a {@link Permission} from this handler.
*
* @param permission the permission to remove
*/
@ -38,55 +38,49 @@ public interface PermissionHandler {
}
/**
* Checks if the given {@link Permission} is possessed by this command sender.
* Simple shortcut to <pre>getAllPermissions().contains(permission) &amp;&amp; permission.isValidFor(this)</pre> for readability.
* Gets if this handler has the permission {@code permission}.
* <p>
* Uses {@link Permission#equals(Object)} internally.
*
* @param p permission to check against
* @return true if the sender has the permission and validate {@link Permission#isValidFor(PermissionHandler, Object)}
* @param permission the permission to check
* @return true if the handler has the permission
*/
default boolean hasPermission(@NotNull Permission p) {
return hasPermission(p, null);
default boolean hasPermission(@NotNull Permission permission) {
for (Permission permissionLoop : getAllPermissions()) {
if (permissionLoop.equals(permission)) {
return true;
}
default <T> boolean hasPermission(@NotNull Permission<T> p, @Nullable T data) {
return getAllPermissions().contains(p) && p.isValidFor(this, data);
}
return false;
}
/**
* Checks if the given {@link Permission} is possessed by this command sender.
* Will call {@link Permission#isValidFor(PermissionHandler, Object)} on all permissions that are an instance of {@code permissionClass}.
* If no matching permission is found, this result returns false.
* Gets if this handler has the permission with the name {@code permissionName} and which verify the optional
* {@link PermissionVerifier}.
*
* @param permissionClass the permission class to check
* @return true if the sender has the permission and validate {@link Permission#isValidFor(PermissionHandler, Object)}
* @see #getAllPermissions()
* @param permissionName the permission name
* @param permissionVerifier the optional verifier
* @return true if the handler has the permission
*/
default boolean hasPermission(@NotNull Class<? extends Permission> permissionClass) {
boolean result = true;
boolean foundPerm = false;
for (Permission p : getAllPermissions()) {
if (permissionClass.isInstance(p)) {
foundPerm = true;
result &= p.isValidFor(this, null);
default boolean hasPermission(@NotNull String permissionName, @Nullable PermissionVerifier permissionVerifier) {
for (Permission permission : getAllPermissions()) {
if (permission.getPermissionName().equals(permissionName)) {
return permissionVerifier != null ?
permissionVerifier.isValid(permission.getNBTData()) :
true;
}
}
if (!foundPerm)
return false;
return result;
}
default <T> boolean hasPermission(@NotNull Class<? extends Permission<T>> permissionClass, @Nullable T data) {
boolean result = true;
boolean foundPerm = false;
for (Permission p : getAllPermissions()) {
if (permissionClass.isInstance(p)) {
foundPerm = true;
result &= p.isValidFor(this, data);
}
}
if (!foundPerm)
return false;
return result;
/**
* Gets if this handler has the permission with the name {@code permissionName}.
*
* @param permissionName the permission name
* @return true if the handler has the permission
*/
default boolean hasPermission(@NotNull String permissionName) {
return hasPermission(permissionName, null);
}
}

View File

@ -0,0 +1,20 @@
package net.minestom.server.permission;
import org.jetbrains.annotations.Nullable;
import org.jglrxavpok.hephaistos.nbt.NBTCompound;
/**
* Interface used to check if the {@link NBTCompound nbt data} of a {@link Permission} is correct.
*/
@FunctionalInterface
public interface PermissionVerifier {
/**
* Called when using {@link PermissionHandler#hasPermission(String, PermissionVerifier)}.
*
* @param nbtCompound the data of the permission, can be null if not any
* @return true if {@link PermissionHandler#hasPermission(String, PermissionVerifier)}
* should return true, false otherwise
*/
boolean isValid(@Nullable NBTCompound nbtCompound);
}

View File

@ -3,8 +3,7 @@ package permissions;
import net.minestom.server.MinecraftServer;
import net.minestom.server.entity.Player;
import net.minestom.server.permission.Permission;
import net.minestom.server.permission.PermissionHandler;
import org.jetbrains.annotations.NotNull;
import org.jglrxavpok.hephaistos.nbt.NBTCompound;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
@ -19,6 +18,8 @@ public class TestPermissions {
private Player player;
private Permission permission1, permission2;
@BeforeEach
public void init() {
MinecraftServer.init(); // for entity manager
@ -32,67 +33,42 @@ public class TestPermissions {
return false;
}
};
permission1 = new Permission("perm.name",
new NBTCompound()
.setString("name", "Minestom")
.setInt("amount", 5));
permission2 = new Permission("perm.name2");
}
@Test
public void noPermission() {
assertFalse(player.hasPermission(Permission.class));
}
class PermTest1 implements Permission<Object> {
@Override
public boolean isValidFor(@NotNull PermissionHandler permissionHandler, Object data) {
return true;
}
}
class PermTest2 implements Permission<Object> {
@Override
public boolean isValidFor(@NotNull PermissionHandler permissionHandler, Object data) {
return true;
}
assertFalse(player.hasPermission(""));
assertFalse(player.hasPermission("random.permission"));
}
@Test
public void hasPermissionClass() {
assertFalse(player.hasPermission(Permission.class));
player.addPermission(new PermTest1());
assertTrue(player.hasPermission(PermTest1.class));
assertFalse(player.hasPermission(PermTest2.class));
assertTrue(player.hasPermission(Permission.class)); // allow superclasses
player.addPermission(new PermTest2());
assertTrue(player.hasPermission(PermTest2.class));
}
assertFalse(player.hasPermission(permission1));
player.addPermission(permission1);
assertTrue(player.hasPermission(permission1));
assertFalse(player.hasPermission(permission2));
class BooleanPerm implements Permission<Object> {
private final boolean value;
BooleanPerm(boolean v) {
this.value = v;
}
@Override
public boolean isValidFor(@NotNull PermissionHandler permissionHandler, Object data) {
return value;
}
player.addPermission(permission2);
assertTrue(player.hasPermission(permission2));
}
@Test
public void hasTwoPermissionsOfSameClassButContradictEachOther() {
player.addPermission(new BooleanPerm(true));
assertTrue(player.hasPermission(BooleanPerm.class));
player.addPermission(new BooleanPerm(false));
assertFalse(player.hasPermission(BooleanPerm.class)); // all permissions must be valid
}
public void hasPermissionNameNbt() {
player.addPermission(permission1);
assertTrue(player.hasPermission("perm.name"));
assertTrue(player.hasPermission("perm.name",
nbtCompound -> nbtCompound != null && nbtCompound.getString("name").equals("Minestom")));
@Test
public void singlePermission() {
Permission p = (commandSender, data) -> true;
player.addPermission(p);
assertTrue(p.isValidFor(player, null));
assertTrue(player.hasPermission(p));
assertTrue(player.hasPermission(Permission.class));
player.addPermission(permission2);
assertFalse(player.hasPermission("perm.name2", nbtCompound -> nbtCompound != null));
}
@AfterEach