path: root/console/src/bin/nogpads.rs

use clap::{arg, Args, Parser, Subcommand};
use nogamepads_core::pad_io::server::nogamepads_server::{PadServer};
use nogamepads_core::DEFAULT_PORT;
use serde::{Deserialize, Serialize};
use std::env::current_dir;
use std::fs::{create_dir, File};
use std::io::{BufReader, Write};
use std::net::{IpAddr, Ipv4Addr};
use std::path::PathBuf;
use nogamepads_core::pad_data::game_profile::game_profile::GameProfile;

/// NoGamePads Console - Server Cli
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct NoGamepadServerCli {
    #[command(subcommand)]
    command: Commands,
}

/// 主要命令
#[derive(Subcommand, Debug)]
enum Commands {

    // 服务端配置
    #[command(about = "Configure the server")]
    Config(ConfigArgs),

    // 运行服务端
    #[command(about = "Run the server")]
    Run(RunArgs)
}

/// 服务端配置 参数
#[derive(Args, Debug)]
struct ConfigArgs {

    // 绑定的端口号
    #[arg(short, long, help = "Server port (0 = Default)")] // <---- DEFAULT_PORT
    port: Option<u16>,

    // 游戏名称
    #[arg(short ='n', long = "name")]
    game_name: Option<String>,

    // 游戏描述
    #[arg(short = 'd', long = "description")]
    game_description: Option<String>,

    // 游戏组织
    #[arg(short = 'o', long = "organization")]
    organization: Option<String>,

    // 游戏版本
    #[arg(short = 'v', long = "version")]
    version: Option<String>,

    // 工作室 & 游戏 主页
    #[arg(short = 'w', long = "website")]
    website: Option<String>,

    // 交流邮箱
    #[arg(short = 'e', long = "email")]
    email: Option<String>
}

/// 运行服务端 参数
#[derive(Args, Debug)]
struct RunArgs {

    // 调试模式
    #[arg(long)]
    debug: bool,
}

/// 本地存储的配置信息
#[derive(Serialize, Deserialize, PartialEq, Debug)]
struct ServerConfig {
    port: u16,
    profile: GameProfile,
}

impl Default for ServerConfig {
    fn default() -> Self {
        ServerConfig {
            port: DEFAULT_PORT,
            profile: GameProfile::default(),
        }
    }
}

/// # 快速生成 更新服务端信息 的宏
macro_rules! update_config {
    ($config:expr, $args:expr, $($field:ident),+) => {
        $(
            if let Some(ref value) = $args.$field {
                $config.profile.$field = value.clone();
                println!("Changed profile \"{}\" to \"{}\"",
                    stringify!($field),
                    value
                );
            }
        )+
    };
}

fn main () {

    // 命令行
    let cli = NoGamepadServerCli::parse();
    match cli.command {

        // 服务端配置
        Commands::Config(args) => {
            // 读取服务端配置
            let mut config = read_config();

            // 端口信息配置：
            // 端口数值被限定在 0 - 65535，但是若端口参数为 0，则会被设置为默认端口
            if args.port.is_some() {
                let port = args.port.unwrap_or(DEFAULT_PORT).clamp(0, 65535);
                config.port = if port == 0 { DEFAULT_PORT } else { port };
            }

            // 其他信息配置
            update_config!(
                config, args,
                game_name,
                game_description,
                organization,
                version,
                website,
                email
            );

            // 写入配置
            write_config(config);
        },

        // 运行服务端
        Commands::Run(args) => {
            // 读取服务端配置
            let config = read_config();

            // 根据调试选项启动服务端
            if args.debug {
                println!("- DEBUG MODE -");
                println!("Server started!");
                PadServer::default()
                    .addr(IpAddr::from(Ipv4Addr::new(127, 0, 0, 1)), config.port)
                    .put_profile(config.profile)
                    .enable_console()
                    .build()
                    .start_server();
            } else {
                println!("Server started!");
                PadServer::default()
                    .addr(IpAddr::from(Ipv4Addr::new(127, 0, 0, 1)), config.port)
                    .put_profile(config.profile)
                    .build()
                    .start_server();
            }
        }
    }
}

/// # 获得配置文件目录地址
///
/// ## 参数 - Parameters
///
/// | Field  | Type                  | Description |
/// | ------ | --------------------- | ----------- |
/// | -> | PathBuf | 地址 |
#[allow(dead_code)]
fn get_config_folder_path () -> PathBuf {
    current_dir().unwrap().join(".nogpads")
}

/// # 获得配置文件地址
///
/// ## 参数 - Parameters
///
/// | Field  | Type                  | Description |
/// | ------ | --------------------- | ----------- |
/// | -> | PathBuf | 地址 |
#[allow(dead_code)]
fn get_config_file_path () -> PathBuf {
    get_config_folder_path().join("config.yaml")
}

/// # 获得布局文件地址
///
/// ## 参数 - Parameters
///
/// | Field  | Type                  | Description |
/// | ------ | --------------------- | ----------- |
/// | -> | PathBuf | 地址 |
#[allow(dead_code)]
fn get_layout_file_path () -> PathBuf {
    get_config_folder_path().join("layout.yaml")
}

/// # 获得皮肤资源地址
///
/// ## 参数 - Parameters
///
/// | Field  | Type                  | Description |
/// | ------ | --------------------- | ----------- |
/// | -> | PathBuf | 地址 |
#[allow(dead_code)]
fn get_assets_package_path () -> PathBuf {
    get_config_folder_path().join("assets.zip")
}

/// # 读取配置信息
///
/// 读取 服务端控制台 的配置信息
///
/// ## 参数 - Parameters
///
/// | Field  | Type                  | Description |
/// | ------ | --------------------- | ----------- |
/// | -> | ServerConfig | 配置信息 |
fn read_config () -> ServerConfig {
    let config_folder_path = get_config_folder_path();
    let config_file_path = get_config_file_path();

    if ! config_folder_path.exists() {
        create_dir(&config_folder_path).unwrap();
    }

    if ! config_file_path.exists() {
        let config = ServerConfig::default();
        let config_text = serde_yaml::to_string(&config).unwrap();
        File::create(&config_file_path).unwrap().write_all(config_text.as_bytes()).unwrap();
        config
    } else {
        let config_file = File::open(&config_file_path).unwrap();
        let config_reader = BufReader::new(config_file);
        serde_yaml::from_reader(config_reader).unwrap()
    }
}

/// # 写入配置信息
///
/// 将 服务端控制台 的配置信息写入本地
///
/// ## 参数 - Parameters
///
/// | Field  | Type                  | Description |
/// | ------ | --------------------- | ----------- |
/// | config | ServerConfig | 配置信息 |
fn write_config (config: ServerConfig) {
    let config_file_path = get_config_file_path();

    let config_text = serde_yaml::to_string(&config).unwrap();
    File::create(config_file_path).unwrap().write_all(config_text.as_bytes()).unwrap();
}