Bazooka/Grenade Training: Make weapon selection detection future-proof
The old check relied on hardcoded slot IDs for the weapons, which is not good.
use netbuf;
use std::io::Read;
use std::io::Result;
use nom::IResult;
pub mod messages;
mod parser;
pub struct ProtocolDecoder {
buf: netbuf::Buf,
consumed: usize,
}
impl ProtocolDecoder {
pub fn new() -> ProtocolDecoder {
ProtocolDecoder {
buf: netbuf::Buf::new(),
consumed: 0,
}
}
pub fn read_from<R: Read>(&mut self, stream: &mut R) -> Result<usize> {
self.buf.read_from(stream)
}
pub fn extract_messages(&mut self) -> Vec<messages::HWProtocolMessage> {
let parse_result = parser::extract_messages(&self.buf[..]);
match parse_result {
IResult::Done(tail, msgs) => {
self.consumed = self.buf.len() - self.consumed - tail.len();
msgs
},
IResult::Incomplete(_) => unreachable!(),
IResult::Error(_) => unreachable!(),
}
}
pub fn sweep(&mut self) {
self.buf.consume(self.consumed);
self.consumed = 0;
}
}