/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

using Amazon;
using Amazon.Polly;
using Amazon.Polly.Model;
using Amazon.SecurityToken;
using Amazon.SecurityToken.Model;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using System.Net;
using System;
using System.Collections.Generic;
using System.IO;
using ThirdParty.Json.LitJson;
using System.Text;

namespace AWSSDK_DotNet.IntegrationTests.Tests.Polly
{
    [TestClass]
    public class PollyTests
    {
        [TestMethod]
        [TestCategory("Polly")]
        public void HappyCaseAPI()
        {
            using (var client = new AmazonPollyClient(RegionEndpoint.USWest2))
            {
                var response = client.SynthesizeSpeech(GetMp3Request());
                Assert.AreEqual(HttpStatusCode.OK, response.HttpStatusCode);
                Assert.IsTrue(response.AudioStream.ReadByte() > -1);
            }
        }

        [TestMethod]
        [TestCategory("Polly")]
        public void APIWithSpeechMarks()
        {
            using (var client = new AmazonPollyClient(RegionEndpoint.USWest2))
            {
                var response = client.SynthesizeSpeech(GetSpeechMarkRequest());
                Assert.AreEqual(HttpStatusCode.OK, response.HttpStatusCode);
                using (var streamReader = new StreamReader(response.AudioStream))
                {
                    AssertSpeechMarks(JsonMapper.ToObject(streamReader));
                }
            }
        }

        [TestMethod]
        [TestCategory("Polly")]
        public void HappyCasePresignedUrl()
        {
            AssertPreSignedUrl(SynthesizeSpeechUtil.GeneratePresignedUrl(RegionEndpoint.USWest2, GetMp3Request()));
        }

        [TestMethod]
        [TestCategory("Polly")]
        public void PresignedUrlWithSpeechMarks()
        {
            var data = AssertPreSignedUrl(SynthesizeSpeechUtil.GeneratePresignedUrl(RegionEndpoint.USWest2, GetSpeechMarkRequest()));
            AssertSpeechMarks(JsonMapper.ToObject(Encoding.UTF8.GetString(data)));
        }

        [TestMethod]
        [TestCategory("Polly")]
        [TestCategory("RequiresIAMUser")]
        public void PresignedUrlWithSessionToken()
        {
            var stsClient = new AmazonSecurityTokenServiceClient(RegionEndpoint.USWest2);
            var response = stsClient.GetSessionToken(new GetSessionTokenRequest
            {
                DurationSeconds = 900
            });
            AssertPreSignedUrl(SynthesizeSpeechUtil.GeneratePresignedUrl(response.Credentials, RegionEndpoint.USWest2, GetMp3Request()));
        }

        [TestMethod]
        [TestCategory("Polly")]
        public void EnsureIsUrlEncoded()
        {
            var request = GetMp3Request();
            request.Text = "hello == hello"; // we will get an exception if the == isn't encoded
            AssertPreSignedUrl(SynthesizeSpeechUtil.GeneratePresignedUrl(RegionEndpoint.USWest2, GetMp3Request()));
        }

        private static void AssertSpeechMarks(JsonData speechMarksJsonData)
        {
            var names = new HashSet<string>(speechMarksJsonData.PropertyNames);
            Assert.IsTrue(names.Contains("time"));
            Assert.IsTrue(names.Contains("type"));
            Assert.IsTrue(names.Contains("start"));
            Assert.IsTrue(names.Contains("end"));
            Assert.IsTrue(names.Contains("value"));
        }

        private static byte[] AssertPreSignedUrl(string url)
        {
            // download using the PreSigned URL in a totally independent WebClient
            byte[] data;
            using (var webClient = new WebClient())
            {
                data = webClient.DownloadData(url);
            }

            // make sure there are no exceptions, and that there's something there
            Assert.IsNotNull(data);
            Assert.IsTrue(data.Length > 0);

            return data;
        }

        private static SynthesizeSpeechRequest GetMp3Request()
        {
            return new SynthesizeSpeechRequest()
            {
                VoiceId = "Joanna",
                Text = "Hello",
                OutputFormat = OutputFormat.Mp3
            };
        }

        private static SynthesizeSpeechRequest GetSpeechMarkRequest()
        {
            return new SynthesizeSpeechRequest()
            {
                VoiceId = "Joanna",
                Text = "Hello",
                OutputFormat = OutputFormat.Json,
                SpeechMarkTypes = new List<string> { "word", "sentence" }
            };
        }
    }
}