Cielonos/Assets/Scripts/MainGame/Characters/Player/Inventory/PlayerEquipmentSubmodule.cs

using System.Collections.Generic;
using Cielonos.MainGame.Inventory;
using Cielonos.UI;
using UnityEngine;

namespace Cielonos.MainGame.Characters
{
    public class PlayerEquipmentSubmodule : SubmoduleBase<PlayerInventorySubcontroller>
    {
        public List<MainWeaponBase> preparedMainWeapons;
        public MainWeaponBase currentMainWeapon;
        
        public PlayerEquipmentSubmodule(PlayerInventorySubcontroller owner) : base(owner)
        {
            preparedMainWeapons = new List<MainWeaponBase>();
        }
        
        public void EquipMainWeapon(MainWeaponBase newWeapon)
        {
            newWeapon.Initialize();
            currentMainWeapon = newWeapon;
            currentMainWeapon.OnEquipped();
            currentMainWeapon.RegisterFullBodyFuncAnims();
            
            PlayerCanvas.Instance.mainWeaponUIArea.Initialize(currentMainWeapon);
        }

        public void RemoveMainWeapon()
        {
            currentMainWeapon.OnUnequipped();
            currentMainWeapon = null;
        }
    }
}