Ludo-3D/Assets/Scripts/UI/Pages/Screens/GameOverScreen.cs

using System;
using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UI;

public class GameOverScreen : ScreenBase
{
    [SerializeField] private Button playAgainBtn;
    [SerializeField] private Button mainMenuBtn;
    
    private GameModeHandler gameModeHandler;

    private void OnEnable()
    {
        playAgainBtn.onClick.AddListener(OnPlayAgainClicked);
        mainMenuBtn.onClick.AddListener(OnMainMenuClicked);
    }

    private void OnDisable()
    {
        playAgainBtn.onClick.RemoveAllListeners();
        mainMenuBtn.onClick.RemoveAllListeners();
    }

    public void OnPlayAgainClicked()
    {
        HideScreen();
    }

    private void HideScreen()
    {
        screenManager.HideScreen(screenType);
    }

    public void OnMainMenuClicked()
    {
        HideScreen();

        screenManager = screenManager == null ? InterfaceManager.Instance.GetInterfaceInstance<ScreenManager>() : screenManager;
        screenManager.ShowScreen(ScreenType.MenuScreen);
        screenManager.ShowScreen(ScreenType.MainMenuScreen);
    }
}