ServerSeekerV2/src/config.rs

use serde::Deserialize;
use std::cmp::max;
use std::fs::File;
use std::io::{ErrorKind, Read};
use tracing::{error, info};

#[derive(Deserialize)]
pub struct Config {
	pub database: Database,
	pub player_tracking: PlayerTracking,
	pub scanner: Scanner,
	pub masscan: Masscan,
}

#[derive(Deserialize)]
pub struct Database {
	pub url: String,
	pub port: u16,
	pub table: String,
	pub user: String,
	pub password: String,
}

#[derive(Deserialize)]
pub struct PlayerTracking {
	pub enabled: bool,
	pub players: Vec<String>,
}

#[derive(Deserialize)]
pub struct Scanner {
	pub repeat: bool,
	pub scan_delay: u64,
	pub port_range_start: u16,
	pub port_range_end: u16,
}

#[derive(Deserialize)]
pub struct Masscan {
	pub config_file: String,
	pub output_file: String,
}

impl Default for Scanner {
	fn default() -> Self {
		Self {
			repeat: true,
			scan_delay: 60,
			port_range_start: 25565,
			port_range_end: 25565,
		}
	}
}

impl Scanner {
	pub fn total_ports(&self) -> u16 {
		let start = self.port_range_start;
		let end = self.port_range_end;

		if start > end {
			error!("port_range_start cannot be greater than port_range_end!");
			std::process::exit(1);
		}

		max(1, end - start)
	}
}

pub fn load_config(path: String) -> Config {
	let mut file = match File::open(&path) {
		Ok(file) => file,
		Err(error) => match error.kind() {
			ErrorKind::NotFound => panic!("{} not found!", &path),
			ErrorKind::PermissionDenied => panic!("File permissions invalid for {}!", &path),
			_ => panic!("Unknown config error!"),
		},
	};

	let mut contents = String::new();
	file.read_to_string(&mut contents).unwrap_or_default();

	info!("Using config file {path}");
	toml::from_str(&contents).expect("Failed to parse config!")
}