refac: переосмысление

This commit is contained in:
2024-02-07 03:21:23 +03:00
commit 25302bd1e6
44 changed files with 2275 additions and 0 deletions

View File

@@ -0,0 +1,41 @@
package mc.nbt;
import java.util.Arrays;
import java.util.Objects;
public class ByteArrayTag extends TagValue {
private final byte[] value;
public ByteArrayTag(byte[] value) {
this.value = Objects.requireNonNull(value);
}
@Override
public TagType getType() {
return TagType.BYTE_ARRAY;
}
public byte[] getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
ByteArrayTag that = (ByteArrayTag) o;
return Arrays.equals(getValue(), that.getValue());
}
@Override
public int hashCode() {
return Arrays.hashCode(getValue());
}
@Override
public String toString() {
return "ByteArrayTag{" +
"value.length=" + value.length +
'}';
}
}

View File

@@ -0,0 +1,40 @@
package mc.nbt;
import java.util.Objects;
public class ByteTag extends TagValue {
private final byte value;
public ByteTag(byte value) {
this.value = value;
}
@Override
public TagType getType() {
return TagType.BYTE;
}
public byte getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
ByteTag byteTag = (ByteTag) o;
return getValue() == byteTag.getValue();
}
@Override
public int hashCode() {
return Objects.hash(getValue());
}
@Override
public String toString() {
return "ByteTag{" +
"value=" + value +
'}';
}
}

View File

@@ -0,0 +1,113 @@
package mc.nbt;
import java.io.Serializable;
import java.util.*;
public class CompoundTag extends TagValue implements Map<String, TagValue>, Serializable {
private final LinkedHashMap<String, TagValue> map;
public CompoundTag() {
this.map = new LinkedHashMap<>();
}
public CompoundTag(String key, TagValue value) {
this.map = new LinkedHashMap<>();
this.map.put(key, value);
}
public CompoundTag(Map<String, TagValue> map) {
this.map = new LinkedHashMap<>(map);
}
@Override
public TagType getType() {
return TagType.COMPOUND;
}
public CompoundTag append(String key, TagValue value) {
this.map.put(key, value);
return this;
}
// Map methods
@Override
public int size() {
return map.size();
}
@Override
public boolean isEmpty() {
return map.isEmpty();
}
@Override
public boolean containsKey(Object key) {
return map.containsKey(key);
}
@Override
public boolean containsValue(Object value) {
return map.containsValue(value);
}
@Override
public TagValue get(Object key) {
return map.get(key);
}
@Override
public TagValue put(String key, TagValue value) {
return map.put(key, value);
}
@Override
public TagValue remove(Object value) {
return map.remove(value);
}
@Override
public void putAll(Map<? extends String, ? extends TagValue> map) {
this.map.putAll(map);
}
@Override
public void clear() {
map.clear();
}
@Override
public Set<String> keySet() {
return map.keySet();
}
@Override
public Collection<TagValue> values() {
return map.values();
}
@Override
public Set<Entry<String, TagValue>> entrySet() {
return map.entrySet();
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
CompoundTag that = (CompoundTag) o;
return map.equals(that.map);
}
@Override
public int hashCode() {
return Objects.hash(map);
}
@Override
public String toString() {
return "CompoundTag{"
+ map
+ '}';
}
}

View File

@@ -0,0 +1,40 @@
package mc.nbt;
import java.util.Objects;
public class DoubleTag extends TagValue {
private final double value;
public DoubleTag(double value) {
this.value = value;
}
@Override
public TagType getType() {
return TagType.DOUBLE;
}
public double getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
DoubleTag doubleTag = (DoubleTag) o;
return Double.compare(doubleTag.getValue(), getValue()) == 0;
}
@Override
public int hashCode() {
return Objects.hash(getValue());
}
@Override
public String toString() {
return "DoubleTag{" +
"value=" + value +
'}';
}
}

View File

@@ -0,0 +1,27 @@
package mc.nbt;
public class EndTag extends TagValue {
private static final EndTag INSTANCE = new EndTag();
private EndTag() {
}
@Override
public TagType getType() {
return TagType.END;
}
@Override
public String getName() {
return null;
}
@Override
public void setName(String name) {
throw new UnsupportedOperationException();
}
public static EndTag getInstance() {
return INSTANCE;
}
}

View File

@@ -0,0 +1,40 @@
package mc.nbt;
import java.util.Objects;
public class FloatTag extends TagValue {
private final float value;
public FloatTag(float value) {
this.value = value;
}
@Override
public TagType getType() {
return TagType.FLOAT;
}
public float getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
FloatTag floatTag = (FloatTag) o;
return Float.compare(floatTag.getValue(), getValue()) == 0;
}
@Override
public int hashCode() {
return Objects.hash(getValue());
}
@Override
public String toString() {
return "FloatTag{" +
"value=" + value +
'}';
}
}

View File

@@ -0,0 +1,41 @@
package mc.nbt;
import java.util.Arrays;
import java.util.Objects;
public class IntegerArrayTag extends TagValue {
private final int[] value;
public IntegerArrayTag(int[] value) {
this.value = Objects.requireNonNull(value);
}
@Override
public TagType getType() {
return TagType.INTEGER_ARRAY;
}
public int[] getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
IntegerArrayTag that = (IntegerArrayTag) o;
return Arrays.equals(getValue(), that.getValue());
}
@Override
public int hashCode() {
return Arrays.hashCode(getValue());
}
@Override
public String toString() {
return "IntegerArrayTag{" +
"value.length=" + value.length +
'}';
}
}

View File

@@ -0,0 +1,40 @@
package mc.nbt;
import java.util.Objects;
public class IntegerTag extends TagValue {
private final int value;
public IntegerTag(int value) {
this.value = value;
}
@Override
public TagType getType() {
return TagType.INTEGER;
}
public int getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
IntegerTag that = (IntegerTag) o;
return getValue() == that.getValue();
}
@Override
public int hashCode() {
return Objects.hash(getValue());
}
@Override
public String toString() {
return "IntegerTag{" +
"value=" + value +
'}';
}
}

View File

@@ -0,0 +1,173 @@
package mc.nbt;
import java.util.*;
public class ListTag extends TagValue implements List<TagValue> {
private final TagType listType;
private final List<TagValue> list;
public ListTag(TagType listType, int initialCapacity) {
this.listType = Objects.requireNonNull(listType);
this.list = new ArrayList<>(initialCapacity);
}
public ListTag(TagType listType) {
this.listType = Objects.requireNonNull(listType);
this.list = new ArrayList<>();
}
@Override
public TagType getType() {
return TagType.LIST;
}
public TagType getListType() {
return listType;
}
public ListTag append(TagValue tag) {
list.add(tag);
return this;
}
// List methods
@Override
public int size() {
return list.size();
}
@Override
public boolean isEmpty() {
return list.isEmpty();
}
@Override
public boolean contains(Object o) {
return list.contains(o);
}
@Override
public Iterator<TagValue> iterator() {
return list.iterator();
}
@Override
public Object[] toArray() {
return list.toArray();
}
@SuppressWarnings("SuspiciousToArrayCall")
@Override
public <A> A[] toArray(A[] array) {
return list.toArray(array);
}
@Override
public boolean add(TagValue tag) {
return list.add(tag);
}
@Override
public boolean remove(Object o) {
return list.remove(o);
}
@SuppressWarnings("SlowListContainsAll")
@Override
public boolean containsAll(Collection<?> collection) {
return list.containsAll(collection);
}
@Override
public boolean addAll(Collection<? extends TagValue> collection) {
return list.addAll(collection);
}
@Override
public boolean addAll(int index, Collection<? extends TagValue> collection) {
return list.addAll(index, collection);
}
@Override
public boolean removeAll(Collection<?> collection) {
return list.removeAll(collection);
}
@Override
public boolean retainAll(Collection<?> collection) {
return list.retainAll(collection);
}
@Override
public void clear() {
list.clear();
}
@Override
public TagValue get(int index) {
return list.get(index);
}
@Override
public TagValue set(int index, TagValue element) {
return list.set(index, element);
}
@Override
public void add(int index, TagValue element) {
list.add(index, element);
}
@Override
public TagValue remove(int index) {
return list.remove(index);
}
@Override
public int indexOf(Object o) {
return list.indexOf(o);
}
@Override
public int lastIndexOf(Object o) {
return list.lastIndexOf(o);
}
@Override
public ListIterator<TagValue> listIterator() {
return list.listIterator();
}
@Override
public ListIterator<TagValue> listIterator(int index) {
return list.listIterator(index);
}
@Override
public List<TagValue> subList(int fromIndex, int toIndex) {
return list.subList(fromIndex, toIndex);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
ListTag listTag = (ListTag) o;
return list.equals(listTag.list) && getListType() == listTag.getListType();
}
@Override
public int hashCode() {
return Objects.hash(list, getListType());
}
@Override
public String toString() {
return "ListTag{" +
"listType=" + listType +
", list=" + list +
'}';
}
}

View File

@@ -0,0 +1,41 @@
package mc.nbt;
import java.util.Arrays;
import java.util.Objects;
public class LongArrayTag extends TagValue {
private final long[] value;
public LongArrayTag(long[] value) {
this.value = Objects.requireNonNull(value);
}
@Override
public TagType getType() {
return TagType.LONG_ARRAY;
}
public long[] getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
LongArrayTag that = (LongArrayTag) o;
return Arrays.equals(getValue(), that.getValue());
}
@Override
public int hashCode() {
return Arrays.hashCode(getValue());
}
@Override
public String toString() {
return "LongArrayTag{" +
"value.length=" + value.length +
'}';
}
}

View File

@@ -0,0 +1,40 @@
package mc.nbt;
import java.util.Objects;
public class LongTag extends TagValue {
private final long value;
public LongTag(long value) {
this.value = value;
}
@Override
public TagType getType() {
return TagType.LONG;
}
public long getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
LongTag longTag = (LongTag) o;
return getValue() == longTag.getValue();
}
@Override
public int hashCode() {
return Objects.hash(getValue());
}
@Override
public String toString() {
return "LongTag{" +
"value=" + value +
'}';
}
}

View File

@@ -0,0 +1,40 @@
package mc.nbt;
import java.util.Objects;
public class ShortTag extends TagValue {
private final short value;
public ShortTag(short value) {
this.value = value;
}
@Override
public TagType getType() {
return TagType.SHORT;
}
public short getValue() {
return value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
ShortTag shortTag = (ShortTag) o;
return getValue() == shortTag.getValue();
}
@Override
public int hashCode() {
return Objects.hash(getValue());
}
@Override
public String toString() {
return "ShortTag{" +
"value=" + value +
'}';
}
}

View File

@@ -0,0 +1,41 @@
package mc.nbt;
import java.util.Objects;
public class StringTag extends TagValue {
private final String value;
public StringTag(String value) {
this.value = Objects.requireNonNull(value);
}
@Override
public TagType getType() {
return TagType.STRING;
}
public String getValue() {
return this.value;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
StringTag stringTag = (StringTag) o;
return getValue().equals(stringTag.getValue());
}
@Override
public int hashCode() {
return Objects.hash(getValue());
}
@Override
public String toString() {
return "StringTag{" +
"value='" + value + '\'' +
'}';
}
}

View File

@@ -0,0 +1,40 @@
package mc.nbt;
public enum TagType {
BYTE(1),
SHORT(2),
INTEGER(3),
LONG(4),
FLOAT(5),
DOUBLE(6),
BYTE_ARRAY(7),
INTEGER_ARRAY(11),
LONG_ARRAY(12),
STRING(8),
LIST(9),
COMPOUND(10),
END(0);
private final int id;
TagType(int id) {
this.id = id;
}
public int getId() {
return id;
}
public static TagType valueOfId(int id) {
for (TagType type : TagType.values()) {
if (type.id == id) {
return type;
}
}
throw new RuntimeException("Unknown tag id: " + id); //TODO семантический exception
}
}

View File

@@ -0,0 +1,135 @@
package mc.nbt;
public abstract class TagValue {
protected String name;
TagValue() {
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public abstract TagType getType();
public ByteTag asByte() {
throwIfInvalidType(TagType.BYTE);
return (ByteTag) this;
}
public ByteArrayTag asByteArray() {
throwIfInvalidType(TagType.BYTE_ARRAY);
return (ByteArrayTag) this;
}
public ShortTag asShort() {
throwIfInvalidType(TagType.SHORT);
return (ShortTag) this;
}
public IntegerTag asInteger() {
throwIfInvalidType(TagType.INTEGER);
return (IntegerTag) this;
}
public IntegerArrayTag asIntegerArray() {
throwIfInvalidType(TagType.INTEGER_ARRAY);
return (IntegerArrayTag) this;
}
public LongTag asLong() {
throwIfInvalidType(TagType.LONG);
return (LongTag) this;
}
public LongArrayTag asLongArray() {
throwIfInvalidType(TagType.LONG_ARRAY);
return (LongArrayTag) this;
}
public FloatTag asFloat() {
throwIfInvalidType(TagType.FLOAT);
return (FloatTag) this;
}
public DoubleTag asDouble() {
throwIfInvalidType(TagType.DOUBLE);
return (DoubleTag) this;
}
public StringTag asString() {
throwIfInvalidType(TagType.STRING);
return (StringTag) this;
}
public ListTag asList() {
throwIfInvalidType(TagType.LIST);
return (ListTag) this;
}
public CompoundTag asCompound() {
throwIfInvalidType(TagType.COMPOUND);
return (CompoundTag) this;
}
public boolean isByte() {
return getType() == TagType.BYTE;
}
public boolean isByteArray() {
return getType() == TagType.BYTE_ARRAY;
}
public boolean isShort() {
return getType() == TagType.SHORT;
}
public boolean isInteger() {
return getType() == TagType.INTEGER;
}
public boolean isIntegerArray() {
return getType() == TagType.INTEGER_ARRAY;
}
public boolean isLong() {
return getType() == TagType.LONG;
}
public boolean isLongArray() {
return getType() == TagType.LONG_ARRAY;
}
public boolean isFloat() {
return getType() == TagType.FLOAT;
}
public boolean isDouble() {
return getType() == TagType.DOUBLE;
}
public boolean isString() {
return getType() == TagType.STRING;
}
public boolean isCompound() {
return getType() == TagType.COMPOUND;
}
public boolean isEnd() {
return getType() == TagType.END;
}
private void throwIfInvalidType(TagType expectedType) {
if (getType() != expectedType) {
//TODO семантический exception
throw new RuntimeException(
"Value expected to be type %s is of unexpected type %s"
.formatted(expectedType, getType()));
}
}
}

View File

@@ -0,0 +1,26 @@
package mc.nbt.codec;
import mc.nbt.ByteArrayTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class ByteArrayTagCodec implements Codec<ByteArrayTag> {
@Override
public void encode(ByteArrayTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeInt(tag.getValue().length);
outputStream.write(tag.getValue());
}
@SuppressWarnings("ResultOfMethodCallIgnored")
@Override
public ByteArrayTag decode(NbtInputStream inputStream) throws IOException {
int len = inputStream.readInt();
byte[] bytes = new byte[len];
inputStream.read(bytes);
return new ByteArrayTag(bytes);
}
}

View File

@@ -0,0 +1,20 @@
package mc.nbt.codec;
import mc.nbt.ByteTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class ByteTagCodec implements Codec<ByteTag> {
@Override
public void encode(ByteTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeByte(tag.getValue());
}
@Override
public ByteTag decode(NbtInputStream inputStream) throws IOException {
return new ByteTag(inputStream.readByte());
}
}

View File

@@ -0,0 +1,14 @@
package mc.nbt.codec;
import mc.nbt.TagValue;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public interface Codec<T extends TagValue> {
void encode(T tag, NbtOutputStream outputStream) throws IOException;
T decode(NbtInputStream inputStream) throws IOException;
}

View File

@@ -0,0 +1,39 @@
package mc.nbt.codec;
import mc.nbt.TagType;
import mc.nbt.TagValue;
import java.util.Map;
import java.util.Objects;
public class CodecRegistry {
private static final CodecRegistry INSTANCE = new CodecRegistry();
private final Map<TagType, Codec<? extends TagValue>> codecs;
private CodecRegistry() {
codecs = Map.ofEntries(
Map.entry(TagType.BYTE, new ByteTagCodec()),
Map.entry(TagType.SHORT, new ShortTagCodec()),
Map.entry(TagType.INTEGER, new IntegerTagCodec()),
Map.entry(TagType.LONG, new LongTagCodec()),
Map.entry(TagType.FLOAT, new FloatTagCodec()),
Map.entry(TagType.DOUBLE, new DoubleTagCodec()),
Map.entry(TagType.BYTE_ARRAY, new ByteArrayTagCodec()),
Map.entry(TagType.INTEGER_ARRAY, new IntegerArrayTagCodec()),
Map.entry(TagType.LONG_ARRAY, new LongArrayTagCodec()),
Map.entry(TagType.STRING, new StringTagCodec()),
Map.entry(TagType.LIST, new ListTagCodec()),
Map.entry(TagType.COMPOUND, new CompoundTagCodec()),
Map.entry(TagType.END, new EndTagCodec())
);
}
public Codec<? extends TagValue> get(TagType tagType) {
return codecs.get(Objects.requireNonNull(tagType));
}
public static CodecRegistry getInstance() {
return INSTANCE;
}
}

View File

@@ -0,0 +1,40 @@
package mc.nbt.codec;
import mc.nbt.CompoundTag;
import mc.nbt.EndTag;
import mc.nbt.TagValue;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
import java.util.Map;
public class CompoundTagCodec implements Codec<CompoundTag> {
@Override
public void encode(CompoundTag tag, NbtOutputStream outputStream) throws IOException {
for (Map.Entry<String, TagValue> entry : tag.entrySet()) {
if (entry.getValue() == null) //TODO по-хорошему, пользователю нужно запретить вставлять null-теги
continue;
outputStream.writeTag(entry.getKey(), entry.getValue());
}
outputStream.writeTag(EndTag.getInstance());
}
@Override
public CompoundTag decode(NbtInputStream inputStream) throws IOException {
var compoundTag = new CompoundTag();
do {
var tagValue = inputStream.readTag();
if (tagValue.isEnd())
break;
compoundTag.append(tagValue.getName(), tagValue);
} while (true);
return compoundTag;
}
}

View File

@@ -0,0 +1,20 @@
package mc.nbt.codec;
import mc.nbt.DoubleTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class DoubleTagCodec implements Codec<DoubleTag> {
@Override
public void encode(DoubleTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeDouble(tag.getValue());
}
@Override
public DoubleTag decode(NbtInputStream inputStream) throws IOException {
return new DoubleTag(inputStream.readDouble());
}
}

View File

@@ -0,0 +1,20 @@
package mc.nbt.codec;
import mc.nbt.EndTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class EndTagCodec implements Codec<EndTag> {
@Override
public void encode(EndTag tag, NbtOutputStream outputStream) throws IOException {
// nothing
}
@Override
public EndTag decode(NbtInputStream inputStream) throws IOException {
return EndTag.getInstance();
}
}

View File

@@ -0,0 +1,20 @@
package mc.nbt.codec;
import mc.nbt.FloatTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class FloatTagCodec implements Codec<FloatTag> {
@Override
public void encode(FloatTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeFloat(tag.getValue());
}
@Override
public FloatTag decode(NbtInputStream inputStream) throws IOException {
return new FloatTag(inputStream.readFloat());
}
}

View File

@@ -0,0 +1,30 @@
package mc.nbt.codec;
import mc.nbt.IntegerArrayTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class IntegerArrayTagCodec implements Codec<IntegerArrayTag> {
@Override
public void encode(IntegerArrayTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeInt(tag.getValue().length);
for (int item : tag.getValue()) {
outputStream.writeInt(item);
}
}
@Override
public IntegerArrayTag decode(NbtInputStream inputStream) throws IOException {
int len = inputStream.readInt();
int[] ints = new int[len];
for (int i = 0; i < len; i++) {
ints[i] = inputStream.readInt();
}
return new IntegerArrayTag(ints);
}
}

View File

@@ -0,0 +1,20 @@
package mc.nbt.codec;
import mc.nbt.IntegerTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class IntegerTagCodec implements Codec<IntegerTag> {
@Override
public void encode(IntegerTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeInt(tag.getValue());
}
@Override
public IntegerTag decode(NbtInputStream inputStream) throws IOException {
return new IntegerTag(inputStream.readInt());
}
}

View File

@@ -0,0 +1,45 @@
package mc.nbt.codec;
import mc.nbt.ListTag;
import mc.nbt.TagType;
import mc.nbt.TagValue;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class ListTagCodec implements Codec<ListTag> {
@SuppressWarnings({"rawtypes", "unchecked"})
@Override
public void encode(ListTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeByte(tag.getListType().getId());
outputStream.writeInt(tag.size());
Codec codec = CodecRegistry.getInstance().get(tag.getListType());
for (TagValue tagValue : tag) {
codec.encode(tagValue, outputStream);
}
}
@SuppressWarnings("rawtypes")
@Override
public ListTag decode(NbtInputStream inputStream) throws IOException {
int listTypeId = inputStream.readByte();
TagType listType = TagType.valueOfId(listTypeId);
int size = inputStream.readInt();
ListTag list = new ListTag(listType, size);
Codec codec = CodecRegistry.getInstance().get(listType);
if (size > 0) {
for (int i = 0; i < size; i++) {
var tagValue = codec.decode(inputStream);
list.add(tagValue);
}
}
return list;
}
}

View File

@@ -0,0 +1,30 @@
package mc.nbt.codec;
import mc.nbt.LongArrayTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class LongArrayTagCodec implements Codec<LongArrayTag> {
@Override
public void encode(LongArrayTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeInt(tag.getValue().length);
for (long value : tag.getValue()) {
outputStream.writeLong(value);
}
}
@Override
public LongArrayTag decode(NbtInputStream inputStream) throws IOException {
int len = inputStream.readInt();
long[] longs = new long[len];
for (int i = 0; i < len; i++) {
longs[i] = inputStream.readLong();
}
return new LongArrayTag(longs);
}
}

View File

@@ -0,0 +1,20 @@
package mc.nbt.codec;
import mc.nbt.LongTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class LongTagCodec implements Codec<LongTag> {
@Override
public void encode(LongTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeLong(tag.getValue());
}
@Override
public LongTag decode(NbtInputStream inputStream) throws IOException {
return new LongTag(inputStream.readLong());
}
}

View File

@@ -0,0 +1,20 @@
package mc.nbt.codec;
import mc.nbt.ShortTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class ShortTagCodec implements Codec<ShortTag> {
@Override
public void encode(ShortTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeShort(tag.getValue());
}
@Override
public ShortTag decode(NbtInputStream inputStream) throws IOException {
return new ShortTag(inputStream.readShort());
}
}

View File

@@ -0,0 +1,19 @@
package mc.nbt.codec;
import mc.nbt.StringTag;
import mc.nbt.io.NbtInputStream;
import mc.nbt.io.NbtOutputStream;
import java.io.IOException;
public class StringTagCodec implements Codec<StringTag> {
@Override
public void encode(StringTag tag, NbtOutputStream outputStream) throws IOException {
outputStream.writeString(tag.getValue());
}
@Override
public StringTag decode(NbtInputStream inputStream) throws IOException {
return new StringTag(inputStream.readString());
}
}

View File

@@ -0,0 +1,91 @@
package mc.nbt.io;
import mc.nbt.EndTag;
import mc.nbt.TagType;
import mc.nbt.TagValue;
import mc.nbt.codec.Codec;
import mc.nbt.codec.CodecRegistry;
import java.io.DataInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.util.zip.GZIPInputStream;
public class NbtInputStream extends InputStream {
private final DataInputStream dataInputStream;
public NbtInputStream(InputStream inputStream, boolean compressed) throws IOException {
this.dataInputStream = new DataInputStream(compressed
? new GZIPInputStream(inputStream)
: inputStream);
}
public NbtInputStream(InputStream inputStream) throws IOException {
this(inputStream, false);
}
@SuppressWarnings("rawtypes")
public TagValue readTag() throws IOException {
var tagType = TagType.valueOfId(read());
if (tagType == TagType.END) {
return EndTag.getInstance();
}
Codec codec = CodecRegistry.getInstance().get(tagType);
String name = readString();
var tagValue = codec.decode(this);
tagValue.setName(name);
return tagValue;
}
public byte readByte() throws IOException {
return dataInputStream.readByte();
}
public short readShort() throws IOException {
return dataInputStream.readShort();
}
public int readInt() throws IOException {
return dataInputStream.readInt();
}
public long readLong() throws IOException {
return dataInputStream.readLong();
}
public float readFloat() throws IOException {
return dataInputStream.readFloat();
}
public double readDouble() throws IOException {
return dataInputStream.readDouble();
}
public String readString() throws IOException {
int length = readShort();
if (length <= 0) {
return "";
}
byte[] buff = new byte[length];
if (read(buff, 0, length) < length) {
throw new IOException("Unexpected end of stream");
}
return new String(buff, StandardCharsets.UTF_8);
}
@Override
public int read() throws IOException {
return dataInputStream.read();
}
@Override
public void close() throws IOException {
dataInputStream.close();
}
}

View File

@@ -0,0 +1,84 @@
package mc.nbt.io;
import mc.nbt.TagValue;
import mc.nbt.codec.Codec;
import mc.nbt.codec.CodecRegistry;
import java.io.DataOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.util.zip.GZIPOutputStream;
public class NbtOutputStream extends OutputStream {
private final DataOutputStream dataOutputStream;
public NbtOutputStream(OutputStream outputStream, boolean compressed) throws IOException {
this.dataOutputStream = new DataOutputStream(compressed
? new GZIPOutputStream(outputStream)
: outputStream);
}
public NbtOutputStream(OutputStream outputStream) throws IOException {
this(outputStream, false);
}
@SuppressWarnings({"unchecked", "rawtypes"})
public void writeTag(String name, TagValue tagValue) throws IOException {
Codec codec = CodecRegistry.getInstance().get(tagValue.getType());
writeByte(tagValue.getType().getId());
writeString(name);
codec.encode(tagValue, this);
}
public void writeTag(TagValue tagValue) throws IOException {
writeTag(tagValue.getName(), tagValue);
}
public void writeByte(int value) throws IOException {
dataOutputStream.writeByte(value);
}
public void writeShort(int value) throws IOException {
dataOutputStream.writeShort(value);
}
public void writeInt(int value) throws IOException {
dataOutputStream.writeInt(value);
}
public void writeLong(long value) throws IOException {
dataOutputStream.writeLong(value);
}
public void writeFloat(float value) throws IOException {
dataOutputStream.writeFloat(value);
}
public void writeDouble(double value) throws IOException {
dataOutputStream.writeDouble(value);
}
public void writeString(String value) throws IOException {
if (value == null || value.isEmpty()) {
writeShort(0);
} else {
writeShort(value.length());
write(value.getBytes(StandardCharsets.UTF_8));
}
}
@Override
public void write(int b) throws IOException {
dataOutputStream.write(b);
}
@Override
public void close() throws IOException {
this.dataOutputStream.close();
}
}

View File

@@ -0,0 +1,38 @@
package mc.nbt.io;
import mc.nbt.CompoundTag;
import mc.nbt.StringTag;
import mc.nbt.TagType;
import mc.nbt.TagValue;
import org.junit.jupiter.api.Test;
import java.io.IOException;
import java.util.Objects;
import static org.junit.jupiter.api.Assertions.*;
class NbtInputStreamTest {
@Test
void testSimple() throws IOException {
try (var stream = Objects.requireNonNull(NbtInputStreamTest.class.getResourceAsStream("/hello_world.nbt"));
NbtInputStream nbtInputStream = new NbtInputStream(stream)) {
TagValue tag = nbtInputStream.readTag();
assertNotNull(tag);
assertEquals(TagType.COMPOUND, tag.getType());
CompoundTag compoundTag = tag.asCompound();
assertEquals("hello world", compoundTag.getName());
assertEquals(1, compoundTag.size());
assertTrue(compoundTag.containsKey("name"));
tag = compoundTag.get("name");
assertEquals(TagType.STRING, tag.getType());
StringTag stringTag = tag.asString();
assertEquals("name", stringTag.getName());
assertEquals("Bananrama", stringTag.getValue());
}
}
}

View File

@@ -0,0 +1,381 @@
package mc.nbt.io;
import mc.nbt.*;
import org.apache.commons.lang3.RandomStringUtils;
import org.apache.commons.lang3.RandomUtils;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.stream.Stream;
import static org.junit.jupiter.api.Assertions.*;
class NbtOutputStreamTest {
ByteArrayOutputStream baos;
NbtOutputStream nbtOutputStream;
@BeforeEach
void before() throws IOException {
baos = new ByteArrayOutputStream();
nbtOutputStream = new NbtOutputStream(baos);
}
@Test
void writeEndTag() throws IOException {
nbtOutputStream.writeTag(EndTag.getInstance());
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.END, tag.getType());
assertNull(tag.getName());
}
}
@ParameterizedTest
@MethodSource("streamOfTagByte")
void writeByteTag(ByteTag byteTag) throws IOException {
nbtOutputStream.writeTag(byteTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.BYTE, tag.getType());
assertEquals(byteTag.getValue(), tag.asByte().getValue());
assertName(byteTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagShort")
void writeShortTag(ShortTag shortTag) throws IOException {
nbtOutputStream.writeTag(shortTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.SHORT, tag.getType());
assertEquals(shortTag.getValue(), tag.asShort().getValue());
assertName(shortTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagInt")
void writeIntegerTag(IntegerTag integerTag) throws IOException {
nbtOutputStream.writeTag(integerTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.INTEGER, tag.getType());
assertEquals(integerTag.getValue(), tag.asInteger().getValue());
assertName(integerTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagLong")
void writeLongTag(LongTag longTag) throws IOException {
nbtOutputStream.writeTag(longTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.LONG, tag.getType());
assertEquals(longTag.getValue(), tag.asLong().getValue());
assertName(longTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagFloat")
void writeFloatTag(FloatTag floatTag) throws IOException {
nbtOutputStream.writeTag(floatTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.FLOAT, tag.getType());
assertEquals(floatTag.getValue(), tag.asFloat().getValue());
assertName(floatTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagDouble")
void writeDoubleTag(DoubleTag doubleTag) throws IOException {
nbtOutputStream.writeTag(doubleTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.DOUBLE, tag.getType());
assertEquals(doubleTag.getValue(), tag.asDouble().getValue());
assertName(doubleTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagByteArray")
void writeByteArrayTag(ByteArrayTag byteArrayTag) throws IOException {
nbtOutputStream.writeTag(byteArrayTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.BYTE_ARRAY, tag.getType());
assertArrayEquals(byteArrayTag.getValue(), tag.asByteArray().getValue());
assertName(byteArrayTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagString")
void writeStringTag(StringTag stringTag) throws IOException {
nbtOutputStream.writeTag(stringTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.STRING, tag.getType());
assertEquals(stringTag.getValue(), tag.asString().getValue());
assertName(stringTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagList")
void writeListTag(ListTag listTag) throws IOException {
nbtOutputStream.writeTag(listTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.LIST, tag.getType());
assertEquals(listTag.size(), tag.asList().size());
assertEquals(listTag.getListType(), tag.asList().getListType());
assertName(listTag, tag);
for (int i = 0; i < listTag.size(); i++) {
LongTag longTag1 = listTag.get(i).asLong();
LongTag longTag2 = tag.asList().get(i).asLong();
assertNull(longTag2.getName());
assertEquals(longTag1.getValue(), longTag2.getValue());
}
}
}
@ParameterizedTest
@MethodSource("streamOfTagCompound")
void writeCompoundTag(CompoundTag compoundTag) throws IOException {
nbtOutputStream.writeTag(compoundTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.COMPOUND, tag.getType());
assertEquals(compoundTag.size(), tag.asCompound().size());
assertName(compoundTag, tag);
for (int i = 0; i < compoundTag.size(); i++) {
compoundTag.forEach((key, tagValue) -> {
assertTrue(tag.asCompound().containsKey(key));
LongTag longTag1 = tagValue.asLong();
LongTag longTag2 = tag.asCompound().get(key).asLong();
assertNotNull(longTag2.getName());
assertEquals(longTag1.getValue(), longTag2.getValue());
});
}
}
}
@ParameterizedTest
@MethodSource("streamOfTagIntArray")
void writeIntegerArrayTag(IntegerArrayTag integerArrayTag) throws IOException {
nbtOutputStream.writeTag(integerArrayTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.INTEGER_ARRAY, tag.getType());
assertArrayEquals(integerArrayTag.getValue(), tag.asIntegerArray().getValue());
assertName(integerArrayTag, tag);
}
}
@ParameterizedTest
@MethodSource("streamOfTagLongArray")
void writeLongArrayTag(LongArrayTag longArrayTag) throws IOException {
nbtOutputStream.writeTag(longArrayTag);
try (var nbtInputStream = createNbtInputStream()) {
TagValue tag = nbtInputStream.readTag();
assertEquals(TagType.LONG_ARRAY, tag.getType());
assertArrayEquals(longArrayTag.getValue(), tag.asLongArray().getValue());
assertName(longArrayTag, tag);
}
}
// region Methods for ParameterizedTest
// -------------------------------------------------------------------------------------------------------------
private static Stream<Arguments> streamOfTagByte() {
final byte value = RandomUtils.nextBytes(1)[0];
return Stream.of(
Arguments.of(new ByteTag(value)),
Arguments.of(new ByteTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagShort() {
final short value = (short) RandomUtils.nextInt(0, Short.MAX_VALUE);
return Stream.of(
Arguments.of(new ShortTag(value)),
Arguments.of(new ShortTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagInt() {
final int value = RandomUtils.nextInt();
return Stream.of(
Arguments.of(new IntegerTag(value)),
Arguments.of(new IntegerTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagLong() {
final long value = RandomUtils.nextLong();
return Stream.of(
Arguments.of(new LongTag(value)),
Arguments.of(new LongTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagFloat() {
final float value = RandomUtils.nextFloat();
return Stream.of(
Arguments.of(new FloatTag(value)),
Arguments.of(new FloatTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagDouble() {
final double value = RandomUtils.nextDouble();
return Stream.of(
Arguments.of(new DoubleTag(value)),
Arguments.of(new DoubleTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagByteArray() {
final byte[] value = RandomUtils.nextBytes(RandomUtils.nextInt(1, 1000));
return Stream.of(
Arguments.of(new ByteArrayTag(value)),
Arguments.of(new ByteArrayTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagString() {
final String value = randomString();
return Stream.of(
Arguments.of(new StringTag(value)),
Arguments.of(new StringTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagList() {
return Stream.of(
Arguments.of(new ListTag(TagType.LONG).append(new LongTag(1L))),
Arguments.of(new ListTag(TagType.LONG) {{
this.setName(randomString());
}}.append(new LongTag(1L)))
);
}
private static Stream<Arguments> streamOfTagCompound() {
return Stream.of(
Arguments.of(new CompoundTag()
.append("TestName1", new LongTag(RandomUtils.nextLong()))
.append("TestName2", new LongTag(RandomUtils.nextLong()))
.append("TestName3", new LongTag(RandomUtils.nextLong()))),
Arguments.of(new CompoundTag() {{
this.setName(randomString());
}}
.append("TestName1", new LongTag(RandomUtils.nextLong()))
.append("TestName2", new LongTag(RandomUtils.nextLong()))
.append("TestName3", new LongTag(RandomUtils.nextLong())))
);
}
private static Stream<Arguments> streamOfTagIntArray() {
final int[] value = new int[RandomUtils.nextInt(1, 1000)];
for (int i = 0; i < value.length; i++) {
value[i] = RandomUtils.nextInt();
}
return Stream.of(
Arguments.of(new IntegerArrayTag(value)),
Arguments.of(new IntegerArrayTag(value) {{
this.setName(randomString());
}})
);
}
private static Stream<Arguments> streamOfTagLongArray() {
final long[] value = new long[RandomUtils.nextInt(1, 1000)];
for (int i = 0; i < value.length; i++) {
value[i] = RandomUtils.nextLong();
}
return Stream.of(
Arguments.of(new LongArrayTag(value)),
Arguments.of(new LongArrayTag(value) {{
this.setName(randomString());
}})
);
}
// -------------------------------------------------------------------------------------------------------------
// endregion
private NbtInputStream createNbtInputStream() throws IOException {
return new NbtInputStream(new ByteArrayInputStream(baos.toByteArray()));
}
private static String randomString() {
return RandomStringUtils.randomAscii(RandomUtils.nextInt(1, Short.MAX_VALUE));
}
private void assertName(TagValue expectedTag, TagValue actualTag) {
if (expectedTag.getName() == null) {
assertTrue(actualTag.getName().isEmpty());
} else {
assertEquals(expectedTag.getName(), actualTag.getName());
}
}
}

Binary file not shown.