using BehaviorDesigner.Runtime.Tasks;
using BlueWater.Npcs.Crews;

namespace BlueWater.BehaviorTrees.Actions
{
    [TaskCategory("Custom/Npc/Crew/Cleaner")]
    public class CompleteCleaning : Action
    {
        private CleanerCrew _cleanerCrew;

        public override void OnAwake()
        {
            _cleanerCrew = GetComponent<CleanerCrew>();
        }

        public override TaskStatus OnUpdate()
        {
            if (_cleanerCrew.CrewInteraction == null && _cleanerCrew.IsCleaningFloor == false && _cleanerCrew.IsCleaningTable == false)
            {
                return TaskStatus.Success;
            }
            
            return TaskStatus.Running;
        }
    }
}