// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.protocol;

import java.util.Objects;
import com.hypixel.hytale.protocol.io.ValidationResult;
import java.util.Iterator;
import java.util.HashMap;
import com.hypixel.hytale.protocol.io.ProtocolException;
import com.hypixel.hytale.protocol.io.VarInt;
import io.netty.buffer.ByteBuf;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Map;

public class InteractionConfiguration
{
    public static final int NULLABLE_BIT_FIELD_SIZE = 1;
    public static final int FIXED_BLOCK_SIZE = 4;
    public static final int VARIABLE_FIELD_COUNT = 2;
    public static final int VARIABLE_BLOCK_START = 12;
    public static final int MAX_SIZE = 1677721600;
    public boolean displayOutlines;
    public boolean debugOutlines;
    @Nullable
    public Map<GameMode, Float> useDistance;
    public boolean allEntities;
    @Nullable
    public Map<InteractionType, InteractionPriority> priorities;
    
    public InteractionConfiguration() {
        this.displayOutlines = true;
    }
    
    public InteractionConfiguration(final boolean displayOutlines, final boolean debugOutlines, @Nullable final Map<GameMode, Float> useDistance, final boolean allEntities, @Nullable final Map<InteractionType, InteractionPriority> priorities) {
        this.displayOutlines = true;
        this.displayOutlines = displayOutlines;
        this.debugOutlines = debugOutlines;
        this.useDistance = useDistance;
        this.allEntities = allEntities;
        this.priorities = priorities;
    }
    
    public InteractionConfiguration(@Nonnull final InteractionConfiguration other) {
        this.displayOutlines = true;
        this.displayOutlines = other.displayOutlines;
        this.debugOutlines = other.debugOutlines;
        this.useDistance = other.useDistance;
        this.allEntities = other.allEntities;
        this.priorities = other.priorities;
    }
    
    @Nonnull
    public static InteractionConfiguration deserialize(@Nonnull final ByteBuf buf, final int offset) {
        final InteractionConfiguration obj = new InteractionConfiguration();
        final byte nullBits = buf.getByte(offset);
        obj.displayOutlines = (buf.getByte(offset + 1) != 0);
        obj.debugOutlines = (buf.getByte(offset + 2) != 0);
        obj.allEntities = (buf.getByte(offset + 3) != 0);
        if ((nullBits & 0x1) != 0x0) {
            final int varPos0 = offset + 12 + buf.getIntLE(offset + 4);
            final int useDistanceCount = VarInt.peek(buf, varPos0);
            if (useDistanceCount < 0) {
                throw ProtocolException.negativeLength("UseDistance", useDistanceCount);
            }
            if (useDistanceCount > 4096000) {
                throw ProtocolException.dictionaryTooLarge("UseDistance", useDistanceCount, 4096000);
            }
            final int varIntLen = VarInt.length(buf, varPos0);
            obj.useDistance = new HashMap<GameMode, Float>(useDistanceCount);
            int dictPos = varPos0 + varIntLen;
            for (int i = 0; i < useDistanceCount; ++i) {
                final GameMode key = GameMode.fromValue(buf.getByte(dictPos));
                ++dictPos;
                final float val = buf.getFloatLE(dictPos);
                dictPos += 4;
                if (obj.useDistance.put(key, val) != null) {
                    throw ProtocolException.duplicateKey("useDistance", key);
                }
            }
        }
        if ((nullBits & 0x2) != 0x0) {
            final int varPos2 = offset + 12 + buf.getIntLE(offset + 8);
            final int prioritiesCount = VarInt.peek(buf, varPos2);
            if (prioritiesCount < 0) {
                throw ProtocolException.negativeLength("Priorities", prioritiesCount);
            }
            if (prioritiesCount > 4096000) {
                throw ProtocolException.dictionaryTooLarge("Priorities", prioritiesCount, 4096000);
            }
            final int varIntLen = VarInt.length(buf, varPos2);
            obj.priorities = new HashMap<InteractionType, InteractionPriority>(prioritiesCount);
            int dictPos = varPos2 + varIntLen;
            for (int i = 0; i < prioritiesCount; ++i) {
                final InteractionType key2 = InteractionType.fromValue(buf.getByte(dictPos));
                ++dictPos;
                final InteractionPriority val2 = InteractionPriority.deserialize(buf, dictPos);
                dictPos += InteractionPriority.computeBytesConsumed(buf, dictPos);
                if (obj.priorities.put(key2, val2) != null) {
                    throw ProtocolException.duplicateKey("priorities", key2);
                }
            }
        }
        return obj;
    }
    
    public static int computeBytesConsumed(@Nonnull final ByteBuf buf, final int offset) {
        final byte nullBits = buf.getByte(offset);
        int maxEnd = 12;
        if ((nullBits & 0x1) != 0x0) {
            final int fieldOffset0 = buf.getIntLE(offset + 4);
            int pos0 = offset + 12 + fieldOffset0;
            final int dictLen = VarInt.peek(buf, pos0);
            pos0 += VarInt.length(buf, pos0);
            for (int i = 0; i < dictLen; ++i) {
                ++pos0;
                pos0 += 4;
            }
            if (pos0 - offset > maxEnd) {
                maxEnd = pos0 - offset;
            }
        }
        if ((nullBits & 0x2) != 0x0) {
            final int fieldOffset2 = buf.getIntLE(offset + 8);
            int pos2 = offset + 12 + fieldOffset2;
            final int dictLen = VarInt.peek(buf, pos2);
            pos2 += VarInt.length(buf, pos2);
            for (int i = 0; i < dictLen; ++i) {
                pos2 = ++pos2 + InteractionPriority.computeBytesConsumed(buf, pos2);
            }
            if (pos2 - offset > maxEnd) {
                maxEnd = pos2 - offset;
            }
        }
        return maxEnd;
    }
    
    public void serialize(@Nonnull final ByteBuf buf) {
        final int startPos = buf.writerIndex();
        byte nullBits = 0;
        if (this.useDistance != null) {
            nullBits |= 0x1;
        }
        if (this.priorities != null) {
            nullBits |= 0x2;
        }
        buf.writeByte(nullBits);
        buf.writeByte(this.displayOutlines ? 1 : 0);
        buf.writeByte(this.debugOutlines ? 1 : 0);
        buf.writeByte(this.allEntities ? 1 : 0);
        final int useDistanceOffsetSlot = buf.writerIndex();
        buf.writeIntLE(0);
        final int prioritiesOffsetSlot = buf.writerIndex();
        buf.writeIntLE(0);
        final int varBlockStart = buf.writerIndex();
        if (this.useDistance != null) {
            buf.setIntLE(useDistanceOffsetSlot, buf.writerIndex() - varBlockStart);
            if (this.useDistance.size() > 4096000) {
                throw ProtocolException.dictionaryTooLarge("UseDistance", this.useDistance.size(), 4096000);
            }
            VarInt.write(buf, this.useDistance.size());
            for (final Map.Entry<GameMode, Float> e : this.useDistance.entrySet()) {
                buf.writeByte(e.getKey().getValue());
                buf.writeFloatLE(e.getValue());
            }
        }
        else {
            buf.setIntLE(useDistanceOffsetSlot, -1);
        }
        if (this.priorities != null) {
            buf.setIntLE(prioritiesOffsetSlot, buf.writerIndex() - varBlockStart);
            if (this.priorities.size() > 4096000) {
                throw ProtocolException.dictionaryTooLarge("Priorities", this.priorities.size(), 4096000);
            }
            VarInt.write(buf, this.priorities.size());
            for (final Map.Entry<InteractionType, InteractionPriority> e2 : this.priorities.entrySet()) {
                buf.writeByte(e2.getKey().getValue());
                e2.getValue().serialize(buf);
            }
        }
        else {
            buf.setIntLE(prioritiesOffsetSlot, -1);
        }
    }
    
    public int computeSize() {
        int size = 12;
        if (this.useDistance != null) {
            size += VarInt.size(this.useDistance.size()) + this.useDistance.size() * 5;
        }
        if (this.priorities != null) {
            int prioritiesSize = 0;
            for (final Map.Entry<InteractionType, InteractionPriority> kvp : this.priorities.entrySet()) {
                prioritiesSize += 1 + kvp.getValue().computeSize();
            }
            size += VarInt.size(this.priorities.size()) + prioritiesSize;
        }
        return size;
    }
    
    public static ValidationResult validateStructure(@Nonnull final ByteBuf buffer, final int offset) {
        if (buffer.readableBytes() - offset < 12) {
            return ValidationResult.error("Buffer too small: expected at least 12 bytes");
        }
        final byte nullBits = buffer.getByte(offset);
        if ((nullBits & 0x1) != 0x0) {
            final int useDistanceOffset = buffer.getIntLE(offset + 4);
            if (useDistanceOffset < 0) {
                return ValidationResult.error("Invalid offset for UseDistance");
            }
            int pos = offset + 12 + useDistanceOffset;
            if (pos >= buffer.writerIndex()) {
                return ValidationResult.error("Offset out of bounds for UseDistance");
            }
            final int useDistanceCount = VarInt.peek(buffer, pos);
            if (useDistanceCount < 0) {
                return ValidationResult.error("Invalid dictionary count for UseDistance");
            }
            if (useDistanceCount > 4096000) {
                return ValidationResult.error("UseDistance exceeds max length 4096000");
            }
            pos += VarInt.length(buffer, pos);
            for (int i = 0; i < useDistanceCount; ++i) {
                ++pos;
                pos += 4;
                if (pos > buffer.writerIndex()) {
                    return ValidationResult.error("Buffer overflow reading value");
                }
            }
        }
        if ((nullBits & 0x2) != 0x0) {
            final int prioritiesOffset = buffer.getIntLE(offset + 8);
            if (prioritiesOffset < 0) {
                return ValidationResult.error("Invalid offset for Priorities");
            }
            int pos = offset + 12 + prioritiesOffset;
            if (pos >= buffer.writerIndex()) {
                return ValidationResult.error("Offset out of bounds for Priorities");
            }
            final int prioritiesCount = VarInt.peek(buffer, pos);
            if (prioritiesCount < 0) {
                return ValidationResult.error("Invalid dictionary count for Priorities");
            }
            if (prioritiesCount > 4096000) {
                return ValidationResult.error("Priorities exceeds max length 4096000");
            }
            pos += VarInt.length(buffer, pos);
            for (int i = 0; i < prioritiesCount; ++i) {
                pos = ++pos + InteractionPriority.computeBytesConsumed(buffer, pos);
            }
        }
        return ValidationResult.OK;
    }
    
    public InteractionConfiguration clone() {
        final InteractionConfiguration copy = new InteractionConfiguration();
        copy.displayOutlines = this.displayOutlines;
        copy.debugOutlines = this.debugOutlines;
        copy.useDistance = ((this.useDistance != null) ? new HashMap<GameMode, Float>(this.useDistance) : null);
        copy.allEntities = this.allEntities;
        if (this.priorities != null) {
            final Map<InteractionType, InteractionPriority> m = new HashMap<InteractionType, InteractionPriority>();
            for (final Map.Entry<InteractionType, InteractionPriority> e : this.priorities.entrySet()) {
                m.put(e.getKey(), e.getValue().clone());
            }
            copy.priorities = m;
        }
        return copy;
    }
    
    @Override
    public boolean equals(final Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj instanceof final InteractionConfiguration other) {
            return this.displayOutlines == other.displayOutlines && this.debugOutlines == other.debugOutlines && Objects.equals(this.useDistance, other.useDistance) && this.allEntities == other.allEntities && Objects.equals(this.priorities, other.priorities);
        }
        return false;
    }
    
    @Override
    public int hashCode() {
        return Objects.hash(this.displayOutlines, this.debugOutlines, this.useDistance, this.allEntities, this.priorities);
    }
}
