// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
#nullable disable
 
using Microsoft.NET.Sdk.WorkloadManifestReader;
 
namespace ManifestReaderTests
{
    internal class MockManifestProvider : IWorkloadManifestProvider
    {
        readonly (string name, string path, string manifestVersion, string featureBand)[] _manifests;
 
        public MockManifestProvider(params string[] manifestPaths)
        {
            _manifests = Array.ConvertAll(manifestPaths, mp =>
            {
                string manifestId = Path.GetFileNameWithoutExtension(Path.GetDirectoryName(mp));
                return (manifestId, mp, (string)null, (string)null);
            });
            SdkFeatureBand = new SdkFeatureBand("6.0.100");
        }
 
        public MockManifestProvider(params (string name, string path, string manifestVersion, string featureBand)[] manifests)
        {
            _manifests = manifests;
            SdkFeatureBand = new SdkFeatureBand("6.0.100");
        }
 
        public SdkFeatureBand SdkFeatureBand { get; set; }
 
        public Dictionary<string, WorkloadSet> GetAvailableWorkloadSets() => new();
 
        public void RefreshWorkloadManifests() { }
 
        public IEnumerable<ReadableWorkloadManifest> GetManifests()
        {
            foreach ((var id, var path, var manifestVersion, var featureBand) in _manifests)
            {
                yield return new(
                    id,
                    Path.GetDirectoryName(path),
                    path,
                    featureBand ?? SdkFeatureBand.ToString(),
                    manifestVersion,
                    () => File.OpenRead(path),
                    () => WorkloadManifestReader.TryOpenLocalizationCatalogForManifest(path)
                );
            }
        }
 
        public string GetSdkFeatureBand() => SdkFeatureBand.ToString();
        public IWorkloadManifestProvider.WorkloadVersionInfo GetWorkloadVersion() => new IWorkloadManifestProvider.WorkloadVersionInfo(SdkFeatureBand.ToString() + ".2");
    }
}