CC-Tweaked/src/main/java/dan200/computercraft/shared/peripheral/monitor/MonitorWatcher.java

/*
 * This file is part of ComputerCraft - http://www.computercraft.info
 * Copyright Daniel Ratcliffe, 2011-2021. Do not distribute without permission.
 * Send enquiries to dratcliffe@gmail.com
 */
package dan200.computercraft.shared.peripheral.monitor;

import dan200.computercraft.ComputerCraft;
import dan200.computercraft.shared.network.NetworkHandler;
import dan200.computercraft.shared.network.client.MonitorClientMessage;
import dan200.computercraft.shared.network.client.TerminalState;
import net.minecraft.entity.player.ServerPlayerEntity;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.ChunkPos;
import net.minecraft.world.World;
import net.minecraft.world.chunk.Chunk;
import net.minecraft.world.chunk.ChunkStatus;
import net.minecraft.world.server.ServerWorld;
import net.minecraftforge.event.TickEvent;
import net.minecraftforge.event.world.ChunkWatchEvent;
import net.minecraftforge.eventbus.api.SubscribeEvent;
import net.minecraftforge.fml.common.Mod;

import java.util.ArrayDeque;
import java.util.Queue;

@Mod.EventBusSubscriber( modid = ComputerCraft.MOD_ID )
public final class MonitorWatcher
{
    private static final Queue<TileMonitor> watching = new ArrayDeque<>();
    private static final Queue<PlayerUpdate> playerUpdates = new ArrayDeque<>();

    private MonitorWatcher()
    {
    }

    static void enqueue( TileMonitor monitor )
    {
        if( monitor.enqueued ) return;

        monitor.enqueued = true;
        monitor.cached = null;
        watching.add( monitor );
    }

    @SubscribeEvent
    public static void onWatch( ChunkWatchEvent.Watch event )
    {
        ChunkPos chunkPos = event.getPos();
        Chunk chunk = (Chunk) event.getWorld().getChunk( chunkPos.x, chunkPos.z, ChunkStatus.FULL, false );
        if( chunk == null ) return;

        for( TileEntity te : chunk.getBlockEntities().values() )
        {
            // Find all origin monitors who are not already on the queue.
            if( !(te instanceof TileMonitor) ) continue;

            TileMonitor monitor = (TileMonitor) te;
            ServerMonitor serverMonitor = getMonitor( monitor );
            if( serverMonitor == null || monitor.enqueued ) continue;

            // The chunk hasn't been sent to the client yet, so we can't send an update. Do it on tick end.
            playerUpdates.add( new PlayerUpdate( event.getPlayer(), monitor ) );
        }
    }

    @SubscribeEvent
    public static void onTick( TickEvent.ServerTickEvent event )
    {
        if( event.phase != TickEvent.Phase.END ) return;

        PlayerUpdate playerUpdate;
        while( (playerUpdate = playerUpdates.poll()) != null )
        {
            TileMonitor tile = playerUpdate.monitor;
            if( tile.enqueued || tile.isRemoved() ) continue;

            ServerMonitor monitor = getMonitor( tile );
            if( monitor == null ) continue;

            // Some basic sanity checks to the player. It's possible they're no longer within range, but that's harder
            // to track efficiently.
            ServerPlayerEntity player = playerUpdate.player;
            if( !player.isAlive() || player.getLevel() != tile.getLevel() ) continue;

            NetworkHandler.sendToPlayer( playerUpdate.player, new MonitorClientMessage( tile.getBlockPos(), getState( tile, monitor ) ) );
        }

        long limit = ComputerCraft.monitorBandwidth;
        boolean obeyLimit = limit > 0;

        TileMonitor tile;
        while( (!obeyLimit || limit > 0) && (tile = watching.poll()) != null )
        {
            tile.enqueued = false;
            ServerMonitor monitor = getMonitor( tile );
            if( monitor == null ) continue;

            BlockPos pos = tile.getBlockPos();
            World world = tile.getLevel();
            if( !(world instanceof ServerWorld) ) continue;

            Chunk chunk = world.getChunkAt( pos );
            if( !((ServerWorld) world).getChunkSource().chunkMap.getPlayers( chunk.getPos(), false ).findAny().isPresent() )
            {
                continue;
            }

            TerminalState state = getState( tile, monitor );
            NetworkHandler.sendToAllTracking( new MonitorClientMessage( pos, state ), chunk );

            limit -= state.size();
        }
    }

    private static ServerMonitor getMonitor( TileMonitor monitor )
    {
        return !monitor.isRemoved() && monitor.getXIndex() == 0 && monitor.getYIndex() == 0 ? monitor.getCachedServerMonitor() : null;
    }

    private static TerminalState getState( TileMonitor tile, ServerMonitor monitor )
    {
        TerminalState state = tile.cached;
        if( state == null ) state = tile.cached = monitor.write();
        return state;
    }

    private static final class PlayerUpdate
    {
        final ServerPlayerEntity player;
        final TileMonitor monitor;

        private PlayerUpdate( ServerPlayerEntity player, TileMonitor monitor )
        {
            this.player = player;
            this.monitor = monitor;
        }
    }
}