geometry-tower-defense/Assets/GameMain/Scripts/Entity/EntityData/MapData.cs

using System;
using System.Collections.Generic;
using GeometryTD.Definition;
using GeometryTD.CustomUtility;
using UnityEngine;

namespace GeometryTD.Entity.EntityData
{
    [Serializable]
    public class MapData : EntityDataBase
    {
        [SerializeField] private int _levelId = 0;
        [SerializeField] private int _initialCoin = 0;
        [SerializeField] private TowerStatsData[] _buildTowerStatsSnapshot = Array.Empty<TowerStatsData>();

        [NonSerialized] private Func<int, bool> _tryConsumeCoin;
        [NonSerialized] private Action<int> _addCoin;

        public MapData(int entityId, int levelId, Vector3 position) : this(entityId, 0, levelId, position)
        {
        }

        public MapData(int entityId, int typeId, int levelId, Vector3 position) : base(entityId, typeId)
        {
            _levelId = levelId;
            Position = position;
        }

        public MapData(
            int entityId,
            int typeId,
            int levelId,
            Vector3 position,
            int initialCoin,
            IReadOnlyList<TowerStatsData> buildTowerStatsSnapshot,
            Func<int, bool> tryConsumeCoin,
            Action<int> addCoin) : base(entityId, typeId)
        {
            _levelId = levelId;
            Position = position;
            _initialCoin = Mathf.Max(0, initialCoin);
            SetBuildTowerStatsSnapshot(buildTowerStatsSnapshot);
            _tryConsumeCoin = tryConsumeCoin;
            _addCoin = addCoin;
        }

        public int LevelId
        {
            get => _levelId;
            set => _levelId = value;
        }

        public int InitialCoin
        {
            get => _initialCoin;
            set => _initialCoin = Mathf.Max(0, value);
        }

        public int CurrentBuildTowerCount => _buildTowerStatsSnapshot != null ? _buildTowerStatsSnapshot.Length : 0;

        public void BindCombatCallbacks(Func<int, bool> tryConsumeCoin, Action<int> addCoin)
        {
            _tryConsumeCoin = tryConsumeCoin;
            _addCoin = addCoin;
        }

        public bool TryConsumeCoin(int coin)
        {
            int requiredCoin = Mathf.Max(0, coin);
            if (requiredCoin <= 0)
            {
                return true;
            }

            return _tryConsumeCoin != null && _tryConsumeCoin.Invoke(requiredCoin);
        }

        public void AddCoin(int coin)
        {
            int amount = Mathf.Max(0, coin);
            if (amount <= 0)
            {
                return;
            }

            _addCoin?.Invoke(amount);
        }

        public bool TryGetBuildTowerStats(int buildIndex, out TowerStatsData stats)
        {
            stats = null;
            if (_buildTowerStatsSnapshot == null || buildIndex < 0 || buildIndex >= _buildTowerStatsSnapshot.Length)
            {
                return false;
            }

            TowerStatsData sourceStats = _buildTowerStatsSnapshot[buildIndex];
            if (sourceStats == null)
            {
                return false;
            }

            stats = InventoryCloneUtility.CloneTowerStats(sourceStats);
            return stats != null;
        }

        public void SetBuildTowerStatsSnapshot(IReadOnlyList<TowerStatsData> buildTowerStatsSnapshot)
        {
            if (buildTowerStatsSnapshot == null || buildTowerStatsSnapshot.Count <= 0)
            {
                _buildTowerStatsSnapshot = Array.Empty<TowerStatsData>();
                return;
            }

            _buildTowerStatsSnapshot = new TowerStatsData[buildTowerStatsSnapshot.Count];
            for (int i = 0; i < buildTowerStatsSnapshot.Count; i++)
            {
                _buildTowerStatsSnapshot[i] = InventoryCloneUtility.CloneTowerStats(buildTowerStatsSnapshot[i]);
            }
        }

        public MapData CloneForEntity(int entityId, Vector3 position)
        {
            return new MapData(
                entityId,
                TypeId,
                _levelId,
                position,
                _initialCoin,
                _buildTowerStatsSnapshot,
                _tryConsumeCoin,
                _addCoin);
        }
    }
}