ProjectDDD/Assets/_DDD/_Scripts/RestaurantController/RestaurantController.cs

using System.Threading.Tasks;
using UnityEngine;

namespace DDD
{
    public class RestaurantController : Singleton<RestaurantController>, IManager, IGameFlowHandler
    {
        public RestaurantEnvironmentStateSo RestaurantEnvironmentStateSo { get; private set; }
        public RestaurantCustomerStateSo RestaurantCustomerStateSo { get; private set; }

        private const string CreateRestaurantPlayerSo = "CreateRestaurantPlayerSo";
        private const string CreateEnvironmentSo = "CreateEnvironmentSo";

        public void PreInit()
        {
            LoadOrCreateRestaurantState();
            RegisterFlowHandler();
        }

        public async Task Init()
        {
            await LoadData();
        }

        public void PostInit()
        {
            GenerateDummyEnvironmentProps();// XXX : DUMMY! REMOVE THIS
        }

        private void RegisterFlowHandler()
        {
            GameFlowManager.Instance.FlowHandlers.Add(this);
        }

        private void LoadOrCreateRestaurantState()
        {
            // TODO : Load states from saved files. if none, create them.
            RestaurantEnvironmentStateSo = ScriptableObject.CreateInstance<RestaurantEnvironmentStateSo>();
        }

        private async Task LoadData()
        {
            RestaurantCustomerStateSo = await AssetManager.LoadAsset<RestaurantCustomerStateSo>(DataConstants.RestaurantCustomerStateSo);
        }

        private void GenerateDummyEnvironmentProps()
        {
            // Make dummy placement data
            foreach (EnvironmentData prop in DataManager.Instance.GetDataSo<EnvironmentDataSo>().GetDataList())
            {
                for (int i = 0; i < 10; i++)
                {
                    // Make random position
                    Vector2 randomPos = new Vector2(
                        Random.Range(-10f, 10f),
                        Random.Range(10f, 20f)
                    );
                    var randomPropData = new RestaurantEnvironmentData(prop.Id, randomPos);
                    RestaurantEnvironmentStateSo.RestaurantEnvironmentProps.Add(randomPropData);
                }
            }
        }


        public async Task OnReadyNewFlow(GameFlowState newFlowState)
        {
            if (newFlowState == GameFlowState.ReadyForRestaurant)
            {
                CreateRestaurantPlayerSo createRestaurantPlayerSoJob = await AssetManager.LoadAsset<CreateRestaurantPlayerSo>(CreateRestaurantPlayerSo);
                //CreateEnvironmentSo createEnvironmentSoJob = await AssetManager.LoadAsset<CreateEnvironmentSo>(CreateEnvironmentSo);
                RestaurantManagementSo restaurantManagementSo = await AssetManager.LoadAsset<RestaurantManagementSo>(DataConstants.RestaurantManagementSo);
                var playerHandle = createRestaurantPlayerSoJob.OnReadyNewFlow(newFlowState);
                //var propHandle = createEnvironmentSoJob.OnReadyNewFlow(newFlowState);
                var todayMenuHandle = restaurantManagementSo.OnReadyNewFlow(newFlowState);
                // Combine handles and return it
                InputManager.Instance.SwitchCurrentActionMap(InputActionMaps.Restaurant);
                await Task.WhenAll(playerHandle, todayMenuHandle);
            }
            else if (newFlowState == GameFlowState.RunRestaurant)
            {
                var restaurantCustomerStateHandle = RestaurantCustomerStateSo.OnReadyNewFlow(newFlowState);

                await Task.WhenAll(restaurantCustomerStateHandle);
            }
        }

        public async Task OnExitCurrentFlow(GameFlowState exitingFlowState)
        {
            if (exitingFlowState == GameFlowState.ReadyForRestaurant)
            {
                
            }
            else if (exitingFlowState == GameFlowState.RunRestaurant)
            {
                var restaurantCustomerStateHandle = RestaurantCustomerStateSo.OnExitCurrentFlow(exitingFlowState);

                await Task.WhenAll(restaurantCustomerStateHandle);
            }
        }
    }
}