/*
 * Copyright (c) 2010, Tess Snider

 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.
*/

using Gtk;
using System;
using System.IO;
using System.IO.Compression;
using System.Collections.Generic;
using System.Text.RegularExpressions;

namespace DesertPaintLab
{
	public class PlayerProfile
	{
		string name;
		string directory;
		string reactFile;
        string reagentFile;
		
        static Regex recipeHeaderRegex = new Regex(@"^--- Recipe: (?<colorname>(\w*\s)*\w+)\s*");
        static Regex recipeIngredientRegex = new Regex(@"(?<ingredient>(\w+\s)?\w+)\s*\|\s*(?<quantity>\d+)\s*");

        ReactionSet reactions = new ReactionSet();
        // ingredient -> [ingredient, reaction]
		//SortedDictionary<string, SortedDictionary<string, Reaction>> reactions =
		//	new SortedDictionary<string, SortedDictionary<string, Reaction>>();
        SortedDictionary<string, PaintRecipe> recipes;

        static PlayerProfile current = null;

        static PlayerProfile Current
        {
            get {
                return current;
            }
        }

        public string LastError { get; private set; }
		
		public PlayerProfile(string name, string directory)
		{
			this.name = name;
			this.directory = directory;
			this.reactFile = System.IO.Path.Combine(directory, "dp_reactions.txt");
            this.reagentFile = System.IO.Path.Combine(directory, "ingredients.txt");
            this.recipes = new SortedDictionary<string, PaintRecipe>();
            foreach (PaintColor color in Palette.Colors)
            {
                this.recipes.Add(color.Name, new PaintRecipe());
            }
		}

        public string Directory
        {
            get {
                return this.directory;
            }
        }

        public ReactionSet Reactions
        {
            get {
                return this.reactions;
            }
        }

        public string ReagentFile
        {
            get {
                return this.reagentFile;
            }
        }

        public SortedDictionary<string, PaintRecipe> Recipes
        {
            get {
                return this.recipes;
            }
        }

        public int RecipeCount
        {
            get {
                int count = 0;
                foreach (PaintRecipe recipe in this.recipes.Values)
                {
                    if (recipe.IsValid)
                    {
                        ++count;
                    }
                }
                return count;
            }
        }
		
		public bool Initialize()
		{
            // Copy template files into new directory.
            string templatePath = FileUtils.FindApplicationResourceDirectory("template");
            
            if (!System.IO.Directory.Exists(templatePath))
            {
                LastError = "Failed to find profile template folder.";
                return false;
            }

			// Create new directory.
			System.IO.Directory.CreateDirectory(directory);
		    
			DirectoryInfo di = new DirectoryInfo(templatePath);
			FileInfo[] templateFiles = di.GetFiles();

			foreach (FileInfo file in templateFiles)
			{
                string destFile = System.IO.Path.Combine(directory, file.Name);
				System.IO.File.Copy(file.FullName, destFile, true);
                if (!File.Exists(destFile))
                {
                    LastError = "Failed to copy template file " + file.Name + ".";
                    return false;
                }					
			}
            return true;
		}
		
		public void ConvertFromPP(string ppFile, string dpFile)
		{
			string line;
			using (StreamReader reader = new StreamReader(ppFile))
			{
				using (StreamWriter writer = new StreamWriter(dpFile))
				{
					while ((line = reader.ReadLine()) != null) 
                	{
						string[] tokens = line.Split(null);
						if ((tokens.Length > 0) && (tokens[0] != "//"))
						{							
							// Write reaction.
							writer.Write(tokens[0] + " " + tokens[2] + " ");
							switch (tokens[4])
							{
							case "W":
								writer.WriteLine(tokens[6] + " " + tokens[6] + " " + tokens[6]);
								break;
							case "R":
								writer.WriteLine(tokens[6] + " 0 0");
								break;
							case "G":
								writer.WriteLine("0 " + tokens[6] + " 0");
								break;
							case "B":
								writer.WriteLine("0 0 " + tokens[6]);
								break;	
							}
							
							// Write reverse reaction.
							writer.Write(tokens[2] + " " + tokens[0] + " ");
							switch (tokens[4])
							{
							case "W":
								writer.WriteLine(tokens[8] + " " + tokens[8] + " " + tokens[8]);
								break;
							case "R":
								writer.WriteLine(tokens[8] + " 0 0");
								break;
							case "G":
								writer.WriteLine("0 " + tokens[8] + " 0");
								break;
							case "B":
								writer.WriteLine("0 0 " + tokens[8]);
								break;	
							}							
						}
					}
				}
			}
		}
		
		public bool SaveToPP(string ppFile)
		{
			Reaction reaction1, reaction2;
			using (StreamWriter writer = new StreamWriter(ppFile))
			{
                foreach (string reagentName1 in ReagentManager.Names)
                {
                    // TODO: could be more efficient by only iterating over the names after reagent1
                    foreach (string reagentName2 in ReagentManager.Names)
                    {
                        if (reagentName1.Equals(reagentName2))
                        {
                            continue;
                        }
                        Reagent reagent1 = ReagentManager.GetReagent(reagentName1);
                        Reagent reagent2 = ReagentManager.GetReagent(reagentName2);
                        reaction1 = reactions.Find(reagent1, reagent2);
                        if (reaction1 != null && !reaction1.Exported)
                        {
                            reaction2 = reactions.Find(reagent2, reagent1);
                            if (reaction2 != null)
                            {
								writer.Write(reagent1.PracticalPaintName + " | " + reagent2.PracticalPaintName + " | ");
								if ((Math.Abs(reaction1.Red) > Math.Abs(reaction1.Green)) ||
								    (Math.Abs(reaction2.Red) > Math.Abs(reaction2.Green)))
								{
									writer.WriteLine("R | " + reaction1.Red + " | " + reaction2.Red);	
								}
								else if ((Math.Abs(reaction1.Green) > Math.Abs(reaction1.Red)) ||
									(Math.Abs(reaction2.Green) > Math.Abs(reaction2.Red)))
								{
									writer.WriteLine("G | " + reaction1.Green + " | " + reaction2.Green);	
								}
								else if ((Math.Abs(reaction1.Blue) > Math.Abs(reaction1.Red)) ||
									(Math.Abs(reaction2.Blue) > Math.Abs(reaction2.Red)))
								{
									writer.WriteLine("B | " + reaction1.Blue + " | " + reaction2.Blue);
								}
								else
								{
									writer.WriteLine("W | " + reaction1.Red + " | " + reaction2.Red);	
								}
								reaction1.Exported = true;
								reaction2.Exported = true;
							}
						}
					}
				}
			}
			
			// Clear Exported flags.
            foreach (string reagentName1 in ReagentManager.Names)
            {
                // TODO: could be more efficient by only iterating over the names after reagent1
                foreach (string reagentName2 in ReagentManager.Names)
                {
                    if (reagentName1.Equals(reagentName2))
                    {
                        continue;
                    }
                    Reagent reagent1 = ReagentManager.GetReagent(reagentName1);
                    Reagent reagent2 = ReagentManager.GetReagent(reagentName2);
                    reaction1 = reactions.Find(reagent1, reagent2);
                    if (reaction1 != null)
                    {   
                        reaction1.Exported = false;
                    }
                    reaction2 = reactions.Find(reagent2, reagent1);
                    if (reaction2 != null)
                    {
                        reaction2.Exported = false;
                    }
				}
			}
			return true;
		}
		
		public void ImportFromPP(string importDir)
		{
			// Convert old file.
			ConvertFromPP(
			    System.IO.Path.Combine(importDir, "reactions.txt"),
				reactFile);
			try
			{
				// If there is an ingredients file, move it in.
				System.IO.File.Copy(
				    System.IO.Path.Combine(importDir, "ingredients.txt"),
				    System.IO.Path.Combine(directory, "ingredients.txt"),
				    true);
			}
			catch (Exception)
			{
				// If there is no ingredients file, we don't really care.	
			}
		}
		
        public void Import(string file)
        {
            ZipFile.ExtractToDirectory(file, directory);
        }

		public void Export(string file)
		{
            ZipFile.CreateFromDirectory(directory, file);
		}
		
		public bool Load()
		{
			string line;
			reactions.Clear();
            if (File.Exists(reagentFile))
            {
    			ReagentManager.LoadProfileReagents(reagentFile);
            }
            else
            {
                LastError = "Failed to find profile reagents file.";
                return false;
            }
			ReagentManager.InitializeReactions(ref reactions);
            if (!File.Exists(reactFile))
            {
                LastError = "Failed to find profile reactions file.";
                return false;
            }
			using (StreamReader reader = new StreamReader(reactFile))
			{
				while ((line = reader.ReadLine()) != null) 
               	{
					string[] tokens = line.Split(null);
                    Reagent reagent1 = ReagentManager.GetReagent(tokens[0]);
                    Reagent reagent2 = ReagentManager.GetReagent(tokens[1]);
					reactions.Set(reagent1, reagent2, new Reaction(int.Parse(tokens[2]), int.Parse(tokens[3]), int.Parse(tokens[4])));
				}
			}
            return true;
		}
		
		public void Save()
		{
			Reaction reaction;
			using (StreamWriter writer = new StreamWriter(reactFile, false))
			{
                foreach (string reagentName1 in ReagentManager.Names)
                {
                    // TODO: could be more efficient by only iterating over the names after reagent1
                    foreach (string reagentName2 in ReagentManager.Names)
                    {
                        if (reagentName1.Equals(reagentName2))
                        {
                            continue;
                        }
                        Reagent reagent1 = ReagentManager.GetReagent(reagentName1);
                        Reagent reagent2 = ReagentManager.GetReagent(reagentName2);
                        reaction = reactions.Find(reagent1, reagent2);
    					if (reaction != null)
    					{
                                writer.WriteLine(reagent1.PracticalPaintName + " " + reagent2.PracticalPaintName + " " +
    							reaction.Red + " " + reaction.Green + " " + reaction.Blue);
    					}
    				}
    			}
            }
		}

        public void LoadRecipes()
        {
            foreach (PaintRecipe recipe in this.recipes.Values)
            {
                recipe.Clear();
            }
            string recipeFile = System.IO.Path.Combine(directory, "dp_recipes.txt");
            string line;
            Match match;
            bool inRecipe = false;
            PaintRecipe testRecipe = new PaintRecipe();
            testRecipe.Reactions = reactions;
            string currentRecipeColor = null;
            if (File.Exists(recipeFile))
            {
                using (StreamReader reader = new StreamReader(recipeFile))
                {
                    while ((line = reader.ReadLine()) != null) 
                    {
                        match = recipeHeaderRegex.Match(line); 
                        if (match.Success)
                        {
                            if (testRecipe != null && currentRecipeColor != null)
                            {
                                recipes[currentRecipeColor].CopyFrom(testRecipe);
                            }
                            testRecipe.Clear();
                            currentRecipeColor = match.Groups["colorname"].Value;
                            inRecipe = true;
                        }
                        else if (inRecipe)
                        {
                            match = recipeIngredientRegex.Match(line);
                            if (match.Success)
                            {
                                string ingredient = match.Groups["ingredient"].Value;
                                uint quantity = uint.Parse(match.Groups["quantity"].Value);
                                testRecipe.AddReagent(ingredient, quantity);
                            }
                        }
                    }
                    if (inRecipe)
                    {
                        recipes[currentRecipeColor].CopyFrom(testRecipe);
                    }
                }
            }
        }

        public void SaveRecipes()
        {
            if (recipes != null)
            {
                string recipeFile = System.IO.Path.Combine(directory, "dp_recipes.txt");
                using (StreamWriter writer = new StreamWriter(recipeFile, false))
                {
                    foreach (KeyValuePair<string, PaintRecipe> pair in recipes)
                    {
                        writer.WriteLine("--- Recipe: {0}", pair.Key);
                        foreach (PaintRecipe.RecipeIngredient ingredient in pair.Value.Ingredients)
                        {
                            writer.WriteLine("{0,-14} | {1}", ingredient.name, ingredient.quantity);
                        }
                    }
                }
            }
        }

        public void ExportWikiRecipes(string file)
        {
            StreamWriter writer = new StreamWriter(file);
            ExportWikiFormat(writer);
        }
		
        public void ExportWikiFormat(TextWriter writer)
        {
            PaintRecipe recipe;
            List<KeyValuePair<string, string>> missing = new List<KeyValuePair<string, string>>();
            using (writer)
            {
                writer.WriteLine("{| class='wikitable sortable' border=\"1\" style=\"background-color:#DEB887;\"");
                writer.WriteLine("! Color !! Recipe !! Verified");    
                foreach (PaintColor color in Palette.Colors)
                {
                    writer.WriteLine("|-");
                    string colorLine = "| ";
                    colorLine += "style=\"font-weight: bold; background-color: #" + color.Red.ToString("X2") + color.Green.ToString("X2") + color.Blue.ToString("X2") + ";";
                    float rPortion = color.Red / 255.0f;
                    float gPortion = color.Green / 255.0f;
                    float bPortion = color.Blue / 255.0f;
                    float maxColor = Math.Max(rPortion, Math.Max(gPortion, bPortion));
                    float minColor = Math.Min(rPortion, Math.Min(gPortion, bPortion));
                    float luminance = (maxColor - minColor) * 0.5f;
                    if (luminance < 0.5f)
                    {
                        // dark color gets light text
                        colorLine += " color: #FFFFFF;";
                    }
                    colorLine += "\" | " + color.Name + " || ";
                    if (recipes.TryGetValue(color.Name, out recipe))
                    {
                        foreach (PaintRecipe.RecipeIngredient ingredient in recipe.Ingredients)
                        {
                            colorLine += " " + ingredient.ToString();
                        }
                    } 
                    else
                    {
                        // no recipe
                    }
                    colorLine += " || ";
                    
                    if (recipe.CheckMissingReactions(ref missing) == false)
                    {
                        colorLine += "Y";
                    }
                    writer.WriteLine(colorLine);
                }
                writer.WriteLine("|}");
            }
        }
        
		public Reaction FindReaction(Reagent reagent1, Reagent reagent2)
		{
            return reactions.Find(reagent1, reagent2);
		}
		
		public void SetReaction(Reagent reagent1, Reagent reagent2, Reaction reaction)
		{
            reactions.Set(reagent1, reagent2, reaction);
		}

        public void SetRecipe(PaintRecipe recipe)
        {
            string colorName = Palette.FindNearest(recipe.ReactedColor);
            recipes[colorName].CopyFrom(recipe);
        }
	}
}