// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

using System;
using System.Collections.Generic;
using Microsoft.TypeSpec.Generator.Input;
using Microsoft.TypeSpec.Generator.Primitives;
using Microsoft.TypeSpec.Generator.Statements;
using Microsoft.TypeSpec.Generator.Tests.Common;
using NUnit.Framework;

namespace Microsoft.TypeSpec.Generator.Tests.Providers
{
    internal class PropertyDescriptionTests
    {
        // Validates that the property description string is constructed correctly based on the property type
        [TestCaseSource(nameof(BuildPropertyDescriptionTestCases))]
        public void BuildPropertyDescription(InputModelProperty inputModelProperty, CSharpType type)
        {
            var propertySummaryStatement = PropertyDescriptionBuilder.BuildPropertyDescription(inputModelProperty, type, SerializationFormat.Default, PropertyDescriptionBuilder.CreateDefaultPropertyDescription(inputModelProperty.Name, false));
            using CodeWriter codeWriter = new CodeWriter();
            propertySummaryStatement.Write(codeWriter);
            string propertyDescription = codeWriter.ToString(false);
            var propertyDescriptionString = string.Join(Environment.NewLine, propertyDescription);
            Assert.IsNotNull(propertyDescription);
            Assert.IsNotEmpty(propertyDescriptionString);

            if (type.ContainsBinaryData)
            {
                Assert.IsTrue(propertyDescriptionString.Contains("Examples:"));
            }
        }

        // Validates that the summary description string is constructed correctly for several types
        [Test]
        public void TestGetUnionTypesDescriptions()
        {
            var dateTime = new DateTimeOffset(1, 2, 3, 4, 5, 6, TimeSpan.Zero);
            var expected = Helpers.GetExpectedFromFile();

            var unionItems = new List<CSharpType>
            {
                new CSharpType(typeof(bool), false),
                new CSharpType(typeof(int), false),
                new CSharpType(typeof(IDictionary<,>), false, typeof(string), new CSharpType(typeof(int))),
                CSharpType.FromLiteral(new CSharpType(typeof(int), false), 21),
                CSharpType.FromLiteral(new CSharpType(typeof(string), false), "test"),
                CSharpType.FromLiteral(new CSharpType(typeof(bool), false), true),
                CSharpType.FromLiteral(new CSharpType(typeof(DateTimeOffset), false), dateTime)
            };

            IReadOnlyList<XmlDocStatement> descriptions = PropertyDescriptionBuilder.GetUnionTypesDescriptions(unionItems);

            Assert.AreEqual(7, descriptions.Count);

            using var codeWriter = new CodeWriter();
            var xmlDoc = new XmlDocStatement("test", [], innerStatements: [.. descriptions]);
            xmlDoc.Write(codeWriter);
            var actual = codeWriter.ToString(false);

            Assert.AreEqual(expected, actual);
        }

        public static IEnumerable<TestCaseData> BuildPropertyDescriptionTestCases
        {
            get
            {
                // list property
                yield return new TestCaseData(InputFactory.Property("prop1", InputFactory.Array(InputPrimitiveType.String)), new CSharpType(typeof(IList<string>)));
                // list of binary data property
                yield return new TestCaseData(InputFactory.Property("prop1", InputFactory.Array(InputPrimitiveType.Any), isReadOnly: true), new CSharpType(typeof(IReadOnlyList<BinaryData>)));
                // dictionary property with binary data value
                yield return new TestCaseData(InputFactory.Property("prop1", InputFactory.Dictionary(InputPrimitiveType.Any)), new CSharpType(typeof(IDictionary<string, BinaryData>)));
                // nullable dictionary property
                yield return new TestCaseData(InputFactory.Property("prop1", InputFactory.Dictionary(InputPrimitiveType.String)), new CSharpType(typeof(IDictionary<string, string>)));
                // primitive type property
                yield return new TestCaseData(InputFactory.Property("prop1", InputPrimitiveType.String), new CSharpType(typeof(string)));
                // binary data property
                yield return new TestCaseData(InputFactory.Property("prop1", InputPrimitiveType.Any, isReadOnly: true), new CSharpType(typeof(BinaryData)));
            }
        }
    }
}
