geometry-tower-defense/Assets/GameMain/Scripts/UI/Game/Controller/RepoFormController.cs

using System.Collections.Generic;
using GeometryTD.CustomEvent;
using GeometryTD.Definition;
using GameFramework.Event;
using UnityGameFramework.Runtime;

namespace GeometryTD.UI
{
    public class RepoFormController : UIFormControllerCommonBase<RepoFormContext, RepoForm>
    {
        private RepoFormUseCase _useCase;
        private readonly Dictionary<long, RepoItemContext> _itemContextMap = new Dictionary<long, RepoItemContext>();

        protected override UIFormType UIFormTypeId => UIFormType.RepoForm;

        protected override void RefreshUI(RepoForm form, RepoFormContext context)
        {
            form.RefreshUI(context);
        }

        protected override void SubscribeCustomEvents()
        {
            GameEntry.Event.Subscribe(RepoItemSelectedEventArgs.EventId, OnRepoItemSelected);
            GameEntry.Event.Subscribe(CombineSlotClickedEventArgs.EventId, OnCombineSlotClicked);
        }

        protected override void UnsubscribeCustomEvents()
        {
            GameEntry.Event.Unsubscribe(RepoItemSelectedEventArgs.EventId, OnRepoItemSelected);
            GameEntry.Event.Unsubscribe(CombineSlotClickedEventArgs.EventId, OnCombineSlotClicked);
        }

        public override int? OpenUI(object userData = null)
        {
            if (userData is RepoFormContext repoFormContext)
            {
                return OpenUIInternal(repoFormContext);
            }

            if (userData is RepoFormRawData rawDataFromUserData)
            {
                return OpenUI(rawDataFromUserData);
            }

            if (userData != null)
            {
                Log.Warning("RepoFormController.OpenUI() userData type is invalid.");
                return null;
            }

            if (_useCase == null)
            {
                Log.Error("RepoFormController.OpenUI() useCase is null.");
                return null;
            }

            RepoFormRawData rawData = _useCase.CreateInitialModel();
            return OpenUI(rawData);
        }

        public int? OpenUI(RepoFormRawData rawData)
        {
            RepoFormContext context = BuildContext(rawData);
            return OpenUIInternal(context);
        }

        public override void BindUseCase(IUIUseCase useCase)
        {
            if (!(useCase is RepoFormUseCase repoFormUseCase))
            {
                Log.Error("RepoFormController.BindUseCase() useCase is invalid.");
                return;
            }

            _useCase = repoFormUseCase;
        }

        private RepoFormContext BuildContext(RepoFormRawData rawData)
        {
            _itemContextMap.Clear();
            if (rawData?.Inventory == null)
            {
                return null;
            }

            List<RepoItemContext> items = new List<RepoItemContext>();

            if (rawData.Inventory.Towers != null)
            {
                foreach (var tower in rawData.Inventory.Towers)
                {
                    AddItemContext(items, new RepoItemContext
                    {
                        Title = $"[Tower] {tower.Name}  ({tower.Rarity})",
                        InstanceId = tower.InstanceId,
                        ComponentSlotType = TowerCompSlotType.None,
                        IconAreaContext = BuildIconAreaContext(tower.Rarity)
                    });
                }
            }

            if (rawData.Inventory.MuzzleComponents != null)
            {
                foreach (var item in rawData.Inventory.MuzzleComponents)
                {
                    AddItemContext(items, new RepoItemContext
                    {
                        Title = $"[Muzzle] {item.Name}  ({item.Rarity})",
                        InstanceId = item.InstanceId,
                        ComponentSlotType = TowerCompSlotType.Muzzle,
                        IconAreaContext = BuildIconAreaContext(item.Rarity)
                    });
                }
            }

            if (rawData.Inventory.BearingComponents != null)
            {
                foreach (var item in rawData.Inventory.BearingComponents)
                {
                    AddItemContext(items, new RepoItemContext
                    {
                        Title = $"[Bearing] {item.Name}  ({item.Rarity})",
                        InstanceId = item.InstanceId,
                        ComponentSlotType = TowerCompSlotType.Bearing,
                        IconAreaContext = BuildIconAreaContext(item.Rarity)
                    });
                }
            }

            if (rawData.Inventory.BaseComponents != null)
            {
                foreach (var item in rawData.Inventory.BaseComponents)
                {
                    AddItemContext(items, new RepoItemContext
                    {
                        Title = $"[Base] {item.Name}  ({item.Rarity})",
                        InstanceId = item.InstanceId,
                        ComponentSlotType = TowerCompSlotType.Base,
                        IconAreaContext = BuildIconAreaContext(item.Rarity)
                    });
                }
            }

            return new RepoFormContext
            {
                CombineAreaContext = new CombineAreaContext(),
                CompAreaContext = new CompAreaContext
                {
                    Items = items.ToArray()
                }
            };
        }

        private void AddItemContext(List<RepoItemContext> items, RepoItemContext itemContext)
        {
            if (itemContext == null)
            {
                return;
            }

            items.Add(itemContext);
            _itemContextMap[itemContext.InstanceId] = itemContext;
        }

        private static IconAreaContext BuildIconAreaContext(RarityType rarity)
        {
            return new IconAreaContext
            {
                Rarity = rarity,
                Icon = null
            };
        }

        private void OnRepoItemSelected(object sender, GameEventArgs e)
        {
            if (!(e is RepoItemSelectedEventArgs args))
            {
                return;
            }

            if (Form == null || !_itemContextMap.TryGetValue(args.ItemId, out RepoItemContext itemContext))
            {
                return;
            }

            bool assigned = Form.TryAssignItemToCombineArea(itemContext);
            if (assigned)
            {
                Form.SetRepoItemSelected(args.ItemId, true);
            }
        }

        private void OnCombineSlotClicked(object sender, GameEventArgs e)
        {
            if (!(e is CombineSlotClickedEventArgs args))
            {
                return;
            }

            if (Form == null)
            {
                return;
            }

            if (Form.TryClearCombineSlot(args.SlotIndex, out long removedItemId) && removedItemId > 0)
            {
                Form.SetRepoItemSelected(removedItemId, false);
            }
        }
    }
}