rust-gamedig/src/protocols/minecraft/protocol/legacy_v1_4.rs

use crate::{GDError, GDResult};
use crate::protocols::minecraft::{LegacyGroup, Response, Server};
use crate::protocols::minecraft::protocol::legacy_v1_6::LegacyV1_6;
use crate::protocols::types::TimeoutSettings;
use crate::socket::{Socket, TcpSocket};
use crate::utils::buffer::{get_string_utf16_be, get_u16_be, get_u8};

pub struct LegacyV1_4 {
    socket: TcpSocket
}

impl LegacyV1_4 {
    fn new(address: &str, port: u16, timeout_settings: Option<TimeoutSettings>) -> GDResult<Self> {
        let socket = TcpSocket::new(address, port)?;
        socket.apply_timeout(timeout_settings)?;

        Ok(Self {
            socket
        })
    }

    fn send_initial_request(&mut self) -> GDResult<()> {
        self.socket.send(&[0xFE, 0x01])
    }

    fn get_info(&mut self) -> GDResult<Response> {
        self.send_initial_request()?;

        let buf = self.socket.receive(None)?;
        let mut pos = 0;

        if get_u8(&buf, &mut pos)? != 0xFF {
            return Err(GDError::PacketBad("Expected 0xFF".to_string()));
        }

        let length = get_u16_be(&buf, &mut pos)? * 2;
        if buf.len() != (length + 3) as usize { //+ 3 because of the first byte and the u16
            return Err(GDError::PacketBad("Not right size".to_string()));
        }

        if LegacyV1_6::is_protocol(&buf, &mut pos)? {
            return LegacyV1_6::get_response(&buf, &mut pos);
        }

        let packet_string = get_string_utf16_be(&buf, &mut pos)?;

        let split: Vec<&str> = packet_string.split("§").collect();
        if split.len() != 3 {
            return Err(GDError::PacketBad("Not right size".to_string()));
        }

        let description = split[0].to_string();
        let online_players = split[1].parse()
            .map_err(|_| GDError::PacketBad("Expected int".to_string()))?;
        let max_players = split[2].parse()
            .map_err(|_| GDError::PacketBad("Expected int".to_string()))?;

        Ok(Response {
            version_name: "1.4+".to_string(),
            version_protocol: -1,
            max_players,
            online_players,
            sample_players: None,
            description,
            favicon: None,
            previews_chat: None,
            enforces_secure_chat: None,
            server_type: Server::Legacy(LegacyGroup::V1_4)
        })
    }

    pub fn query(address: &str, port: u16, timeout_settings: Option<TimeoutSettings>) -> GDResult<Response> {
        LegacyV1_4::new(address, port, timeout_settings)?.get_info()
    }
}