/*
 * Minosoft
 * Copyright (C) 2020-2023 Moritz Zwerger
 *
 * This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with this program. If not, see <https://www.gnu.org/licenses/>.
 *
 * This software is not affiliated with Mojang AB, the original developer of Minecraft.
 */
package de.bixilon.minosoft.protocol.packets.s2c.play.container

import de.bixilon.minosoft.data.container.Container
import de.bixilon.minosoft.data.container.IncompleteContainer
import de.bixilon.minosoft.data.container.stack.ItemStack
import de.bixilon.minosoft.protocol.network.connection.play.PlayConnection
import de.bixilon.minosoft.protocol.packets.factory.LoadPacket
import de.bixilon.minosoft.protocol.packets.s2c.PlayS2CPacket
import de.bixilon.minosoft.protocol.protocol.ProtocolVersions.V_1_17_1_PRE1
import de.bixilon.minosoft.protocol.protocol.buffers.play.PlayInByteBuffer
import de.bixilon.minosoft.util.logging.Log
import de.bixilon.minosoft.util.logging.LogLevels
import de.bixilon.minosoft.util.logging.LogMessageType
import java.util.*

@LoadPacket
class ContainerItemsS2CP(buffer: PlayInByteBuffer) : PlayS2CPacket {
    val containerId = buffer.readUnsignedByte()
    val revision: Int = if (buffer.versionId >= V_1_17_1_PRE1) {
        buffer.readVarInt()
    } else {
        -1
    }
    val items: Array<ItemStack?> = buffer.readArray(
        if (buffer.versionId >= V_1_17_1_PRE1) {
            buffer.readVarInt()
        } else {
            buffer.readUnsignedShort()
        }
    ) { buffer.readItemStack() }
    val floatingItem = if (buffer.versionId >= V_1_17_1_PRE1) {
        buffer.readItemStack()?.let { Optional.of(it) } ?: Optional.empty()
    } else {
        null
    }

    private fun pushIncompleteContainer(connection: PlayConnection) {
        val container = IncompleteContainer()


        for ((slotId, stack) in this.items.withIndex()) {
            if (stack == null) {
                continue
            }
            container.slots[slotId] = stack
        }
        container.floating = floatingItem?.let { if (it.isEmpty) null else it.get() }

        connection.player.items.incomplete[containerId] = container
    }

    private fun updateContainer(container: Container) {
        container.lock()
        container.clear()

        for ((slotId, stack) in this.items.withIndex()) {
            if (stack == null) {
                continue
            }
            container[slotId] = stack
        }
        container.serverRevision = revision
        this.floatingItem?.let { container.floatingItem = if (it.isEmpty) null else it.get() }
        container.commit()
    }

    override fun handle(connection: PlayConnection) {
        val container = connection.player.items.containers[containerId]
        if (container == null) {
            pushIncompleteContainer(connection)
        } else {
            updateContainer(container)
        }
    }

    override fun log(reducedLog: Boolean) {
        Log.log(LogMessageType.NETWORK_PACKETS_IN, level = LogLevels.VERBOSE) { "Container items (containerId=$containerId, items=${items.contentToString()}, floating=$floatingItem)" }
    }
}
