using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using Rage;
using RAGENativeUI;
using RAGENativeUI.Elements;
using SceneManager.Objects;
using SceneManager.Utils;

namespace SceneManager.Menus
{
    internal static class PathMainMenu
    {
        private static int MAX_PATH_LIMIT { get; } = 10;
        internal static List<string> ImportedPaths { get; } = new List<string>();
        private static string[] DismissOptions { get; } = new string[] { "From path", "From waypoint", "From world" };
        internal static UIMenu Menu { get; } = new UIMenu("Scene Manager", "~o~Path Manager Main Menu");
        internal static UIMenuItem CreateNewPath { get; } = new UIMenuItem("Create New Path");
        internal static UIMenuListScrollerItem<string> ImportPath { get; } = new UIMenuListScrollerItem<string>("Import Path", "Import a saved path", ImportedPaths);
        internal static UIMenuItem DeleteAllPaths { get; } = new UIMenuItem("Delete All Paths");
        internal static UIMenuNumericScrollerItem<int> EditPath { get; private set; }
        internal static UIMenuListScrollerItem<string> DirectOptions { get; } = new UIMenuListScrollerItem<string>("Direct driver to path's", "", new[] { "First waypoint", "Nearest waypoint" });
        internal static UIMenuNumericScrollerItem<int> DirectDriver { get; private set; } 
        internal static UIMenuListScrollerItem<string> DismissDriver { get; } = new UIMenuListScrollerItem<string>("Dismiss nearest driver", $"~b~From path: ~w~AI will be released from the path\n~b~From waypoint: ~w~AI will skip their current waypoint task\n~b~From world: ~w~AI will be removed from the world.", DismissOptions);
        internal static UIMenuCheckboxItem DisableAllPaths { get; } = new UIMenuCheckboxItem("Disable All Paths", false);

        internal static void Initialize()
        {
            Menu.ParentMenu = MainMenu.Menu;
            MenuManager.MenuPool.Add(Menu);

            Menu.OnItemSelect += PathMenu_OnItemSelected;
            Menu.OnCheckboxChange += PathMenu_OnCheckboxChange;
            Menu.OnMenuOpen += PathMenu_OnMenuOpen;
        }

        internal static void BuildPathMenu()
        {
            MenuManager.MenuPool.CloseAllMenus();
            Menu.Clear();

            Menu.AddItem(CreateNewPath);
            CreateNewPath.ForeColor = Color.Gold;
            Menu.AddItem(ImportPath);
            ImportPath.ForeColor = Color.Gold;
            ImportPath.Enabled = true;
            Menu.AddItem(EditPath = new UIMenuNumericScrollerItem<int>("Edit Path", "", 1, PathManager.Paths.Count, 1));
            EditPath.ForeColor = Color.Gold;
            Menu.AddItem(DisableAllPaths);
            DisableAllPaths.Enabled = true;
            Menu.AddItem(DeleteAllPaths);
            DeleteAllPaths.Enabled = true;
            DeleteAllPaths.ForeColor = Color.Gold;
            Menu.AddItem(DirectOptions);
            Menu.AddItem(DirectDriver = new UIMenuNumericScrollerItem<int>("Direct nearest driver to path", "", 1, PathManager.Paths.Count, 1)); // This must instantiate here because the Paths.Count value changes
            DirectDriver.ForeColor = Color.Gold;
            DirectDriver.Enabled = true;
            Menu.AddItem(DismissDriver);
            DismissDriver.ForeColor = Color.Gold;

            if (PathManager.Paths.Count == MAX_PATH_LIMIT)
            {
                CreateNewPath.Enabled = false;
                ImportPath.Enabled = false;
            }
            if (PathManager.Paths.Count == 0)
            {
                EditPath.Enabled = false;
                DeleteAllPaths.Enabled = false;
                DisableAllPaths.Enabled = false;
                DirectDriver.Enabled = false;
            }
            if(Settings.ImportedPaths.Count == 0)
            {
                ImportPath.Enabled = false;
            }

            MenuManager.MenuPool.RefreshIndex();
        }

        private static void PathMenu_OnItemSelected(UIMenu sender, UIMenuItem selectedItem, int index)
        {
            if (selectedItem == CreateNewPath)
            {
                GoToPathCreationMenu();
            }

            if(selectedItem == ImportPath)
            {
                GoToImportMenu();
            }

            if (selectedItem == EditPath)
            {
                GoToEditPathMenu();
            }

            if (selectedItem == DeleteAllPaths)
            {
                Utils.DeleteAllPaths.Delete();
                DisableAllPaths.Checked = false;
                BuildPathMenu();
                Menu.Visible = true;
            }

            if (selectedItem == DirectDriver)
            {
                if(Utils.DirectDriver.ValidateOptions(DirectOptions, PathManager.Paths[DirectDriver.Index], out Vehicle nearbyVehicle, out Waypoint targetWaypoint))
                {
                    Utils.DirectDriver.Direct(nearbyVehicle, PathManager.Paths[DirectDriver.Index], targetWaypoint);
                }
            }

            if (selectedItem == DismissDriver)
            {
                Utils.DismissDriver.Dismiss(DismissDriver.Index);
            }
        }

        private static void PathMenu_OnCheckboxChange(UIMenu sender, UIMenuCheckboxItem checkboxItem, bool @checked)
        {
            if (checkboxItem == DisableAllPaths)
            {
                TogglePaths.Toggle(DisableAllPaths.Checked);
            }
        }

        private static void PathMenu_OnMenuOpen(UIMenu menu)
        {
            var scrollerItems = new List<UIMenuScrollerItem> { DirectOptions, DirectDriver, DismissDriver, EditPath };
            GameFiber.StartNew(() => UserInput.InitializeMenuMouseControl(menu, scrollerItems), "RNUI Mouse Input Fiber");
        }

        private static void GoToPathCreationMenu()
        {
            if (PathCreationMenu.PathCreationState == State.Creating)
            {
                Menu.Visible = false;
                PathCreationMenu.Menu.Visible = true;
                Path currentPath = PathManager.Paths.FirstOrDefault(x => x.State == State.Creating);
                Game.DisplayNotification($"~o~Scene Manager~y~[Creating]\n~w~Resuming path {currentPath.Number}");
            }
            else
            {
                PathCreationMenu.BuildPathCreationMenu();
                Menu.Visible = false;
                PathCreationMenu.Menu.Visible = true;
            }
        }

        private static void GoToImportMenu()
        {
            Menu.Visible = false;
            ImportPathMenu.Menu.Visible = true;
        }

        private static void GoToEditPathMenu()
        {
            Menu.Visible = false;
            EditPathMenu.Menu.Visible = true;
        }
    }
}