diff options
| author | 1992414357@qq.com <1992414357@qq.com> | 2025-06-09 01:44:36 +0800 |
|---|---|---|
| committer | 1992414357@qq.com <1992414357@qq.com> | 2025-06-09 01:44:36 +0800 |
| commit | 49192dbb98e0ab1f2a66b4786fac86d63f69be64 (patch) | |
| tree | 7cdf27c7cab5fb6b1aabc2ac7c44b7b492558945 /core/src/pad_service | |
| parent | c9dbba0d288becb7f05cebe526be25c76e5a850a (diff) | |
重构所有部分
Diffstat (limited to 'core/src/pad_service')
| -rw-r--r-- | core/src/pad_service/client.rs | 520 | ||||
| -rw-r--r-- | core/src/pad_service/client_debug_cli.rs | 92 | ||||
| -rw-r--r-- | core/src/pad_service/mod.rs | 5 | ||||
| -rw-r--r-- | core/src/pad_service/server.rs | 819 | ||||
| -rw-r--r-- | core/src/pad_service/server_debug_cli.rs | 212 |
5 files changed, 0 insertions, 1648 deletions
diff --git a/core/src/pad_service/client.rs b/core/src/pad_service/client.rs deleted file mode 100644 index b0b05a2..0000000 --- a/core/src/pad_service/client.rs +++ /dev/null @@ -1,520 +0,0 @@ -pub mod nogamepads_client { - use std::collections::VecDeque; - use crate::pad_data::pad_messages::nogamepads_message_transfer::{read_msg, send_msg}; - use crate::pad_data::pad_messages::nogamepads_messages::{ConnectionCallbackMessage, ConnectionErrorType, ConnectionMessage, ControlMessage, GameMessage, LeaveReason}; - use crate::pad_data::pad_player_info::nogamepads_player_info::PlayerInfo; - use log::{error, info}; - use std::net::{IpAddr, Ipv4Addr}; - use std::process::exit; - use std::sync::atomic::AtomicBool; - use std::sync::atomic::Ordering::SeqCst; - use std::sync::{Arc, Mutex}; - use std::time::Duration; - use clap::CommandFactory; - use tokio::io::{AsyncReadExt, AsyncWriteExt, ReadHalf, WriteHalf}; - use tokio::net::TcpStream; - use tokio::{io, spawn}; - use tokio::runtime::Runtime; - use nogamepads::console_utils::debug_console::read_cli; - use nogamepads::convert_utils::convert_deque_to_vec; - use nogamepads::logger_utils::logger_build; - use crate::pad_service::client_debug_cli::{process_debug_cmd, Pcc}; - use crate::DEFAULT_PORT; - use crate::pad_data::game_profile::game_profile::GameProfile; - use crate::pad_data::pad_messages::nogamepads_message_encoder::NgpdMessageEncoder; - use crate::pad_data::pad_messages::nogamepads_messages::ControlMessage::{Axis, Dir, Msg, Pressed, Released}; - - type WriteList = Arc<Mutex<VecDeque<ControlMessage>>>; - type ReadList = Arc<Mutex<VecDeque<GameMessage>>>; - - #[repr(C)] - pub struct PadClient { - - // --- 主要参数 --- - - // 目标地址 - target_address: IpAddr, - - // 目标端口 - #[allow(dead_code)] - target_port: u16, - - // 绑定的玩家 - bind_player: PlayerInfo, - - // 调试模式 - enable_console: bool, - - // 保持安静,不初始化 env_logger - quiet: bool, - - // --- 运行时参数 --- - - // 发送信息列表 - write_list: WriteList, - - // 读取信息列表 - read_list: ReadList, - - // 是否退出 - exit: AtomicBool, - } - - impl Clone for PadClient { - fn clone(&self) -> Self { - PadClient { - enable_console: self.enable_console.clone(), - target_address: self.target_address.clone(), - target_port: self.target_port.clone(), - bind_player: self.bind_player.clone(), - quiet: self.quiet.clone(), - - write_list: self.write_list.clone(), - read_list: self.read_list.clone(), - exit: AtomicBool::new(self.exit.load(SeqCst)), - } - } - } - - impl Default for PadClient { - fn default() -> Self { - PadClient { - enable_console: false, - target_address: IpAddr::from(Ipv4Addr::new(127, 0, 0, 1)), - target_port: DEFAULT_PORT, - bind_player: PlayerInfo::new(), - quiet: false, - - write_list: WriteList::default(), - read_list: ReadList::default(), - exit: AtomicBool::new(false) - } - } - } - - // 客户端构建部分 - impl PadClient { - - pub fn bind_addr(address: IpAddr) -> PadClient { - PadClient { - target_address: address, - ..PadClient::default() - } - } - - pub fn bind_addr_with_port(address: IpAddr, port: u16) -> PadClient { - PadClient { - target_address: address, - target_port: port, - ..PadClient::default() - } - } - - pub fn enable_console(&mut self) -> &mut PadClient { - self.enable_console = true; - self - } - - pub fn quiet(&mut self) -> &mut PadClient { - self.quiet = true; - self - } - - pub fn bind_player(&mut self, player: PlayerInfo) -> &mut PadClient { - self.bind_player = player; - self - } - - pub fn clone_addr(&self) -> (IpAddr, u16) { - (self.target_address, self.target_port) - } - - pub fn is_quiet(&self) -> bool { - self.quiet - } - - pub fn is_enable_console(&self) -> bool { - self.enable_console - } - - pub fn unbind_player(&mut self) -> PlayerInfo { - let player = self.bind_player.clone(); - self.bind_player = PlayerInfo::new(); - player - } - } - - // 客户端消息管理 - impl PadClient { - - pub fn key_press(&self, key_id: u8) { - self.put_msg(Pressed(key_id)); - } - - pub fn key_release(&self, key_id: u8) { - self.put_msg(Released(key_id)); - } - - pub fn change_axis(&self, axis_id: u8, axis: f64) { - self.put_msg(Axis(axis_id, axis)); - } - - pub fn change_direction(&self, direction_id: u8, x: f64, y: f64) { - self.put_msg(Dir(direction_id, (x.clamp(0.0, 1.0), y.clamp(0.0, 1.0)))); - } - - pub fn say_str(&self, msg: &str) { - self.put_msg(Msg(msg.to_owned())); - } - - pub fn say(&self, msg: String) { - self.put_msg(Msg(msg)); - } - - pub fn put_msg(&self, msg: ControlMessage) { - let mut guard = self.write_list.lock().unwrap(); - guard.push_back(msg); - } - - pub fn pop_a_msg(&self) -> Option<GameMessage> { - let mut guard = self.read_list.lock().unwrap(); - if !guard.is_empty() { - guard.pop_front() - } else { - None - } - } - - pub fn pop_msg_or(&self, or: GameMessage) -> GameMessage { - self.pop_a_msg().unwrap_or(or) - } - - pub fn list_received(&self) -> Vec<GameMessage> { - match self.read_list.lock() { - Ok(guard) => { - convert_deque_to_vec(&guard.to_owned()) - } - Err(_) => { Vec::new() } - } - } - } - - // 客户端状态控制 - impl PadClient { - - pub fn connect(self) { - self.connect_in_runtime(None); - } - - pub fn connect_in_runtime(self, tokio_runtime: Option<Runtime>) { - - self.exit.store(false, SeqCst); - - // 构建 Logger - if !self.quiet { - logger_build(); - } - - info!("Starting \"NoGamepads Client\"."); - - // 入口 - let entry = self.get_connect_entry(); - - // 阻塞运行 - if tokio_runtime.is_some() { - tokio_runtime.unwrap().block_on(entry); - } else { - let runtime = tokio::runtime::Builder::new_multi_thread() - .thread_name("nogpad-pad_service") - .thread_stack_size(32 * 1024 * 1024) - .enable_time() - .enable_io() - .build() - .unwrap(); - runtime.block_on(entry); - } - } - - pub fn get_connect_entry(self) -> impl Future<Output = ()> + Send + 'static { - - // 调试模式 - let debug = self.enable_console; - - // Arc - let arc_client = Arc::new(self); - - async move { - let main_thread = spawn({ - let client = Arc::clone(&arc_client); - async move { - Self::main_client_thread(client).await - } - }); - - let background_thread = spawn({ - let client = Arc::clone(&arc_client); - async move { - Self::background_thread(client).await - } - }); - - if debug { - let debug_cli = spawn({ - let client = Arc::clone(&arc_client); - async move { - Self::process_debug_cli(client).await - } - }); - let _ = tokio::join!(debug_cli, main_thread, background_thread); - } else { - let _ = tokio::join!(main_thread, background_thread); - } - } - } - - pub fn exit_server(&self) { - self.exit.store(true, SeqCst); - } - - async fn main_client_thread(self: Arc<Self>) { - let mut buffer : [u8; 1024] = [0; 1024]; - let addr_str = format!("{}:{}", self.target_address.to_string(), DEFAULT_PORT); - - info!("Connected to {}", &addr_str); - - // 下载服务端配置文件 - { - info!("Check: Downloaded game profile."); - let profile = self.check_server_profile(&mut buffer, addr_str.clone()).await; - if profile.is_some() { - info!("Success: Downloaded."); - let profile = profile.unwrap_or(GameProfile::default()); - for line in profile.to_string().split('\n') { - info!("{}", line); - } - } - else { - error!("Failed: Can't download profile!"); - self.exit_server(); - } - } - - // 尝试加入服务端,并建立长连接 - { - if !self.try_join_game(&mut buffer, addr_str.clone()).await { - error!("Failed: Can't join the game!"); - self.exit_server(); - return; - } - } - } - - async fn check_server_profile(self: &Arc<Self>, buffer: &mut [u8], addr_str: String) -> Option<GameProfile> { - match TcpStream::connect(&addr_str).await { - Ok(mut stream) => { - send_msg(&mut stream, ConnectionMessage::RequestProfile).await; - let callback : ConnectionCallbackMessage = read_msg(buffer, &mut stream).await; - match callback { - ConnectionCallbackMessage::Profile(profile) => { - Some(profile) - } - ConnectionCallbackMessage::Deny(err_type) => { - error!("Request failed: Server denied your request! ({:?})", err_type); - None - } - ConnectionCallbackMessage::Err => { - error!("Connection failed: Can't connect to server!"); - None - } - _ => { None } - } - } - Err(_err) => { - None - } - } - } - - async fn try_join_game(self: &Arc<Self>, buffer: &mut [u8], addr_str: String) -> bool { - - match TcpStream::connect(&addr_str).await { - Ok(mut stream) => { - - // 发送连接请求 - let info = self.bind_player.clone(); - send_msg(&mut stream, ConnectionMessage::Connection(info)).await; - - // 读取回调 - let callback : ConnectionCallbackMessage = read_msg(buffer, &mut stream).await; - match callback { - ConnectionCallbackMessage::Deny(error) => { - match error { - ConnectionErrorType::ContainSamePlayer => { - error!("Connection failed: Contains same player!"); - false - } - ConnectionErrorType::PlayerBanned => { - error!("Connection failed: You are banned!"); - false - } - ConnectionErrorType::Timeout => { - error!("Connection failed: Timeout!"); - false - } - ConnectionErrorType::GameLocked => { - error!("Connection failed: Game was locked!"); - false - } - _ => { false } - } - } - ConnectionCallbackMessage::Ok => { - - // 服务端检查完毕,发送 Ready 以示加入游戏 - send_msg(&mut stream, ConnectionMessage::Ready).await; - let callback : ConnectionCallbackMessage = read_msg(buffer, &mut stream).await; - match callback { - ConnectionCallbackMessage::Welcome => { - info!("Welcome!"); - Self::long_connection(Arc::clone(&self), stream).await; - } - ConnectionCallbackMessage::Deny(_error) => { - error!("Request failed: Server denied your request",); - } - _ => {} - } - true - } - _ => { false } - } - } - Err(err) => { - error!("Failed to connect to server: {}", err); - false - } - } - } - - async fn long_connection(self: Arc<Self>, stream: TcpStream) { - let (reader, writer) = io::split(stream); - spawn(Self::read_task(Arc::clone(&self), reader)); - spawn(Self::write_task(Arc::clone(&self), writer)); - } - - async fn read_task(self: Arc<Self>, mut reader: ReadHalf<TcpStream>) { - let mut buf = [0u8; 1024]; - loop { - match reader.read(&mut buf).await { - Ok(0) => break, - Ok(n) => { - let msg = GameMessage::de(buf[0..n].to_vec()); - { - match self.read_list.lock() { - Ok(mut guard) => { - match &msg { - GameMessage::Leave(reason) => { - match reason { - LeaveReason::GameOver => { - info!("Leave Game: Game Over!"); - self.exit_server(); - } - LeaveReason::ServerClosed => { - info!("Leave Game: Server closed!"); - self.exit_server(); - } - LeaveReason::YouAreKicked => { - error!("Kick Game: You are kicked!"); - self.exit_server(); - } - LeaveReason::YouAreBanned => { - error!("Kick Game: You are banned!"); - self.exit_server(); - } - } - } - _ => { - info!("{:?}", &msg); - guard.push_back(msg); - } - } - } - Err(_) => {} - } - } - } - Err(e) => { - error!("Error reading from stream: {}", e); - self.exit_server(); - break; - } - } - } - } - - async fn write_task(self: Arc<Self>, mut writer: WriteHalf<TcpStream>) { - loop { - let msg : Option<ControlMessage>; - { - let lock = self.write_list.lock(); - match lock { - Ok(mut guard) => { - if ! guard.is_empty() { - msg = guard.pop_front(); - } else { - msg = None; - } - } - Err(_) => { - msg = None; - } - } - } - if msg.is_some() { - let msg = msg.unwrap(); - match &writer.write_all(NgpdMessageEncoder::en(&msg).as_slice()).await { - Ok(_) => { - info!("Sent {:?}", msg); - } - Err(_error) => { - error!("Sent {:?} failed!", msg); - } - } - } - } - } - - async fn background_thread(self: Arc<Self>) { - loop { - // 退出程序的监听 - if self.exit.load(SeqCst) { - tokio::time::sleep(Duration::from_secs(1)).await; - info!("Main thread exited."); - exit(0); - } - } - } - - async fn process_debug_cli(self: Arc<Self>) { - loop { - if self.exit.load(SeqCst) { - info!("Debug console exited"); - break - } - tokio::time::sleep(Duration::from_secs_f64(0.2)).await; - let option: Option<Pcc> = read_cli( - format!("CLIENT {}/{}> ", - self.target_address.to_string(), - self.bind_player.account.id).as_str(), - "pcc".to_string(), - Pcc::command() - ).await; - match option { - None => {} - Some(cmd) => { - process_debug_cmd(cmd, Arc::clone(&self)); - } - } - } - } - } -}
\ No newline at end of file diff --git a/core/src/pad_service/client_debug_cli.rs b/core/src/pad_service/client_debug_cli.rs deleted file mode 100644 index 988bf39..0000000 --- a/core/src/pad_service/client_debug_cli.rs +++ /dev/null @@ -1,92 +0,0 @@ -use crate::pad_service::client::nogamepads_client::PadClient; -use crate::pad_data::pad_messages::nogamepads_messages::{ControlMessage, GameMessage}; -use clap::{Args, Parser, Subcommand}; -use std::sync::Arc; -use log::info; - -/// NoGamePads Client - Cli -#[derive(Parser, Debug)] -#[command(author, version, about, long_about = None)] -pub struct Pcc { - #[command(subcommand)] - command: Commands, -} - -/// 主要命令 -#[derive(Subcommand, Debug)] -enum Commands { - - // 清屏 - #[command(about = "Clean the screen")] - Clear, - - // 断开当前连接 - #[command(about = "Exit from server")] - Exit, - - // 检查收到的消息 - #[command(about = "Check received")] - Received(ReceivedArgs), - - // 取出一条消息 - #[command(about = "Pop a message")] - Pop(PopArgs), - - // 发送消息 - #[command(about = "Send Message")] - Msg(MsgArgs), -} - -#[derive(Args, Debug)] -struct ReceivedArgs { - - #[arg(long)] - list: bool -} - -/// 发送消息 参数 -#[derive(Args, Debug)] -struct MsgArgs { - - // 消息内容 - #[arg(value_name = "CONTENT")] - message: String, -} - -#[derive(Args, Debug)] -struct PopArgs { } - -pub fn process_debug_cmd (cmd: Pcc, client: Arc<PadClient>) { - match cmd.command { - Commands::Clear => { - clearscreen::clear().expect("Failed to clear screen"); - } - - Commands::Exit => { - client.exit_server(); - } - - Commands::Received(args) => { - if args.list { - for msg in client.list_received() { - info!("{:?}", msg); - } - } else { - info!("Total {} messsage(s)!", client.list_received().iter().count()); - } - } - - Commands::Pop(_args) => { - info!("{:?}", client.pop_msg_or(GameMessage::Err)); - } - - Commands::Msg(args) => { - client.put_msg(ControlMessage::Msg(args.message)); - } - } -} - -#[allow(dead_code)] -fn put_to_list(client: Arc<PadClient>, message: ControlMessage) { - client.put_msg(message); -}
\ No newline at end of file diff --git a/core/src/pad_service/mod.rs b/core/src/pad_service/mod.rs deleted file mode 100644 index ae10f2d..0000000 --- a/core/src/pad_service/mod.rs +++ /dev/null @@ -1,5 +0,0 @@ -pub mod client; -pub mod client_debug_cli; - -pub mod server; -pub mod server_debug_cli;
\ No newline at end of file diff --git a/core/src/pad_service/server.rs b/core/src/pad_service/server.rs deleted file mode 100644 index eeadb01..0000000 --- a/core/src/pad_service/server.rs +++ /dev/null @@ -1,819 +0,0 @@ -pub mod nogamepads_server { - use std::collections::{HashMap, VecDeque}; - use crate::pad_data::pad_messages::nogamepads_message_encoder::NgpdMessageEncoder; - use crate::pad_data::pad_messages::nogamepads_message_transfer::{read_msg, send_msg}; - use crate::pad_data::pad_messages::nogamepads_messages::{ConnectionCallbackMessage, ConnectionMessage, ControlMessage, GameMessage, LeaveReason}; - use log::{error, info, warn}; - use std::net::{IpAddr, Ipv4Addr, SocketAddr}; - use std::ops::Deref; - use std::process::exit; - use std::sync::atomic::AtomicBool; - use std::sync::atomic::Ordering::SeqCst; - use std::sync::{Arc, Mutex, MutexGuard, PoisonError}; - use std::time::Duration; - use clap::CommandFactory; - use clearscreen::clear; - use prettytable::{Row, Table}; - use tokio::io::{AsyncReadExt, AsyncWriteExt, ReadHalf, WriteHalf}; - use tokio::net::{TcpListener, TcpStream}; - use tokio::{io, signal, spawn}; - use tokio::runtime::Runtime; - use nogamepads::console_utils::debug_console::read_cli; - use nogamepads::logger_utils::logger_build; - use crate::DEFAULT_PORT; - use crate::pad_data::game_profile::game_profile::GameProfile; - use crate::pad_data::layout::layout_data::{LayoutKeyRegisters, LayoutKeyRuntimeData}; - use crate::pad_data::pad_messages::nogamepads_messages::ConnectionErrorType::{ContainSamePlayer, GameLocked, PlayerBanned, WhatTheHell}; - use crate::pad_data::pad_messages::nogamepads_messages::GameMessage::Leave; - use crate::pad_data::pad_messages::nogamepads_messages::LeaveReason::ServerClosed; - use crate::pad_data::pad_player_info::nogamepads_player_info::PlayerInfo; - use crate::pad_service::server_debug_cli::{process_debug_cmd, Psc}; - - type PlayerMap = Arc<Mutex<HashMap<String, PlayerInfo>>>; - type WriteList = Arc<Mutex<HashMap<String, VecDeque<GameMessage>>>>; - - #[repr(C)] - pub struct PadServer { - - // --- 主要参数 --- - - // 本地监听地址 - address: IpAddr, - - // 游戏信息 - game_profile: GameProfile, - - // 绑定端口 - port: u16, - - // 调试模式 - enable_console: bool, - - // 保持安静,不初始化 env_logger - quiet: bool, - - // 键位表 - keys: LayoutKeyRegisters, - - // --- 运行时参数 --- - - - // 发送信息列表 - write_list: WriteList, - - // 运行环境信息 - control_data: Arc<Mutex<LayoutKeyRuntimeData>>, - - // 在线玩家 - online_players: PlayerMap, - - // 被封禁的玩家 - banned_players: PlayerMap, - - // 是否锁定该游戏:禁止后续玩家加入 - game_locked: AtomicBool, - - // 是否停止服务器 - stop: AtomicBool, - - // 调试模式 - 实时信息显示模式 - monitor_view: AtomicBool, - } - - impl Clone for PadServer { - fn clone(&self) -> Self { - PadServer { - address: self.address.clone(), - game_profile: self.game_profile.clone(), - port: self.port.clone(), - enable_console: self.enable_console, - quiet: self.quiet, - keys: self.keys.clone(), - - write_list: self.write_list.clone(), - control_data: self.control_data.clone(), - online_players: self.online_players.clone(), - banned_players: self.banned_players.clone(), - game_locked: AtomicBool::new((&self.game_locked.load(SeqCst)).clone()), - stop: AtomicBool::new((&self.stop.load(SeqCst)).clone()), - monitor_view: AtomicBool::new((&self.monitor_view.load(SeqCst)).clone()) - } - } - } - - impl Default for PadServer { - fn default() -> Self { - PadServer { - address: IpAddr::from(Ipv4Addr::new(127, 0, 0, 1)), - game_profile: GameProfile::default(), - port: DEFAULT_PORT, - enable_console: false, - quiet: false, - keys: LayoutKeyRegisters::default(), - - write_list: WriteList::default(), - control_data: Arc::new(Mutex::new(LayoutKeyRuntimeData::default())), - online_players: PlayerMap::default(), - banned_players: PlayerMap::default(), - game_locked: AtomicBool::new(false), - stop: AtomicBool::new(false), - monitor_view: AtomicBool::new(false), - } - } - } - - // 服务端构建部分 - impl PadServer { - - pub fn build_simple() -> Arc<PadServer> { - Arc::new(Self::default() - .addr(IpAddr::from(Ipv4Addr::new(127, 0, 0, 1)), DEFAULT_PORT) - .put_profile(GameProfile::default()).to_owned()) - } - - pub fn addr(&mut self, ip_addr: IpAddr, port: u16) -> &mut PadServer { - self.ip_addr(ip_addr).port(port) - } - - pub fn socket_addr(&mut self, socket_addr: SocketAddr) -> &mut PadServer { - self.ip_addr(socket_addr.ip()).port(socket_addr.port()) - } - - pub fn port(&mut self, port: u16) -> &mut PadServer { - self.port = port; - self - } - - pub fn ip_addr(&mut self, ip_addr: IpAddr) -> &mut PadServer { - self.address = ip_addr; - self - } - - pub fn put_profile(&mut self, profile: GameProfile) -> &mut PadServer { - self.game_profile = profile; - self - } - - pub fn enable_console(&mut self) -> &mut PadServer { - self.enable_console = true; - self - } - - pub fn quiet(&mut self) -> &mut PadServer { - self.quiet = true; - self - } - - pub fn register_keys(&mut self, keys: LayoutKeyRegisters) -> &mut PadServer { - self.keys = keys; - self - } - - pub fn register_button(&mut self, key: u8, name: &str) -> &mut PadServer { - self.keys.button_keys.insert(key, name.to_string()); - self - } - - pub fn register_axis(&mut self, key: u8, name: &str) -> &mut PadServer { - self.keys.axis_keys.insert(key, name.to_string()); - self - } - - pub fn register_direction(&mut self, key: u8, name: &str) -> &mut PadServer { - self.keys.direction_keys.insert(key, name.to_string()); - self - } - - pub fn build(&self) -> Arc<PadServer> { - Arc::new(self.clone()) - } - } - - // 服务端消息管理 - impl PadServer { - - pub fn put_msg_to(&self, msg: GameMessage, player: &PlayerInfo) { - match self.write_list.lock() { - Ok(mut guard) => { - let hash = &player.account.player_hash.clone(); - if ! guard.contains_key(hash.as_str()) { - guard.entry(player.account.player_hash.clone()) - .or_insert_with(VecDeque::new) - .push_back(msg); - } - } - Err(_) => { - error!("Cannot lock \"{:?}\" in write_list", player.account.player_hash); - } - } - } - - pub fn put_msg_to_all(&self, msg: &GameMessage) { - match self.list_players() { - Ok(list) => { - for player in list { - self.put_msg_to(msg.clone(), &player); - } - } - Err(_) => { - error!("Cannot put GameMessage with no players."); - } - } - } - - pub fn pop_a_msg(&self) -> Option<(PlayerInfo, ControlMessage)> { - match self.control_data.lock() { - Ok(mut guard) => { - guard.pop_control_event(self) - } - Err(_) => { - None - } - } - } - } - - // 操控信息管理 - impl PadServer { - pub fn get_player_direction(&self, player: &PlayerInfo, key: &u8) -> Option<(f64, f64)> { - match self.control_data.lock() { - Ok(guard) => { - guard.get_direction(player, &key) - } - Err(_) => { - None - } - } - } - - pub fn get_player_axis(&self, player: &PlayerInfo, key: &u8) -> Option<f64> { - match self.control_data.lock() { - Ok(guard) => { - guard.get_axis(player, &key) - } - Err(_) => { - None - } - } - } - - pub fn get_player_button_status(&self, player: &PlayerInfo, key: &u8) -> Option<bool> { - match self.control_data.lock() { - Ok(guard) => { - guard.get_button_status(player, &key) - } - Err(_) => { - None - } - } - } - } - - // 服务端玩家管理 - impl PadServer { - - pub fn is_player_online (&self, player: &PlayerInfo) -> bool { - let guard = self.online_players.lock().unwrap(); - guard.contains_key(&player.account.player_hash) - } - - fn set_player_online (&self, player: &PlayerInfo, online: bool) { - let online_current = self.is_player_online(player); - if online_current && !online { - let mut guard = self.online_players.lock().unwrap(); - guard.remove(&player.account.player_hash); - info!("{} is OFFLINE!", player.account.id); - } else if !online_current && online { - let mut guard = self.online_players.lock().unwrap(); - guard.insert(player.account.player_hash.clone(), player.clone()); - info!("{} is ONLINE!", player.account.id); - } - } - - pub fn is_player_banned (&self, player: &PlayerInfo) -> bool { - let guard = self.banned_players.lock().unwrap(); - guard.contains_key(&player.account.player_hash) - } - - pub fn kick_player(&self, player: &PlayerInfo) { - if self.is_player_online(player) { - self.put_msg_to(Leave(LeaveReason::YouAreKicked), player); - } - } - - pub fn ban_player(&self, player: &PlayerInfo) { - self.set_player_banned(player, true); - if self.is_player_online(player) { - self.put_msg_to(Leave(LeaveReason::YouAreBanned), player); - } - } - - pub fn pardon_player(&self, player: &PlayerInfo) { - self.set_player_banned(player, false); - } - - fn set_player_banned (&self, player: &PlayerInfo, banned: bool) { - let banned_current = self.is_player_banned(player); - if banned_current && !banned { - let mut guard = self.banned_players.lock().unwrap(); - guard.remove(&player.account.player_hash); - info!("Pardoned player {}", player.account.id); - } else if !banned_current && banned { - let mut guard = self.banned_players.lock().unwrap(); - guard.insert(player.account.player_hash.clone(), player.clone()); - info!("Banned player {}!", player.account.id); - } - } - - pub fn list_players(&self) -> Result<Vec<PlayerInfo>, PoisonError<MutexGuard<HashMap<String, PlayerInfo>>>> { - match self.online_players.lock() { - Ok(guard) => { - Ok(guard.values().cloned().collect()) - } - Err(err) => Err(err) - } - } - - pub fn list_players_banned(&self) -> Result<Vec<PlayerInfo>, PoisonError<MutexGuard<HashMap<String, PlayerInfo>>>> { - match self.banned_players.lock() { - Ok(guard) => { - Ok(guard.values().cloned().collect()) - } - Err(err) => Err(err) - } - } - - pub fn find_online_player(&self, hash: String) -> Option<PlayerInfo> { - match self.online_players.lock() { - Ok(guard) => { - guard.get(&hash).cloned() - } - Err(_) => None - } - } - } - - // 服务端状态控制 - #[allow(dead_code)] - impl PadServer { - - pub fn stop_server(&self) { - self.put_msg_to_all(&Leave(ServerClosed)); - self.stop.store(true, SeqCst); - } - - pub fn start_server(self: Arc<Self>) { - - // 构建 Logger - if ! self.quiet { - logger_build(); - } - - // 运行时 - let runtime = Self::get_runtime(); - - info!("Starting \"NoGamepads Server\"."); - - // 入口 - let console = self.enable_console; - let entry = self.get_entry(console); - - // 阻塞运行 - runtime.block_on(entry); - } - - fn get_runtime() -> Runtime { - tokio::runtime::Builder::new_multi_thread() - .thread_name("nogpad-server") - .thread_stack_size(32 * 1024 * 1024) - .enable_time() - .enable_io() - .build() - .unwrap() - } - - fn get_entry(self: Arc<Self>, debug: bool) -> impl Future<Output = ()> + Send + 'static { - async move { - let main_thread = spawn({ - let client = Arc::clone(&self); - async move { - Self::main_request_thread(client).await - } - }); - - let background_thread = spawn({ - let client = Arc::clone(&self); - async move { - Self::background_thread(client).await - } - }); - - let ctrl_c_thread = spawn({ - let client = Arc::clone(&self); - async move { - Self::process_ctrl_c(client).await - } - }); - - if debug { - let debug_cli = spawn({ - let client = Arc::clone(&self); - async move { - Self::process_debug_cli(client).await - } - }); - - let _ = tokio::join!(ctrl_c_thread, debug_cli, main_thread, background_thread); - } else { - let _ = tokio::join!(ctrl_c_thread, main_thread, background_thread); - } - } - } - - pub fn lock_game(&self) { - self.game_locked.store(true, SeqCst); - } - - pub fn unlock_game(&self) { - self.game_locked.store(false, SeqCst); - } - - pub fn is_game_locked(&self) -> bool { - self.game_locked.load(SeqCst) - } - - pub fn enter_monitor(&self) { - self.monitor_view.store(true, SeqCst); - } - - pub fn exit_monitor(&self) { - self.monitor_view.store(false, SeqCst); - } - - async fn main_request_thread(self: Arc<Self>) { - - let addr_str = format!("{}:{}", self.address.to_string(), self.port); - info!("Server listening at {}", addr_str); - - // Tcp 监听器 - let listener : TcpListener; - match TcpListener::bind(&addr_str).await { - Ok(result) => { - info!("Listener created."); - listener = result; - } - Err(_) => { - error!("Server listening at {} failed!", addr_str); - exit(1); - } - } - - // 请求信息循环 - loop { - match listener.accept().await { - Ok((stream, _)) => { - spawn(Self::process_request(Arc::clone(&self), stream)); - } - Err(error) => { - error!("Error: {}", error); - } - } - } - } - - async fn process_request(self: Arc<Self>, mut stream: TcpStream) { - let mut buffer = [0; 1024]; - let connection_msg : ConnectionMessage = read_msg(&mut buffer, &mut stream).await; - match connection_msg { - - // 客户端请求加入游戏,并建立长连接 - ConnectionMessage::Connection(info) => { - - // 加入游戏资格检测 - info!("Account {} trying to connect.", info.account.player_hash); - - // 0. 当前游戏是否已经锁定? - if self.is_game_locked() { - // 当前游戏已经锁定,禁止加入玩家,发送失败信息,并断开连接 - send_msg(&mut stream, ConnectionCallbackMessage::Deny(GameLocked)).await; - return; - } - - // 1. 是否存在重复玩家? - let online = self.is_player_online(&info); - if online { - // 当前玩家已在线,发送失败信息,并断开连接 - send_msg(&mut stream, ConnectionCallbackMessage::Deny(ContainSamePlayer)).await; - return; - } - - // 2. 该玩家是否被封禁? - let banned = self.is_player_banned(&info); - if banned { - // 当前玩家已被封禁,发送失败信息,并断开连接 - send_msg(&mut stream, ConnectionCallbackMessage::Deny(PlayerBanned)).await; - return; - } - - // OK!若执行到此处,说明该玩家具有加入资格,Welcome! - - send_msg(&mut stream, ConnectionCallbackMessage::Ok).await; - let callback : ConnectionMessage = read_msg(&mut buffer, &mut stream).await; - - match callback { - // 玩家已就绪,发送 Welcome 信息以邀请该玩家加入游戏 - ConnectionMessage::Ready => { - info!("Player \"{}\" is ready!", info.account.id); - - // 发送 Welcome - send_msg(&mut stream, ConnectionCallbackMessage::Welcome).await; - - // 注册该玩家到在线列表 - self.set_player_online(&info, true); - - // 启动控制循环 - - spawn(Self::long_connection(Arc::clone(&self), stream, info)); - }, - _ => { - send_msg(&mut stream, ConnectionCallbackMessage::Deny(WhatTheHell)).await; // WTH ? - } - } - } - - // 客户端请求获得游戏信息 - ConnectionMessage::RequestProfile => { - // 发送游戏信息到客户端 - send_msg(&mut stream, ConnectionCallbackMessage::Profile(self.game_profile.clone())).await; - } - - // 客户端发来了错误信息 - ConnectionMessage::Err => { - match stream.peer_addr() { - Ok(addr) => { - warn!("Received an error message from {}.", addr.to_string()); - } - Err(_) => { - warn!("Received an error message from unknown pad_service."); - } - } - } - - // 客户端发来了不相干的信息 - _ => { - match stream.peer_addr() { - Ok(addr) => { - warn!("Received unknown connection message from {}.", addr.to_string()); - } - Err(_) => { - warn!("Received unknown connection message from unknown pad_service."); - } - } - } - } - } - - async fn long_connection(self: Arc<Self>, stream: TcpStream, player_info: PlayerInfo) { - let player_info_arc = Arc::new(player_info); - let (reader, writer) = io::split(stream); - spawn(Self::read_task(Arc::clone(&self), reader, Arc::clone(&player_info_arc))); - spawn(Self::write_task(Arc::clone(&self), writer, Arc::clone(&player_info_arc))); - } - - async fn read_task(self: Arc<Self>, - mut reader: ReadHalf<TcpStream>, - player_info: Arc<PlayerInfo>) { - let player_hash = player_info.account.player_hash.clone(); - let mut buf = [0u8; 1024]; - loop { - match reader.read(&mut buf).await { - Ok(0) => break, - Ok(n) => { - let msg = ControlMessage::de(buf[0..n].to_vec()); - match self.control_data.lock() { - Ok(mut guard) => { - info!("{:?} from {}({})", &msg, player_info.customize.nickname, player_info.account.id); - let player_info = self.find_online_player(player_hash.clone()); - if player_info.is_some() { - let player_info = player_info.unwrap(); - guard.insert_control(player_info, msg); - } else { - warn!("Player \"{}\" not found!", player_hash); - } - } - Err(_) => { - } - } - } - Err(e) => { - warn!("Error reading from stream: {}", e); - - self.set_player_online(&player_info, false); - - // 放入一条错误信息到队列,使 write_task 及时发现该玩家离开 - self.put_msg_to(GameMessage::Err, &player_info); - - break; - } - } - } - } - - async fn write_task(self: Arc<Self>, - mut writer: WriteHalf<TcpStream>, - player_info: Arc<PlayerInfo>) { - let player_hash = player_info.account.player_hash.clone(); - let mut exit = false; - loop { - let msg : Option<GameMessage>; - match self.write_list.lock() { - Ok(mut hash_map) => { - if ! hash_map.is_empty() { - match hash_map.get_mut(&player_hash) { - None => { - msg = None; - } - Some(queue) => { - if ! queue.is_empty() { - msg = queue.pop_front(); - } else { - msg = None; - hash_map.remove(&player_hash); - } - } - } - } - else { msg = None; } - } - Err(_) => { - msg = None; - } - } - if msg.is_some() { - let msg = msg.unwrap(); - match &writer.write_all(NgpdMessageEncoder::en(&msg).as_slice()).await { - Ok(_) => { - - info!("Sent {:?} to {}", msg, &player_info.account.id); - } - Err(error) => { - warn!("Sent {:?} to {} failed!", msg, &player_info.account.id); - warn!("{:?}", error); - - exit = true; - } - } - } - if exit { - warn!("Long connection between \"{}\" closed.", &player_info.account.id); - break - } - } - } - - async fn background_thread(self: Arc<Self>) { - loop { - // 退出程序的监听 - if self.stop.load(SeqCst) { - tokio::time::sleep(Duration::from_secs(1)).await; - - info!("Main thread exited."); - exit(0); - } - } - } - - async fn process_ctrl_c(self: Arc<Self>) { - loop { - signal::ctrl_c().await.unwrap(); - if self.monitor_view.load(SeqCst) { - self.exit_monitor(); - } else { - self.stop_server(); - info!("Stopping server..."); - break; - } - } - } - - async fn process_debug_cli(self: Arc<Self>) { - loop { - if self.stop.load(SeqCst) { - info!("Debug console exited"); - return; - } - - if ! self.monitor_view.load(SeqCst) { - tokio::time::sleep(Duration::from_secs_f64(0.2)).await; - - // 控制台模式 - let option: Option<Psc> = read_cli( - format!("SERVER {}> ", self.address.to_string()).as_str(), - "psc".to_string(), - Psc::command() - ).await; - match option { - None => {} - Some(cmd) => { - process_debug_cmd(cmd, Arc::clone(&self)); - } - } - } else { - tokio::time::sleep(Duration::from_secs_f64(0.2)).await; - - // 实时信息模式 - - // 表头文本 - let mut header : Vec<String> = Vec::new(); - header.push("PLAYER \\ KEYS".to_string()); - self.all_keys(|key, name, i|{ - if i == 0 { - header.push(format!("{}(btn:{})", name, key)); - } else if i == 1 { - header.push(format!("{}(dir:{})", name, key)); - } else if i == 2 { - header.push(format!("{}(ax:{})", name, key)); - } - }); - - let mut info_table = Table::new(); - - // 添加表头 - info_table.add_row(Row::from_iter(header)); - - // 玩家 - let players = &self.deref().online_players; - match players.lock() { - Ok(guard) => { - for (_player_hash, player) in guard.iter() { - - // 行 - let mut line = Vec::new(); - line.push(player.account.id.to_string()); - self.all_keys(|key, _, i|{ - if i == 0 { - // 按钮 - let r = self.get_player_button_status(player, key); - if r.is_some() { - let r = r.unwrap(); - if r { - line.push("TRUE".to_string()); - } else { - line.push("FALSE".to_string()); - } - } else { - line.push("UNKNOWN".to_string()); - } - - } else if i == 1 { - // 方向 - let r = self.get_player_direction(player, key); - if r.is_some() { - let r = r.unwrap(); - line.push(format!("{}, {}", r.0, r.1)); - } else { - line.push("UNKNOWN".to_string()); - } - - } else if i == 2 { - // 轴向 - let r = self.get_player_axis(player, key); - if r.is_some() { - let r = r.unwrap(); - line.push(format!("{}", r)); - } else { - line.push("UNKNOWN".to_string()); - } - } - }); - - // 添加行文本 - info_table.add_row(Row::from_iter(line)); - } - } - Err(_) => {} - } - - let result = info_table.to_string(); - let _ = clear(); - println!("[HELP] Ctrl + C to exit monitor.\n{}", result); - } - } - } - - fn all_keys<F>(self: &Arc<Self>, mut f: F) - where F: FnMut(&u8, &String, i32){ - let mut i = 0; - for iter in [ - self.keys.button_keys.iter(), - self.keys.direction_keys.iter(), - self.keys.axis_keys.iter(), - ] { - for (key, name) in iter { - f(key, name, i); - } - i += 1; - } - } - } -}
\ No newline at end of file diff --git a/core/src/pad_service/server_debug_cli.rs b/core/src/pad_service/server_debug_cli.rs deleted file mode 100644 index ac52253..0000000 --- a/core/src/pad_service/server_debug_cli.rs +++ /dev/null @@ -1,212 +0,0 @@ -use std::collections::HashMap; -use crate::pad_data::pad_messages::nogamepads_messages::{GameMessage}; -use crate::pad_service::server::nogamepads_server::PadServer; -use clap::{Args, Parser, Subcommand}; -use std::sync::{Arc, MutexGuard, PoisonError}; -use log::{error, info}; -use crate::pad_data::pad_player_info::nogamepads_player_info::PlayerInfo; - -/// NoGamePads Server - Cli -#[derive(Parser, Debug)] -#[command(author, version, about, long_about = None)] -pub struct Psc { - #[command(subcommand)] - command: Commands, -} - -#[derive(Subcommand, Debug)] -enum Commands { - - // 清屏 - #[command(about = "Clean the screen")] - Clear, - - #[command(about = "Enter monitor")] - Monitor, - - // 关闭服务器 - #[command(about = "Close the server")] - Stop, - - // 展示所有玩家 - #[command(about = "List all online players")] - List, - - // 展示所有封禁的玩家 - #[command(about = "List all banned players")] - Banned, - - // 取出一条消息 - #[command(about = "Pop a message")] - Pop, - - // 踢出玩家 - #[command(about = "Kick a player")] - Kick(PlayerArgs), - - // 封禁玩家 - #[command(about = "Ban a player")] - Ban(PlayerArgs), - - // 解封(赦免)玩家 - #[command(about = "Pardon a player")] - Pardon(PlayerArgs), - - // 锁定游戏 - #[command(about = "Lock this game")] - Lock, - - // 解锁游戏 - #[command(about = "Unlock this game")] - Unlock, - - // 激活事件触发器 - #[command(about = "Send SkinEventTrigger")] - Event(EventArgs) -} - -// 检查收到的消息 -#[derive(Args, Debug)] -struct ReceivedArgs { - - #[arg(default_value = "0")] - player: usize, - - #[arg(long)] - list: bool, -} - -/// 激活事件触发器 参数 -#[derive(Args, Debug)] -struct EventArgs { - - // 玩家序号 - #[arg(value_name = "PLAYER_INDEX")] - index: usize, - - // 事件编号 - #[arg(value_name = "CONTENT")] - message: u8, -} - -#[derive(Args, Debug)] -struct PlayerArgs { - - // 玩家序号 - #[arg(value_name = "PLAYER_INDEX")] - index: usize -} - -pub fn process_debug_cmd (cmd: Psc, server: Arc<PadServer>) { - match cmd.command { - - Commands::Clear => { - clearscreen::clear().expect("Failed to clear screen"); - } - - Commands::Monitor => { - server.enter_monitor(); - } - - Commands::Stop => { - server.stop_server(); - } - - Commands::List => { - print_player_list(server.list_players()); - } - - Commands::Banned => { - print_player_list(server.list_players_banned()); - } - - Commands::Pop => { - let message = server.pop_a_msg(); - if message.is_some() { - let (player, msg) = message.unwrap(); - info!("{:?} from \"{}\"({})", msg, player.customize.nickname, player.account.id); - } - } - - Commands::Kick(args) => { - let player = get_player_by_index(&server, args.index); - if player.is_some() { - let player = player.unwrap(); - server.kick_player(&player); - } - } - - Commands::Ban(args) => { - let player = get_player_by_index(&server, args.index); - if player.is_some() { - let player = player.unwrap(); - server.ban_player(&player); - } - } - - Commands::Pardon(args) => { - let player = get_player_by_ban_index(&server, args.index); - if player.is_some() { - let player = player.unwrap(); - server.pardon_player(&player); - } - } - - Commands::Lock => { - if ! server.is_game_locked() { - server.lock_game(); - info!("Game locked"); - } - } - - Commands::Unlock => { - if server.is_game_locked() { - server.unlock_game(); - info!("Game unlocked"); - } - } - - Commands::Event(args) => { - put_to_list(server, args.index, GameMessage::SkinEventTrigger(args.message)); - } - } -} - -fn put_to_list(server: Arc<PadServer>, player_index: usize, message: GameMessage) { - match get_player_by_index(&server, player_index) { - None => { - error!("Put message failed : Player index \"{}\" not found!", player_index); - } - Some(player) => { - server.put_msg_to(message, &player); - } - } -} - -fn get_player_by_index(server: &Arc<PadServer>, index: usize) -> Option<PlayerInfo> { - let list = server.list_players().unwrap_or(Vec::new()); - let max = list.iter().count(); - let index = if max > 0 { index.clamp(0, max - 1) } else { 0 }; - - let result = list.get(index).cloned(); - result -} - -fn get_player_by_ban_index(server: &Arc<PadServer>, index: usize) -> Option<PlayerInfo> { - let list = server.list_players_banned().unwrap_or(Vec::new()); - let max = list.iter().count(); - let index = if max > 0 { index.clamp(0, max - 1) } else { 0 }; - - let result = list.get(index).cloned(); - result -} - -fn print_player_list(list: Result<Vec<PlayerInfo>, PoisonError<MutexGuard<HashMap<String, PlayerInfo>>>>) { - let list = list.unwrap_or(Vec::new()); - let mut i = 0; - for player in list { - let n = player.customize.nickname; - info!("({}){} ", i, n); - i += 1; - } -}
\ No newline at end of file |
