using System;

using System.Collections.Generic;

using System.IO;

using System.Linq;

using System.Reflection;

using System.Text.RegularExpressions;

using KSP;

using UnityEngine;

namespace ModuleManager

{

// Once MUST be true for the election process to work when 2+ dll of the same version are loaded

// But I need it to be false for the reload database thingy

[KSPAddonFixed(KSPAddon.Startup.Instantly, false, typeof(ConfigManager))]

public class ConfigManager : MonoBehaviour

{

//FindConfigNodeIn finds and returns a ConfigNode in src of type nodeType.

//If nodeName is not null, it will only find a node of type nodeType with the value name=nodeName.

//If nodeTag is not null, it will only find a node of type nodeType with the value name=nodeName and tag=nodeTag.

public static ConfigNode FindConfigNodeIn(ConfigNode src, string nodeType,

string nodeName = null, string nodeTag = null)

{

#if DEBUG

if (nodeTag == null)

print ("Searching node for " + nodeType + "[" + nodeName + "]");

else

print ("Searching node for " + nodeType + "[" + nodeName + "," + nodeTag + "]");

#endif

foreach (ConfigNode n in src.GetNodes(nodeType)) {

if (nodeName == null && nodeTag == null)

return n;

if (n.HasValue("name") && WildcardMatch(n.GetValue("name"), nodeName) &&

(nodeTag == null ||

(n.HasValue("tag") && WildcardMatch(n.GetValue("tag"), nodeTag)))) {

#if DEBUG

print ("found node!");

#endif

return n;

}

}

return null;

}

// Added that to prevent a crash in KSP 'CopyTo' when a subnode has an empty name like

// Like a pair of curly bracket without a name before them

public static bool IsSane(ConfigNode node)

{

if (node.name.Length == 0)

return false;

foreach (ConfigNode subnode in node.nodes)

if (!IsSane(subnode))

return false;

return true;

}

public static string RemoveWS(string withWhite)

{ // Removes ALL whitespace of a string.

return new string(withWhite.ToCharArray().Where(c => !Char.IsWhiteSpace(c)).ToArray());

}

// ModifyNode applies the ConfigNode mod as a 'patch' to ConfigNode original, then returns the patched ConfigNode.

// it uses FindConfigNodeIn(src, nodeType, nodeName, nodeTag) to recurse.

public static ConfigNode ModifyNode(ConfigNode original, ConfigNode mod)

{

if (!IsSane(original) || !IsSane(mod)) {

print("[ModuleManager] A node has an empty name. Skipping it. Original: " + original.name);

return original;

}

ConfigNode newNode = original.CreateCopy();

foreach (ConfigNode.Value val in mod.values) {

if (val.name[0] != '@' && val.name[0] != '!' && val.name[0] != '%')

newNode.AddValue(val.name, val.value);

else {

// Parsing: Format is @key = value or @key,index = value

string valName = val.name.Substring(1);

int index = 0;

if (valName.Contains(",")) {

int.TryParse(valName.Split(',')[1], out index);

valName = valName.Split(',')[0];

} // index is useless right now, but some day it might not be.

if (val.name[0] == '@')

newNode.SetValue(valName, val.value, index);

else if (val.name[0] == '!')

newNode.RemoveValues(valName);

else if (val.name[0] == '%') {

newNode.RemoveValues(valName);

newNode.AddValue(valName, val.value);

}

}

}

foreach (ConfigNode subMod in mod.nodes) {

subMod.name = RemoveWS(subMod.name);

if (subMod.name[0] != '@' && subMod.name[0] != '!' && subMod.name[0] != '%' && subMod.name[0] != '$')

newNode.AddNode(subMod);

else {

ConfigNode subNode;

//if (subMod.name[0] == '@' && subMod.name[0] != '%')

// subNode = null;

if (subMod.name.Contains("[")) {

// format @NODETYPE[Name] {...} or @NODETYPE[Name, Tag] {...} or ! instead of @

string nodeType = subMod.name.Substring(1).Split('[')[0];

string nodeName = subMod.name.Split('[')[1].Replace("]", "");

string nodeTag = null;

if (nodeName.Contains(",")) {

// format @NODETYPE[Name, Tag] {...} or ! instead of @

nodeTag = nodeName.Split(',')[1];

nodeName = nodeName.Split(',')[0];

}

subNode = FindConfigNodeIn(newNode, nodeType, nodeName, nodeTag);

} else {

// format @NODETYPE {...} or ! instead of @

string nodeType = subMod.name.Substring(1);

// format @NODETYPE,N {...} or ! instead of @

// The problem with ! is that the index is messed up

// So the patch need to take that into account

// and lower the index for the next search

int index = 0;

if (nodeType.Contains(",")) {

int.TryParse(nodeType.Split(',')[1], out index);

nodeType = nodeType.Split(',')[0];

}

ConfigNode[] subNodes = newNode.GetNodes(nodeType);

if (subNodes.Length > index)

subNode = subNodes[index];

else

subNode = null;

}

if (subMod.name[0] == '@') {

// find the original subnode to modify, modify it and add the modified.

if (subNode != null) {

ConfigNode newSubNode = ModifyNode(subNode, subMod);

newNode.nodes.Add(newSubNode);

} else

print("[ModuleManager] Could not find node to modify: " + subMod.name);

}

if (subMod.name[0] == '%') {

// if the original node exist add it

if (subNode != null) {

ConfigNode newSubNode = ModifyNode(subNode, subMod);

newNode.nodes.Add(newSubNode);

}

else { // if not add the mod node without the % in its name

// This part is messy. ialdabaoth is right, i need to rewrite.

string type;

string name;

if (subMod.name.Contains("[")) {

type = subMod.name.Substring(1).Split('[')[0];

name = subMod.name.Split('[')[1].Replace("]", "");

}

else

{

type = subMod.name.Substring(1);

name = null;

}

ConfigNode copy = new ConfigNode(type);

if (name != null)

copy.AddValue("name", name);

ConfigNode newSubNode = ModifyNode(copy, subMod);

newNode.nodes.Add(newSubNode);

}

}

if (subMod.name[0] == '$')

{

// find the original subnode to copy, add the original, add the the modified copy.

if (subNode != null) {

newNode.nodes.Add(subNode);

ConfigNode newSubNode = ModifyNode(subNode, subMod);

newNode.nodes.Add(newSubNode);

}

else

print("[ModuleManager] Could not find node to copy: " + subMod.name);

}

if (subNode != null)

newNode.nodes.Remove(subNode);

}

}

return newNode;

}

public static List<UrlDir.UrlConfig> AllConfigsStartingWith(string match)

{

List<UrlDir.UrlConfig> nodes = new List<UrlDir.UrlConfig>();

foreach (UrlDir.UrlConfig url in GameDatabase.Instance.root.AllConfigs) {

if (url.type.StartsWith(match))

url.config.name = url.type;

nodes.Add(url);

}

return nodes;

}

bool loaded = false;

int patchCount = 0;

public void OnGUI()

{

/*

* It should be a code to reload when the Reload Database debug button is used.

* But it seem to go balistic after the 2nd reload.

*

if (PartLoader.Instance.Recompile == waitingReload)

{

waitingReload = !waitingReload;

print("[ModuleManager] waitingReload change " + waitingReload + " loaded " + loaded);

if (!waitingReload)

{

loaded = false;

print("[ModuleManager] loaded = false ");

}

}

*/

if (!GameDatabase.Instance.IsReady() && ((HighLogic.LoadedScene == GameScenes.MAINMENU) || (HighLogic.LoadedScene == GameScenes.SPACECENTER)))

{

return;

}

if (loaded)

return;

// Check for old version and MMSarbianExt

var oldMM = AssemblyLoader.loadedAssemblies.Where(a => a.assembly.GetName().Name == Assembly.GetExecutingAssembly().GetName().Name).Where(a => a.assembly.GetName().Version.CompareTo(new System.Version(1, 5)) == -1);

var oldAssemblies = oldMM.Concat(AssemblyLoader.loadedAssemblies.Where(a => a.assembly.GetName().Name == "MMSarbianExt"));

if (oldAssemblies.Any()) {

var badPaths = oldAssemblies.Select(a => a.path).Select(p => Uri.UnescapeDataString(new Uri(Path.GetFullPath(KSPUtil.ApplicationRootPath)).MakeRelativeUri(new Uri(p)).ToString().Replace('/', Path.DirectorySeparatorChar)));

PopupDialog.SpawnPopupDialog("Old versions of Module Manager", "You have old versions of Module Manager (older than 1.5) or MMSarbianExt.\nYou will need to remove them for Module Manager to work\nExit KSP and delete those files :\n" + String.Join("\n", badPaths.ToArray()), "OK", false, HighLogic.Skin);

loaded = true;

print("[ModuleManager] Old version of Module Manager present. Stopping");

return;

}

Assembly currentAssembly = Assembly.GetExecutingAssembly();

var eligible = AssemblyLoader.loadedAssemblies.Where(a => a.assembly.GetName().Name == currentAssembly.GetName().Name);

//print("[ModuleManager] starting \n" + String.Join("\n", eligible.Select(a => a.path).ToArray()));

// Elect the newest loaded version of MM to process all patch files.

// If there is a newer version loaded then don't do anything

if (eligible.Any(a => a.assembly.GetName().Version.CompareTo(currentAssembly.GetName().Version) == 1)) {

loaded = true;

print("[ModuleManager] version " + currentAssembly.GetName().Version + " at " + currentAssembly.Location + " lost the election");

return;

} else {

string candidates = "";

foreach (AssemblyLoader.LoadedAssembly a in eligible)

if (currentAssembly.Location != a.path)

candidates += "Version " + a.assembly.GetName().Version + " " + a.path + " " + "\n";

if (candidates.Length > 0)

print("[ModuleManager] version " + currentAssembly.GetName().Version + " at " + currentAssembly.Location + " won the election against\n" + candidates);

}

/* No longer usefull but kept as reference in case I need it

List<String> processedPath = new List<string>();

// Generate the list of subdirectory of Gamedata where we process patch file

foreach (AssemblyLoader.LoadedAssembly a in eligible)

{

string fullPath = Path.GetDirectoryName(a.path);

string relPath = Uri.UnescapeDataString(new Uri(Path.GetFullPath(KSPUtil.ApplicationRootPath)).MakeRelativeUri(new Uri(fullPath)).ToString() + "/").Remove(0, "GameData/".Length);

if (relPath == "/")

{

PopupDialog.SpawnPopupDialog("Module Manager Path", "This Module Manager will only work if installed in a subfolder of GameData.\nAbording patch loading", "OK", false, HighLogic.Skin);

loaded = true;

return;

}

processedPath.Add(relPath);

}

print("[ModuleManager] will procces patch in" + String.Join("\n",processedPath.ToArray()));

*/

// Build a list of subdirectory that won't be processed

List<String> excludePaths = new List<string>();

foreach (UrlDir.UrlConfig mod in GameDatabase.Instance.root.AllConfigs) {

if (mod.name == "MODULEMANAGER[LOCAL]") {

string fullpath = mod.url.Substring(0, mod.url.LastIndexOf('/'));

string excludepath = fullpath.Substring(0, fullpath.LastIndexOf('/'));

excludePaths.Add(excludepath);

print("excludepath: " + excludepath);

}

}

if (excludePaths.Any())

print("[ModuleManager] will not procces patch in those subdirectories:\n" + String.Join("\n", excludePaths.ToArray()));

patchCount = 0;

ApplyNodesSwitch();

ApplyPatch(excludePaths, false);

// :Final node

ApplyPatch(excludePaths, true);

print("[ModuleManager] Applied " + patchCount + " patch");

loaded = true;

}

// Apply patch to all relevent nodes

public void ApplyPatch(List<String> excludePaths, bool final)

{

foreach (UrlDir.UrlConfig mod in GameDatabase.Instance.root.AllConfigs) {

if (mod.type[0] == '@' || (mod.type[0] == '$' )) {

try {

if (!final ^ mod.name.EndsWith(":Final")) {

char[] sep = new char[] { '[', ']' };

string name = RemoveWS(mod.name);

string cond = "";

if (final)

name = name.Substring(0, name.LastIndexOf(":Final"));

if (name.Contains(":HAS[")) {

int start = name.IndexOf(":HAS[");

cond = name.Substring(start + 5, name.LastIndexOf(']') - start - 5);

name = name.Substring(0, start);

}

string[] splits = name.Split(sep, 3);

string pattern = splits.Length>1?splits[1]:null;

string type = splits[0].Substring(1);

foreach (UrlDir.UrlConfig url in GameDatabase.Instance.root.AllConfigs) {

if (url.type == type

&& WildcardMatch(url.name, pattern)

&& CheckCondition(url.config, cond)

&& !IsPathInList(mod.url, excludePaths)) {

if (mod.type[0] == '@') {

print("[ModuleManager] Applying node " + mod.url + " to " + url.url);

patchCount++;

url.config = ConfigManager.ModifyNode(url.config, mod.config);

}

else {

// Here we would duplicate an Node if we had the mean to do it

//ConfigNode newNode = ConfigManager.ModifyNode(url.config, mod.config);

//UrlDir.UrlConfig newurl = new UrlDir.UrlConfig(mod.parent, newNode);

//print("[ModuleManager] Copying Node " + newurl.url + " " + newurl.name);

}

}

}

}

} catch (Exception e) {

print("[ModuleManager] Exception while processing node : " + mod.url + "\n" + e.ToString());

}

}

}

}

public bool IsPathInList(string modPath, List<String> pathList)

{

return pathList.Any(modPath.StartsWith);

}

// Split condiction while not getting lost in embeded brackets

public static List<string> SplitCondition(string cond)

{

cond = RemoveWS(cond) + ",";

List<string> conds = new List<string>();

int start = 0;

int level = 0;

for (int end = 0; end < cond.Length; end++) {

if (cond[end] == ',' && level == 0) {

conds.Add(cond.Substring(start, end - start));

start = end + 1;

} else if (cond[end] == '[')

level++;

else if (cond[end] == ']')

level--;

}

return conds;

}

public static bool CheckCondition(ConfigNode node, string conds)

{

conds = RemoveWS(conds);

if (conds.Length == 0)

return true;

List<string> condsList = SplitCondition(conds);

if (condsList.Count == 1) {

conds = condsList[0];

string remainCond = "";

if (conds.Contains("HAS[")) {

int start = conds.IndexOf("HAS[") + 4;

remainCond = conds.Substring(start, condsList[0].LastIndexOf(']') - start);

conds = conds.Substring(0, start - 5);

}

char[] sep = new char[] { '[', ']' };

string[] splits = conds.Split(sep, 3);

string type = splits[0].Substring(1);

string name = splits.Length > 1 ? splits[1] : null;

switch (conds[0]) {

case '@':

case '!':

// @MODULE[ModuleAlternator] or !MODULE[ModuleAlternator]

bool not = (conds[0] == '!');

ConfigNode subNode = ConfigManager.FindConfigNodeIn(node, type, name);

if (subNode != null)

return not ^ CheckCondition(subNode, remainCond);

return not ^ false;

case '#':

// #module[Winglet]

if (node.HasValue(type) && node.GetValue(type).Equals(name))

return CheckCondition(node, remainCond);

return false;

case '~':

// ~breakingForce[] breakingForce is not present

if (!(node.HasValue(type)))

return CheckCondition(node, remainCond);

return false;

default:

return false;

}

}

return condsList.TrueForAll(c => CheckCondition(node, c));

}

public static bool WildcardMatch(String s, String wildcard)

{

if (wildcard == null) return true;

String pattern = "^" + Regex.Escape(wildcard).Replace(@"\*", ".*").Replace(@"\?", ".") + "$";

Regex regex;

regex = new Regex(pattern);

return (regex.IsMatch(s));

}

public static void ApplyNodesSwitch()

{

log("Processing Node Switch. Avaiable DLL are "+ String.Join(" ", AssemblyLoader.loadedAssemblies.Select(a => a.assembly.GetName().Name).ToArray()));

foreach (UrlDir.UrlConfig url in GameDatabase.Instance.root.AllConfigs)

{

//log("processing " + url.name);

url.config = SearchNodesSwitch(url.config, url.type + " " + url.name);

}

}

public static ConfigNode SearchNodesSwitch(ConfigNode node, String rootName)

{

ConfigNode newNode = new ConfigNode(node.name);

foreach (ConfigNode.Value val in node.values)

newNode.AddValue(val.name, val.value);

foreach (ConfigNode child in node.nodes)

{

if (child.name == "MM_SWITCH")

{

foreach (ConfigNode cas in child.nodes)

{

if (cas.name == "DEFAULT" || (cas.name == "CASE" && TestSwitchCase(cas)))

{

string list = "";

foreach (ConfigNode.Value value in cas.values)

list += " " + value.name + "=" + value.value;

log("MM_SWITCH processed for " + rootName + " using " + cas.name + list);

foreach (ConfigNode n in cas.nodes)

newNode.AddNode(SearchNodesSwitch(n, rootName));

break;

}

}

}

else

{

newNode.nodes.Add(SearchNodesSwitch(child, rootName));

}

}

return newNode;

}

public static bool TestSwitchCase(ConfigNode node)

{

bool state = true;

foreach (ConfigNode.Value value in node.values) {

// Switch case so we can extend the number of test easily

switch (value.name)

{

case "dll_loaded":

state = state && AssemblyLoader.loadedAssemblies.Any(a => a.assembly.GetName().Name == value.value);

break;

}

}

return state;

}

public static void log(String s)

{

print("[ModuleManager] " + s);

}

}

/// <summary>

/// KSPAddon with equality checking using an additional type parameter. Fixes the issue where AddonLoader prevents multiple start-once addons with the same start scene.

/// </summary>

public class KSPAddonFixed : KSPAddon, IEquatable<KSPAddonFixed>

{

private readonly Type type;

public KSPAddonFixed(KSPAddon.Startup startup, bool once, Type type)

: base(startup, once)

{

this.type = type;

}

public override bool Equals(object obj)

{

if (obj.GetType() != this.GetType()) { return false; }

return Equals((KSPAddonFixed)obj);

}

public bool Equals(KSPAddonFixed other)

{

if (this.once != other.once) { return false; }

if (this.startup != other.startup) { return false; }

if (this.type != other.type) { return false; }

return true;

}

public override int GetHashCode()

{

return this.startup.GetHashCode() ^ this.once.GetHashCode() ^ this.type.GetHashCode();

}

}

}