Add ISession provider.

AzureAD · Jul 11, 2020 · 2f133d1 · 2f133d1
1 parent 7cfc0cb
commit 2f133d1
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/Microsoft.Identity.Web/TokenCacheProviders/Session/SessionTokenCacheProviderExtension.cs b/src/Microsoft.Identity.Web/TokenCacheProviders/Session/SessionTokenCacheProviderExtension.cs
@@ -5,7 +5,9 @@
 using System.Linq;
 using Microsoft.AspNetCore.Authentication;
 using Microsoft.AspNetCore.Builder;
+using Microsoft.AspNetCore.Http;
 using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.DependencyInjection.Extensions;
 
 namespace Microsoft.Identity.Web.TokenCacheProviders.Session
 {
@@ -64,6 +66,7 @@ public static IServiceCollection AddSessionTokenCaches(this IServiceCollection s
 
             services.AddHttpContextAccessor();
             services.AddScoped<IMsalTokenCacheProvider, MsalSessionTokenCacheProvider>();
+            services.TryAddScoped(provider => provider.GetService<IHttpContextAccessor>().HttpContext.Session);
 
             return services;
         }
@@ -129,7 +132,13 @@ public static IServiceCollection AddSessionAppTokenCache(this IServiceCollection
             }
 
             services.AddHttpContextAccessor();
+            services.AddSession(option =>
+            {
+                option.Cookie.IsEssential = true;
+            });
             services.AddScoped<IMsalTokenCacheProvider, MsalSessionTokenCacheProvider>();
+            services.TryAddScoped(provider => provider.GetService<IHttpContextAccessor>().HttpContext.Session);
+
             return services;
         }
 
@@ -167,6 +176,8 @@ public static IServiceCollection AddSessionPerUserTokenCache(this IServiceCollec
                     option.Cookie.IsEssential = true;
                 });
             services.AddScoped<IMsalTokenCacheProvider, MsalSessionTokenCacheProvider>();
+            services.TryAddScoped(provider => provider.GetService<IHttpContextAccessor>().HttpContext.Session);
+
             return services;
         }
     }