using System; using VNLib.Net.Http; using VNLib.Net.Messaging.FBM.Client; using VNLib.Plugins.Sessions.Cache.Client; using static VNLib.Plugins.Essentials.Sessions.ISessionExtensions; namespace VNLib.Plugins.Essentials.Sessions.OAuth { /// /// The implementation of the OAuth2 session container for HTTP sessions /// internal sealed class OAuth2Session : RemoteSession { private readonly Action InvalidateCache; /// /// Initalizes a new /// /// The session id (or token) /// The used as the backing cache provider /// The ammount of time to wait for a background operation (delete, update, get) /// Called when the session has been marked as invalid and the close even hook is being executed public OAuth2Session(string sessionId, FBMClient client, TimeSpan backgroundTimeOut, Action invalidCache) : base(sessionId, client, backgroundTimeOut) { InvalidateCache = invalidCache; IsInvalid = false; } public bool IsInvalid { get; private set; } /// /// public override string Token { get => throw new NotSupportedException("Token property is not supported for OAuth2 sessions"); set => throw new NotSupportedException("Token property is not supported for OAuth2 sessions"); } /// protected override void IndexerSet(string key, string value) { //Guard protected entires switch (key) { case TOKEN_ENTRY: case LOGIN_TOKEN_ENTRY: throw new InvalidOperationException("Token entry may not be changed!"); } base.IndexerSet(key, value); } /// /// public override async Task WaitAndLoadAsync(IHttpEvent entity, CancellationToken token = default) { //Wait to enter lock await base.WaitAndLoadAsync(entity, token); if (IsInvalid) { //Release lock MainLock.Release(); throw new SessionStatusException("The session has been invalidated"); } //Set session type if (IsNew) { SessionType = SessionType.OAuth2; } } /// protected override async ValueTask UpdateResource(bool isAsync, IHttpEvent state) { Task? result = null; //Invalid flag is set, so exit if (IsInvalid) { result = Task.CompletedTask; } //Check flags in priority level, Invalid is highest state priority else if (Flags.IsSet(INVALID_MSK)) { //Clear all stored values DataStore!.Clear(); //Delete the entity syncronously await ProcessDeleteAsync(); //Set invalid flag IsInvalid = true; //Invlidate cache InvalidateCache(this); result = Task.CompletedTask; } else if (Flags.IsSet(MODIFIED_MSK)) { //Send update to server result = Task.Run(ProcessUpdateAsync); } //Clear all flags Flags.ClearAll(); return result; } } }