// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System;
using System.Net.Http;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Sockets.Client;
using Microsoft.Extensions.Logging;
using Microsoft.AspNetCore.Sockets;

namespace ClientSample
{
    internal class RawSample
    {
        public static async Task MainAsync(string[] args)
        {
            var baseUrl = "http://localhost:5000/chat";
            if (args.Length > 0)
            {
                baseUrl = args[0];
            }

            var loggerFactory = new LoggerFactory();
            loggerFactory.AddConsole(LogLevel.Debug);
            var logger = loggerFactory.CreateLogger<Program>();

            using (var httpClient = new HttpClient(new LoggingMessageHandler(loggerFactory, new HttpClientHandler())))
            {
                logger.LogInformation("Connecting to {0}", baseUrl);
                var transport = new LongPollingTransport(httpClient, loggerFactory);
                using (var connection = new Connection(new Uri(baseUrl), loggerFactory))
                {
                    await connection.StartAsync(transport, httpClient);

                    logger.LogInformation("Connected to {0}", baseUrl);

                    var cts = new CancellationTokenSource();
                    Console.CancelKeyPress += (sender, a) =>
                    {
                        a.Cancel = true;
                        logger.LogInformation("Stopping loops...");
                        cts.Cancel();
                    };

                    // Ready to start the loops
                    var receive =
                        StartReceiving(loggerFactory.CreateLogger("ReceiveLoop"), connection, cts.Token).ContinueWith(_ => cts.Cancel());
                    var send =
                        StartSending(loggerFactory.CreateLogger("SendLoop"), connection, cts.Token).ContinueWith(_ => cts.Cancel());

                    await Task.WhenAll(receive, send);
                    await connection.StopAsync();
                }
            }
        }

        private static async Task StartSending(ILogger logger, Connection connection, CancellationToken cancellationToken)
        {
            logger.LogInformation("Send loop starting");
            while (!cancellationToken.IsCancellationRequested)
            {
                var line = Console.ReadLine();
                logger.LogInformation("Sending: {0}", line);

                await connection.SendAsync(Encoding.UTF8.GetBytes("Hello World"), MessageType.Text);
            }
            logger.LogInformation("Send loop terminated");
        }

        private static async Task StartReceiving(ILogger logger, Connection connection, CancellationToken cancellationToken)
        {
            logger.LogInformation("Receive loop starting");
            try
            {
                var receiveData = new ReceiveData();
                while (await connection.ReceiveAsync(receiveData, cancellationToken))
                {
                    logger.LogInformation($"Received: {Encoding.UTF8.GetString(receiveData.Data)}");
                }
            }
            catch (OperationCanceledException)
            {
                logger.LogInformation("Connection is closing");
            }
            catch (Exception ex)
            {
                logger.LogError(0, ex, "Connection terminated due to an exception");
            }

            logger.LogInformation("Receive loop terminated");
        }
    }
}