rust-gamedig/crates/lib/examples/generic.rs

use gamedig::{
    protocols::types::{CommonResponse, ExtraRequestSettings, TimeoutSettings},
    query_with_timeout_and_extra_settings,
    GDResult,
    GAMES,
};

use std::net::{IpAddr, SocketAddr, ToSocketAddrs};

/// Make a query given the name of a game
fn generic_query(
    game_name: &str,
    addr: &IpAddr,
    port: Option<u16>,
    timeout_settings: Option<TimeoutSettings>,
    extra_settings: Option<ExtraRequestSettings>,
) -> GDResult<Box<dyn CommonResponse>> {
    let game = GAMES
        .get(game_name)
        .expect("Game doesn't exist, run without arguments to see a list of games");

    println!("Querying {:#?} with game {:#?}.", addr, game);

    let response = query_with_timeout_and_extra_settings(game, addr, port, timeout_settings, extra_settings)?;
    println!("Response: {:#?}", response.as_json());

    let common = response.as_original();
    println!("Common response: {:#?}", common);

    Ok(response)
}

fn main() {
    let mut args = std::env::args().skip(1);

    // Handle arguments
    if let Some(game_name) = args.next() {
        let hostname = args.next().expect("Must provide an address");
        // Use to_socket_addrs to resolve hostname to IP
        let addr: SocketAddr = format!("{}:0", hostname)
            .to_socket_addrs()
            .unwrap()
            .next()
            .expect("Could not lookup host");
        let port: Option<u16> = args.next().map(|s| s.parse().unwrap());

        let timeout_settings = TimeoutSettings::new(
            TimeoutSettings::default().get_read(),
            TimeoutSettings::default().get_write(),
            2,
        )
        .unwrap();

        let extra_settings = ExtraRequestSettings::default()
            .set_hostname(hostname.to_string())
            .set_gather_rules(true)
            .set_gather_players(true)
            .set_check_app_id(false);

        generic_query(
            &game_name,
            &addr.ip(),
            port,
            Some(timeout_settings),
            Some(extra_settings),
        )
        .unwrap();
    } else {
        // Without arguments print a list of games

        for (name, game) in gamedig::games::GAMES.entries() {
            println!("{}\t{}", name, game.name);
        }
    }
}

#[cfg(test)]
mod test {
    use gamedig::{protocols::types::TimeoutSettings, GAMES};
    use std::{
        net::{IpAddr, Ipv4Addr},
        time::Duration,
    };

    use super::generic_query;

    const ADDR: IpAddr = IpAddr::V4(Ipv4Addr::LOCALHOST);

    fn test_game(game_name: &str) {
        let timeout_settings = Some(
            TimeoutSettings::new(
                Some(Duration::from_nanos(1)),
                Some(Duration::from_nanos(1)),
                0,
            )
            .unwrap(),
        );
        assert!(generic_query(game_name, &ADDR, None, timeout_settings, None).is_err());
    }

    #[test]
    fn battlefield1942() { test_game("battlefield1942"); }

    #[test]
    fn minecraft() { test_game("minecraft"); }

    #[test]
    fn teamfortress2() { test_game("teamfortress2"); }

    #[test]
    fn quake() { test_game("quake3"); }

    #[test]
    fn all_games() {
        for game_name in GAMES.keys() {
            test_game(game_name);
        }
    }
}