131 lines
3.8 KiB
C#
131 lines
3.8 KiB
C#
using System;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using BlueWest.Cryptography;
|
|
using BlueWest.Data.Application;
|
|
using BlueWest.Data.Auth;
|
|
using Microsoft.Extensions.Hosting;
|
|
using Redis.OM;
|
|
using Redis.OM.Searching;
|
|
|
|
namespace BlueWest.WebApi.Session
|
|
{
|
|
/// <summary>
|
|
/// Session Provider Context
|
|
/// </summary>
|
|
public sealed class SessionManager : ISessionCache
|
|
{
|
|
private readonly RedisConnectionProvider _provider;
|
|
private readonly RedisCollection<SessionToken> _sessionTokens;
|
|
/// <summary>
|
|
/// Index Creation Device
|
|
/// </summary>
|
|
/// <param name="provider">Redis connection</param>
|
|
public SessionManager(
|
|
RedisConnectionProvider provider)
|
|
{
|
|
_provider = provider;
|
|
_sessionTokens = (RedisCollection<SessionToken>)provider.RedisCollection<SessionToken>();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Empty constructor
|
|
/// </summary>
|
|
public SessionManager() { }
|
|
|
|
/// <summary>
|
|
/// Get a session token by the respective Id.
|
|
/// </summary>
|
|
/// <param name="tokenId"></param>
|
|
/// <returns></returns>
|
|
public async Task<SessionToken> GetSessionTokenByIdAsync(string tokenId)
|
|
{
|
|
return await _sessionTokens.Where(x => x.Id == tokenId)
|
|
.FirstOrDefaultAsync();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Create a new session token
|
|
/// </summary>
|
|
/// <param name="token"></param>
|
|
public async Task AddSessionToken(SessionToken token)
|
|
{
|
|
await _sessionTokens.InsertAsync(token);
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public async Task SaveAsync()
|
|
{
|
|
await _sessionTokens.SaveAsync();
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// Save session data
|
|
/// </summary>
|
|
public void Save()
|
|
{
|
|
_sessionTokens.Save();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets a Bearer By Access Token Id
|
|
/// </summary>
|
|
/// <param name="sessionTokenId"></param>
|
|
public async Task<bool> IsSessionValidAsync(string sessionTokenId)
|
|
{
|
|
var accessToken = await _sessionTokens
|
|
.Where(t => t.Id == sessionTokenId)
|
|
.FirstOrDefaultAsync();
|
|
|
|
if (accessToken == null) return false;
|
|
|
|
if (accessToken.IsValid)
|
|
{
|
|
// Check if it's not expired
|
|
var expirationDate = accessToken.CreatedDate + accessToken.ValidFor;
|
|
var timeNow = DateTimeOffset.Now.ToUnixTimeMilliseconds();
|
|
if (expirationDate >= timeNow) return accessToken.IsValid;
|
|
accessToken.IsValid = false;
|
|
await _sessionTokens.SaveAsync();
|
|
}
|
|
|
|
return accessToken.IsValid;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Checks if a session is valid
|
|
/// </summary>
|
|
/// <param name="sessionTokenId"></param>
|
|
/// <returns></returns>
|
|
public bool IsSessionValid(string sessionTokenId)
|
|
{
|
|
var accessToken = _sessionTokens
|
|
.FirstOrDefault(t => t.Id == sessionTokenId);
|
|
|
|
if (accessToken == null) return false;
|
|
|
|
if (!accessToken.IsValid) return accessToken.IsValid;
|
|
|
|
accessToken.Validate();
|
|
_sessionTokens.Save();
|
|
|
|
return accessToken.IsValid;
|
|
}
|
|
|
|
|
|
/// <inheritdoc />
|
|
public async Task StartAsync(CancellationToken cancellationToken)
|
|
{
|
|
await _provider.Connection.CreateIndexAsync(typeof(SessionToken));
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public Task StopAsync(CancellationToken cancellationToken)
|
|
{
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|
|
}
|
|
|