MovementTests/addons/forge/resources/ForgeEffectData.cs

// Copyright © Gamesmiths Guild.

using System.Collections.Generic;
using System.Diagnostics;
using Gamesmiths.Forge.Core;
using Gamesmiths.Forge.Cues;
using Gamesmiths.Forge.Effects;
using Gamesmiths.Forge.Effects.Calculator;
using Gamesmiths.Forge.Effects.Components;
using Gamesmiths.Forge.Effects.Duration;
using Gamesmiths.Forge.Effects.Magnitudes;
using Gamesmiths.Forge.Effects.Modifiers;
using Gamesmiths.Forge.Effects.Periodic;
using Gamesmiths.Forge.Effects.Stacking;
using Gamesmiths.Forge.Godot.Resources.Calculators;
using Gamesmiths.Forge.Godot.Resources.Components;
using Gamesmiths.Forge.Godot.Resources.Magnitudes;
using Godot;
using Godot.Collections;

namespace Gamesmiths.Forge.Godot.Resources;

[Tool]
[GlobalClass]
[Icon("uid://obsk7rrtq1xd")]
public partial class ForgeEffectData : Resource
{
	private EffectData? _data;

	private DurationType _durationType;
	private bool _hasPeriodicApplication;
	private bool _canStack;

	private StackPolicy _sourcePolicy;
	private StackLevelPolicy _levelPolicy;
	private StackOwnerOverridePolicy _instigatorOverridePolicy;
	private LevelComparison _levelOverridePolicy;

	[Export]
	public string Name { get; set; } = string.Empty;

	[Export]
	public bool SnapshotLevel { get; set; } = true;

	[ExportGroup("Modifier Data")]

	[Export(PropertyHint.ResourceType, "ForgeModifier")]
	public Array<ForgeModifier>? Modifiers { get; set; }

	[ExportGroup("Components")]

	[Export(PropertyHint.ResourceType, "ForgeEffectComponent")]
	public Array<ForgeEffectComponent>? Components { get; set; }

	[ExportGroup("Executions")]

	[Export(PropertyHint.ResourceType, "ForgeCustomExecution")]
	public Array<ForgeCustomExecution>? Executions { get; set; }

	[ExportGroup("Duration Data")]
	[Export]
	public DurationType DurationType
	{
		get => _durationType;

		set
		{
			_durationType = value;

			if (value == DurationType.Instant)
			{
				_hasPeriodicApplication = false;
				_canStack = false;
			}

			NotifyPropertyListChanged();
		}
	}

	[Export]
	public ForgeModifierMagnitude? Duration { get; set; }

	[ExportGroup("Periodic Data")]

	[Export]
	public bool HasPeriodicApplication
	{
		get => _hasPeriodicApplication;

		set
		{
			_hasPeriodicApplication = value;
			NotifyPropertyListChanged();
		}
	}

	[Export]
	public ForgeScalableFloat? Period { get; set; }

	[Export]
	public bool ExecuteOnApplication { get; set; }

	[ExportGroup("Stacking Data")]
	[Export]
	public bool CanStack
	{
		get => _canStack;

		set
		{
			_canStack = value;
			NotifyPropertyListChanged();
		}
	}

	[Export]
	public ForgeScalableInt StackLimit { get; set; } = new(1);

	[Export]
	public ForgeScalableInt InitialStack { get; set; } = new(1);

	[Export]
	public StackPolicy SourcePolicy
	{
		get => _sourcePolicy;

		set
		{
			_sourcePolicy = value;
			NotifyPropertyListChanged();
		}
	}

	[Export]
	[ExportSubgroup("Aggregate by Target", "Instigator")]
	public StackOwnerDenialPolicy InstigatorDenialPolicy { get; set; }

	[Export]
	public StackOwnerOverridePolicy InstigatorOverridePolicy
	{
		get => _instigatorOverridePolicy;

		set
		{
			_instigatorOverridePolicy = value;
			NotifyPropertyListChanged();
		}
	}

	[Export]
	public StackOwnerOverrideStackCountPolicy InstigatorOverrideStackCountPolicy { get; set; }

	[Export]
	public StackLevelPolicy LevelPolicy
	{
		get => _levelPolicy;

		set
		{
			_levelPolicy = value;
			NotifyPropertyListChanged();
		}
	}

	[Export]
	[ExportSubgroup("Aggregate Levels", "Level")]
	public LevelComparison LevelDenialPolicy { get; set; }

	[Export]
	public LevelComparison LevelOverridePolicy
	{
		get => _levelOverridePolicy;

		set
		{
			_levelOverridePolicy = value;
			NotifyPropertyListChanged();
		}
	}

	[Export]
	public StackLevelOverrideStackCountPolicy LevelOverrideStackCountPolicy { get; set; }

	[Export]
	public StackMagnitudePolicy MagnitudePolicy { get; set; }

	[Export]
	public StackOverflowPolicy OverflowPolicy { get; set; }

	[Export]
	public StackExpirationPolicy ExpirationPolicy { get; set; }

	[Export]
	[ExportSubgroup("Has Duration")]
	public StackApplicationRefreshPolicy ApplicationRefreshPolicy { get; set; }

	[Export]
	[ExportSubgroup("Periodic Effects")]
	public StackApplicationResetPeriodPolicy ApplicationResetPeriodPolicy { get; set; }

	[Export]
	public bool ExecuteOnSuccessfulApplication { get; set; }

	[ExportGroup("Cues")]
	[Export(PropertyHint.ResourceType, "ForgeCue")]
	public Array<ForgeCue>? Cues { get; set; }

	[Export]
	public CueTriggerRequirement RequireModifierSuccessToTriggerCue { get; set; }

	[Export]
	public bool SuppressStackingCues { get; set; }

	public EffectData GetEffectData()
	{
		if (_data.HasValue)
		{
			return _data.Value;
		}

		Modifiers ??= [];
		Components ??= [];
		Executions ??= [];
		Cues ??= [];

		var modifiers = new List<Modifier>();
		foreach (ForgeModifier modifier in Modifiers)
		{
			modifiers.Add(modifier.GetModifier());
		}

		var components = new List<IEffectComponent>();
		foreach (ForgeEffectComponent component in Components)
		{
			components.Add(component.GetComponent());
		}

		var executions = new List<CustomExecution>();
		foreach (ForgeCustomExecution execution in Executions)
		{
			executions.Add(execution.GetExecutionClass());
		}

		var cues = new List<CueData>();
		foreach (ForgeCue cue in Cues)
		{
			cues.Add(cue.GetCueData());
		}

		Debug.Assert(Name is not null, $"{nameof(Duration)} is not set.");

		_data = new EffectData(
			Name,
			GetDurationData(),
			[.. modifiers],
			GetStackingData(),
			GetPeriodicData(),
			SnapshotLevel,
			[.. components],
			RequireModifierSuccessToTriggerCue,
			SuppressStackingCues,
			[.. executions],
			[.. cues]);

		return _data.Value;
	}

#if TOOLS
	public override void _ValidateProperty(Dictionary property)
	{
		if (property["name"].AsStringName() == PropertyName.SnapshotLevel)
		{
			if (DurationType == DurationType.Instant)
			{
				property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
				SnapshotLevel = true;
			}

			// This is used to sure Modifiers are set as snapshot for instant effects.
			if (Modifiers is not null)
			{
				foreach (ForgeModifier modifier in Modifiers)
				{
					modifier.IsInstantEffect = DurationType == DurationType.Instant;
				}
			}
		}

		if (property["name"].AsStringName() == PropertyName.Duration && DurationType != DurationType.HasDuration)
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}

		if (DurationType == DurationType.Instant
			&& property["name"].AsStringName() == PropertyName.HasPeriodicApplication)
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}

		if (!HasPeriodicApplication
			&& (property["name"].AsStringName() == PropertyName.Period ||
				property["name"].AsStringName() == PropertyName.ExecuteOnApplication ||
				property["name"].AsStringName() == PropertyName.ApplicationResetPeriodPolicy ||
				property["name"].AsStringName() == PropertyName.ExecuteOnSuccessfulApplication))
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}

		if ((DurationType == DurationType.Instant || !CanStack)
			&& (property["name"].AsStringName() == PropertyName.StackLimit ||
				property["name"].AsStringName() == PropertyName.InitialStack ||
				property["name"].AsStringName() == PropertyName.SourcePolicy ||
				property["name"].AsStringName() == PropertyName.InstigatorDenialPolicy ||
				property["name"].AsStringName() == PropertyName.InstigatorOverridePolicy ||
				property["name"].AsStringName() == PropertyName.InstigatorOverrideStackCountPolicy ||
				property["name"].AsStringName() == PropertyName.LevelPolicy ||
				property["name"].AsStringName() == PropertyName.LevelDenialPolicy ||
				property["name"].AsStringName() == PropertyName.LevelOverridePolicy ||
				property["name"].AsStringName() == PropertyName.LevelOverrideStackCountPolicy ||
				property["name"].AsStringName() == PropertyName.MagnitudePolicy ||
				property["name"].AsStringName() == PropertyName.OverflowPolicy ||
				property["name"].AsStringName() == PropertyName.ExpirationPolicy ||
				property["name"].AsStringName() == PropertyName.ApplicationRefreshPolicy ||
				property["name"].AsStringName() == PropertyName.ApplicationResetPeriodPolicy ||
				property["name"].AsStringName() == PropertyName.ExecuteOnSuccessfulApplication))
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}

		if (SourcePolicy == StackPolicy.AggregateBySource
			&& (property["name"].AsStringName() == PropertyName.InstigatorDenialPolicy ||
				property["name"].AsStringName() == PropertyName.InstigatorOverridePolicy ||
				property["name"].AsStringName() == PropertyName.InstigatorOverrideStackCountPolicy))
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}

		if (InstigatorOverridePolicy != StackOwnerOverridePolicy.Override
			&& property["name"].AsStringName() == PropertyName.InstigatorOverrideStackCountPolicy)
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}

		if (LevelPolicy == StackLevelPolicy.SegregateLevels
			&& (property["name"].AsStringName() == PropertyName.LevelDenialPolicy ||
				property["name"].AsStringName() == PropertyName.LevelOverridePolicy ||
				property["name"].AsStringName() == PropertyName.LevelOverrideStackCountPolicy))
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}

		if (LevelOverridePolicy == 0 && property["name"].AsStringName() == PropertyName.LevelOverrideStackCountPolicy)
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}

		if (DurationType != DurationType.HasDuration &&
			property["name"].AsStringName() == PropertyName.ApplicationRefreshPolicy)
		{
			property["usage"] = (int)(PropertyUsageFlags.Default | PropertyUsageFlags.ReadOnly);
		}
	}
#endif

	private DurationData GetDurationData()
	{
		return new DurationData(DurationType, GetDurationModifier());
	}

	private ModifierMagnitude? GetDurationModifier()
	{
		if (DurationType != DurationType.HasDuration)
		{
			return null;
		}

		Debug.Assert(Duration is not null, $"{nameof(Duration)} reference is missing.");

		return Duration.GetModifier();
	}

	private StackingData? GetStackingData()
	{
		if (!CanStack)
		{
			return null;
		}

		return new StackingData(
			StackLimit.GetScalableInt(),
			InitialStack.GetScalableInt(),
			SourcePolicy,
			LevelPolicy,
			MagnitudePolicy,
			OverflowPolicy,
			ExpirationPolicy,
			GetOwnerDenialPolicy(),
			GetOwnerOverridePolicy(),
			GetOwnerOverrideStackCountPolicy(),
			GetLevelDenialPolicy(),
			GetLevelOverridePolicy(),
			GetLevelOverrideStackCountPolicy(),
			GetApplicationRefreshPolicy(),
			GetApplicationResetPeriodPolicy(),
			GetExecuteOnSuccessfulApplication());
	}

	private StackOwnerDenialPolicy? GetOwnerDenialPolicy()
	{
		if (SourcePolicy != StackPolicy.AggregateByTarget)
		{
			return null;
		}

		return InstigatorDenialPolicy;
	}

	private StackOwnerOverridePolicy? GetOwnerOverridePolicy()
	{
		if (SourcePolicy != StackPolicy.AggregateByTarget)
		{
			return null;
		}

		return InstigatorOverridePolicy;
	}

	private StackOwnerOverrideStackCountPolicy? GetOwnerOverrideStackCountPolicy()
	{
		if (SourcePolicy != StackPolicy.AggregateByTarget ||
			InstigatorOverridePolicy != StackOwnerOverridePolicy.Override)
		{
			return null;
		}

		return InstigatorOverrideStackCountPolicy;
	}

	private LevelComparison? GetLevelDenialPolicy()
	{
		if (LevelPolicy != StackLevelPolicy.AggregateLevels)
		{
			return null;
		}

		return LevelDenialPolicy;
	}

	private LevelComparison? GetLevelOverridePolicy()
	{
		if (LevelPolicy != StackLevelPolicy.AggregateLevels)
		{
			return null;
		}

		return LevelOverridePolicy;
	}

	private StackLevelOverrideStackCountPolicy? GetLevelOverrideStackCountPolicy()
	{
		if (LevelPolicy != StackLevelPolicy.AggregateLevels ||
			LevelOverridePolicy == 0)
		{
			return null;
		}

		return LevelOverrideStackCountPolicy;
	}

	private StackApplicationRefreshPolicy? GetApplicationRefreshPolicy()
	{
		if (DurationType != DurationType.HasDuration)
		{
			return null;
		}

		return ApplicationRefreshPolicy;
	}

	private StackApplicationResetPeriodPolicy? GetApplicationResetPeriodPolicy()
	{
		if (!HasPeriodicApplication)
		{
			return null;
		}

		return ApplicationResetPeriodPolicy;
	}

	private bool? GetExecuteOnSuccessfulApplication()
	{
		if (!HasPeriodicApplication)
		{
			return null;
		}

		return ExecuteOnSuccessfulApplication;
	}

	private PeriodicData? GetPeriodicData()
	{
		if (!HasPeriodicApplication)
		{
			return null;
		}

		Debug.Assert(Period is not null, $"{nameof(Period)} reference is missing.");

		return new PeriodicData(
			Period.GetScalableFloat(),
			ExecuteOnApplication,
			PeriodInhibitionRemovedPolicy.NeverReset);
	}
}