All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
net.minestom.server.component.DataComponentMapImpl Maven / Gradle / Ivy
package net.minestom.server.component;
import it.unimi.dsi.fastutil.ints.Int2ObjectArrayMap;
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import net.kyori.adventure.nbt.BinaryTag;
import net.kyori.adventure.nbt.CompoundBinaryTag;
import net.minestom.server.network.NetworkBuffer;
import net.minestom.server.utils.nbt.BinaryTagSerializer;
import net.minestom.server.utils.validate.Check;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.util.Map;
import java.util.function.Function;
import java.util.function.IntFunction;
/**
* A component list, always stored as a patch of added and removed components (even if none are removed).
*
* The inner map contains the value for added components, null for removed components, and no entry for unmodified components.
*
* @param components The component patch.
*/
record DataComponentMapImpl(@NotNull Int2ObjectMap components) implements DataComponentMap {
private static final char REMOVAL_PREFIX = '!';
@Override
public boolean has(@NotNull DataComponent component) {
return components.containsKey(component.id()) && components.get(component.id()) != null;
}
@Override
public @Nullable T get(@NotNull DataComponent component) {
//noinspection unchecked
return (T) components.get(component.id());
}
@Override
public boolean has(@NotNull DataComponentMap prototype, @NotNull DataComponent component) {
if (components.containsKey(component.id())) {
return components.get(component.id()) != null;
} else {
return prototype.has(component);
}
}
@Override
public @Nullable T get(@NotNull DataComponentMap prototype, @NotNull DataComponent component) {
if (components.containsKey(component.id())) {
//noinspection unchecked
return (T) components.get(component.id());
} else {
return prototype.get(component);
}
}
@Override
public @NotNull DataComponentMap set(@NotNull DataComponent component, @NotNull T value) {
Int2ObjectMap newComponents = new Int2ObjectArrayMap<>(components);
newComponents.put(component.id(), value);
return new DataComponentMapImpl(newComponents);
}
@Override
public @NotNull DataComponentMap remove(@NotNull DataComponent component) {
Int2ObjectMap newComponents = new Int2ObjectArrayMap<>(components);
newComponents.put(component.id(), null);
return new DataComponentMapImpl(newComponents);
}
@Override
public @NotNull Builder toBuilder() {
return new BuilderImpl(new Int2ObjectArrayMap<>(components));
}
@Override
public @NotNull PatchBuilder toPatchBuilder() {
return new PatchBuilderImpl(new Int2ObjectArrayMap<>(components));
}
record BuilderImpl(@NotNull Int2ObjectMap components) implements DataComponentMap.Builder {
@Override
public boolean has(@NotNull DataComponent component) {
return components.get(component.id()) != null;
}
@Override
public @Nullable T get(@NotNull DataComponent component) {
//noinspection unchecked
return (T) components.get(component.id());
}
@Override
public @NotNull Builder set(@NotNull DataComponent component, @NotNull T value) {
components.put(component.id(), value);
return this;
}
@Override
public @NotNull DataComponentMap build() {
return new DataComponentMapImpl(new Int2ObjectArrayMap<>(components));
}
}
record PatchBuilderImpl(@NotNull Int2ObjectMap components) implements DataComponentMap.PatchBuilder {
@Override
public boolean has(@NotNull DataComponent component) {
return components.get(component.id()) != null;
}
@Override
public @Nullable T get(@NotNull DataComponent component) {
//noinspection unchecked
return (T) components.get(component.id());
}
@Override
public @NotNull PatchBuilder set(@NotNull DataComponent component, @NotNull T value) {
components.put(component.id(), value);
return this;
}
@Override
public @NotNull PatchBuilder remove(@NotNull DataComponent component) {
components.put(component.id(), null);
return this;
}
@Override
public @NotNull DataComponentMap build() {
return new DataComponentMapImpl(new Int2ObjectArrayMap<>(components));
}
}
record PatchNetworkType(@NotNull IntFunction> idToType) implements NetworkBuffer.Type {
@Override
public void write(@NotNull NetworkBuffer buffer, DataComponentMap value) {
final DataComponentMapImpl patch = (DataComponentMapImpl) value;
int added = 0;
for (Object o : patch.components.values()) {
if (o != null) added++;
}
buffer.write(NetworkBuffer.VAR_INT, added);
buffer.write(NetworkBuffer.VAR_INT, patch.components.size() - added);
for (Int2ObjectMap.Entry entry : patch.components.int2ObjectEntrySet()) {
if (entry.getValue() != null) {
buffer.write(NetworkBuffer.VAR_INT, entry.getIntKey());
//noinspection unchecked
DataComponent type = (DataComponent) this.idToType.apply(entry.getIntKey());
assert type != null;
type.write(buffer, entry.getValue());
}
}
for (Int2ObjectMap.Entry entry : patch.components.int2ObjectEntrySet()) {
if (entry.getValue() == null) {
buffer.write(NetworkBuffer.VAR_INT, entry.getIntKey());
}
}
}
@Override
public DataComponentMap read(@NotNull NetworkBuffer buffer) {
int added = buffer.read(NetworkBuffer.VAR_INT);
int removed = buffer.read(NetworkBuffer.VAR_INT);
Check.stateCondition(added + removed > 256, "Item component patch too large: {0}", added + removed);
Int2ObjectMap patch = new Int2ObjectArrayMap<>(added + removed);
for (int i = 0; i < added; i++) {
int id = buffer.read(NetworkBuffer.VAR_INT);
//noinspection unchecked
DataComponent type = (DataComponent) this.idToType.apply(id);
Check.notNull(type, "Unknown component: {0}", id);
patch.put(type.id(), type.read(buffer));
}
for (int i = 0; i < removed; i++) {
int id = buffer.read(NetworkBuffer.VAR_INT);
patch.put(id, null);
}
return new DataComponentMapImpl(patch);
}
}
record NbtType(
@NotNull IntFunction> idToType,
@NotNull Function> nameToType,
boolean isPatch
) implements BinaryTagSerializer {
@Override
public @NotNull BinaryTag write(@NotNull Context context, @NotNull DataComponentMap value) {
final DataComponentMapImpl patch = (DataComponentMapImpl) value;
if (patch.components.isEmpty()) return CompoundBinaryTag.empty();
CompoundBinaryTag.Builder builder = CompoundBinaryTag.builder();
for (Int2ObjectMap.Entry entry : patch.components.int2ObjectEntrySet()) {
//noinspection unchecked
DataComponent type = (DataComponent) this.idToType.apply(entry.getIntKey());
Check.notNull(type, "Unknown item component id: {0}", entry.getIntKey());
if (entry.getValue() == null) {
if (isPatch) builder.put(REMOVAL_PREFIX + type.name(), CompoundBinaryTag.empty());
// Removing a component in an absolute (non-patch) builder is a noop because it is not yet present.
} else {
builder.put(type.name(), type.write(context, entry.getValue()));
}
}
return builder.build();
}
@Override
public @NotNull DataComponentMap read(@NotNull Context context, @NotNull BinaryTag anyTag) {
if (!(anyTag instanceof CompoundBinaryTag tag))
throw new IllegalArgumentException("Component patch must be a compound tag, was: " + anyTag.type());
if (tag.size() == 0) return EMPTY;
Int2ObjectMap patch = new Int2ObjectArrayMap<>(tag.size());
for (Map.Entry entry : tag) {
String key = entry.getKey();
boolean remove = false;
if (!key.isEmpty() && key.charAt(0) == REMOVAL_PREFIX) {
key = key.substring(1);
remove = true;
}
DataComponent type = this.nameToType.apply(key);
Check.notNull(type, "Unknown item component: {0}", key);
if (remove) {
if (isPatch) patch.put(type.id(), null);
// Removing a component in an absolute (non-patch) builder is a noop because it is not yet present.
} else {
Object value = type.read(context, entry.getValue());
patch.put(type.id(), value);
}
}
return new DataComponentMapImpl(patch);
}
}
}
