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

using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.Configuration;
using System;
using System.IO;
using System.Security.Cryptography.X509Certificates;
using System.Threading;
using Microsoft.AspNetCore.Builder;
using Microsoft.Extensions.DependencyInjection;

namespace AspnetCoreModule.TestSites.Standard
{
    public static class Program
    {
        public static IApplicationLifetime AappLifetime;
        public static bool AappLifetimeStopping = false;
        public static int GracefulShutdownDelayTime = 0;

        private static X509Certificate2 _x509Certificate2;

        public static void Main(string[] args)
        {
            var config = new ConfigurationBuilder()
                .AddCommandLine(args)
                .Build();

            string startUpClassString = Environment.GetEnvironmentVariable("ANCMTestStartupClassName");
            IWebHostBuilder builder = null;
            if (!string.IsNullOrEmpty(startUpClassString))
            {
                if (startUpClassString == "StartupHTTPS")
                {
                    // load .\testresources\testcert.pfx
                    string pfxPassword = "testPassword";
                    if (File.Exists(@".\TestResources\testcert.pfx"))
                    {
                        _x509Certificate2 = new X509Certificate2(@".\TestResources\testcert.pfx", pfxPassword);
                    }
                    else
                    {
                        throw new Exception(@"Certificate file not found: .\TestResources\testcert.pfx of which password should " + pfxPassword);
                    }

                    builder = new WebHostBuilder()
                        .UseConfiguration(config)
                        .UseIISIntegration()
                        .UseKestrel()
                        .UseStartup<Startup>();
                }
                else if (startUpClassString == "StartupCompressionCaching")
                {
                    builder = new WebHostBuilder()
                        .UseConfiguration(config)
                        .UseIISIntegration()
                        .UseContentRoot(Directory.GetCurrentDirectory())
                        .UseStartup<StartupCompressionCaching>();
                }
                else if (startUpClassString == "StartupNoCompressionCaching")
                {
                    StartupCompressionCaching.CompressionMode = false;
                    builder = new WebHostBuilder()
                        .UseConfiguration(config)
                        .UseIISIntegration()
                        .UseContentRoot(Directory.GetCurrentDirectory())
                        .UseStartup<StartupCompressionCaching>();
                }
                else if (startUpClassString == "StartupHelloWorld")
                {
                    builder = new WebHostBuilder()
                        .UseConfiguration(config)
                        .UseIISIntegration()
                        .UseStartup<StartupHelloWorld>();
                }
                else if (startUpClassString == "StartupNtlmAuthentication")
                {
                    builder = new WebHostBuilder()
                        .UseConfiguration(config)
                        .UseIISIntegration()
                        .UseStartup<StartupNtlmAuthentication>();
                }
                else
                {
                    throw new System.Exception("Invalid startup class name : " + startUpClassString);
                }
            }
            else
            {
                builder = new WebHostBuilder()
                    .ConfigureServices(services =>
                    {
                        const string PairingToken = "TOKEN";
                        string paringToken = builder.GetSetting(PairingToken) ?? Environment.GetEnvironmentVariable($"ASPNETCORE_{PairingToken}");
                        services.AddSingleton<IStartupFilter>(
                            new IISSetupFilter(paringToken)
                        );
                    })
                    .UseConfiguration(config)
                    .UseStartup<Startup>();
            }

            string startupDelay = Environment.GetEnvironmentVariable("ANCMTestStartUpDelay");
            if (!string.IsNullOrEmpty(startupDelay))
            {
                Startup.SleeptimeWhileStarting = Convert.ToInt32(startupDelay);
            }

            if (Startup.SleeptimeWhileStarting != 0)
            {
                Thread.Sleep(Startup.SleeptimeWhileStarting);
            }

            string shutdownDelay = Environment.GetEnvironmentVariable("ANCMTestShutdownDelay");
            if (!string.IsNullOrEmpty(shutdownDelay))
            {
                Startup.SleeptimeWhileClosing = Convert.ToInt32(shutdownDelay);
            }

            builder.UseKestrel();

            var host = builder.Build();
            AappLifetime = (IApplicationLifetime)host.Services.GetService(typeof(IApplicationLifetime));

            string gracefulShutdownDelay = Environment.GetEnvironmentVariable("GracefulShutdownDelayTime");
            if (!string.IsNullOrEmpty(gracefulShutdownDelay))
            {
                GracefulShutdownDelayTime = Convert.ToInt32(gracefulShutdownDelay);
            }
            AappLifetime.ApplicationStarted.Register(
                () => {
                    Thread.Sleep(1000);
                }
            );
            AappLifetime.ApplicationStopping.Register(
                () => {
                    AappLifetimeStopping = true;
                    Thread.Sleep(Startup.SleeptimeWhileClosing / 2);
                }
            );
            AappLifetime.ApplicationStopped.Register(
                () => {
                    Thread.Sleep(Startup.SleeptimeWhileClosing / 2);
                    Startup.SleeptimeWhileClosing = 0;   // All of SleeptimeWhileClosing is used now
                }
            );
            try
            {
                host.Run();
            }
            catch
            {
                // ignore
            }

            if (Startup.SleeptimeWhileClosing != 0)
            {
                Thread.Sleep(Startup.SleeptimeWhileClosing);
            }
        }
    }
}