use crate::network::*;
use crate::packet::*;
use basin2_lib::result::*;
use bytes::BytesMut;

#[derive(PartialEq, Clone, Debug)]
pub struct CustomQueryPacket {
    pub transactionId: i32,
    pub data: Option<BytesMut>,
}

impl CodablePacket for CustomQueryPacket {
    fn encode(self, buf: &mut BytesMut) {
        buf.set_mc_var_int(self.transactionId);
        match self.data {
            Some(bytes) => {
                buf.set_mc_bool(true);
                buf.unsplit(bytes);
            }
            None => {
                buf.set_mc_bool(false);
            }
        }
    }

    fn decode(buf: &mut BytesMut) -> Result<Self>
    where
        Self: Sized,
    {
        let transactionId = buf.get_mc_var_int()?;
        let data = if buf.get_mc_bool()? {
            Some(buf.clone_bounded(1048576)?)
        } else {
            None
        };
        return Ok(CustomQueryPacket {
            transactionId,
            data,
        });
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::packet::test::*;

    #[test]
    fn test_cycle() -> Result<()> {
        cycle(CustomQueryPacket {
            transactionId: 128,
            data: Some(BytesMut::from(&vec![0x0a, 0x0b][..])),
        })
        .and_then(|()| {
            cycle(CustomQueryPacket {
                transactionId: 128,
                data: None,
            })
        })
    }
}
