77 lines
1.9 KiB
Rust
77 lines
1.9 KiB
Rust
#![feature(if_let_guard)]
|
|
|
|
use crate::game_handler::*;
|
|
use anyhow::{Context, Result};
|
|
use axum::extract::ws::Message;
|
|
use game_handler::*;
|
|
use lib::*;
|
|
use std::{
|
|
collections::{HashMap, HashSet},
|
|
fs::File,
|
|
io::{BufRead, BufReader},
|
|
net::SocketAddr,
|
|
sync::{Arc, RwLock},
|
|
};
|
|
use tokio::sync::mpsc::Sender;
|
|
use tokio::sync::{broadcast, mpsc};
|
|
use user_handler::*;
|
|
use uuid::Uuid;
|
|
pub mod game_handler;
|
|
pub mod message_handler;
|
|
pub mod user_handler;
|
|
pub mod websocket;
|
|
|
|
/// User
|
|
#[derive(Debug)]
|
|
pub struct User {
|
|
pub uuid: String,
|
|
pub name: String,
|
|
pub tx: Sender<String>,
|
|
}
|
|
|
|
impl User {
|
|
/// Create a new user object from incoming data
|
|
pub fn new(name: String, tx: Sender<String>) -> User {
|
|
User {
|
|
uuid: Uuid::now_v7().to_string(),
|
|
name,
|
|
tx,
|
|
}
|
|
}
|
|
|
|
pub fn change_name(&mut self, new_name: String) {
|
|
self.name = new_name;
|
|
}
|
|
}
|
|
|
|
/// Parse name list
|
|
pub fn load_names(path: &str) -> Result<Vec<String>> {
|
|
let f = File::open(path).with_context(|| format!("Invalid names path: \"{}\"", path))?;
|
|
let f = BufReader::new(f);
|
|
|
|
let mut buf = vec![];
|
|
|
|
for line in f.lines() {
|
|
buf.push(line?)
|
|
}
|
|
|
|
Ok(buf)
|
|
}
|
|
|
|
// Our shared state
|
|
pub struct AppState {
|
|
pub white_cards_by_id: HashMap<String, Arc<CardWhiteWithID>>,
|
|
pub broadcast_tx: broadcast::Sender<String>,
|
|
pub users_tx: mpsc::Sender<UserHandlerMessage>,
|
|
pub messages_tx: mpsc::Sender<(SocketAddr, Message)>,
|
|
pub games_tx: mpsc::Sender<GameHandlerMessage>,
|
|
pub first_names: Vec<String>,
|
|
pub last_names: Vec<String>,
|
|
pub reserved_names: RwLock<HashSet<String>>,
|
|
pub users_by_id: RwLock<HashMap<String, Arc<RwLock<User>>>>,
|
|
pub online_users: RwLock<HashMap<SocketAddr, Arc<RwLock<User>>>>,
|
|
pub offline_users: RwLock<HashMap<String, Arc<RwLock<User>>>>,
|
|
pub packs: CardPacks,
|
|
pub packs_meta: CardPacksMeta,
|
|
pub games: RwLock<HashMap<String, Arc<RwLock<Game>>>>,
|
|
}
|