package net.shadowfacts.simplemultipart.container; import com.google.common.collect.ImmutableSet; import net.fabricmc.fabric.api.util.NbtType; import net.fabricmc.fabric.block.entity.ClientSerializable; import net.minecraft.block.Block; import net.minecraft.block.BlockState; import net.minecraft.block.Blocks; import net.minecraft.block.entity.BlockEntity; import net.minecraft.block.entity.BlockEntityType; import net.minecraft.item.ItemStack; import net.minecraft.nbt.CompoundTag; import net.minecraft.nbt.ListTag; import net.minecraft.nbt.Tag; import net.minecraft.server.world.ServerWorld; import net.minecraft.util.Tickable; import net.minecraft.util.math.BlockPos; import net.minecraft.util.shape.VoxelShape; import net.minecraft.world.loot.context.LootContext; import net.minecraft.world.loot.context.Parameters; import net.shadowfacts.simplemultipart.SimpleMultipart; import net.shadowfacts.simplemultipart.multipart.MultipartState; import net.shadowfacts.simplemultipart.multipart.entity.MultipartEntity; import net.shadowfacts.simplemultipart.multipart.entity.MultipartEntityProvider; import net.shadowfacts.simplemultipart.util.MultipartHelper; import net.shadowfacts.simplemultipart.multipart.MultipartView; import net.shadowfacts.simplemultipart.util.ShapeUtils; import java.util.*; import java.util.stream.Collectors; /** * @author shadowfacts */ public abstract class AbstractContainerBlockEntity extends BlockEntity implements MultipartContainer, ClientSerializable { protected Set parts = new HashSet<>(); public AbstractContainerBlockEntity(BlockEntityType type) { super(type); } @Override public Set getParts() { return ImmutableSet.copyOf(parts); } @Override public boolean hasParts() { return !parts.isEmpty(); } @Override public boolean canInsert(MultipartState partState) { VoxelShape newShape = partState.getBoundingShape(null); for (Entry e : parts) { VoxelShape existingShape = e.state.getBoundingShape(e); if (ShapeUtils.intersect(newShape, existingShape)) { return false; } } return true; } @Override public void insert(MultipartState partState) { if (!canInsert(partState)) { return; } MultipartEntity entity = null; if (partState.getMultipart() instanceof MultipartEntityProvider) { entity = ((MultipartEntityProvider)partState.getMultipart()).createMultipartEntity(partState, this); } parts.add(new Entry(this, partState, entity)); updateWorld(); } @Override public void remove(MultipartView view) { if (view.getContainer() != this) { return; } parts.removeIf(e -> e.state == view.getState() && e.entity == view.getEntity()); if (parts.isEmpty()) { world.setBlockState(pos, Blocks.AIR.getDefaultState()); } else { updateWorld(); } } @Override public boolean breakPart(MultipartView view) { Optional entry = parts.stream().filter(e -> e.state == view.getState() && e.entity == view.getEntity()).findFirst(); if (!entry.isPresent()) { return false; } if (world instanceof ServerWorld) { List drops = getDroppedStacks(entry.get(), (ServerWorld)world, pos); drops.forEach(stack -> Block.dropStack(world, pos, stack)); // TODO: don't drop if player is creative } remove(view); updateWorld(); return true; } @Override public void schedulePartSave() { markDirty(); // see yarn #360 } private void updateWorld() { boolean hasTickableParts = parts.stream().anyMatch(e -> e.getEntity() != null && e.getEntity() instanceof Tickable); boolean currentlyTickable = this instanceof Tickable; if (hasTickableParts != currentlyTickable) { Block newBlock = hasTickableParts ? SimpleMultipart.tickableContainerBlock : SimpleMultipart.containerBlock; world.setBlockState(pos, newBlock.getDefaultState(), 3); AbstractContainerBlockEntity newContainer = (AbstractContainerBlockEntity)world.getBlockEntity(pos); newContainer.parts = parts.stream() .map(e -> new Entry(newContainer, e.state, e.entity)) .collect(Collectors.toSet()); newContainer.parts.stream().filter(e -> e.entity != null).forEach(e -> e.entity.container = newContainer); } world.markDirty(pos, world.getBlockEntity(pos)); world.scheduleBlockRender(pos); BlockState blockState = world.getBlockState(pos); world.updateListeners(pos, blockState, blockState, 3); } private List getDroppedStacks(Entry e, ServerWorld world, BlockPos pos) { LootContext.Builder builder = new LootContext.Builder(world); builder.setRandom(world.random); builder.put(SimpleMultipart.MULTIPART_STATE_PARAMETER, e.state); builder.put(Parameters.POSITION, pos); return e.state.getDroppedStacks(e, builder); } private ListTag partsToTag() { ListTag list = new ListTag(); for (Entry e : parts) { CompoundTag tag = new CompoundTag(); tag.put("part", MultipartHelper.serializeMultipartState(e.state)); if (e.entity != null) { tag.put("entity", e.entity.toTag(new CompoundTag())); } list.add(tag); } return list; } private void partsFromTag(ListTag list) { parts.clear(); for (Tag tag : list) { CompoundTag compound = (CompoundTag)tag; MultipartState state = MultipartHelper.deserializeMultipartState(compound.getCompound("part")); MultipartEntity entity = null; if (state.getMultipart() instanceof MultipartEntityProvider && compound.containsKey("entity", NbtType.COMPOUND)) { entity = ((MultipartEntityProvider)state.getMultipart()).createMultipartEntity(state, this); entity.fromTag(compound.getCompound("entity")); } parts.add(new Entry(this, state, entity)); } } @Override public CompoundTag toTag(CompoundTag tag) { tag.put("parts", partsToTag()); return super.toTag(tag); } @Override public void fromTag(CompoundTag tag) { super.fromTag(tag); ListTag list = tag.getList("parts", NbtType.COMPOUND); partsFromTag(list); } @Override public CompoundTag toClientTag(CompoundTag tag) { tag.put("parts", partsToTag()); return tag; } @Override public void fromClientTag(CompoundTag tag) { ListTag list = tag.getList("parts", NbtType.COMPOUND); partsFromTag(list); updateWorld(); } public static class Entry implements MultipartView { public final MultipartContainer container; public MultipartState state; public MultipartEntity entity; private Entry(MultipartContainer container, MultipartState state, MultipartEntity entity) { this.container = container; this.state = state; this.entity = entity; } @Override public MultipartContainer getContainer() { return container; } @Override public MultipartState getState() { return state; } @Override public void setState(MultipartState state) { this.state = state; } @Override public MultipartEntity getEntity() { return entity; } @Override public void setEntity(MultipartEntity entity) { this.entity = entity; } @Override public String toString() { return "Entry{" + state + "}"; } } }