2022-09-20 15:53:53 -05:00
|
|
|
use std::net::TcpListener;
|
|
|
|
use std::thread;
|
2022-09-29 15:28:38 -05:00
|
|
|
use std::sync::{Arc, Mutex};
|
2022-10-03 13:18:40 -05:00
|
|
|
use std::collections::HashMap;
|
|
|
|
|
2022-10-03 14:18:25 -05:00
|
|
|
mod message;
|
|
|
|
use message::Message;
|
|
|
|
|
2022-10-03 14:42:15 -05:00
|
|
|
mod code_generator;
|
|
|
|
use code_generator::CodeGenerator;
|
2022-09-20 15:53:53 -05:00
|
|
|
|
2022-10-03 14:18:25 -05:00
|
|
|
use tungstenite::protocol::Message as WsMessage;
|
2022-09-20 15:53:53 -05:00
|
|
|
|
|
|
|
fn main() {
|
2022-10-03 13:18:40 -05:00
|
|
|
let code_generator = Arc::new(Mutex::new(CodeGenerator::default()));
|
2022-09-20 15:53:53 -05:00
|
|
|
let server = TcpListener::bind("127.0.0.1:8080").unwrap();
|
2022-10-03 13:18:40 -05:00
|
|
|
|
|
|
|
let rooms = Arc::new(Mutex::new(HashMap::new()));
|
|
|
|
|
2022-09-20 15:53:53 -05:00
|
|
|
for stream in server.incoming() {
|
2022-10-03 13:18:40 -05:00
|
|
|
let code_generator = Arc::clone(&code_generator);
|
|
|
|
let rooms = Arc::clone(&rooms);
|
|
|
|
thread::spawn (move || {
|
2022-09-20 15:53:53 -05:00
|
|
|
let mut ws = tungstenite::accept(stream.unwrap()).unwrap();
|
|
|
|
println!("New client!");
|
2022-10-03 14:18:25 -05:00
|
|
|
ws.write_message(WsMessage::Text("HOSTJOIN:".to_string())).unwrap();
|
2022-09-20 15:53:53 -05:00
|
|
|
loop {
|
|
|
|
let message = ws.read_message();
|
|
|
|
println!("{:?}", message);
|
2022-09-29 15:28:38 -05:00
|
|
|
match message {
|
|
|
|
Err(_) => break,
|
2022-10-03 14:18:25 -05:00
|
|
|
Ok(WsMessage::Close(_)) => break,
|
|
|
|
Ok(WsMessage::Text(msg)) => {
|
|
|
|
let msg = Message::parse(msg.as_str().trim());
|
|
|
|
match msg {
|
|
|
|
Ok(msg) =>
|
|
|
|
match msg.command {
|
|
|
|
"HOST" => {
|
|
|
|
let code = code_generator.lock().unwrap().generate();
|
|
|
|
let mut room = Room::default();
|
|
|
|
let player = Player { name: "Guest".to_string() };
|
|
|
|
room.players.push(player);
|
|
|
|
rooms.lock().unwrap().insert(code.clone(), room);
|
|
|
|
ws.write_message(WsMessage::Text(code)).unwrap();
|
|
|
|
}
|
|
|
|
_ => todo!(),
|
|
|
|
}
|
|
|
|
Err(_) => todo!(),
|
|
|
|
}
|
|
|
|
}
|
2022-09-29 15:28:38 -05:00
|
|
|
_ => unimplemented!(),
|
2022-09-20 15:53:53 -05:00
|
|
|
}
|
|
|
|
}
|
2022-10-03 13:18:40 -05:00
|
|
|
});
|
2022-09-20 15:53:53 -05:00
|
|
|
}
|
|
|
|
}
|
2022-10-03 13:18:40 -05:00
|
|
|
#[derive(Default)]
|
2022-09-20 15:53:53 -05:00
|
|
|
struct Room {
|
|
|
|
players: Vec<Player>,
|
|
|
|
}
|
|
|
|
|
|
|
|
struct Player {
|
|
|
|
name: String,
|
|
|
|
}
|