// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using Microsoft.NET.Sdk.WorkloadManifestReader;

namespace ManifestReaderTests
{
    internal class MockManifestProvider : IWorkloadManifestProvider
    {
        readonly (string name, string path, 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);
            });
            SdkFeatureBand = new SdkFeatureBand("6.0.100");
        }

        public MockManifestProvider(params (string name, string path, 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 featureBand) in _manifests)
            {
                yield return new(
                    id,
                    Path.GetDirectoryName(path),
                    path,
                    featureBand ?? SdkFeatureBand.ToString(),
                    string.Empty,
                    () => File.OpenRead(path),
                    () => WorkloadManifestReader.TryOpenLocalizationCatalogForManifest(path)
                );
            }
        }

        public string GetSdkFeatureBand() => SdkFeatureBand.ToString();
    }
}
