99 lines
4.4 KiB
C#
99 lines
4.4 KiB
C#
using Jellyfin.Api;
|
|
using Jellyfin.Api.Auth;
|
|
using Jellyfin.Api.Auth.FirstTimeSetupOrElevatedPolicy;
|
|
using Jellyfin.Api.Auth.RequiresElevationPolicy;
|
|
using Jellyfin.Api.Constants;
|
|
using Jellyfin.Api.Controllers;
|
|
using Jellyfin.Server.Formatters;
|
|
using Microsoft.AspNetCore.Authentication;
|
|
using Microsoft.AspNetCore.Authorization;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.OpenApi.Models;
|
|
|
|
namespace Jellyfin.Server.Extensions
|
|
{
|
|
/// <summary>
|
|
/// API specific extensions for the service collection.
|
|
/// </summary>
|
|
public static class ApiServiceCollectionExtensions
|
|
{
|
|
/// <summary>
|
|
/// Adds jellyfin API authorization policies to the DI container.
|
|
/// </summary>
|
|
/// <param name="serviceCollection">The service collection.</param>
|
|
/// <returns>The updated service collection.</returns>
|
|
public static IServiceCollection AddJellyfinApiAuthorization(this IServiceCollection serviceCollection)
|
|
{
|
|
serviceCollection.AddSingleton<IAuthorizationHandler, FirstTimeSetupOrElevatedHandler>();
|
|
serviceCollection.AddSingleton<IAuthorizationHandler, RequiresElevationHandler>();
|
|
return serviceCollection.AddAuthorizationCore(options =>
|
|
{
|
|
options.AddPolicy(
|
|
Policies.RequiresElevation,
|
|
policy =>
|
|
{
|
|
policy.AddAuthenticationSchemes(AuthenticationSchemes.CustomAuthentication);
|
|
policy.AddRequirements(new RequiresElevationRequirement());
|
|
});
|
|
options.AddPolicy(
|
|
Policies.FirstTimeSetupOrElevated,
|
|
policy =>
|
|
{
|
|
policy.AddAuthenticationSchemes(AuthenticationSchemes.CustomAuthentication);
|
|
policy.AddRequirements(new FirstTimeSetupOrElevatedRequirement());
|
|
});
|
|
});
|
|
}
|
|
|
|
/// <summary>
|
|
/// Adds custom legacy authentication to the service collection.
|
|
/// </summary>
|
|
/// <param name="serviceCollection">The service collection.</param>
|
|
/// <returns>The updated service collection.</returns>
|
|
public static AuthenticationBuilder AddCustomAuthentication(this IServiceCollection serviceCollection)
|
|
{
|
|
return serviceCollection.AddAuthentication(AuthenticationSchemes.CustomAuthentication)
|
|
.AddScheme<AuthenticationSchemeOptions, CustomAuthenticationHandler>(AuthenticationSchemes.CustomAuthentication, null);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Extension method for adding the jellyfin API to the service collection.
|
|
/// </summary>
|
|
/// <param name="serviceCollection">The service collection.</param>
|
|
/// <param name="baseUrl">The base url for the API.</param>
|
|
/// <returns>The MVC builder.</returns>
|
|
public static IMvcBuilder AddJellyfinApi(this IServiceCollection serviceCollection, string baseUrl)
|
|
{
|
|
return serviceCollection.AddMvc(opts =>
|
|
{
|
|
opts.UseGeneralRoutePrefix(baseUrl);
|
|
opts.OutputFormatters.Insert(0, new CamelCaseJsonProfileFormatter());
|
|
opts.OutputFormatters.Insert(0, new PascalCaseJsonProfileFormatter());
|
|
})
|
|
|
|
// Clear app parts to avoid other assemblies being picked up
|
|
.ConfigureApplicationPartManager(a => a.ApplicationParts.Clear())
|
|
.AddApplicationPart(typeof(StartupController).Assembly)
|
|
.AddJsonOptions(options =>
|
|
{
|
|
// Setting the naming policy to null leaves the property names as-is when serializing objects to JSON.
|
|
options.JsonSerializerOptions.PropertyNamingPolicy = null;
|
|
})
|
|
.AddControllersAsServices();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Adds Swagger to the service collection.
|
|
/// </summary>
|
|
/// <param name="serviceCollection">The service collection.</param>
|
|
/// <returns>The updated service collection.</returns>
|
|
public static IServiceCollection AddJellyfinApiSwagger(this IServiceCollection serviceCollection)
|
|
{
|
|
return serviceCollection.AddSwaggerGen(c =>
|
|
{
|
|
c.SwaggerDoc("v1", new OpenApiInfo { Title = "Jellyfin API", Version = "v1" });
|
|
});
|
|
}
|
|
}
|
|
}
|