#region Copyright & License Information
/*
 * Copyright (c) The OpenRA Developers and Contributors
 * 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, either version 3 of
 * the License, or (at your option) any later version. For more
 * information, see COPYING.
 */
#endregion

using OpenRA.Widgets;

namespace OpenRA.Mods.Common.Widgets.Logic
{
	public class MapEditorTabsLogic : ChromeLogic
	{
		readonly Widget widget;
		readonly EditorViewportControllerWidget editor;

		protected enum MenuType { Select, Tiles, Layers, Actors, Tools, History }
		protected MenuType menuType = MenuType.Tiles;
		readonly Widget tabContainer;

		MenuType lastSelectedTab = MenuType.Tiles;

		[ObjectCreator.UseCtor]
		public MapEditorTabsLogic(Widget widget)
		{
			this.widget = widget;
			editor = widget.Parent.Parent.Get<EditorViewportControllerWidget>("MAP_EDITOR");
			editor.DefaultBrush.UpdateSelectedTab += HandleUpdateSelectedTab;

			tabContainer = widget.Get("MAP_EDITOR_TAB_CONTAINER");

			SetupTab("SELECT_TAB", "SELECT_WIDGETS", MenuType.Select);
			SetupTab("TILES_TAB", "TILE_WIDGETS", MenuType.Tiles);
			SetupTab("OVERLAYS_TAB", "LAYER_WIDGETS", MenuType.Layers);
			SetupTab("ACTORS_TAB", "ACTOR_WIDGETS", MenuType.Actors);
			SetupTab("TOOLS_TAB", "TOOLS_WIDGETS", MenuType.Tools);
			SetupTab("HISTORY_TAB", "HISTORY_WIDGETS", MenuType.History);
		}

		protected override void Dispose(bool disposing)
		{
			editor.DefaultBrush.UpdateSelectedTab -= HandleUpdateSelectedTab;

			base.Dispose(disposing);
		}

		void SetupTab(string buttonId, string tabId, MenuType tabType)
		{
			if (buttonId != null)
			{
				var tab = tabContainer.Get<ButtonWidget>(buttonId);
				tab.IsHighlighted = () => menuType == tabType;
				tab.OnClick = () => menuType = SelectTab(tabType);

				if (tabType == MenuType.Select)
					tab.IsDisabled = () => !editor.DefaultBrush.Selection.HasSelection;
			}

			var container = widget.Parent.Get<ContainerWidget>(tabId);
			container.IsVisible = () => menuType == tabType;
		}

		MenuType SelectTab(MenuType newMenuType)
		{
			if (newMenuType != MenuType.Select)
				lastSelectedTab = newMenuType;

			return newMenuType;
		}

		void HandleUpdateSelectedTab()
		{
			var hasSelection = editor.DefaultBrush.Selection.HasSelection;

			if (menuType != MenuType.Select && hasSelection)
				menuType = MenuType.Select;
			else if (menuType == MenuType.Select && !hasSelection)
				menuType = lastSelectedTab;
		}
	}
}