// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.

namespace WixToolset.Mba.Core
{
    using System;
    using System.Collections.Generic;
    using System.IO;
    using System.Xml;
    using System.Xml.XPath;

    /// <summary>
    /// Default implementation of <see cref="IBundleInfo"/>.
    /// </summary>
    public class BundleInfo : IBundleInfo
    {
        /// <inheritdoc/>
        public bool PerMachine { get; internal set; }

        /// <inheritdoc/>
        public string Name { get; internal set; }

        /// <inheritdoc/>
        public string LogVariable { get; internal set; }

        /// <inheritdoc/>
        public IOverridableVariables OverridableVariables { get; internal set; }

        /// <inheritdoc/>
        public IDictionary<string, IPackageInfo> Packages { get; internal set; }

        internal BundleInfo()
        {
            this.Packages = new Dictionary<string, IPackageInfo>();
        }

        /// <inheritdoc/>
        public IPackageInfo AddRelatedBundleAsPackage(DetectRelatedBundleEventArgs e)
        {
            var package = PackageInfo.GetRelatedBundleAsPackage(e.ProductCode, e.RelationType, e.PerMachine, e.Version);
            this.Packages.Add(package.Id, package);
            return package;
        }

        /// <inheritdoc/>
        public IPackageInfo AddUpdateBundleAsPackage(SetUpdateCompleteEventArgs e)
        {
            var package = PackageInfo.GetUpdateBundleAsPackage(e.NewPackageId);
            this.Packages.Add(package.Id, package);
            return package;
        }

        /// <summary>
        /// Parses BA manifest from the given stream.
        /// </summary>
        /// <param name="stream"></param>
        /// <returns></returns>
        public static IBundleInfo ParseBundleFromStream(Stream stream)
        {
            XPathDocument manifest = new XPathDocument(stream);
            XPathNavigator root = manifest.CreateNavigator();
            return ParseBundleFromXml(root);
        }

        /// <summary>
        /// Parses BA manifest from the given <see cref="XPathNavigator"/>.
        /// </summary>
        /// <param name="root">The root of the BA manifest.</param>
        /// <returns></returns>
        public static IBundleInfo ParseBundleFromXml(XPathNavigator root)
        {
            BundleInfo bundle = new BundleInfo();

            XmlNamespaceManager namespaceManager = new XmlNamespaceManager(root.NameTable);
            namespaceManager.AddNamespace("p", BootstrapperApplicationData.XMLNamespace);
            XPathNavigator bundleNode = root.SelectSingleNode("/p:BootstrapperApplicationData/p:WixBundleProperties", namespaceManager);

            if (bundleNode == null)
            {
                throw new Exception("Failed to select bundle information.");
            }

            bool? perMachine = BootstrapperApplicationData.GetYesNoAttribute(bundleNode, "PerMachine");
            if (perMachine.HasValue)
            {
                bundle.PerMachine = perMachine.Value;
            }

            bundle.Name = BootstrapperApplicationData.GetAttribute(bundleNode, "DisplayName");

            bundle.LogVariable = BootstrapperApplicationData.GetAttribute(bundleNode, "LogPathVariable");

            bundle.OverridableVariables = OverridableVariablesInfo.ParseFromXml(root);

            bundle.Packages = PackageInfo.ParsePackagesFromXml(root);

            return bundle;
        }
    }
}