112 lines
3.9 KiB
C#
112 lines
3.9 KiB
C#
// 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.Collections.Generic;
|
|
using System.Reflection;
|
|
using Microsoft.AspNet.Mvc;
|
|
using Microsoft.AspNet.Mvc.Formatters;
|
|
using Microsoft.AspNet.Mvc.Internal;
|
|
|
|
namespace Microsoft.Extensions.DependencyInjection
|
|
{
|
|
/// <summary>
|
|
/// Extensions for configuring MVC using an <see cref="IMvcBuilder"/>.
|
|
/// </summary>
|
|
public static class MvcCoreMvcBuilderExtensions
|
|
{
|
|
/// <summary>
|
|
/// Registers an action to configure <see cref="MvcOptions"/>.
|
|
/// </summary>
|
|
/// <param name="builder">The <see cref="IMvcBuilder"/>.</param>
|
|
/// <param name="setupAction">An <see cref="Action{MvcOptions}"/>.</param>
|
|
/// <returns>The <see cref="IMvcBuilder"/>.</returns>
|
|
public static IMvcBuilder AddMvcOptions(
|
|
this IMvcBuilder builder,
|
|
Action<MvcOptions> setupAction)
|
|
{
|
|
if (builder == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(builder));
|
|
}
|
|
|
|
if (setupAction == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(setupAction));
|
|
}
|
|
|
|
builder.Services.Configure<MvcOptions>(setupAction);
|
|
return builder;
|
|
}
|
|
|
|
public static IMvcBuilder AddFormatterMappings(
|
|
this IMvcBuilder builder,
|
|
Action<FormatterMappings> setupAction)
|
|
{
|
|
if (builder == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(builder));
|
|
}
|
|
|
|
if (setupAction == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(setupAction));
|
|
}
|
|
|
|
builder.Services.Configure<MvcOptions>((options) => setupAction(options.FormatterMappings));
|
|
return builder;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Register the specified <paramref name="controllerTypes"/> as services and as a source for controller
|
|
/// discovery.
|
|
/// </summary>
|
|
/// <param name="builder">The <see cref="IMvcBuilder"/>.</param>
|
|
/// <param name="controllerTypes">A sequence of controller <see cref="Type"/>s to register in the
|
|
/// <paramref name="services"/> and used for controller discovery.</param>
|
|
/// <returns>The <see cref="IMvcBuilder"/>.</returns>
|
|
public static IMvcBuilder AddControllersAsServices(
|
|
this IMvcBuilder builder,
|
|
IEnumerable<Type> controllerTypes)
|
|
{
|
|
if (builder == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(builder));
|
|
}
|
|
|
|
if (controllerTypes == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(controllerTypes));
|
|
}
|
|
|
|
ControllersAsServices.AddControllersAsServices(builder.Services, controllerTypes);
|
|
return builder;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Registers controller types from the specified <paramref name="assemblies"/> as services and as a source
|
|
/// for controller discovery.
|
|
/// </summary>
|
|
/// <param name="builder">The <see cref="IMvcBuilder"/>.</param>
|
|
/// <param name="controllerAssemblies">Assemblies to scan.</param>
|
|
/// <returns>The <see cref="IMvcBuilder"/>.</returns>
|
|
public static IMvcBuilder AddControllersAsServices(
|
|
this IMvcBuilder builder,
|
|
IEnumerable<Assembly> controllerAssemblies)
|
|
{
|
|
if (builder == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(builder));
|
|
}
|
|
|
|
if (controllerAssemblies == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(controllerAssemblies));
|
|
}
|
|
|
|
ControllersAsServices.AddControllersAsServices(builder.Services, controllerAssemblies);
|
|
return builder;
|
|
}
|
|
}
|
|
}
|