joshuafhiggins/connect4-moderator-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Releases 2 Activity

feat(demo mode): allow for random color assignments

Browse Source
This commit is contained in:
Joshua Higgins
2026-01-02 17:26:30 -05:00
Unverified
parent 2b1ff2f98b
commit 8ad6b0868d
4 changed files with 98 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/lib.rs
View File

@@ -23,6 +23,9 @@ pub type Observers = Arc<RwLock<HashMap<SocketAddr, UnboundedSender<Message>>>>;
pub type Matches = Arc<RwLock<HashMap<u32, Arc<RwLock<Match>>>>>;
pub type WrappedTournament = Arc<RwLock<Option<Arc<RwLock<dyn Tournament + Send + Sync>>>>>;
pub const SERVER_PLAYER_USERNAME: &str = "The Server";
pub const SERVER_PLAYER_ADDR: &str = "127.0.0.1:6666";
pub async fn broadcast_message(observers: &Observers, addrs: &Vec<SocketAddr>, msg: &str) {
for addr in addrs {
let observers_guard = observers.read().await;

8
src/main.rs
View File

@@ -193,7 +193,9 @@ async fn handle_connection(
}
"GET" => {
if let Some(data_id) = parts.get(1) {
if let Err(e) = sd.handle_get_data(tx.clone(), data_id.to_string()).await {
if let Err(e) =
sd.handle_get_data(tx.clone(), data_id.to_string()).await
{
error!("handle_get_data: {}", e);
let _ = send(&tx, e.to_string().as_str());
}
@@ -205,7 +207,9 @@ async fn handle_connection(
if parts.len() > 2 {
let data_id = parts[1].to_string();
let data_value = parts[2].to_string();
if let Err(e) = sd.handle_set_data(tx.clone(), addr, data_id, data_value).await {
if let Err(e) =
sd.handle_set_data(tx.clone(), addr, data_id, data_value).await
{
error!("handle_set_data: {}", e);
let _ = send(&tx, e.to_string().as_str());
}

56
src/server.rs
View File

@@ -41,6 +41,13 @@ impl Server {
)));
}
if requested_username == SERVER_PLAYER_USERNAME {
return Err(anyhow::anyhow!(format!(
"ERROR:INVALID:ID:{}",
requested_username
)));
}
let clients_guard = self.clients.read().await;
for client in clients_guard.values() {
if requested_username == client.read().await.username {
@@ -129,14 +136,19 @@ impl Server {
let new_match = Arc::new(RwLock::new(Match::new(
match_id,
addr.to_string().parse()?,
addr.to_string().parse()?,
SERVER_PLAYER_ADDR.to_string().parse()?,
is_demo_mode,
)));
self.matches.write().await.insert(match_id, new_match.clone());
client.ready = false;
client.current_match = Some(match_id);
client.color = Color::Red;
client.color = if new_match.read().await.player1 == addr {
let _ = send(&tx, "GAME:START:1");
Color::Red
} else {
let _ = send(&tx, "GAME:START:0");
Color::Yellow
};
}
Ok(())
@@ -305,6 +317,7 @@ impl Server {
client.color = Color::None;
drop(client);
if !is_demo_mode {
let mut opponent = clients_guard.get(&opponent_addr).unwrap().write().await;
if opponent.color == winner {
opponent.score += 1;
@@ -312,9 +325,11 @@ impl Server {
opponent.current_match = None;
opponent.color = Color::None;
drop(opponent);
}
matches_guard.remove(&current_match_id).unwrap();
if !is_demo_mode && matches_guard.is_empty() {
if self.tournament.read().await.is_some() && matches_guard.is_empty() {
drop(matches_guard);
drop(clients_guard);
@@ -324,6 +339,11 @@ impl Server {
if tourney.read().await.is_completed() {
*tournament_guard = None;
}
} else if self.tournament.read().await.is_none() {
let _ = send(&tx, "TOURNAMENT:END");
if !is_demo_mode {
let _ = send(&opponent_connection, "TOURNAMENT:END");
}
}
return Ok(());
@@ -342,8 +362,8 @@ impl Server {
if current_match.demo_mode {
let move_to_dispatch = current_match.move_to_dispatch.clone();
current_match.ledger.push(move_to_dispatch);
current_match.move_to_dispatch = (Color::Yellow, column_to_use);
current_match.place_token(Color::Yellow, column_to_use);
current_match.move_to_dispatch = (!client.color, column_to_use);
current_match.place_token(!client.color, column_to_use);
}
let waiting =
@@ -452,12 +472,19 @@ impl Server {
let clients_guard = self.clients.read().await;
let matches_guard = self.matches.read().await;
let the_match = matches_guard.get(&match_id).unwrap().read().await;
let player1 = clients_guard.get(&the_match.player1).unwrap().read().await.username.clone();
let mut player2 =
clients_guard.get(&the_match.player2).unwrap().read().await.username.clone();
if the_match.demo_mode {
player2 = "demo".to_string();
}
let player1 = if !the_match.player1.to_string().eq(SERVER_PLAYER_ADDR) {
clients_guard.get(&the_match.player1).unwrap().read().await.username.clone()
} else {
SERVER_PLAYER_USERNAME.to_string()
};
let player2 = if !the_match.player2.to_string().eq(SERVER_PLAYER_ADDR) {
clients_guard.get(&the_match.player2).unwrap().read().await.username.clone()
} else {
SERVER_PLAYER_USERNAME.to_string()
};
let ledger = the_match.ledger.clone();
drop(clients_guard);
@@ -733,20 +760,19 @@ impl Server {
self.broadcast_message(&the_match.viewers, "GAME:TERMINATED").await;
let clients_guard = self.clients.read().await;
if the_match.player1 != SERVER_PLAYER_ADDR.to_string().parse().unwrap() {
let mut player1 = clients_guard.get(&the_match.player1).unwrap().write().await;
let _ = send(&player1.connection, "GAME:TERMINATED");
player1.current_match = None;
player1.color = Color::None;
drop(player1);
}
if !the_match.demo_mode {
if the_match.player2 != SERVER_PLAYER_ADDR.to_string().parse().unwrap() {
let mut player2 = clients_guard.get(&the_match.player2).unwrap().write().await;
let _ = send(&player2.connection, "GAME:TERMINATED");
player2.current_match = None;
player2.color = Color::None;
drop(player2);
}
drop(clients_guard);
drop(the_match);

16
src/types.rs
View File

@@ -1,16 +1,28 @@
use rand::Rng;
use std::net::SocketAddr;
use std::vec;
use std::{ops, vec};
use tokio::sync::mpsc::UnboundedSender;
use tokio_tungstenite::tungstenite::Message;
#[derive(PartialEq, Clone)]
#[derive(PartialEq, Clone, Copy)]
pub enum Color {
Red,
Yellow,
None,
}
impl ops::Not for Color {
type Output = Color;
fn not(self) -> Color {
match self {
Color::Red => Color::Yellow,
Color::Yellow => Color::Red,
Color::None => Color::None,
}
}
}
#[derive(Clone)]
pub struct Client {
pub username: String,