Added support for Counter-Strike: Source

This commit is contained in:
CosminPerRam 2022-10-21 12:35:33 +03:00
parent b5141e8196
commit aefd8cc43c
5 changed files with 98 additions and 1 deletions

View file

@ -4,7 +4,8 @@
|----------|----------------------------------|----------------|----------------------------------|
| TF2 | Team Fortress 2 | Valve Protocol | |
| The_Ship | The Ship | Valve Protocol | |
| CSGO | Counter Strike: Global Offensive | Valve Protocol | Rules doesnt work on this title. |
| CSGO | Counter-Strike: Global Offensive | Valve Protocol | Rules doesnt work on this title. |
| CSS | Counter-Strike: Source | Valve Protocol | |
## Planned to add support:
All Valve titles.

10
examples/css.rs Normal file
View file

@ -0,0 +1,10 @@
use gamedig::games::css;
fn main() {
let response = css::query("104.128.58.206", None);
match response {
Err(error) => println!("Couldn't query, error: {error}"),
Ok(r) => println!("{:?}", r)
}
}

83
src/games/css.rs Normal file
View file

@ -0,0 +1,83 @@
use crate::{GDResult, valve};
use crate::valve::{ValveProtocol, App, GatheringSettings, Server, ServerRule, ServerPlayer};
#[derive(Debug)]
pub struct Player {
pub name: String,
pub score: u32,
pub duration: f32
}
impl Player {
fn from_valve_response(player: &ServerPlayer) -> Self {
Self {
name: player.name.clone(),
score: player.score,
duration: player.duration
}
}
}
#[derive(Debug)]
pub struct Response {
pub protocol: u8,
pub name: String,
pub map: String,
pub game: String,
pub players: u8,
pub players_details: Vec<Player>,
pub max_players: u8,
pub bots: u8,
pub server_type: Server,
pub has_password: bool,
pub vac_secured: bool,
pub version: String,
pub port: Option<u16>,
pub steam_id: Option<u64>,
pub tv_port: Option<u16>,
pub tv_name: Option<String>,
pub keywords: Option<String>,
pub rules: Vec<ServerRule>
}
impl Response {
pub fn new_from_valve_response(response: valve::Response) -> Self {
let (port, steam_id, tv_port, tv_name, keywords) = match response.info.extra_data {
None => (None, None, None, None, None),
Some(ed) => (ed.port, ed.steam_id, ed.tv_port, ed.tv_name, ed.keywords)
};
Self {
protocol: response.info.protocol,
name: response.info.name,
map: response.info.map,
game: response.info.game,
players: response.info.players,
players_details: response.players.unwrap().iter().map(|p| Player::from_valve_response(p)).collect(),
max_players: response.info.max_players,
bots: response.info.bots,
server_type: response.info.server_type,
has_password: response.info.has_password,
vac_secured: response.info.vac_secured,
version: response.info.version,
port,
steam_id,
tv_port,
tv_name,
keywords,
rules: response.rules.unwrap()
}
}
}
pub fn query(address: &str, port: Option<u16>) -> GDResult<Response> {
let valve_response = ValveProtocol::query(App::CSS, address, match port {
None => 27015,
Some(port) => port
}, GatheringSettings {
players: true,
rules: true
})?;
Ok(Response::new_from_valve_response(valve_response))
}

View file

@ -4,3 +4,4 @@
pub mod tf2;
pub mod the_ship;
pub mod csgo;
pub mod css;

View file

@ -124,6 +124,7 @@ pub enum Request {
/// Supported app id's
#[derive(PartialEq)]
pub enum App {
CSS = 240,
TF2 = 440,
CSGO = 730,
TheShip = 2400
@ -134,6 +135,7 @@ impl TryFrom<u16> for App {
fn try_from(value: u16) -> GDResult<Self> {
match value {
x if x == App::CSS as u16 => Ok(App::CSS),
x if x == App::TF2 as u16 => Ok(App::TF2),
x if x == App::CSGO as u16 => Ok(App::CSGO),
x if x == App::TheShip as u16 => Ok(App::TheShip),