1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

use std::io::{Read, Result as Res, Write};

use endio::{Deserialize, LE, LERead, LEWrite, Serialize};
use endio_bit::{BEBitReader, BEBitWriter};
use lu_packets_derive::{BitVariantTests, ReplicaSerde};

use crate::common::{LVec, ObjId};
use super::{ReplicaD, ComponentConstruction, ComponentProtocol, ComponentSerialization};

// so close to being able to do serialization automatically...if not for the irregularity with `added_by_teammate`...
#[derive(Debug, PartialEq)]
pub struct BuffInfo {
	pub buff_id: u32,
	pub time_left: Option<u32>,
	pub cancel_on_death: bool,
	pub cancel_on_zone: bool,
	pub cancel_on_damaged: bool,
	pub cancel_on_remove_buff: bool,
	pub cancel_on_ui: bool,
	pub cancel_on_logout: bool,
	pub cancel_on_unequip: bool,
	pub cancel_on_damage_absorb_ran_out: bool,
	pub added_by_teammate: Option<ObjId>,
	pub apply_on_teammates: bool,
	pub ref_count: i32,
}

impl<R: Read> Deserialize<LE, BEBitReader<R>> for BuffInfo {
	fn deserialize(reader: &mut BEBitReader<R>) -> Res<Self> {
		let buff_id = LERead::read(reader)?;
		let time_left = ReplicaD::deserialize(reader)?;
		let cancel_on_death = reader.read_bit()?;
		let cancel_on_zone = reader.read_bit()?;
		let cancel_on_damaged = reader.read_bit()?;
		let cancel_on_remove_buff = reader.read_bit()?;
		let cancel_on_ui = reader.read_bit()?;
		let cancel_on_logout = reader.read_bit()?;
		let cancel_on_unequip = reader.read_bit()?;
		let cancel_on_damage_absorb_ran_out = reader.read_bit()?;
		let added_by_teammate = reader.read_bit()?;
		let apply_on_teammates = reader.read_bit()?;
		let added_by_teammate = if added_by_teammate { Some(LERead::read(reader)?) } else { None };
		let ref_count = LERead::read(reader)?;
		Ok(Self { buff_id, time_left, cancel_on_death, cancel_on_zone, cancel_on_damaged, cancel_on_remove_buff, cancel_on_ui, cancel_on_logout, cancel_on_unequip, cancel_on_damage_absorb_ran_out, added_by_teammate, apply_on_teammates, ref_count })
	}
}
impl<'a, W: Write> Serialize<LE, BEBitWriter<W>> for &'a BuffInfo {
	fn serialize(self, writer: &mut BEBitWriter<W>) -> Res<()> {
		LEWrite::write(writer, self.buff_id)?;
		crate::raknet::client::replica::ReplicaS::serialize(&self.time_left, writer)?;
		writer.write_bit(self.cancel_on_death)?;
		writer.write_bit(self.cancel_on_zone)?;
		writer.write_bit(self.cancel_on_damaged)?;
		writer.write_bit(self.cancel_on_remove_buff)?;
		writer.write_bit(self.cancel_on_ui)?;
		writer.write_bit(self.cancel_on_logout)?;
		writer.write_bit(self.cancel_on_unequip)?;
		writer.write_bit(self.cancel_on_damage_absorb_ran_out)?;
		writer.write_bit(self.added_by_teammate.is_some())?;
		writer.write_bit(self.apply_on_teammates)?;
		if let Some(x) = self.added_by_teammate {
			LEWrite::write(writer, x)?;
		}
		LEWrite::write(writer, self.ref_count)?;
		Ok(())
	}
}

#[derive(BitVariantTests, Debug, PartialEq, ReplicaSerde)]
pub struct BuffConstruction {
	pub buffs: Option<LVec<u32, BuffInfo>>,
	pub immunities: Option<LVec<u32, BuffInfo>>,
}

#[derive(BitVariantTests, Debug, PartialEq, ReplicaSerde)]
pub struct BuffSerialization {}

impl ComponentConstruction for BuffConstruction {
	fn ser(&self, writer: &mut BEBitWriter<Vec<u8>>) -> Res<()> {
		self.serialize(writer)
	}
}

impl ComponentSerialization for BuffSerialization {
	fn ser(&self, writer: &mut BEBitWriter<Vec<u8>>) -> Res<()> {
		self.serialize(writer)
	}
}

pub struct BuffProtocol;

impl ComponentProtocol for BuffProtocol {
	type Construction = BuffConstruction;
	type Serialization = BuffSerialization;
}