// 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.Core.Burn.Bundles
{
    using System;
    using System.Collections.Generic;
    using System.Diagnostics;
    using System.IO;
    using System.Linq;
    using WixToolset.Core.Cab;
    using WixToolset.Data;
    using WixToolset.Data.Rows;

    /// <summary>
    /// Creates cabinet files.
    /// </summary>
    internal class CreateContainerCommand
    {
        public CompressionLevel DefaultCompressionLevel { private get; set; }

        public IEnumerable<WixBundlePayloadRow> Payloads { private get; set; }

        public string ManifestFile { private get; set; }

        public string OutputPath { private get; set; }

        public string Hash { get; private set; }

        public long Size { get; private set; }

        public void Execute()
        {
            int payloadCount = this.Payloads.Count(); // The number of embedded payloads

            if (!String.IsNullOrEmpty(this.ManifestFile))
            {
                ++payloadCount;
            }

            using (var cab = new WixCreateCab(Path.GetFileName(this.OutputPath), Path.GetDirectoryName(this.OutputPath), payloadCount, 0, 0, this.DefaultCompressionLevel))
            {
                // If a manifest was provided always add it as "payload 0" to the container.
                if (!String.IsNullOrEmpty(this.ManifestFile))
                {
                    cab.AddFile(this.ManifestFile, "0");
                }

                foreach (WixBundlePayloadRow payload in this.Payloads)
                {
                    Debug.Assert(PackagingType.Embedded == payload.Packaging);

                    Messaging.Instance.OnMessage(WixVerboses.LoadingPayload(payload.FullFileName));

                    cab.AddFile(payload.FullFileName, payload.EmbeddedId);
                }

                cab.Complete();
            }

            // Now that the container is created, set the outputs of the command.
            FileInfo fileInfo = new FileInfo(this.OutputPath);

            this.Hash = Common.GetFileHash(fileInfo.FullName);

            this.Size = fileInfo.Length;
        }
    }
}