From bd704008f238ac4e51ed7a7568529c6f4225cf26 Mon Sep 17 00:00:00 2001
From: WanderingPenwing <nicolas.pinson31@gmail.com>
Date: Mon, 7 Oct 2024 21:27:53 +0200
Subject: [PATCH] better hints

---
 src/explo_state.rs |   9 +++-
 src/game_state.rs  | 110 ++++++++++++++++++++++++++++-----------------
 src/main.rs        |  31 +------------
 3 files changed, 78 insertions(+), 72 deletions(-)

diff --git a/src/explo_state.rs b/src/explo_state.rs
index 10d9ba4..2c8fe43 100644
--- a/src/explo_state.rs
+++ b/src/explo_state.rs
@@ -7,8 +7,15 @@ pub fn player_mouse_move (
     buttons: Res<ButtonInput<MouseButton>>,
     mut player_query: Query<(&mut Player, &Camera, &mut GlobalTransform)>,
     window_query: Query<&Window, With<bevy::window::PrimaryWindow>>,
+    ui_query: Query<&Interaction, With<Button>>,
 ) {
-	info!("-");
+    for interaction in ui_query.iter() {
+        if *interaction == Interaction::Pressed {
+        	// Button clicked
+            return;
+        }
+    }
+    
 	let Ok((mut player, camera, global_transform)) = player_query.get_single_mut() else {
 	    return;
 	};
diff --git a/src/game_state.rs b/src/game_state.rs
index 72c3f73..d6c6dd0 100644
--- a/src/game_state.rs
+++ b/src/game_state.rs
@@ -8,8 +8,6 @@ use crate::MainGame;
 use crate::Sky;
 use crate::Constellation;
 use crate::ConstellationLine;
-use crate::PlayerState;
-use crate::GameData;
 
 use crate::celestial_to_cartesian;
 use crate::spawn_cons_lines;
@@ -30,6 +28,37 @@ pub struct ScoreLabel;
 #[derive(Component)]
 pub struct HintLabel;
 
+#[derive(Resource)]
+pub struct GameData {
+    content: Vec<String>,
+	pub score: usize,
+	health: usize,
+	state: PlayerState,
+	target_cons_name: Option<String>,
+	target_cons_focused: bool,
+}
+
+impl Default for GameData {
+    fn default() -> Self {
+         GameData {
+         	content: vec![],
+   	    	score: 0,
+   	    	health: 3,
+   	    	state: PlayerState::Playing,
+   	    	target_cons_name: None,
+   	    	target_cons_focused: false,
+   	    }
+    }
+}
+
+#[derive(Default, PartialEq, Debug)]
+enum PlayerState {
+	#[default]
+	Playing,
+	Hinted,
+	Answered,
+}
+
 pub fn setup(
 	mut commands: Commands, 
 	mut game_data: ResMut<GameData>,
@@ -185,6 +214,10 @@ pub fn player_interact(
     let Ok(mut player) = player_query.get_single_mut() else {
 		return
     };
+
+    if player.dragging_pos.is_some() {
+    	game_data.target_cons_focused = false;
+    }
   
     if keys.just_pressed(KeyCode::Space) || game_data.target_cons_name.is_none() {
         choose_constellation(&mut player, sky, text_query, button_query, constellation_line_query, commands, game_state, game_data);
@@ -194,62 +227,54 @@ pub fn player_interact(
     if keys.just_pressed(KeyCode::Escape) {
    		game_state.set(GameState::Start);
    	}
-
-    if keys.pressed(KeyCode::KeyI) && game_data.state == PlayerState::Playing {
-    	if let Some(target_cons) = game_data.target_cons_name.clone() {
-    		game_data.state = PlayerState::Hinted;
-			spawn_cons_lines(commands, meshes, materials, sky, target_cons);
-			return
-  		}
+	
+    if keys.pressed(KeyCode::KeyI) {
+    	if game_data.state != PlayerState::Playing {
+    		info!("Invalid state : {:?}", game_data.state);
+    	}
+    	let Some(target_cons) = game_data.target_cons_name.clone() else {
+			return;
+  		};
+  		game_data.state = PlayerState::Hinted;
+		spawn_cons_lines(commands, meshes, materials, sky, target_cons);
+		return;
 	}
 
-    if keys.pressed(KeyCode::KeyR) {
-		if let Some(target_constellation_name) = game_data.target_cons_name.clone() {
-			let mut target_constellation = sky.content[0].clone();
-			for constellation in sky.content.clone() {
-				if constellation.name == target_constellation_name {
-					target_constellation = constellation
-				}
-			}
-			player.target_rotation = Some(constellation_center(target_constellation));
-		}
-    }
-
     if keys.pressed(KeyCode::KeyW) {
-        let target_constellation_name: String = "Ursa Minor".into();
-          	
+		game_data.target_cons_focused = true;
+		let Some(target_constellation_name) = game_data.target_cons_name.clone() else {
+			return;
+		};
 		let mut target_constellation = sky.content[0].clone();
 		for constellation in sky.content.clone() {
 			if constellation.name == target_constellation_name {
 				target_constellation = constellation
 			}
 		}
-		
 		player.target_rotation = Some(constellation_center(target_constellation));
-	}
+    }
 }
 
 pub fn ui_labels(
-    mut param_set: ParamSet<(
-        Query<&mut Text, With<HealthLabel>>,
-        Query<&mut Text, With<ScoreLabel>>,
-        Query<&mut Text, With<HintLabel>>,
-    )>,
+    mut label_query: Query<(&mut Text, Option<&HealthLabel>, Option<&ScoreLabel>, Option<&HintLabel>)>,
     game_data: Res<GameData>
 ) {
-	if let Ok(mut health_text) = param_set.p0().get_single_mut() {
-		health_text.sections[0].value = "# ".repeat(game_data.health);
-	}
 
-	if let Ok(mut score_text) = param_set.p1().get_single_mut() {
-		score_text.sections[0].value = format!("{}", game_data.score);
-	}
-
-	if let Ok(mut hint_text) = param_set.p2().get_single_mut() {
-		if game_data.state == PlayerState::Answered {
-			hint_text.sections[0].value = "press space to continue".into();
-		} else {
-			hint_text.sections[0].value = "press i to get an hint".into();
+	for (mut text, health_label, score_label, hint_label) in label_query.iter_mut() {
+		if health_label.is_some() {
+			text.sections[0].value = "# ".repeat(game_data.health);
+		} else if score_label.is_some() {
+			text.sections[0].value = format!("{}", game_data.score);
+		} else if hint_label.is_some() {
+			if !game_data.target_cons_focused {
+				text.sections[0].value = "press z to re-center".into();
+			} else if game_data.state == PlayerState::Playing {
+				text.sections[0].value = "press i to get an hint".into();
+			} else if game_data.state == PlayerState::Answered {
+				text.sections[0].value = "press space to continue".into();
+			} else {
+				text.sections[0].value = "guess the constellation".into();
+			}
 		}
 	}
 }
@@ -386,6 +411,7 @@ fn choose_constellation(
     }
 
     game_data.state = PlayerState::Playing;
+    game_data.target_cons_focused = true;
 }
 
 fn constellation_center(target_constellation: Constellation) -> Quat {
diff --git a/src/main.rs b/src/main.rs
index b34e4b6..9760e41 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -10,6 +10,8 @@ mod start_state;
 mod game_state;
 mod explo_state;
 
+use game_state::GameData;
+
 const NORMAL_BUTTON: Color = Color::srgb(0.15, 0.15, 0.15);
 const WRONG_BUTTON: Color = Color::srgb(0.50, 0.15, 0.15);
 const RIGHT_BUTTON: Color = Color::srgb(0.15, 0.50, 0.15);
@@ -109,35 +111,6 @@ struct Player {
 	dragging_pos: Option<Vec2>,
 }
 
-#[derive(Resource)]
-struct GameData {
-    content: Vec<String>,
-	score: usize,
-	health: usize,
-	state: PlayerState,
-	target_cons_name: Option<String>,
-}
-
-impl Default for GameData {
-    fn default() -> Self {
-         GameData {
-         	content: vec![],
-   	    	score: 0,
-   	    	health: 3,
-   	    	state: PlayerState::Playing,
-   	    	target_cons_name: None,
-   	    }
-    }
-}
-
-#[derive(Default, PartialEq)]
-enum PlayerState {
-	#[default]
-	Playing,
-	Hinted,
-	Answered,
-}
-
 #[derive(Clone, Copy, Default, Eq, PartialEq, Debug, Hash, States)]
 enum GameState {
     #[default]