From 098d30348443c53b321de879071f6b3541c86d91 Mon Sep 17 00:00:00 2001
From: PhantomMorrigan <phantom.morrigan@proton.me>
Date: Fri, 14 Nov 2025 18:12:51 +0100
Subject: [PATCH] The Ghosts have eaten from the Tree of Knowledge, but for
 their greed they were punished with stealing from each other like the little
 shits they are.

---
 src/berries.rs | 23 +++++++++++++++--------
 src/main.rs    | 36 +++++++++++++++++++++++++++---------
 2 files changed, 42 insertions(+), 17 deletions(-)

diff --git a/src/berries.rs b/src/berries.rs
index 3ba3107..2425271 100644
--- a/src/berries.rs
+++ b/src/berries.rs
@@ -6,22 +6,29 @@ use rand::Rng;
 #[derive(Component)]
 pub struct Berry;
 
+#[derive(Message)]
+pub struct NewBerry(pub Entity);
+
 pub fn spawn_berries(
     mut commands: Commands,
     mut timer: Local<Timer>,
     time: Res<Time>,
     asset_server: Res<AssetServer>,
+    mut new_berries: MessageWriter<NewBerry>,
 ) {
-    timer.set_duration(Duration::from_secs(2));
+    timer.set_duration(Duration::from_secs(1));
     timer.tick(time.delta());
     if timer.is_finished() {
-        commands.spawn((
-            Berry,
-            Transform::from_translation(
-                (rand::rng().random::<Vec2>() - Vec2::splat(0.5)).extend(0.0) * 1000.0,
-            ),
-            Sprite::from_image(asset_server.load("berry.png")),
-        ));
+        let new = commands
+            .spawn((
+                Berry,
+                Transform::from_translation(
+                    (rand::rng().random::<Vec2>() - Vec2::splat(0.5)).extend(0.0) * 1000.0,
+                ),
+                Sprite::from_image(asset_server.load("berry.png")),
+            ))
+            .id();
+        new_berries.write(NewBerry(new));
         timer.reset();
     }
 }
diff --git a/src/main.rs b/src/main.rs
index bcc5208..22d4de2 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,7 +3,7 @@
 use bevy::prelude::*;
 use bevy_bae::prelude::*;
 
-use crate::berries::Berry;
+use crate::berries::{Berry, NewBerry};
 
 mod berries;
 
@@ -13,6 +13,7 @@ fn main() {
     App::new()
         .add_plugins((DefaultPlugins, BaePlugin::default()))
         .add_systems(Startup, setup)
+        .add_message::<NewBerry>()
         .add_systems(Update, berries::spawn_berries)
         .run();
 }
@@ -44,17 +45,12 @@ fn setup(mut commands: Commands, asset_server: Res<AssetServer>) {
 }
 
 #[derive(Component)]
-#[relationship(relationship_target = TargetedBerry)]
 pub struct TargetBerry(pub Entity);
 
-#[derive(Component)]
-#[relationship_target(relationship = TargetBerry)]
-pub struct TargetedBerry(Entity);
-
 fn find_closest_berry(
     In(input): In<OperatorInput>,
     mut commands: Commands,
-    berries: Query<(Entity, &Transform), (With<Berry>, Without<TargetedBerry>)>,
+    berries: Query<(Entity, &Transform), With<Berry>>,
     planner: Query<&Transform, With<Plan>>,
 ) -> OperatorStatus {
     let pos = planner.get(input.entity).unwrap().translation.xy();
@@ -78,11 +74,33 @@ fn find_closest_berry(
 fn go_to_berry(
     In(input): In<OperatorInput>,
     mut planners: Query<(&mut Transform, &TargetBerry), With<Plan>>,
-    berries: Query<&Transform, (With<TargetedBerry>, Without<Plan>)>,
+    berries: Query<&Transform, (With<Berry>, Without<Plan>)>,
     time: Res<Time>,
+    mut news: MessageReader<NewBerry>,
+    mut commands: Commands,
 ) -> OperatorStatus {
-    let (mut trans, target_entity) = planners.get_mut(input.entity).unwrap();
+    let Ok((mut trans, target_entity)) = planners.get_mut(input.entity) else {
+        return OperatorStatus::Failure;
+    };
+
     if let Ok(target) = berries.get(target_entity.0) {
+        for new in news.read() {
+            let new_trans = berries.get(new.0).unwrap();
+
+            if new_trans
+                .translation
+                .xy()
+                .distance_squared(trans.translation.xy())
+                < target
+                    .translation
+                    .xy()
+                    .distance_squared(trans.translation.xy())
+            {
+                commands.entity(input.entity).insert(TargetBerry(new.0));
+                return OperatorStatus::Ongoing;
+            }
+        }
+
         let dir = (target.translation.xy() - trans.translation.xy()).normalize();
         let mov = dir * SPEED * time.delta_secs();
         if (target.translation.xy() - (trans.translation.xy() + mov)).length() < mov.length() {
-- 
2.51.2