package util.network.epidemic

import util.network.simulator.Address
import util.network.simulator.packet.Packet
import util.network.simulator.packet.PayloadPacket

data class EpidemicPacket<T : Content<*>>(
    override val senderAddress: Address,
    override val receiverAddress: Address,
    override val payload: T,
    val type: Type
) : PayloadPacket<T> {

    fun clone(
        senderAddress: Address = this.senderAddress,
        receiverAddress: Address = this.receiverAddress,
        payload: T = this.payload,
        type: Type = this.type
    ): EpidemicPacket<T> =
        EpidemicPacket(senderAddress, receiverAddress, payload, type)

    enum class Type {
        PUSH, PULL, PUSHPULL, REPLY
    }

    companion object {
        fun <T> createFromNetworkPacket(
            packet: Packet,
            payload: Content<T>,
            type: Type
        ): EpidemicPacket<Content<T>> = EpidemicPacket(
            packet.senderAddress,
            packet.receiverAddress,
            payload,
            type
        )

        fun <T> createFromPayloadPacket(
            packet: PayloadPacket<Content<T>>,
            type: Type
        ): EpidemicPacket<Content<T>> = EpidemicPacket(
            packet.senderAddress,
            packet.receiverAddress,
            packet.payload,
            type
        )
    }
}