test/Assets/Scripts/SelecScreenCharacter.cs

using CSNetwork.Protocols.RPCData;
using System;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UI;
using BrewMonster;

namespace BrewMonster.UI
{
    public class SelecScreenCharacter : MonoBehaviour
    {
        [SerializeField] private GameObject characterItemPrefab;
        [SerializeField] private GameObject addCharacterItemPrefab;
        [SerializeField] private RectTransform parentItems;
        [SerializeField] private Button createCharacterButton;
        [SerializeField] private CreateCharacterScreen createCharacterScreen;

        private Action<RoleInfo> _onClickItemChar;
        private Action<RoleInfo> _onCreateCharacterComplete;

        private void Start()
        {
            if (createCharacterButton != null)
            {
                createCharacterButton.onClick.AddListener(OnCreateCharacterClicked);
            }
        }

        public void InitScreen(List<RoleInfo> roleInfos, Action<RoleInfo> OnClickItemChar, Action<RoleInfo> onCreateCharacterComplete = null)
        {
            _onClickItemChar = OnClickItemChar;
            _onCreateCharacterComplete = onCreateCharacterComplete;

            // Clear existing items
            if (parentItems != null)
            {
                foreach (Transform child in parentItems)
                {
                    Destroy(child.gameObject);
                }
            }

            // Create character items
            if (roleInfos != null)
            {
                foreach (RoleInfo info in roleInfos)
                {
                    if (characterItemPrefab != null && parentItems != null)
                    {
                        CharacterItemUI item = Instantiate(characterItemPrefab, parentItems).GetComponent<CharacterItemUI>();
                        item.InitItem(info, OnClickItemChar);
                    }
                }

                // If number of roles < 8, spawn addCharacterItemPrefab and hide createCharacterButton
                if (roleInfos.Count < 8)
                {
                    // Hide the createCharacterButton
                    if (createCharacterButton != null)
                    {
                        createCharacterButton.gameObject.SetActive(false);
                    }

                    // Spawn addCharacterItemPrefab
                    if (addCharacterItemPrefab != null && parentItems != null)
                    {
                        GameObject addCharacterItem = Instantiate(addCharacterItemPrefab, parentItems);
                        // Set up click handler for the add character item
                        Button addButton = addCharacterItem.GetComponent<Button>();
                        if (addButton == null)
                        {
                            addButton = addCharacterItem.GetComponentInChildren<Button>();
                        }
                        if (addButton != null)
                        {
                            addButton.onClick.RemoveAllListeners();
                            addButton.onClick.AddListener(OnCreateCharacterClicked);
                        }
                    }
                }
                else
                {
                    // Show the createCharacterButton if we have 8 or more roles
                    if (createCharacterButton != null)
                    {
                        createCharacterButton.gameObject.SetActive(true);
                    }
                }
            }
            else
            {
                // If roleInfos is null, show createCharacterButton
                if (createCharacterButton != null)
                {
                    createCharacterButton.gameObject.SetActive(true);
                }
            }
        }

        private void OnCreateCharacterClicked()
        {
            if (createCharacterScreen != null)
            {
                gameObject.SetActive(false);
                createCharacterScreen.Show(OnCreateCharacterComplete, OnCreateCharacterCancel);
            }
        }

        private void OnCreateCharacterComplete(RoleInfo newRole)
        {
            if (createCharacterScreen != null)
            {
                createCharacterScreen.Hide();
            }

            gameObject.SetActive(true);
            _onCreateCharacterComplete?.Invoke(newRole);
        }

        private void OnCreateCharacterCancel()
        {
            if (createCharacterScreen != null)
            {
                createCharacterScreen.Hide();
            }

            gameObject.SetActive(true);
        }
    }
}