using Asp.Versioning;
using FluentValidation;
using Microsoft.AspNetCore.Authentication.JwtBearer;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Options;
using OpenTelemetry.Trace;
using System.Diagnostics;
using Vegasco.WebApi.Authentication;
using Vegasco.WebApi.Endpoints;
using Vegasco.WebApi.Endpoints.OpenApi;
using Vegasco.WebApi.Persistence;

namespace Vegasco.WebApi.Common;

public static class DependencyInjectionExtensions
{
	/// <summary>
	///     Adds all the WebApi related services to the Dependency Injection container.
	/// </summary>
	/// <param name="services"></param>
	/// <param name="configuration"></param>
	/// <param name="environment"></param>
	public static void AddWebApiServices(this IServiceCollection services, IConfiguration configuration, IHostEnvironment environment)
	{
		services
			.AddMiscellaneousServices()
			.AddOpenApi()
			.AddApiVersioning()
			.AddOtel()
			.AddAuthenticationAndAuthorization(environment)
			.AddDbContext(configuration);
	}

	private static IServiceCollection AddMiscellaneousServices(this IServiceCollection services)
	{
		services.AddResponseCompression();

		services.AddValidatorsFromAssemblies(
		[
			typeof(IWebApiMarker).Assembly
		], ServiceLifetime.Singleton);

		services.AddHealthChecks();
		services.AddEndpointsFromAssemblyContaining<IWebApiMarker>();

		services.AddHttpContextAccessor();

		return services;
	}

	private static IServiceCollection AddOpenApi(this IServiceCollection services)
	{
		services.ConfigureOptions<ConfigureSwaggerGenOptions>();

		services.AddEndpointsApiExplorer();
		services.AddSwaggerGen(o =>
		{
			o.CustomSchemaIds(type =>
			{
				if (string.IsNullOrEmpty(type.FullName))
				{
					return type.Name;
				}

				var fullClassName = type.FullName;

				if (!string.IsNullOrEmpty(type.Namespace))
				{
					fullClassName = fullClassName
						.Replace(type.Namespace, "")
						.TrimStart('.');
				}

				return fullClassName;
			});
		});

		return services;
	}

	private static IServiceCollection AddApiVersioning(this IServiceCollection services)
	{
		services.AddApiVersioning(o =>
		{
			o.DefaultApiVersion = new ApiVersion(1);
			o.ApiVersionReader = new UrlSegmentApiVersionReader();
			o.ReportApiVersions = true;
		})
			.AddApiExplorer(o =>
			{
				o.GroupNameFormat = "'v'V";
				o.SubstituteApiVersionInUrl = true;
			});

		return services;
	}

	private static IServiceCollection AddOtel(this IServiceCollection services)
	{
		Activity.DefaultIdFormat = ActivityIdFormat.W3C;

		ActivitySource activitySource = new(Constants.AppOtelName);
		services.AddSingleton(activitySource);

		services.AddOpenTelemetry()
			.WithTracing(t =>
			{
				t.AddAspNetCoreInstrumentation()
					.AddHttpClientInstrumentation()
					.AddOtlpExporter()
					.AddSource(activitySource.Name);
			})
			.WithMetrics();

		return services;
	}

	private static IServiceCollection AddAuthenticationAndAuthorization(this IServiceCollection services, IHostEnvironment environment)
	{
		services.AddOptions<JwtOptions>()
			.BindConfiguration(JwtOptions.SectionName)
			.ValidateFluently()
			.ValidateOnStart();

		var jwtOptions = services.BuildServiceProvider().GetRequiredService<IOptions<JwtOptions>>();

		services.AddAuthentication(JwtBearerDefaults.AuthenticationScheme)
			.AddJwtBearer(JwtBearerDefaults.AuthenticationScheme, o =>
			{
				o.MetadataAddress = jwtOptions.Value.MetadataUrl;

				o.TokenValidationParameters.ValidAudience = jwtOptions.Value.ValidAudience;
				o.TokenValidationParameters.ValidateAudience = true;

				if (!string.IsNullOrWhiteSpace(jwtOptions.Value.NameClaimType))
				{
					o.TokenValidationParameters.NameClaimType = jwtOptions.Value.NameClaimType;
				}

				o.RequireHttpsMetadata = !jwtOptions.Value.AllowHttpMetadataUrl && !environment.IsDevelopment();
			});

		services.AddAuthorizationBuilder()
			.AddPolicy(Constants.Authorization.RequireAuthenticatedUserPolicy, p => p
				.RequireAuthenticatedUser()
				.AddAuthenticationSchemes(JwtBearerDefaults.AuthenticationScheme));

		services.AddScoped<UserAccessor>();

		return services;
	}

	private static IServiceCollection AddDbContext(this IServiceCollection services, IConfiguration configuration)
	{
		services.AddDbContext<ApplicationDbContext>(o =>
		{
			o.UseNpgsql(configuration.GetConnectionString("Database"), c =>
			{
				c.EnableRetryOnFailure();
			});
		});

		return services;
	}
}