worldofcraft/LuckPerms-DBA
10
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Migrate temporary permissions / parents in PEX migration

Browse Source
This commit is contained in:
Luck 2018-02-01 20:03:20 +00:00
parent fe200ca1ec
commit 5799a26cb7
No known key found for this signature in database
GPG Key ID: EFA9B3EC5FD90F8B
1 changed files with 98 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

127
bukkit/src/main/java/me/lucko/luckperms/bukkit/migration/MigrationPermissionsEx.java
View File

@ -25,6 +25,8 @@
package me.lucko.luckperms.bukkit.migration;
import com.google.common.base.Strings;
import me.lucko.luckperms.api.event.cause.CreationCause;
import me.lucko.luckperms.common.commands.CommandPermission;
import me.lucko.luckperms.common.commands.CommandResult;
@ -48,8 +50,13 @@ import org.bukkit.Bukkit;
import ru.tehkode.permissions.PermissionEntity;
import ru.tehkode.permissions.PermissionGroup;
import ru.tehkode.permissions.PermissionManager;
import ru.tehkode.permissions.PermissionUser;
import ru.tehkode.permissions.PermissionsData;
import ru.tehkode.permissions.bukkit.PermissionsEx;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
@ -60,6 +67,24 @@ import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
public class MigrationPermissionsEx extends SubCommand<Object> {
private static final Method GET_DATA_METHOD;
private static final Field TIMED_PERMISSIONS_FIELD;
private static final Field TIMED_PERMISSIONS_TIME_FIELD;
static {
try {
GET_DATA_METHOD = PermissionEntity.class.getDeclaredMethod("getData");
GET_DATA_METHOD.setAccessible(true);
TIMED_PERMISSIONS_FIELD = PermissionEntity.class.getDeclaredField("timedPermissions");
TIMED_PERMISSIONS_FIELD.setAccessible(true);
TIMED_PERMISSIONS_TIME_FIELD = PermissionEntity.class.getDeclaredField("timedPermissionsTime");
TIMED_PERMISSIONS_TIME_FIELD.setAccessible(true);
} catch (NoSuchMethodException | NoSuchFieldException e) {
throw new ExceptionInInitializerError(e);
}
}
public MigrationPermissionsEx(LocaleManager locale) {
super(CommandSpec.MIGRATION_COMMAND.spec(locale), "permissionsex", CommandPermission.MIGRATION, Predicates.alwaysFalse());
}
@ -174,37 +199,69 @@ public class MigrationPermissionsEx extends SubCommand<Object> {
return CommandResult.SUCCESS;
}
private static void migrateEntity(PermissionEntity entity, PermissionHolder holder, int weight) {
// migrate permissions
Map<String, List<String>> permissions = entity.getAllPermissions();
for (Map.Entry<String, List<String>> worldData : permissions.entrySet()) {
String world = worldData.getKey();
if (world != null && (world.equals("") || world.equals("*"))) {
world = null;
}
if (world != null) {
world = world.toLowerCase();
}
private static Map<String, List<String>> getPermanentPermissions(PermissionEntity entity) {
try {
//noinspection unchecked
PermissionsData data = (PermissionsData) GET_DATA_METHOD.invoke(entity);
return data.getPermissionsMap();
} catch (IllegalAccessException | InvocationTargetException e) {
throw new RuntimeException(e);
}
}
private static void migrateEntity(PermissionEntity entity, PermissionHolder holder, int weight) {
// migrate permanent permissions
for (Map.Entry<String, List<String>> worldData : getPermanentPermissions(entity).entrySet()) {
String world = standardizeWorld(worldData.getKey());
for (String node : worldData.getValue()) {
if (node.isEmpty()) continue;
holder.setPermission(MigrationUtils.parseNode(node, true).setWorld(world).build());
}
}
// migrate temporary permissions
Map<String, List<String>> timedPermissions;
Map<String, Long> timedPermissionsTime;
try {
//noinspection unchecked
timedPermissions = (Map<String, List<String>>) TIMED_PERMISSIONS_FIELD.get(entity);
//noinspection unchecked
timedPermissionsTime = (Map<String, Long>) TIMED_PERMISSIONS_TIME_FIELD.get(entity);
} catch (IllegalAccessException e) {
throw new RuntimeException(e);
}
for (Map.Entry<String, List<String>> worldData : timedPermissions.entrySet()) {
String world = standardizeWorld(worldData.getKey());
for (String node : worldData.getValue()) {
if (node.isEmpty()) continue;
long expiry = timedPermissionsTime.getOrDefault(Strings.nullToEmpty(world) + ":" + node, 0L);
holder.setPermission(MigrationUtils.parseNode(node, true).setWorld(world).setExpiry(expiry).build());
}
}
// migrate parents
Map<String, List<PermissionGroup>> parents = entity.getAllParents();
for (Map.Entry<String, List<PermissionGroup>> worldData : parents.entrySet()) {
String world = worldData.getKey();
if (world != null && (world.equals("") || world.equals("*"))) {
world = null;
}
if (world != null) {
world = world.toLowerCase();
}
String world = standardizeWorld(worldData.getKey());
for (PermissionGroup parent : worldData.getValue()) {
holder.setPermission(NodeFactory.buildGroupNode(MigrationUtils.standardizeName(parent.getName())).setWorld(world).build());
String parentName = parent.getName();
long expiry = 0L;
// check for temporary parent
if (entity instanceof PermissionUser) {
String expiryOption = entity.getOption("group-" + parentName + "-until", world);
if (expiryOption != null) {
try {
expiry = Long.parseLong(expiryOption);
} catch (NumberFormatException e) {
// ignore
}
}
}
holder.setPermission(NodeFactory.buildGroupNode(MigrationUtils.standardizeName(parentName)).setWorld(world).setExpiry(expiry).build());
}
}
@ -223,21 +280,23 @@ public class MigrationPermissionsEx extends SubCommand<Object> {
// migrate options
Map<String, Map<String, String>> options = entity.getAllOptions();
for (Map.Entry<String, Map<String, String>> worldData : options.entrySet()) {
String world = worldData.getKey();
if (world != null && (world.isEmpty() || world.equals("*"))) {
world = null;
}
if (world != null) {
world = world.toLowerCase();
}
String world = standardizeWorld(worldData.getKey());
for (Map.Entry<String, String> opt : worldData.getValue().entrySet()) {
if (opt.getKey() == null || opt.getKey().isEmpty() || opt.getValue() == null || opt.getValue().isEmpty()) {
continue;
}
String key = opt.getKey().toLowerCase();
if (key.equals(NodeFactory.PREFIX_KEY) || key.equals(NodeFactory.SUFFIX_KEY) || key.equals(NodeFactory.WEIGHT_KEY) || key.equals("rank") || key.equals("rank-ladder") || key.equals("name") || key.equals("username")) {
boolean ignore = key.equals(NodeFactory.PREFIX_KEY) ||
key.equals(NodeFactory.SUFFIX_KEY) ||
key.equals(NodeFactory.WEIGHT_KEY) ||
key.equals("rank") ||
key.equals("rank-ladder") ||
key.equals("name") ||
key.equals("username") ||
(key.startsWith("group-") && key.endsWith("-until"));
if (ignore) {
continue;
}
@ -245,4 +304,14 @@ public class MigrationPermissionsEx extends SubCommand<Object> {
}
}
}
private static String standardizeWorld(String world) {
if (world != null && (world.isEmpty() || world.equals("*"))) {
world = null;
}
if (world != null) {
world = world.toLowerCase();
}
return world;
}
}