From 7605200ae100ea95b7e0493e359763c47d94ff78 Mon Sep 17 00:00:00 2001
From: szymon <szymon_jozef@proton.me>
Date: Wed, 19 Aug 2026 21:14:44 +0200
Subject: [PATCH] feat: add mode selection screen

---
 src/bin/macroquad_client.rs   | 49 +++++++++++++++++++++++++++++------
 src/ui/macroquad/main_menu.rs |  2 +-
 src/ui/macroquad/mod.rs       |  1 +
 src/ui/state.rs               |  5 ++++
 4 files changed, 48 insertions(+), 9 deletions(-)

diff --git a/src/bin/macroquad_client.rs b/src/bin/macroquad_client.rs
index c16894f..5ced9ee 100644
--- a/src/bin/macroquad_client.rs
+++ b/src/bin/macroquad_client.rs
@@ -1,13 +1,16 @@
 use checkers::ui::{
-    macroquad::main_menu::{draw_main_menu, get_main_menu_style},
+    macroquad::{
+        main_menu::{draw_main_menu, get_main_menu_style},
+        mode_selection::draw_mode_selection,
+    },
     state::GuiState,
 };
 use macroquad::{
-    color::{BLUE, RED, WHITE},
+    color::{BLACK, BLUE, RED, WHITE},
     math::vec2,
     miniquad::{self, conf::LinuxBackend::WaylandWithX11Fallback},
     texture::{DrawTextureParams, draw_texture_ex, load_texture},
-    ui::root_ui,
+    ui::{Skin, root_ui},
     window::{Conf, clear_background, next_frame, screen_height, screen_width},
 };
 
@@ -23,6 +26,26 @@ fn window_conf() -> Conf {
     }
 }
 
+pub async fn get_general_style() -> Skin {
+    let label_style = root_ui()
+        .style_builder()
+        .text_color(WHITE)
+        .font_size(64)
+        .build();
+
+    let button_style = root_ui()
+        .style_builder()
+        .text_color(BLACK)
+        .font_size(32)
+        .build();
+
+    Skin {
+        label_style,
+        button_style,
+        ..root_ui().default_skin()
+    }
+}
+
 #[macroquad::main(window_conf)]
 pub async fn main() {
     env_logger::init();
@@ -30,7 +53,7 @@ pub async fn main() {
     let mut state = Default::default();
     let background = load_texture("assets/background.png").await.unwrap();
 
-    let main_menu_style = get_main_menu_style().await;
+    let main_menu_style = get_general_style().await;
     root_ui().push_skin(&main_menu_style);
 
     loop {
@@ -50,13 +73,23 @@ pub async fn main() {
                 draw_main_menu(&mut state).await;
             }
 
+            GuiState::ModeSelection => {
+                draw_mode_selection(&mut state).await;
+            }
+
+            GuiState::DificultySelection => {
+                todo!();
+            }
+
+            GuiState::ServerSelection => {
+                todo!();
+            }
+
             GuiState::Settings => {
-                clear_background(RED);
-                next_frame().await;
+                todo!();
             }
             GuiState::Game => {
-                clear_background(BLUE);
-                next_frame().await;
+                todo!();
             }
 
             GuiState::Exit => {
diff --git a/src/ui/macroquad/main_menu.rs b/src/ui/macroquad/main_menu.rs
index ad88bc5..c60d940 100644
--- a/src/ui/macroquad/main_menu.rs
+++ b/src/ui/macroquad/main_menu.rs
@@ -45,7 +45,7 @@ pub async fn draw_main_menu(state: &mut GuiState) {
             menu_builder.label(ui, "Main menu");
 
             if menu_builder.button(ui, "Play") {
-                *state = GuiState::Game;
+                *state = GuiState::ModeSelection;
             }
 
             if menu_builder.button(ui, "Settings") {
diff --git a/src/ui/macroquad/mod.rs b/src/ui/macroquad/mod.rs
index 85282d5..024a4de 100644
--- a/src/ui/macroquad/mod.rs
+++ b/src/ui/macroquad/mod.rs
@@ -1,2 +1,3 @@
 pub mod main_menu;
+pub mod mode_selection;
 mod menu_builder;
diff --git a/src/ui/state.rs b/src/ui/state.rs
index 1161859..72ed73e 100644
--- a/src/ui/state.rs
+++ b/src/ui/state.rs
@@ -2,6 +2,11 @@
 pub enum GuiState {
     #[default]
     MainMenu,
+
+    ModeSelection,
+    DificultySelection,
+    ServerSelection,
+
     Settings,
     Game,
     Exit,