Attempt to simplify the token cache providers

DO NOT MERGE

Ideally we'd want the implementation of the token cache
providers to be simplified. However this is not yet possible
because the MSAL Abstract token cache provider contains a boolean telling
if the token cache is an app or user token cache (so that the key
can be computed).

Ideally we'd want the ITokenCache to tell the application if it's an app token cache or a user token cache.

Author: jmprieur

2-WebApp-graph-user/2-2-TokenCache/Startup.cs

@@ -51,8 +51,7 @@ public void ConfigureServices(IServiceCollection services)
             // and chosen token cache implementation
             services.AddMicrosoftIdentityPlatformAuthentication(Configuration)
                     .AddMsal(Configuration, new string[] { Constants.ScopeUserRead })
-                    .AddSqlAppTokenCache(msalSqlTokenCacheOptions)
-                    .AddSqlPerUserTokenCache(msalSqlTokenCacheOptions);
+                    .AddSqlTokenCaches(msalSqlTokenCacheOptions);
 
 
             // Add Graph

Microsoft.Identity.Web/TokenAcquisition.cs

@@ -29,8 +29,7 @@ public class TokenAcquisition : ITokenAcquisition
         private readonly AzureADOptions _azureAdOptions;
         private readonly ConfidentialClientApplicationOptions _applicationOptions;
 
-        private readonly IMsalAppTokenCacheProvider _appTokenCacheProvider;
-        private readonly IMsalUserTokenCacheProvider _userTokenCacheProvider;
+        private readonly IMsalTokenCacheProvider _tokenCacheProvider;
 
         private IConfidentialClientApplication application;
         private readonly IHttpContextAccessor _httpContextAccessor;
@@ -42,20 +41,18 @@ public class TokenAcquisition : ITokenAcquisition
         /// This constructor is called by ASP.NET Core dependency injection
         /// </summary>
         /// <param name="configuration"></param>
-        /// <param name="appTokenCacheProvider">The App token cache provider</param>
+        /// <param name="tokenCacheProvider">The App token cache provider</param>
         /// <param name="userTokenCacheProvider">The User token cache provider</param>
         public TokenAcquisition(
-            IMsalAppTokenCacheProvider appTokenCacheProvider,
-            IMsalUserTokenCacheProvider userTokenCacheProvider,
+            IMsalTokenCacheProvider tokenCacheProvider,
             IHttpContextAccessor httpContextAccessor,
             IOptions<AzureADOptions> azureAdOptions,
             IOptions<ConfidentialClientApplicationOptions> applicationOptions)
         {
             _httpContextAccessor = httpContextAccessor;
             _azureAdOptions = azureAdOptions.Value;
             _applicationOptions = applicationOptions.Value;
-            _appTokenCacheProvider = appTokenCacheProvider;
-            _userTokenCacheProvider = userTokenCacheProvider;
+            _tokenCacheProvider = tokenCacheProvider;
         }
 
         /// <summary>
@@ -283,7 +280,7 @@ public async Task RemoveAccountAsync(RedirectContext context)
             if (account != null)
             {
                 await app.RemoveAsync(account).ConfigureAwait(false);
-                _userTokenCacheProvider?.ClearAsync().ConfigureAwait(false);
+                _tokenCacheProvider?.ClearAsync().ConfigureAwait(false);
             }
         }
 
@@ -328,8 +325,8 @@ private IConfidentialClientApplication BuildConfidentialClientApplication()
                 .Build();
 
             // Initialize token cache providers
-            _appTokenCacheProvider?.InitializeAsync(app.AppTokenCache);
-            _userTokenCacheProvider?.InitializeAsync(app.UserTokenCache);
+            _tokenCacheProvider?.InitializeAsync(app.AppTokenCache, true);
+            _tokenCacheProvider?.InitializeAsync(app.UserTokenCache, false);
 
             return app;
         }

Microsoft.Identity.Web/TokenCacheProviders/IMSALUserTokenCacheProvider.cs

@@ -16,33 +16,10 @@ public static class InMemoryTokenCacheProviderExtension
         /// <returns></returns>
         public static IServiceCollection AddInMemoryTokenCaches(
             this IServiceCollection services)
-        {
-            AddInMemoryAppTokenCache(services);
-            AddInMemoryPerUserTokenCache(services);
-            return services;
-        }
-
-        /// <summary>Adds the in-memory based application token cache to the service collection.</summary>
-        /// <param name="services">The services collection to add to.</param>
-        /// <param name="cacheOptions">The MSALMemoryTokenCacheOptions allows the caller to set the token cache expiration</param>
-        public static IServiceCollection AddInMemoryAppTokenCache(
-            this IServiceCollection services)
-        {
-            services.AddMemoryCache();
-            services.AddSingleton<IMsalAppTokenCacheProvider, MsalAppMemoryTokenCacheProvider>();
-            return services;
-        }
-
-        /// <summary>Adds the in-memory based per user token cache to the service collection.</summary>
-        /// <param name="services">The services collection to add to.</param>
-        /// <param name="cacheOptions">The MSALMemoryTokenCacheOptions allows the caller to set the token cache expiration</param>
-        /// <returns></returns>
-        public static IServiceCollection AddInMemoryPerUserTokenCache(
-            this IServiceCollection services)
         {
             services.AddMemoryCache();
             services.AddHttpContextAccessor();
-            services.AddSingleton<IMsalUserTokenCacheProvider, MsalPerUserMemoryTokenCacheProvider>();
+            services.AddSingleton<IMsalTokenCacheProvider, MsalMemoryTokenCacheProvider>();
             return services;
         }
     }

Microsoft.Identity.Web/TokenCacheProviders/IMsalAppTokenCacheProvider.cs

@@ -50,59 +50,10 @@ public static IServiceCollection AddSessionTokenCaches(this IServiceCollection s
                 });
             }
 
-            AddSessionAppTokenCache(services);
-            AddSessionPerUserTokenCache(services);
+            services.AddHttpContextAccessor();;
+            services.AddScoped<IMsalTokenCacheProvider, MsalSessionTokenCacheProvider>();
 
             return services;
         }
-
-        /// <summary>Adds the Http session based application token cache to the service collection.</summary>
-        /// For this session cache to work effectively the aspnetcore session has to be configured properly.
-        /// The latest guidance is provided at https://docs.microsoft.com/en-us/aspnet/core/fundamentals/app-state
-        ///
-        /// // In the method - public void ConfigureServices(IServiceCollection services) in startup.cs, add the following
-        /// services.AddSession(option =>
-        /// {
-        ///	    option.Cookie.IsEssential = true;
-        /// });
-        ///
-        /// In the method - public void Configure(IApplicationBuilder app, IHostingEnvironment env) in startup.cs, add the following
-        ///
-        /// app.UseSession(); // Before UseMvc()
-        ///
-        /// <param name="services">The services collection to add to.</param>
-        /// <returns></returns>
-        public static IServiceCollection AddSessionAppTokenCache(this IServiceCollection services)
-        {
-            services.AddHttpContextAccessor();
-            services.AddScoped<IMsalAppTokenCacheProvider, MsalAppSessionTokenCacheProvider>();
-            return services;
-        }
-
-        /// <summary>Adds the http session based per user token cache to the service collection.</summary>
-        /// For this session cache to work effectively the aspnetcore session has to be configured properly.
-        /// The latest guidance is provided at https://docs.microsoft.com/en-us/aspnet/core/fundamentals/app-state
-        ///
-        /// // In the method - public void ConfigureServices(IServiceCollection services) in startup.cs, add the following
-        /// services.AddSession(option =>
-        /// {
-        ///	    option.Cookie.IsEssential = true;
-        /// });
-        ///
-        /// In the method - public void Configure(IApplicationBuilder app, IHostingEnvironment env) in startup.cs, add the following
-        ///
-        /// app.UseSession(); // Before UseMvc()
-        ///
-        /// <param name="services">The services collection to add to.</param>
-        /// <returns></returns>
-        public static IServiceCollection AddSessionPerUserTokenCache(this IServiceCollection services)
-        {
-            services.AddHttpContextAccessor();
-            services.AddSession(option =>
-            { option.Cookie.IsEssential = true; }
-            );
-            services.AddScoped<IMsalUserTokenCacheProvider, MsalPerUserSessionTokenCacheProvider>();
-            return services;
-        }
     }
 }