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

using System.Collections.Generic;
using System.Linq;
using System.Xml.Linq;

namespace Microsoft.Extensions.FileProviders.Embedded.Manifest
{
    class TestEntry
    {
        public bool IsFile => ResourcePath != null;
        public string Name { get; set; }
        public TestEntry[] Children { get; set; }
        public string ResourcePath { get; set; }

        public static TestEntry Directory(string name, params TestEntry[] entries) =>
            new TestEntry() { Name = name, Children = entries };

        public static TestEntry File(string name, string path = null) =>
            new TestEntry() { Name = name, ResourcePath = path ?? name };

        public XElement ToXElement() => IsFile ?
            new XElement("File", new XAttribute("Name", Name), new XElement("ResourcePath", ResourcePath)) :
            new XElement("Directory", new XAttribute("Name", Name), Children.Select(c => c.ToXElement()));

        public IEnumerable<TestEntry> GetFiles()
        {
            if (IsFile)
            {
                return Enumerable.Empty<TestEntry>();
            }

            var files = Children.Where(c => c.IsFile).ToArray();
            var otherFiles = Children.Where(c => !c.IsFile).SelectMany(d => d.GetFiles()).ToArray();

            return files.Concat(otherFiles).ToArray();
        }

    }
}
