import Vino.Movement.Capabilities.Standard.CharacterMovementCapability;
import Cake.LevelSpecific.Tree.Wasps.Behaviours.WaspBehaviourComponent;

class UWaspLocalFlyingMovementCapability : UCharacterMovementCapability
{
	default CapabilityTags.Add(n"Flying");
	default CapabilityTags.Add(CapabilityTags::Movement);

	default TickGroup = ECapabilityTickGroups::LastMovement;
	default TickGroupOrder = 20.f;

    UWaspBehaviourComponent BehaviourComp;
    FHazeAcceleratedRotator AcceleratedRotation;
	UWaspComposableSettings Settings;

	UFUNCTION(BlueprintOverride)
	void Setup(FCapabilitySetupParams SetupParams)
	{
		Super::Setup(SetupParams);
        BehaviourComp = UWaspBehaviourComponent::Get(Owner);
		Settings = UWaspComposableSettings::GetSettings(Owner);
	}

	UFUNCTION(BlueprintOverride)
	EHazeNetworkActivation ShouldActivate() const
	{
		if (!MoveComp.CanCalculateMovement())
    		return EHazeNetworkActivation::DontActivate;
		return EHazeNetworkActivation::ActivateLocal;
	}

	UFUNCTION(BlueprintOverride)
	EHazeNetworkDeactivation ShouldDeactivate() const
	{
		if (!MoveComp.CanCalculateMovement())
			return RemoteLocalControlCrumbDeactivation();
		return EHazeNetworkDeactivation::DontDeactivate;
	}
	
	UFUNCTION(BlueprintOverride)
	void TickActive(float DeltaSeconds)
	{
		// Moves independently on remote and control side
		FRotator TargetRotation = (BehaviourComp.MovementDestination - Owner.GetActorLocation()).Rotation(); 
		if (BehaviourComp.bHasFocus)
			TargetRotation = (BehaviourComp.FocusLocation - Owner.GetActorLocation()).Rotation();
		AcceleratedRotation.Value = Owner.GetActorRotation();
		AcceleratedRotation.AccelerateTo(TargetRotation, Settings.FlyingRotationDuration, DeltaSeconds);
		MoveComp.SetTargetFacingRotation(AcceleratedRotation.Value); 
	
		// Behaviour comp needs to have focus set every tick or it will revert to default
		BehaviourComp.bHasFocus = false;
		
		FHazeFrameMovement MoveData = MoveComp.MakeFrameMovement(n"WaspFlyingMovement");
		float Acceleration = BehaviourComp.Acceleration;
		if (Acceleration < 0.f)
			Acceleration = 0.f;

		FVector Destination = BehaviourComp.GetMoveDestination();
		Destination += BehaviourComp.TargetGroundVelocity.Value * DeltaSeconds; // One tick behind

		// Accelerate towards destination with friction 
		// As it's air movement it can be nice to get some oscillation, 
		// so don't use critically dampened spring acceleration. 
		FVector Velocity = MoveComp.GetVelocity();
		FVector ToTargetDir = (Destination - Owner.GetActorLocation()).GetSafeNormal();
		FVector AccleratedVelocity = ToTargetDir * Acceleration;
		FVector Dampening = Velocity * 1.3f;
		Velocity += (AccleratedVelocity - Dampening) * DeltaSeconds;

		MoveData.ApplyVelocity(Velocity);
		MoveData.ApplyTargetRotationDelta();

		if (BehaviourComp.MovementLockHeightStrength > 0.f) 
			MoveData.ApplyDelta(FVector(0.f, 0.f, FMath::Lerp(0.f, BehaviourComp.MovementLockedHeight - Owner.ActorLocation.Z, DeltaSeconds * BehaviourComp.MovementLockHeightStrength)));

		// We expect behaviours to set acceleration each tick, or we will just drift to a stop
		BehaviourComp.Acceleration = 0.f;
		BehaviourComp.MovementLockHeightStrength = 0.f;

		MoveData.OverrideStepDownHeight(0.f);
		MoveData.OverrideGroundedState(EHazeGroundedState::Airborne);
		MoveCharacter(MoveData, FeatureName::AirMovement);
	}
};