/*
 * 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("virus")]
    public class VirusTestCommands {
        [Command("infect")]
        public static void InfectFileWithMUDVirus(Dictionary<string, object> args) {
            string signature = "";

            if (args.ContainsKey("signature"))
                signature = args["signature"] as string;
            else
                throw new Exception("Virus signature not provided.");

            string script = "";

            Action<string> scriptFound = new Action<string>((s) => {
                script = s;
                Virus v = new LuaVirus(s, 1);
                VirusEngine.Add(v);
                VirusEngine.Infect($"lua.{v.Signature}.1");
            });

            ServerManager.MessageReceived += (srv) => {
                if (srv.Name == "mud_virus") {
                    scriptFound?.Invoke(srv.Contents);
                    scriptFound = null;
                }
            };

            ServerManager.SendMessage("getvirus", signature);
        }
    }


    [RequiresUpgrade("mud_fundamentals")]
    [Namespace("mud")]
    public static class MUDCommands {
        [Command("addvirus")]
        public static void Virus_AddToDatabase(Dictionary<string, object> args) {
            string file = "";
            int threatLevel = 0;


            if (args.ContainsKey("file"))
                file = args["file"] as string;
            else
                throw new Exception("No 'file' argument provided.");

            if (args.ContainsKey("threatlevel"))
                threatLevel = Convert.ToInt32(args["threatlevel"].ToString());

            Virus lua = new LuaVirus(Utils.ReadAllText(file), threatLevel);

            Console.WriteLine("Virus uploaded to current multi-user domain successfully.");

        }


        [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;
        }

    }

    [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
    [Namespace("cheats", true)]
    public static class CheatCommands {
        [Command("freecp", true)]
        [CommandObsolete("{OBSOLETE_CHEATS_FREECP}", "dev.freecp", true)]
        public static bool FreeCodepoints(Dictionary<string, object> args) {
            // never called
            return true;
        }
    }
    [Namespace("dev")]
    public static class ShiftOSDevCommands {
        [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 GetAllUpgrades() {
            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() {
            try {
                throw new Exception("ShiftOS was sent a command to forcefully crash.");
            } catch (Exception e) {
                CrashHandler.Start(e);
                return true;
            }
        }
    }
#endif

    [Namespace("sos")]
    public static class ShiftOSCommands {

        [Command("shutdown")]
        public static bool Shutdown() {
            SaveSystem.ShuttingDown = true;
            TerminalBackend.InvokeCommand("sos.save");
            AppearanceManager.Exit();
            return true;
        }
        [Command("verify")]
        public static bool Verify(Dictionary<string, object> args) {
            if (SaveSystem.CurrentSave.StoryPosition == 4) {
                if (args.ContainsKey(Localization.Parse("{ARGS_PASSWORD}"))) {
                    if (args[Localization.Parse("{ARGS_PASSWORD}")] as string == "theepicwin") {
                        SaveSystem.CurrentSave.StoryPosition++;
                    } else {
                        Console.WriteLine("{SENTIENCE_INVALIDPASSWORD}");
                    }
                } else {
                    Console.WriteLine("{SENTIENCE_INVALIDPASSWORD}");
                }
                return true;
            } else {
                return false;
            }
        }


        [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 {




        [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;
        }

        [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
                {
                    Console.WriteLine("Please specify a valid 'app' param.");
                    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;
            }
        }


    }
}