using System;
using System.Collections.Generic;

using System.Linq;
using System.Text;
using System.Xml;

namespace Roblox.XmlFormat
{
    public class XmlRobloxFile : IRobloxFile
    {
        // IRobloxFile
        internal readonly Instance XmlContents = new Instance("Folder", "XmlRobloxFile");
        public Instance Contents => XmlContents;

        // Runtime Specific
        public readonly XmlDocument Root = new XmlDocument();
        public Dictionary<string, Instance> Instances = new Dictionary<string, Instance>();
        
        public void ReadFile(byte[] buffer)
        {
            try
            {
                string xml = Encoding.UTF8.GetString(buffer);
                Root.LoadXml(xml);
            }
            catch
            {
                throw new Exception("XmlRobloxFile: Could not read XML!");
            }

            XmlNode roblox = Root.FirstChild;

            if (roblox != null && roblox.Name == "roblox")
            {
                // Verify the version we are using.
                XmlNode version = roblox.Attributes.GetNamedItem("version");
                int schemaVersion;

                if (version == null || !int.TryParse(version.Value, out schemaVersion))
                    throw new Exception("XmlRobloxFile: No version number defined!");
                else if (schemaVersion < 4)
                    throw new Exception("XmlRobloxFile: Provided version must be at least 4!");

                // Process the instances.
                foreach (XmlNode child in roblox.ChildNodes)
                {
                    if (child.Name == "Item")
                    {
                        Instance item = XmlDataReader.ReadInstance(child, ref Instances);
                        item.Parent = XmlContents;
                    }
                }

                // Resolve references for Ref properties.
                var refProps = Instances.Values
                    .SelectMany(inst => inst.Properties)
                    .Where(prop => prop.Type == PropertyType.Ref);
                    
                foreach (Property refProp in refProps)
                {
                    string refId = refProp.Value as string;
                    if (Instances.ContainsKey(refId))
                    {
                        Instance refInst = Instances[refId];
                        refProp.Value = refInst;
                    }
                    else if (refId != "null")
                    {
                        Console.WriteLine("XmlRobloxFile: Could not resolve reference for " + refProp.GetFullName());
                    }
                }
            }
            else
            {
                throw new Exception("XmlRobloxFile: No `roblox` tag found!");
            }
        }
    }
}