kubi/kubi-server/src/auth.rs

use shipyard::{UniqueView, UniqueViewMut};
use kubi_shared::networking::messages::{ClientToServerMessage, ServerToClientMessage, InitData};
use kubi_udp::server::ServerEvent;
use crate::{server::{ServerEvents, UdpServer}, config::ConfigTable, util::log_error};

pub fn authenticate_players(
  mut server: UniqueViewMut<UdpServer>,
  events: UniqueView<ServerEvents>,
  config: UniqueView<ConfigTable>
) {
  for event in &events.0 {
    if let ServerEvent::MessageReceived {
      from, 
      message: ClientToServerMessage::ClientHello {
        username,
        password
      } 
    } = event {
      // Handle password auth
      if let Some(server_password) = &config.server.password {
        if let Some(user_password) = &password {
          if server_password != user_password {
            server.0.send_message(*from, ServerToClientMessage::ServerFuckOff {
              reason: "Passwords don't match".into()
            }).map_err(log_error).ok();
            continue
          }
        } else {
          server.0.send_message(*from, ServerToClientMessage::ServerFuckOff {
            reason: "This server is password-protected".into()
          }).map_err(log_error).ok();
          continue
        }
      }
      //Approve the user
      server.0.send_message(*from, ServerToClientMessage::ServerHello {
        init: InitData {}
      }).map_err(log_error).ok();
    }
  }
}