BITFALL/Assets/Artists/Scripts/Entities/Equipment/EntityEquipmentContainer.cs

120 lines
3.6 KiB
C#

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using BITFALL.Entities.Equipment;
using BITFALL.Entities.Inventory;
using BITFALL.Player.Inventory;
using UnityEngine;
using BITKit;
using BITKit.Entities;
namespace BITFALL
{
/// <summary>
/// 玩家装备容器
/// 支持,护甲,头盔和背包等
/// </summary>
[CustomType(typeof(IEntityEquipmentContainer))]
public class EntityEquipmentContainer : EntityBehavior, IEntityEquipmentContainer
{
public IDictionary<IEquipmentSlot, IBasicItem> Equipment { get; } =
new Dictionary<IEquipmentSlot, IBasicItem>();
[Inject] private IEntityInventory _inventory;
public override void OnAwake()
{
var health = UnityEntity.Get<IHealth>();
health.OnSetAlive += OnSetAlive;
}
private void OnSetAlive(bool obj)
{
if (Data.Get<bool>(BITConstant.Environment.sp_keepInventory)) return;
if (obj) return;
foreach (var x in Equipment.ToArray())
{
OnDeEquip?.Invoke(x.Key, x.Value);
_inventory.Add(x.Value);
}
Equipment.Clear();
}
public override void OnStart()
{
base.OnStart();
_inventory.TryUseItemFactory += TryExecute;
}
public Action<IEquipmentSlot, IBasicItem> OnEquip { get; set; }
public Action<IEquipmentSlot, IBasicItem> OnDeEquip { get; set; }
public bool TryDeEquip<T>(T slot) where T : IEquipmentSlot
{
if (!Equipment.TryGetAny(x => x.Key.GetType().IsInstanceOfType(slot), out var pair)) return false;
if (_inventory.Add(pair.Value))
{
DeEquip(slot, pair.Value);
}
return false;
}
public bool TryUseEquip<T>(T slot) where T : IEquipmentSlot
{
return TryUseEquip(System.Activator.CreateInstance(typeof(T)) as IEquipmentSlot);
}
public bool TryUseEquip(IEquipmentSlot slot)
{
if (!Equipment.TryGetAny(x => x.Key.GetType().IsInstanceOfType(slot), out var pair)) return false;
if (!Equipment.TryRemove(pair.Key)) return false;
OnDeEquip?.Invoke(pair.Key, pair.Value);
_inventory.UseItem(pair.Value);
if (_inventory.TryGetItem(x => x.AddressablePath == pair.Value.AddressablePath, out var item))
{
Equip(pair.Key, item);
_inventory.Remove(item);
}
return true;
}
private bool Equip(IEquipmentSlot slot, IBasicItem item)
{
if (Equipment.TryAdd(slot, item) is false) return false;
OnEquip?.Invoke(slot, item);
return true;
}
private bool DeEquip(IEquipmentSlot slot, IBasicItem item)
{
if (Equipment.TryRemove(slot) is false) return false;
OnDeEquip?.Invoke(slot, item);
return true;
}
public int Priority => 0;
public bool TryExecute(IBasicItem value)
{
var asset = value.GetAssetable();
//尝试获取可装备信息
if (!asset.TryGetProperty<EquipmentAsSlot>(out var equipmentAsSlot)) return false;
//已装备物品
if (Equipment.TryGetValue(equipmentAsSlot.slot, out _))
{
return false;
}
//装配物品
Equip(equipmentAsSlot.slot, value);
_inventory.Remove(value);
return true;
}
}
}