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

#[derive(PartialEq, Clone, Debug)]
pub struct SetScorePacket {
    pub owner: String,
    pub objectiveName: String,
    pub score: i32,
    pub method: ServerScoreboardMethod,
}

impl CodablePacket for SetScorePacket {
    fn encode(self, buf: &mut BytesMut) {
        use ServerScoreboardMethod::*;

        buf.set_mc_string(self.owner);
        buf.set_mc_var_int(self.method as i32);
        buf.set_mc_string(self.objectiveName);
        match self.method {
            Remove => (),
            _ => buf.set_mc_var_int(self.score),
        }
    }

    fn decode(buf: &mut BytesMut) -> Result<Self>
    where
        Self: Sized,
    {
        use ServerScoreboardMethod::*;

        let owner = buf.get_mc_string(40)?;
        let method: ServerScoreboardMethod = buf.get_mc_enum()?;
        let objectiveName = buf.get_mc_string(16)?;
        let score = match method {
            Remove => 0,
            _ => buf.get_mc_var_int()?,
        };
        return Ok(SetScorePacket {
            owner,
            objectiveName,
            score,
            method,
        });
    }
}

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

    #[test]
    fn test_cycle_change() -> Result<()> {
        cycle(SetScorePacket {
            owner: "owner".to_string(),
            objectiveName: "objective name".to_string(),
            score: 125,
            method: ServerScoreboardMethod::Change,
        })
    }

    #[test]
    fn test_cycle_remove() -> Result<()> {
        cycle(SetScorePacket {
            owner: "owner".to_string(),
            objectiveName: "objective name".to_string(),
            score: 0,
            method: ServerScoreboardMethod::Remove,
        })
    }
}
