rude/src/game/state/login.rs

use chrono::Utc;
use log::warn;
use mio::Token;

use crate::command::Command;
use crate::game::{Game, PlayerCheck};
use crate::player::Player;
use crate::queue::SendQueue;
use crate::state::*;

impl Game {
	pub fn login(&mut self, token: Token, message: String, state: &Login) -> SendQueue {
		let mut send_queue = SendQueue::new();

		let mut client = if let Some(client) = self.clients.remove(&token) {
			client
		} else {
			warn!("Can't find a client with token: {:?}", &token);
			self.tokens.push_back(token);
			return send_queue;
		};

		match state {
			// get the username
			Login::Username => {
				if message.is_empty() {
					send_queue.push(token, "\n\nUsername: ", false, None);
				} else {
					match self.check_player_name(message) {
						PlayerCheck::Ok(name) => match self.db.find_player_by_name(&name) {
							Ok(Some(_)) => {
								send_queue.push(token, "\nPassword: ", false, None);
								client.state = State::Login(Login::Password(name));
							}
							Ok(None) => {
								send_queue.push(
									token,
									format!("\nCreate {}? [y/N]: ", name),
									false,
									None,
								);
								client.state = State::Login(Login::CreateUser(name));
							}
							Err(_) => {
								send_queue.push(token, "\nError\n\nUsername: ", false, None);
							}
						},
						PlayerCheck::Err(err) => {
							send_queue.push(token, "\nInvalid username:\n", false, None);
							for line in err {
								send_queue.push(token, format!("{}\n", line), false, None);
							}
							send_queue.push(token, "\n\nUsername: ", false, None);
						}
					};
				}
			}

			// username not found
			Login::CreateUser(username) => {
				if !message.clone().is_empty() && message == "y" {
					send_queue.push(token, "\nNew password: ", false, None);
					client.state = State::Login(Login::CreatePassword(username.to_owned()));
				} else {
					send_queue.push(token, "\n\nUsername: ", false, None);
					client.state = State::Login(Login::Username);
				}
			}

			// first new user password
			Login::CreatePassword(username) => {
				if message.is_empty() {
					send_queue.push(token, "\n\nUsername: ", false, None);
					client.state = State::Login(Login::Username);
				} else {
					send_queue.push(token, "\nNew password again: ", false, None);
					client.state =
						State::Login(Login::CreatePassword2((username.to_owned(), message)));
				}
			}

			Login::CreatePassword2((username, pass)) => {
				let pass = pass.to_owned();
				if message.is_empty() || message != pass {
					send_queue.push(token, "\n\nUsername: ", false, None);
					client.state = State::Login(Login::Username);
				} else {
					if let Ok(id) = self.db.new_player_id() {
						let player = Player {
							id,
							name: username.clone(),
							password: pass,
							created: Utc::now(),
							location: self.config.player.starting_location.clone(),
						};

						if self.db.single_save_player(token, &player).is_ok() {
							send_queue.push(token, format!("Welcome, {}\n", username), false, None);
							send_queue.push(token, "", true, None);

							send_queue.append(&mut Command::dispatch_look(
								&Command::default(),
								String::new(),
								token,
								&mut self.db,
							));

							client.state = State::Action;
						} else {
							send_queue.push(token, "Error", true, None);
						}
					} else {
						send_queue.push(token, "Error", true, None);
					}
				}
			}

			Login::Password(username) => {
				if message.is_empty() {
					send_queue.push(token, "\n\nUsername: ", false, None);
					client.state = State::Login(Login::Username);
				} else {
					match self.db.find_player_by_name(username) {
						Ok(Some(player)) => {
							if message == player.password {
								if self.db.save_connected_player(token, &player).is_ok() {
									send_queue.push(
										token,
										format!("Welcome back, {}\n\n", username),
										false,
										None,
									);
									client.state = State::Action;

									send_queue.append(&mut Command::dispatch_look(
										&Command::default(),
										String::new(),
										token,
										&mut self.db,
									));
								} else {
									send_queue.push(token, "Unable to login\n", false, None);
									send_queue.push(token, "\n\nUsername: ", false, None);
									client.state = State::Login(Login::Username);
								}
							} else {
								send_queue.push(token, "Incorrect password\n", false, None);
								send_queue.push(token, "\n\nUsername: ", false, None);
								client.state = State::Login(Login::Username);
							}
						}
						Ok(None) | Err(_) => {
							send_queue.push(token, "Error\n", false, None);
							send_queue.push(token, "\n\nUsername: ", false, None);
							client.state = State::Login(Login::Username);
						}
					}
				}
			}
		};

		self.clients.insert(token, client);
		send_queue
	}
}