From 0bc42a9b21acb5ade5c305263c5949d021794158 Mon Sep 17 00:00:00 2001
From: jacalvar <jacalvar@microsoft.com>
Date: Mon, 28 Mar 2016 14:59:08 -0700
Subject: [PATCH] Return IServiceCollection from AddAntiforgery extension
 methods

---
 .../ServiceCollectionExtensions.cs                       | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs b/src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs
index dfabbe80e3..c16ac5c255 100644
--- a/src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs
+++ b/src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs
@@ -19,7 +19,8 @@ namespace Microsoft.Extensions.DependencyInjection
         /// Adds antiforgery services to the specified <see cref="IServiceCollection" />.
         /// </summary>
         /// <param name="services">The <see cref="IServiceCollection" /> to add services to.</param>
-        public static void AddAntiforgery(this IServiceCollection services)
+        /// <returns>The <see cref="IServiceCollection"/> so that additional calls can be chained.</returns>
+        public static IServiceCollection AddAntiforgery(this IServiceCollection services)
         {
             if (services == null)
             {
@@ -45,6 +46,8 @@ namespace Microsoft.Extensions.DependencyInjection
                 var policy = new AntiforgerySerializationContextPooledObjectPolicy();
                 return provider.Create(policy);
             });
+
+            return services;
         }
 
         /// <summary>
@@ -52,7 +55,8 @@ namespace Microsoft.Extensions.DependencyInjection
         /// </summary>
         /// <param name="services">The <see cref="IServiceCollection" /> to add services to.</param>
         /// <param name="setupAction">An <see cref="Action{AntiforgeryOptions}"/> to configure the provided <see cref="AntiforgeryOptions"/>.</param>
-        public static void AddAntiforgery(this IServiceCollection services, Action<AntiforgeryOptions> setupAction)
+        /// <returns>The <see cref="IServiceCollection"/> so that additional calls can be chained.</returns>
+        public static IServiceCollection AddAntiforgery(this IServiceCollection services, Action<AntiforgeryOptions> setupAction)
         {
             if (services == null)
             {
@@ -66,6 +70,7 @@ namespace Microsoft.Extensions.DependencyInjection
 
             services.AddAntiforgery();
             services.Configure(setupAction);
+            return services;
         }
     }
 }