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

using System;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;

// HostingStartup's in the primary assembly are run automatically.
[assembly: HostingStartup(typeof(SampleStartups.StartupInjection))]

namespace SampleStartups
{
    public class StartupInjection : IHostingStartup
    {
        public void Configure(IWebHostBuilder builder)
        {
            builder.UseStartup<InjectedStartup>();
        }

        // Entry point for the application.
        public static Task Main(string[] args)
        {
            var host = new HostBuilder()
                .ConfigureWebHost(webHostBuilder =>
                {
                    webHostBuilder
                        .UseKestrel()
                        // Each of these three sets ApplicationName to the current assembly, which is needed in order to
                        // scan the assembly for HostingStartupAttributes.
                        // .UseSetting(WebHostDefaults.ApplicationKey, "SampleStartups")
                        // .Configure(_ => { })
                        .UseStartup<NormalStartup>();
                })
                .Build();

            return host.RunAsync();
        }
    }

    public class NormalStartup
    {
        public void ConfigureServices(IServiceCollection services)
        {
            Console.WriteLine("NormalStartup.ConfigureServices");
        }

        public void Configure(IApplicationBuilder app)
        {
            Console.WriteLine("NormalStartup.Configure");
            app.Run(async (context) =>
            {
                await context.Response.WriteAsync("Hello World!");
            });
        }
    }

    public class InjectedStartup
    {
        public void ConfigureServices(IServiceCollection services)
        {
            Console.WriteLine("InjectedStartup.ConfigureServices");
        }

        public void Configure(IApplicationBuilder app)
        {
            Console.WriteLine("InjectedStartup.Configure");
            app.Run(async (context) =>
            {
                await context.Response.WriteAsync("Hello World!");
            });
        }
    }
}
