#region Copyright & License Information
/*
 * Copyright 2007-2011 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.Drawing;
using OpenRA.Graphics;

namespace OpenRA.Widgets
{
	public class ButtonWidget : Widget
	{
		public string Key = null;
		public string Text = "";
		public bool Depressed = false;
		public int VisualHeight = ChromeMetrics.Get<int>("ButtonDepth");
		public string Font = ChromeMetrics.Get<string>("ButtonFont");
		public string ClickSound = null;
		public string ClickDisabledSound = null;
		public Func<string> GetText;
		public Func<bool> IsDisabled = () => false;
		public Action<MouseInput> OnMouseDown = _ => {};
        public Action<MouseInput> OnMouseUp = _ => {};

		// Equivalent to OnMouseUp, but without an input arg
		public Action OnClick = () => {};
		public Action<KeyInput> OnKeyPress = _ => {};
		
		public ButtonWidget()
			: base()
		{
			GetText = () => { return Text; };
			OnMouseUp = _ => OnClick();
			OnKeyPress = _ => OnClick();
		}

		protected ButtonWidget(ButtonWidget widget)
			: base(widget)
		{
			Text = widget.Text;
			Font = widget.Font;
			Depressed = widget.Depressed;
			VisualHeight = widget.VisualHeight;
			GetText = widget.GetText;
			OnMouseDown = widget.OnMouseDown;

			OnMouseUp = mi => OnClick();
			OnKeyPress = _ => OnClick();
		}

		public override bool LoseFocus(MouseInput mi)
		{
			Depressed = false;
			return base.LoseFocus(mi);
		}
		
		public override bool HandleKeyPress(KeyInput e)
		{
			if (e.KeyName != Key || e.Event != KeyInputEvent.Down)
				return false;
			
			if (!IsDisabled())
			{
				OnKeyPress(e);
				Sound.Play(ClickSound);
			}
			else
				Sound.Play(ClickDisabledSound);

			return true;
		}
		
		public override bool HandleMouseInput(MouseInput mi)
		{
			if (mi.Button != MouseButton.Left)
				return false;
			
			if (mi.Event == MouseInputEvent.Down && !TakeFocus(mi))
				return false;

			var disabled = IsDisabled();
			// Only fire the onMouseUp event if we successfully lost focus, and were pressed
			if (Focused && mi.Event == MouseInputEvent.Up)
			{
				if (Depressed && !disabled)
					OnMouseUp(mi);
				
				return LoseFocus(mi);
			}
			if (mi.Event == MouseInputEvent.Down)
			{
				// OnMouseDown returns false if the button shouldn't be pressed
				if (!disabled)
				{
					OnMouseDown(mi);
					Depressed = true;
					Sound.Play(ClickSound);
				}
				else
				{
					LoseFocus(mi);
					Sound.Play(ClickDisabledSound);
				}
			}
			else if (mi.Event == MouseInputEvent.Move && Focused)
				Depressed = RenderBounds.Contains(mi.Location);

			return Depressed;
		}

		public override int2 ChildOrigin { get { return RenderOrigin + 
				((Depressed) ? new int2(VisualHeight, VisualHeight) : new int2(0, 0)); } }
		
		public override void Draw()
		{
			var rb = RenderBounds;
			var disabled = IsDisabled();
			
			var font = Game.Renderer.Fonts[Font];
			var text = GetText();
			var s = font.Measure(text);
			var stateOffset = (Depressed) ? new int2(VisualHeight, VisualHeight) : new int2(0, 0);
			
			DrawBackground("button", rb, disabled, Depressed, Widget.MouseOverWidget == this);
			font.DrawText(text, new int2(rb.X + (UsableWidth - s.X)/ 2, rb.Y + (Bounds.Height - s.Y) / 2) + stateOffset,
			              disabled ? Color.Gray : Color.White);
		}

		public override Widget Clone() { return new ButtonWidget(this); }
		public virtual int UsableWidth { get { return Bounds.Width; } }

		public static void DrawBackground(string baseName, Rectangle rect, bool disabled, bool pressed, bool hover)
		{
			var state = disabled ? "-disabled" : 
						pressed ? "-pressed" : 
						hover ? "-hover" : 
						"";
			
			WidgetUtils.DrawPanel(baseName + state, rect);
		}
	}
}