-
-
Notifications
You must be signed in to change notification settings - Fork 102
/
EntityMetadataCommandHelper.java
126 lines (117 loc) · 5.34 KB
/
EntityMetadataCommandHelper.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package com.denizenscript.denizen.utilities.entity;
import com.denizenscript.denizen.nms.NMSHandler;
import com.denizenscript.denizen.objects.EntityTag;
import com.denizenscript.denizen.objects.PlayerTag;
import com.denizenscript.denizen.utilities.Utilities;
import com.denizenscript.denizen.utilities.packets.NetworkInterceptHelper;
import com.denizenscript.denizencore.exceptions.InvalidArgumentsRuntimeException;
import com.denizenscript.denizencore.scripts.ScriptEntry;
import org.bukkit.entity.Entity;
import org.bukkit.entity.Player;
import java.util.*;
import java.util.function.BiConsumer;
import java.util.function.Predicate;
public record EntityMetadataCommandHelper(Predicate<Entity> getter, BiConsumer<EntityTag, Boolean> setter, Map<UUID, Map<UUID, Boolean>> packetOverrides) {
public EntityMetadataCommandHelper(Predicate<Entity> getter, BiConsumer<EntityTag, Boolean> setter) {
this(getter, setter, new HashMap<>());
}
public void setForPlayers(List<PlayerTag> players, EntityTag target, Predicate<PlayerTag> stateSupplier) {
if (target == null || target.getUUID() == null || players == null) {
return;
}
NetworkInterceptHelper.enable();
boolean wasEntityAdded = !packetOverrides.containsKey(target.getUUID());
Map<UUID, Boolean> playerMap = packetOverrides.computeIfAbsent(target.getUUID(), k -> new HashMap<>());
for (PlayerTag player : players) {
boolean state = stateSupplier.test(player);
Boolean oldState = playerMap.put(player.getUUID(), state);
if ((wasEntityAdded || oldState == null || oldState != state) && player.isOnline()) {
NMSHandler.packetHelper.sendEntityMetadataFlagsUpdate(player.getPlayerEntity(), target.getBukkitEntity());
}
}
}
public Boolean getState(Entity entity, UUID player, boolean fakeOnly) {
if (entity == null) {
return null;
}
if (player != null) {
Map<UUID, Boolean> playerMap = packetOverrides.get(entity.getUniqueId());
if (playerMap != null && playerMap.containsKey(player)) {
return playerMap.get(player);
}
}
if (fakeOnly) {
return null;
}
return getter.test(entity);
}
public boolean noOverrides() {
return packetOverrides.isEmpty();
}
public enum Action {TRUE, FALSE, TOGGLE, RESET}
public void execute(ScriptEntry scriptEntry, List<EntityTag> targets, Action action, List<PlayerTag> forPlayers) {
if (targets == null) {
targets = Utilities.entryDefaultEntityList(scriptEntry, true);
if (targets == null) {
throw new InvalidArgumentsRuntimeException("Must specify valid targets.");
}
}
switch (action) {
case TRUE, FALSE -> {
boolean state = action == Action.TRUE;
if (forPlayers == null) {
for (EntityTag target : targets) {
setter.accept(target, state);
}
}
else {
for (EntityTag target : targets) {
setForPlayers(forPlayers, target, player -> state);
}
}
}
case TOGGLE -> {
if (forPlayers == null) {
for (EntityTag target : targets) {
setter.accept(target, !getState(target.getBukkitEntity(), null, false));
}
}
else {
for (EntityTag target : targets) {
setForPlayers(forPlayers, target, player -> !getState(target.getBukkitEntity(), player.getUUID(), false));
}
}
}
case RESET -> {
for (EntityTag target : targets) {
Map<UUID, Boolean> playerMap = packetOverrides.get(target.getUUID());
if (playerMap == null) {
return;
}
Set<UUID> playersToUpdate = new HashSet<>();
if (forPlayers == null) {
playersToUpdate.addAll(playerMap.keySet());
packetOverrides.remove(target.getUUID());
}
else {
for (PlayerTag player : forPlayers) {
playerMap.remove(player.getUUID());
playersToUpdate.add(player.getUUID());
}
if (playerMap.isEmpty()) {
packetOverrides.remove(target.getUUID());
}
}
if (playersToUpdate.isEmpty()) {
return;
}
for (Player player : NMSHandler.entityHelper.getPlayersThatSee(target.getBukkitEntity())) {
if (playersToUpdate.contains(player.getUniqueId())) {
NMSHandler.packetHelper.sendEntityMetadataFlagsUpdate(player, target.getBukkitEntity());
}
}
}
}
}
}
}