// This code is part of the Fungus library (https://github.com/snozbot/fungus)
// It is released for free under the MIT open source license (https://github.com/snozbot/fungus/blob/master/LICENSE)

using UnityEngine;

namespace Fungus
{
    /// <summary>
    /// Force a loop to terminate immediately.
    /// </summary>
    [CommandInfo("Flow", 
                 "Break", 
                 "Force a loop to terminate immediately.")]
    [AddComponentMenu("")]
    public class Break : Command
    {
        #region Public members

        public override void OnEnter()
        {
            // Find index of previous while command
            int whileIndex = -1;
            int whileIndentLevel = -1;
            for (int i = CommandIndex - 1; i >=0; --i)
            {
                While whileCommand = ParentBlock.CommandList[i] as While;
                if (whileCommand != null)
                {
                    whileIndex = i;
                    whileIndentLevel = whileCommand.IndentLevel;
                    break;
                }
            }

            if (whileIndex == -1)
            {
                // No enclosing While command found, just continue
                Continue();
                return;
            }

            // Find matching End statement at same indent level as While
            for (int i = whileIndex + 1; i < ParentBlock.CommandList.Count; ++i)
            {
                End endCommand = ParentBlock.CommandList[i] as End;
                
                if (endCommand != null && 
                    endCommand.IndentLevel == whileIndentLevel)
                {
                    // Sanity check that break command is actually between the While and End commands
                    if (CommandIndex > whileIndex && CommandIndex < endCommand.CommandIndex)
                    {
                        // Continue at next command after End
                        Continue (endCommand.CommandIndex + 1);
                        return;
                    }
                    else
                    {
                        break;
                    }
                }
            }

            // No matching End command found so just continue
            Continue();
        }

        public override Color GetButtonColor()
        {
            return new Color32(253, 253, 150, 255);
        }

        #endregion
    }    
}