ContainerProvider.cs

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE.txt file in the project root for more information.

using Microsoft.Extensions.Logging;

namespace Sign.Core
{
    internal sealed class ContainerProvider : IContainerProvider
    {
        private readonly HashSet<string> _appxBundleExtensions;
        private readonly HashSet<string> _appxExtensions;
        private readonly IDirectoryService _directoryService;
        private readonly IFileMatcher _fileMatcher;
        private readonly ICertificateProvider _certificateProvider;
        private readonly ILogger<IContainerProvider> _logger;
        private readonly IMakeAppxCli _makeAppxCli;
        private readonly HashSet<string> _nuGetExtensions;
        private readonly HashSet<string> _zipExtensions;
        private const string _cabExtension = ".cab";

        // Dependency injection requires a public constructor.
        public ContainerProvider(
            ICertificateProvider certificateProvider,
            IDirectoryService directoryService,
            IFileMatcher fileMatcher,
            IMakeAppxCli makeAppxCli,
            ILogger<IContainerProvider> logger)
        {
            ArgumentNullException.ThrowIfNull(certificateProvider, nameof(certificateProvider));
            ArgumentNullException.ThrowIfNull(directoryService, nameof(directoryService));
            ArgumentNullException.ThrowIfNull(fileMatcher, nameof(fileMatcher));
            ArgumentNullException.ThrowIfNull(makeAppxCli, nameof(makeAppxCli));
            ArgumentNullException.ThrowIfNull(logger, nameof(logger));

            _certificateProvider = certificateProvider;
            _directoryService = directoryService;
            _fileMatcher = fileMatcher;
            _makeAppxCli = makeAppxCli;
            _logger = logger;

            _appxBundleExtensions = new HashSet<string>(StringComparer.OrdinalIgnoreCase)
            {
                ".appxbundle",
                ".eappxbundle",
                ".emsixbundle",
                ".msixbundle"
            };

            _appxExtensions = new HashSet<string>(StringComparer.OrdinalIgnoreCase)
            {
                ".appx",
                ".eappx",
                ".emsix",
                ".msix"
            };

            _nuGetExtensions = new HashSet<string>(StringComparer.OrdinalIgnoreCase)
            {
                ".nupkg",
                ".snupkg"
            };

            _zipExtensions = new HashSet<string>(StringComparer.OrdinalIgnoreCase)
            {
                ".appxupload",
                ".clickonce",
                ".msixupload",
                ".vsix",
                ".zip"
            };
        }

        public bool IsAppxBundleContainer(FileInfo file)
        {
            ArgumentNullException.ThrowIfNull(file, nameof(file));

            return _appxBundleExtensions.Contains(file.Extension);
        }

        public bool IsAppxContainer(FileInfo file)
        {
            ArgumentNullException.ThrowIfNull(file, nameof(file));

            return _appxExtensions.Contains(file.Extension);
        }

        public bool IsNuGetContainer(FileInfo file)
        {
            ArgumentNullException.ThrowIfNull(file, nameof(file));

            return _nuGetExtensions.Contains(file.Extension);
        }

        public bool IsZipContainer(FileInfo file)
        {
            ArgumentNullException.ThrowIfNull(file, nameof(file));

            return _zipExtensions.Contains(file.Extension);
        }

        public bool IsCabContainer(FileInfo file)
        {
            ArgumentNullException.ThrowIfNull(file, nameof(file));

            return string.Equals(file.Extension, _cabExtension, StringComparison.OrdinalIgnoreCase);
        }

        public IContainer? GetContainer(FileInfo file)
        {
            ArgumentNullException.ThrowIfNull(file, nameof(file));

            if (IsAppxBundleContainer(file))
            {
                return new AppxBundleContainer(file, _directoryService, _fileMatcher, _makeAppxCli, _logger);
            }

            if (IsAppxContainer(file))
            {
                return new AppxContainer(file, _certificateProvider, _directoryService, _fileMatcher, _makeAppxCli, _logger);
            }

            if (IsZipContainer(file))
            {
                return new ZipContainer(file, _directoryService, _fileMatcher, _logger);
            }

            if (IsNuGetContainer(file))
            {
                return new NuGetContainer(file, _directoryService, _fileMatcher, _logger);
            }

            if (IsCabContainer(file))
            {
                return new CabContainer(file, _directoryService, _fileMatcher, _logger);
            }

            return null;
        }
    }
}