/*
 * Decompiled with CFR 0.152.
 */
package net.minecraft.network.protocol.game;

import java.util.Optional;
import net.minecraft.core.Holder;
import net.minecraft.network.FriendlyByteBuf;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.network.protocol.Packet;
import net.minecraft.network.protocol.PacketType;
import net.minecraft.network.protocol.game.ClientGamePacketListener;
import net.minecraft.network.protocol.game.GamePacketTypes;
import net.minecraft.world.damagesource.DamageSource;
import net.minecraft.world.damagesource.DamageType;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.level.Level;
import net.minecraft.world.phys.Vec3;

public record ClientboundDamageEventPacket(int entityId, Holder<DamageType> sourceType, int sourceCauseId, int sourceDirectId, Optional<Vec3> sourcePosition) implements Packet<ClientGamePacketListener>
{
    public static final StreamCodec<RegistryFriendlyByteBuf, ClientboundDamageEventPacket> STREAM_CODEC = Packet.codec(ClientboundDamageEventPacket::write, ClientboundDamageEventPacket::new);

    public ClientboundDamageEventPacket(Entity entity, DamageSource source) {
        this(entity.getId(), source.typeHolder(), source.getEntity() != null ? source.getEntity().getId() : -1, source.getDirectEntity() != null ? source.getDirectEntity().getId() : -1, Optional.ofNullable(source.sourcePositionRaw()));
    }

    private ClientboundDamageEventPacket(RegistryFriendlyByteBuf input) {
        this(input.readVarInt(), (Holder)DamageType.STREAM_CODEC.decode(input), ClientboundDamageEventPacket.readOptionalEntityId(input), ClientboundDamageEventPacket.readOptionalEntityId(input), input.readOptional(i -> new Vec3(i.readDouble(), i.readDouble(), i.readDouble())));
    }

    private static void writeOptionalEntityId(FriendlyByteBuf output, int id) {
        output.writeVarInt(id + 1);
    }

    private static int readOptionalEntityId(FriendlyByteBuf input) {
        return input.readVarInt() - 1;
    }

    private void write(RegistryFriendlyByteBuf output) {
        output.writeVarInt(this.entityId);
        DamageType.STREAM_CODEC.encode(output, this.sourceType);
        ClientboundDamageEventPacket.writeOptionalEntityId(output, this.sourceCauseId);
        ClientboundDamageEventPacket.writeOptionalEntityId(output, this.sourceDirectId);
        output.writeOptional(this.sourcePosition, (o, pos) -> {
            o.writeDouble(pos.x());
            o.writeDouble(pos.y());
            o.writeDouble(pos.z());
        });
    }

    @Override
    public PacketType<ClientboundDamageEventPacket> type() {
        return GamePacketTypes.CLIENTBOUND_DAMAGE_EVENT;
    }

    @Override
    public void handle(ClientGamePacketListener listener) {
        listener.handleDamageEvent(this);
    }

    public DamageSource getSource(Level level) {
        if (this.sourcePosition.isPresent()) {
            return new DamageSource(this.sourceType, this.sourcePosition.get());
        }
        Entity cause = level.getEntity(this.sourceCauseId);
        Entity direct = level.getEntity(this.sourceDirectId);
        return new DamageSource(this.sourceType, direct, cause);
    }
}