Add distinction between managed awake and managed start

Assets/Scripts/DamianExperiments/Picnic/PicnicBehaviour.cs

@@ -1,11 +1,12 @@
 using UnityEngine;
-using Pixelplacement;
+using System;
 using System.Collections;
 using Core;
+using Core.Lifecycle;
 using Core.SaveLoad;
 using UnityEngine.Audio;
 
-public class PicnicBehaviour : MonoBehaviour
+public class PicnicBehaviour : ManagedBehaviour
 {
     [Header("Random Call Settings")]
     public float getDistractedMin = 2f;
@@ -17,7 +18,8 @@ public class PicnicBehaviour : MonoBehaviour
     private Animator animator;
 
     [Header("The FakeChocolate to destroy!")]
-    [SerializeField] private GameObject fakeChocolate; // Assign in Inspector
+    [SerializeField] private GameObject fakeChocolate;
+    [SerializeField] private GameObject realChocolate;
 
     private AppleAudioSource _audioSource;
     public AudioResource distractedAudioClips;
@@ -25,32 +27,44 @@ public class PicnicBehaviour : MonoBehaviour
     public AudioResource feederClips;
     public AudioResource moanerClips;
 
-    // Start is called once before the first execution of Update after the MonoBehaviour is created
-    void Start()
-    {
-        StartCoroutine(StateCycleRoutine());
-    }
+    // Save system configuration
+    public override bool AutoRegisterForSave => true;
 
-    void Awake()
+    // Runtime state tracking
+    private bool _fakeChocolateDestroyed;
+
+    protected override void OnManagedAwake()
     {
         stateMachine = GetComponent<AppleMachine>();
         animator = GetComponent<Animator>();
         _audioSource = GetComponent<AppleAudioSource>();
     }
 
+    protected override void OnSceneRestoreCompleted()
+    {
+        if (_fakeChocolateDestroyed)
+        {
+            DestroyChocolateObjects();
+        }
+        else
+        {
+            StartCoroutine(StateCycleRoutine());
+        }
+    }
+
     private IEnumerator StateCycleRoutine()
     {
         while (true)
         {
             // Distracted state
-            float distractedWait = Random.Range(getDistractedMin, getDistractedMax);
+            float distractedWait = UnityEngine.Random.Range(getDistractedMin, getDistractedMax);
             stateMachine.ChangeState("Picnic PPL Distracted");
             animator.SetBool("theyDistracted", true);
             _audioSource.Stop();
             yield return new WaitForSeconds(distractedWait);
 
             // Chilling state
-            float chillingWait = Random.Range(getFlirtyMin, getFlirtyMax);
+            float chillingWait = UnityEngine.Random.Range(getFlirtyMin, getFlirtyMax);
             stateMachine.ChangeState("Picnic PPL Chilling");
             animator.SetBool("theyDistracted", false);
             _audioSource.Stop();
@@ -58,27 +72,33 @@ public class PicnicBehaviour : MonoBehaviour
         }
     }
 
-    void StopAudio()
-    {
-       _audioSource.Stop();  
-    }
-
     public void triedToStealChocolate()
     {
         _audioSource.Stop();
         animator.SetTrigger("theyAngry");
-        //stateMachine.ChangeState("Picnic PPL Angry");
         Logging.Debug("Hey! Don't steal my chocolate!");
         _audioSource.audioSource.resource = angryAudioClips;
         _audioSource.Play(0);
     }
 
     public void destroyFakeChocolate()
+    {
+        _fakeChocolateDestroyed = true;
+        Destroy(fakeChocolate);
+        Destroy(realChocolate);
+    }
+
+    private void DestroyChocolateObjects()
     {
         if (fakeChocolate != null)
         {
             Destroy(fakeChocolate);
-            fakeChocolate = null; // Optional: clear reference
+            fakeChocolate = null;
+        }
+        
+        if (realChocolate != null)
+        {
+            realChocolate.SetActive(true);
         }
     }
 
@@ -100,5 +120,33 @@ public class PicnicBehaviour : MonoBehaviour
         _audioSource.Play(0);
     }
 
+    protected override string OnSceneSaveRequested()
+    {
+        var state = new PicnicBehaviourState { fakeChocolateDestroyed = _fakeChocolateDestroyed };
+        return JsonUtility.ToJson(state);
+    }
 
+    protected override void OnSceneRestoreRequested(string serializedData)
+    {
+        if (string.IsNullOrEmpty(serializedData)) return;
+
+        try
+        {
+            var state = JsonUtility.FromJson<PicnicBehaviourState>(serializedData);
+            if (state != null)
+            {
+                _fakeChocolateDestroyed = state.fakeChocolateDestroyed;
+            }
+        }
+        catch (Exception ex)
+        {
+            Debug.LogWarning($"[PicnicBehaviour] Failed to restore state: {ex.Message}");
+        }
+    }
+}
+
+[Serializable]
+public class PicnicBehaviourState
+{
+    public bool fakeChocolateDestroyed;
 }