/*
 * MIT License
 * 
 * Copyright (c) 2017 Michael VanOverbeek and ShiftOS devs
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#define DEVEL

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Windows.Forms;
using ShiftOS.Engine.Properties;
using System.IO;
using Newtonsoft.Json;
using System.IO.Compression;

using ShiftOS.Objects;
using Discoursistency.Base.Models.Authentication;
using ShiftOS.Engine.Scripting;
using ShiftOS.Objects.ShiftFS;

namespace ShiftOS.Engine
{
    [Namespace("infobox", hide = true)]
    [RequiresUpgrade("desktop;wm_free_placement")]
    public static class InfoboxDebugCommands
    {

        [RequiresArgument("title")]
        [RequiresArgument("msg")]
        [Command("show")]
        public static bool ShowInfo(Dictionary<string, object> args)
        {
            Desktop.InvokeOnWorkerThread(new Action(() =>
            {
                Infobox.Show(args["title"].ToString(), args["msg"].ToString());
            }));
            return true;
        }

        [RequiresArgument("title")]
        [RequiresArgument("msg")]
        [Command("yesno")]
        public static bool ShowYesNo(Dictionary<string, object> args)
        {
            bool forwarding = TerminalBackend.IsForwardingConsoleWrites;
            var fGuid = TerminalBackend.ForwardGUID;
            Action<bool> callback = (result) =>
            {
                TerminalBackend.IsForwardingConsoleWrites = forwarding;
                TerminalBackend.ForwardGUID = (forwarding == true) ? fGuid : null;
                string resultFriendly = (result == true) ? "yes" : "no";
                Console.WriteLine($"{SaveSystem.CurrentSave.Username} says {resultFriendly}.");
                TerminalBackend.IsForwardingConsoleWrites = false;
            };
            Desktop.InvokeOnWorkerThread(new Action(() =>
            {
                Infobox.PromptYesNo(args["title"].ToString(), args["msg"].ToString(), callback);

            }));
            return true;
        }

        [RequiresArgument("title")]
        [RequiresArgument("msg")]
        [Command("text")]
        public static bool ShowText(Dictionary<string, object> args)
        {
            bool forwarding = TerminalBackend.IsForwardingConsoleWrites;
            var fGuid = TerminalBackend.ForwardGUID;
            Action<string> callback = (result) =>
            {
                TerminalBackend.IsForwardingConsoleWrites = forwarding;
                TerminalBackend.ForwardGUID = (forwarding == true) ? fGuid : null;
                Console.WriteLine($"{SaveSystem.CurrentSave.Username} says \"{result}\".");
                TerminalBackend.IsForwardingConsoleWrites = false;
            };
            Desktop.InvokeOnWorkerThread(new Action(() =>
            {
                Infobox.PromptText(args["title"].ToString(), args["msg"].ToString(), callback);
            }));
                return true;
        }

    }

    [Namespace("audio")]
    public static class AudioCommands
    {
        [Command("setvol", description = "Set the volume of the system audio to anywhere between 0 and 100.")]
        [RequiresArgument("value")]
        [RequiresUpgrade("audio_volume")]
        public static bool SetVolume(Dictionary<string,object> args)
        {
            int val = Convert.ToInt32(args["value"].ToString());
            float volume = (val / 100F);
            AudioManager.SetVolume(volume);
            return true;
        }
        [RequiresUpgrade("audio_volume")]
        [Command("mute", description = "Sets the volume of the system audio to 0")]
        public static bool MuteAudio()
        {
            AudioManager.SetVolume(0);
            return true;
        }
    }

    [RequiresUpgrade("mud_fundamentals")]
    [Namespace("mud")]
    public static class MUDCommands
    {
        [Command("status")]
        public static bool Status()
        {
            ServerManager.PrintDiagnostics();
            return true;
        }

        [Command("connect")]
        public static bool Connect(Dictionary<string, object> args)
        {
            try
            {
                string ip = (args.ContainsKey("addr") == true) ? args["addr"] as string : "michaeltheshifter.me";
                int port = (args.ContainsKey("port") == true) ? Convert.ToInt32(args["port"] as string) : 13370;
                try
                {
                    ServerManager.Initiate(ip, port);
                }
                catch (Exception ex)
                {
                    Console.WriteLine("{ERROR}: " + ex.Message);
                }
                return true;
            }
            catch (Exception ex)
            {
                Console.WriteLine("Error running script:" + ex);
                return false;
            }
        }


    }

    [RequiresUpgrade("mud_fundamentals")]
    [Namespace("chat")]
    public static class ChatCommands
    {
        [RequiresArgument("id")]
        [RequiresArgument("name")]
        [RequiresArgument("topic")]
        [Command("create")]
        public static bool CreateChat(Dictionary<string, object> args)
        {
            string id = "";
            string topic = "";
            string name = "";
            int max_users = 0;

            id = args["id"] as string;
            name = args["topic"] as string;
            topic = args["name"] as string;

            bool valid = true;

            if (string.IsNullOrEmpty(id) || string.IsNullOrEmpty(name) || string.IsNullOrEmpty(topic))
                valid = false;

            if (valid)
            {
                ServerManager.SendMessage("chat_create", $@"{{
    id: ""{id}"",
    name: ""{name}"",
    topic: ""{topic}"",
    max_users: {max_users}
}}");
            }
            else
            {
                Console.WriteLine("{CHAT_PLEASE_PROVIDE_VALID_CHANNEL_DATA}");
            }
            return true;
        }

    }

    [TutorialLock]
    [Namespace("trm")]
    public static class TerminalCommands
    {
        [Command("clear")]
        public static bool Clear()
        {
            AppearanceManager.ConsoleOut.Clear();
            return true;
        }

        [Command("echo")]
        [RequiresArgument("text")]
        public static bool Echo(Dictionary<string, object> args)
        {
            Console.WriteLine(args["text"]);
            return true;
        }
    }

#if DEVEL
    internal class Rock : Exception
    {
        internal Rock() : base("Someone threw a rock at the window, and the Terminal shattered.")
        {

        }
    }
    
    [Namespace("dev")]
    public static class ShiftOSDevCommands
    {
        [Command("rock", description = "A little surprise for unstable builds...")]
        public static bool ThrowAFuckingRock()
        {
            Infobox.Show("He who lives in a glass house shouldn't throw stones...", new Rock().Message);
            return false;
        }


        [Command("unbuy")]
        [RequiresArgument("upgrade")]
        public static bool UnbuyUpgrade(Dictionary<string, object> args)
        {
            try
            {
                SaveSystem.CurrentSave.Upgrades[args["upgrade"] as string] = false;
            }
            catch
            {
                Console.WriteLine("Upgrade not found.");
            }
            return true;
        }

        [Command("getallupgrades")]
        public static bool GetAllUpgrades()
        {
            Console.WriteLine(JsonConvert.SerializeObject(SaveSystem.CurrentSave.Upgrades, Formatting.Indented));
            return true;
        }

        [Command("multarg")]
        [RequiresArgument("id")]
        [RequiresArgument("name")]
        [RequiresArgument("type")]
        public static bool MultArg(Dictionary<string, object> args)
        {
            return true;
        }


        [Command("freecp")]
        public static bool FreeCodepoints(Dictionary<string, object> args)
        {
            if (args.ContainsKey("amount"))
                try
                {
                    int codepointsToAdd = Convert.ToInt32(args["amount"].ToString());
                    SaveSystem.TransferCodepointsFrom("dev", codepointsToAdd);
                    return true;
                }
                catch (Exception ex)
                {
                    Console.WriteLine("{ERROR}: " + ex.Message);
                    return true;
                }

            SaveSystem.TransferCodepointsFrom("dev", 1000);
            return true;
        }

        [Command("unlockeverything")]
        public static bool UnlockAllUpgrades()
        {
            foreach (var upg in Shiftorium.GetDefaults())
            {
                Shiftorium.Buy(upg.ID, 0);
            }
            return true;
        }

        [Command("info")]
        public static bool DevInformation()
        {
            Console.WriteLine("{SHIFTOS_PLUS_MOTTO}");
            Console.WriteLine("{SHIFTOS_VERSION_INFO}" + Assembly.GetExecutingAssembly().GetName().Version);
            return true;
        }
        [Command("pullfile")]
        public static bool PullFile(Dictionary<string, object> args)
        {
            if (args.ContainsKey("physical") && args.ContainsKey("virtual"))
            {
                string file = (string)args["physical"];
                string dest = (string)args["virtual"];
                if (System.IO.File.Exists(file))
                {
                    Console.WriteLine("Pulling physical file to virtual drive...");
                    byte[] filebytes = System.IO.File.ReadAllBytes(file);
                    ShiftOS.Objects.ShiftFS.Utils.WriteAllBytes(dest, filebytes);
                }
                else
                {
                    Console.WriteLine("The specified file does not exist on the physical drive.");
                }
            }
            else
            {
                Console.WriteLine("You must supply a physical path.");
            }
            return true;
        }
        [Command("crash")]
        public static bool CrashInstantly()
        {
            CrashHandler.Start(new Exception("ShiftOS was sent a command to forcefully crash."));
            return true;
        }
    }
#endif

    [Namespace("sos")]
    public static class ShiftOSCommands
    {
        [RemoteLock]
        [Command("shutdown")]
        public static bool Shutdown()
        {
            SaveSystem.ShuttingDown = true;
            TerminalBackend.InvokeCommand("sos.save");
            AppearanceManager.Exit();
            return true;
        }

        [Command("help", "{COMMAND_HELP_USAGE}", "{COMMAND_HELP_DESCRIPTION}")]
        public static bool Help()
        {
            var asm = Assembly.GetExecutingAssembly();

            var types = asm.GetTypes();

            foreach (var type in types)
            {
                if (Shiftorium.UpgradeAttributesUnlocked(type))
                {
                    foreach (var a in type.GetCustomAttributes(false))
                    {
                        if (a is Namespace)
                        {
                            var ns = a as Namespace;

                            if (!ns.hide)
                            {
                                string descp = "{NAMESPACE_" + ns.name.ToUpper() + "_DESCRIPTION}";
                                if (descp == Localization.Parse(descp))
                                    descp = "";
                                else
                                    descp = Shiftorium.UpgradeInstalled("help_description") ? Localization.Parse("{SEPERATOR}" + descp) : "";

                                Console.WriteLine($"{{NAMESPACE}}{ns.name}" + descp);

                                foreach (var method in type.GetMethods(BindingFlags.Public | BindingFlags.Static))
                                {
                                    if (Shiftorium.UpgradeAttributesUnlocked(method))
                                    {
                                        foreach (var ma in method.GetCustomAttributes(false))
                                        {
                                            if (ma is Command)
                                            {
                                                var cmd = ma as Command;

                                                if (!cmd.hide)
                                                {
                                                    string descriptionparse = "{COMMAND_" + ns.name.ToUpper() + "_" + cmd.name.ToUpper() + "_DESCRIPTION}";
                                                    string usageparse = "{COMMAND_" + ns.name.ToUpper() + "_" + cmd.name.ToUpper() + "_USAGE}";
                                                    if (descriptionparse == Localization.Parse(descriptionparse))
                                                        descriptionparse = "";
                                                    else
                                                        descriptionparse = Shiftorium.UpgradeInstalled("help_description") ? Localization.Parse("{SEPERATOR}" + descriptionparse) : "";

                                                    if (usageparse == Localization.Parse(usageparse))
                                                        usageparse = "";
                                                    else
                                                        usageparse = Shiftorium.UpgradeInstalled("help_usage") ? Localization.Parse("{SEPERATOR}" + usageparse, new Dictionary<string, string>() {
                                                            {"%ns", ns.name},
                                                            {"%cmd", cmd.name}
                                                        }) : "";

                                                    Console.WriteLine($"{{COMMAND}}{ns.name}.{cmd.name}" + usageparse + descriptionparse);
                                                }
                                            }
                                        }
                                    }

                                }
                            }

                        }
                    }
                }
            }

            return true;
        }

        [Command("save")]
        public static bool Save()
        {
            SaveSystem.SaveGame();
            return true;
        }

        [Command("status")]
        public static bool Status()
        {
            Console.WriteLine($@"ShiftOS version {Assembly.GetExecutingAssembly().GetName().Version.ToString()}

Codepoints:       {SaveSystem.CurrentSave.Codepoints}
Upgrades:         {SaveSystem.CurrentSave.CountUpgrades()} installed,
                  {Shiftorium.GetAvailable().Length} available");
            return true;
        }
    }


    [Namespace("shiftorium")]
    public static class ShiftoriumCommands
    {
        [Command("buy")]
        [RequiresArgument("upgrade")]
        public static bool BuyUpgrade(Dictionary<string, object> userArgs)
        {
            try
            {
                string upgrade = "";

                if (userArgs.ContainsKey("upgrade"))
                    upgrade = (string)userArgs["upgrade"];
                else
                    throw new Exception("You must specify a valid 'upgrade' value.");

                foreach (var upg in Shiftorium.GetAvailable())
                {
                    if (upg.ID == upgrade)
                    {
                        Shiftorium.Buy(upgrade, upg.Cost);
                        return true;
                    }
                }

                throw new Exception($"Couldn't find upgrade with ID: {upgrade}");
            }
            catch
            {
                return false;
            }
        }

        [RequiresUpgrade("shiftorium_bulk_buy")]
        [Command("bulkbuy")]
        [RequiresArgument("upgrades")]
        public static bool BuyBulk(Dictionary<string, object> args)
        {
            if (args.ContainsKey("upgrades"))
            {
                string[] upgrade_list = (args["upgrades"] as string).Split(new[] { "," }, StringSplitOptions.RemoveEmptyEntries);
                foreach (var upg in upgrade_list)
                {
                    var dict = new Dictionary<string, object>();
                    dict.Add("upgrade", upg);
                    BuyUpgrade(dict);
                }
            }
            else
            {
                throw new Exception("Please specify a list of upgrades in the 'upgrades' argument. Each upgrade is separated by a comma.");
            }
            return true;
        }


        [Command("info")]
        public static bool ViewInfo(Dictionary<string, object> userArgs)
        {
            try
            {
                string upgrade = "";

                if (userArgs.ContainsKey("upgrade"))
                    upgrade = (string)userArgs["upgrade"];
                else
                    throw new Exception("You must specify a valid 'upgrade' value.");

                foreach (var upg in Shiftorium.GetDefaults())
                {
                    if (upg.ID == upgrade)
                    {
                        Console.WriteLine($@"Information for {upgrade}:

{upg.Name} - {upg.Cost} Codepoints
------------------------------------------------------

{upg.Description}

To buy this upgrade, run:
shiftorium.buy{{upgrade:""{upg.ID}""}}");
                        return true;
                    }
                }

                throw new Exception($"Couldn't find upgrade with ID: {upgrade}");
            }
            catch
            {
                return false;
            }
        }
        [Command("list")]
        public static bool ListAll()
        {
            try
            {
                Dictionary<string, int> upgrades = new Dictionary<string, int>();
                int maxLength = 5;

                foreach (var upg in Shiftorium.GetAvailable())
                {
                    if (upg.ID.Length > maxLength)
                    {
                        maxLength = upg.ID.Length;
                    }

                    upgrades.Add(upg.ID, upg.Cost);
                }

                Console.WriteLine("ID".PadRight((maxLength + 5) - 2) + "Cost (Codepoints)");

                foreach (var upg in upgrades)
                {
                    Console.WriteLine(upg.Key.PadRight((maxLength + 5) - upg.Key.Length) + "  " + upg.Value.ToString());
                }
                return true;
            }
            catch (Exception e)
            {
                CrashHandler.Start(e);
                return false;
            }
        }
    }

    [Namespace("win")]
    public static class WindowCommands
    {



        [RemoteLock]
        [Command("list")]
        public static bool List()
        {
            Console.WriteLine("{ID}\t{WINDOW}");
            foreach (var app in AppearanceManager.OpenForms)
            {
                //All .NET object instances have a unique hash code. Good for fake process management.
                Console.WriteLine($"{app.GetHashCode()}\t{app.Text}");
            }
            return true;
        }

        [RemoteLock]
        [Command("open")]
        public static bool Open(Dictionary<string, object> args)
        {
            try
            {
                if (args.ContainsKey("app"))
                {
                    var app = args["app"] as string;
                    //ANNND now we start reflecting...
                    foreach (var asmExec in System.IO.Directory.GetFiles(Environment.CurrentDirectory))
                    {
                        if (asmExec.EndsWith(".exe") || asmExec.EndsWith(".dll"))
                        {
                            var asm = Assembly.LoadFile(asmExec);

                            foreach (var type in asm.GetTypes())
                            {
                                if (type.BaseType == typeof(UserControl))
                                {
                                    foreach (var attr in type.GetCustomAttributes(false))
                                    {
                                        if (attr is WinOpenAttribute)
                                        {
                                            if (app == (attr as WinOpenAttribute).ID)
                                            {
                                                if (SaveSystem.CurrentSave.Upgrades.ContainsKey(app))
                                                {
                                                    if (Shiftorium.UpgradeInstalled(app))
                                                    {
                                                        IShiftOSWindow frm = Activator.CreateInstance(type) as IShiftOSWindow;
                                                        AppearanceManager.SetupWindow(frm);
                                                        return true;
                                                    }
                                                    else
                                                    {
                                                        throw new Exception($"{app} was not found on your system! Try looking in the shiftorium...");
                                                    }
                                                }
                                                else
                                                {
                                                    IShiftOSWindow frm = Activator.CreateInstance(type) as IShiftOSWindow;
                                                    AppearanceManager.SetupWindow(frm);
                                                    return true;
                                                }
                                            }
                                        }
                                    }
                                }
                            }
                        }
                    }

                }
                else
                {
                    foreach (var asmExec in System.IO.Directory.GetFiles(Environment.CurrentDirectory))
                    {
                        if (asmExec.EndsWith(".exe") || asmExec.EndsWith(".dll"))
                        {
                            var asm = Assembly.LoadFile(asmExec);

                            foreach (var type in asm.GetTypes())
                            {
                                if (type.GetInterfaces().Contains(typeof(IShiftOSWindow)))
                                {
                                    foreach (var attr in type.GetCustomAttributes(false))
                                    {
                                        if (attr is WinOpenAttribute)
                                        {
                                            if (Shiftorium.UpgradeAttributesUnlocked(type))
                                            {
                                                Console.WriteLine("win.open{app:\"" + (attr as WinOpenAttribute).ID + "\"}");
                                            }
                                        }
                                    }
                                }
                            }
                        }
                    }


                    return true;
                }
                Console.WriteLine("Couldn't find the specified app on your system.");
                return true;
            }
            catch (Exception ex)
            {
                Console.WriteLine("Error running script:" + ex);
                return false;
            }
        }


    }
}