68 lines
2.7 KiB
Java
68 lines
2.7 KiB
Java
package com.hypixel.hytale.server.flock;
|
|
|
|
import com.hypixel.hytale.component.CommandBuffer;
|
|
import com.hypixel.hytale.component.Ref;
|
|
import com.hypixel.hytale.component.Store;
|
|
import com.hypixel.hytale.component.query.Query;
|
|
import com.hypixel.hytale.server.core.entity.entities.Player;
|
|
import com.hypixel.hytale.server.core.modules.entity.AllLegacyLivingEntityTypesQuery;
|
|
import com.hypixel.hytale.server.core.modules.entity.damage.Damage;
|
|
import com.hypixel.hytale.server.core.modules.entity.damage.DeathComponent;
|
|
import com.hypixel.hytale.server.core.modules.entity.damage.DeathSystems;
|
|
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
|
|
import com.hypixel.hytale.server.npc.entities.NPCEntity;
|
|
import com.hypixel.hytale.server.npc.role.Role;
|
|
import javax.annotation.Nonnull;
|
|
|
|
public class FlockDeathSystems {
|
|
public static class EntityDeath extends DeathSystems.OnDeathSystem {
|
|
@Nonnull
|
|
private final Query<EntityStore> query = Query.and(AllLegacyLivingEntityTypesQuery.INSTANCE, Query.not(Player.getComponentType()));
|
|
|
|
@Nonnull
|
|
@Override
|
|
public Query<EntityStore> getQuery() {
|
|
return this.query;
|
|
}
|
|
|
|
public void onComponentAdded(
|
|
@Nonnull Ref<EntityStore> ref, @Nonnull DeathComponent component, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer
|
|
) {
|
|
NPCEntity npcComponent = commandBuffer.getComponent(ref, NPCEntity.getComponentType());
|
|
if (npcComponent != null) {
|
|
Role role = npcComponent.getRole();
|
|
if (role != null && !role.isCorpseStaysInFlock()) {
|
|
commandBuffer.tryRemoveComponent(ref, FlockMembership.getComponentType());
|
|
}
|
|
}
|
|
|
|
Damage damageInfo = component.getDeathInfo();
|
|
Ref<EntityStore> attackerRef = null;
|
|
if (damageInfo != null && damageInfo.getSource() instanceof Damage.EntitySource entitySource) {
|
|
attackerRef = entitySource.getRef();
|
|
}
|
|
|
|
if (attackerRef != null) {
|
|
Flock attackerFlock = FlockPlugin.getFlock(commandBuffer, attackerRef);
|
|
if (attackerFlock != null) {
|
|
attackerFlock.onTargetKilled(commandBuffer, ref);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public static class PlayerDeath extends DeathSystems.OnDeathSystem {
|
|
@Nonnull
|
|
@Override
|
|
public Query<EntityStore> getQuery() {
|
|
return Player.getComponentType();
|
|
}
|
|
|
|
public void onComponentAdded(
|
|
@Nonnull Ref<EntityStore> ref, @Nonnull DeathComponent component, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer
|
|
) {
|
|
commandBuffer.tryRemoveComponent(ref, FlockMembership.getComponentType());
|
|
}
|
|
}
|
|
}
|