#region Copyright & License Information
/*
 * Copyright 2007-2013 The OpenRA Developers (see AUTHORS)
 * This file is part of OpenRA, which is free software. It is made
 * available to you under the terms of the GNU General Public License
 * as published by the Free Software Foundation. For more information,
 * see COPYING.
 */
#endregion

using System;
using System.Collections.Generic;
using System.Linq;
using OpenRA.Traits;
using XRandom = OpenRA.Thirdparty.Random;

namespace OpenRA.Mods.RA.AI
{
	public enum SquadType { Assault, Air, Rush, Protection }

	public class Squad
	{
		public List<Actor> units = new List<Actor>();
		public SquadType type;

		internal World world;
		internal HackyAI bot;
		internal XRandom random;

		internal Actor target;
		internal StateMachine fsm;

		//fuzzy
		internal AttackOrFleeFuzzy attackOrFleeFuzzy = new AttackOrFleeFuzzy();

		public Squad(HackyAI bot, SquadType type) : this(bot, type, null) { }

		public Squad(HackyAI bot, SquadType type, Actor target)
		{
			this.bot = bot;
			this.world = bot.world;
			this.random = bot.random;
			this.type = type;
			this.target = target;
			fsm = new StateMachine(this);

			switch (type)
			{
				case SquadType.Assault:
				case SquadType.Rush:
					fsm.ChangeState(new GroundUnitsIdleState(), true);
					break;
				case SquadType.Air:
					fsm.ChangeState(new AirIdleState(), true);
					break;
				case SquadType.Protection:
					fsm.ChangeState(new UnitsForProtectionIdleState(), true);
					break;
			}
		}

		public void Update()
		{
			if (IsEmpty) return;
			fsm.UpdateFsm();
		}

		public bool IsEmpty
		{
			get { return !units.Any(); }
		}

		public Actor Target
		{
			get { return target; }
			set { target = value; }
		}

		public bool TargetIsValid
		{
			get { return (target != null && !target.IsDead() && !target.Destroyed
				&& target.IsInWorld && !target.HasTrait<Husk>()); }
		}
	}
}