mirror of
https://github.com/bitwarden/server.git
synced 2025-07-01 08:02:49 -05:00
67 lines
1.9 KiB
C#
67 lines
1.9 KiB
C#
using Microsoft.AspNetCore.Authentication;
|
|
using Microsoft.AspNetCore.Authentication.Cookies;
|
|
using Microsoft.Extensions.Caching.Distributed;
|
|
using Microsoft.Extensions.Caching.Memory;
|
|
using Microsoft.Extensions.Caching.Redis;
|
|
|
|
namespace Bit.Core.IdentityServer
|
|
{
|
|
public class RedisCacheTicketStore : ITicketStore
|
|
{
|
|
private const string _keyPrefix = "auth-";
|
|
private readonly IDistributedCache _cache;
|
|
|
|
public RedisCacheTicketStore(RedisCacheOptions options)
|
|
{
|
|
_cache = new RedisCache(options);
|
|
}
|
|
|
|
public async Task<string> StoreAsync(AuthenticationTicket ticket)
|
|
{
|
|
var key = $"{_keyPrefix}{Guid.NewGuid()}";
|
|
await RenewAsync(key, ticket);
|
|
|
|
return key;
|
|
}
|
|
|
|
public Task RenewAsync(string key, AuthenticationTicket ticket)
|
|
{
|
|
var options = new DistributedCacheEntryOptions();
|
|
var expiresUtc = ticket.Properties.ExpiresUtc ??
|
|
DateTimeOffset.UtcNow.AddMinutes(15);
|
|
options.SetAbsoluteExpiration(expiresUtc);
|
|
|
|
var val = SerializeToBytes(ticket);
|
|
_cache.Set(key, val, options);
|
|
|
|
return Task.FromResult(0);
|
|
}
|
|
|
|
public Task<AuthenticationTicket> RetrieveAsync(string key)
|
|
{
|
|
AuthenticationTicket ticket;
|
|
var bytes = _cache.Get(key);
|
|
ticket = DeserializeFromBytes(bytes);
|
|
|
|
return Task.FromResult(ticket);
|
|
}
|
|
|
|
public Task RemoveAsync(string key)
|
|
{
|
|
_cache.Remove(key);
|
|
|
|
return Task.FromResult(0);
|
|
}
|
|
|
|
private static byte[] SerializeToBytes(AuthenticationTicket source)
|
|
{
|
|
return TicketSerializer.Default.Serialize(source);
|
|
}
|
|
|
|
private static AuthenticationTicket DeserializeFromBytes(byte[] source)
|
|
{
|
|
return source == null ? null : TicketSerializer.Default.Deserialize(source);
|
|
}
|
|
}
|
|
}
|